Here I meet a problem when use Conv3d to process a [32,32,32] pixel sized cube,and here is the code:
layer1=nn.Sequential(
nn.AvgPool3d((2,1,1),stride=(2,1,1))
)
layer2_1=nn.Sequential(
nn.Conv3d(1,64,(3,3,3),(1,1,1),(1,1,1)),
nn.LeakyReLU(RL_leakyrate),
)
Here my input x is a Variable with data type: torch.tensor.double,and the size is [4,1,32,32,32].
Then I try 2 operations:
out=layer1(x)
The step is normal and returns a [4,1,16,32,32] size’s tensor with type double ,and I try the next operation:
out2=layer2_1(out)
And it returns the error message:
“Process finished with exit code 139 (interrupted by signal 11: SIGSEGV).”
Then I try to debug step by step, and I notice that in the source code “functional.py”, function"conv3d", line 116:
return f(input,weight,bias)
When I run this code, the program exit with all variables disappeared. I’m sure nobody has raised this question before, so I raise the problem and find some help here.
If more information is needed, please tell me, and I will show the details as clear as possible.
Thanks