Pytorch出现Process finished with exit code 139 (interrupted by signal 11: SIGSEGV)

只是提出问题,并没有解决

简介

最近一直在修改一个3D的网络,在设计网络的过程中出现了上面的问题,具体的原因还没有找到,我有一个习惯是设计好网络结构后,需要使用Variable数据输入到网络里面测试,结果出现代码一里面的问题。但是正常训练的时候却没有这个问题。有点奇怪

代码1

from torch.autograd import Variable
import torch

data = Variable(torch.rand(1, 256, 12, 12, 12))
model = torch.nn.Conv3d(256, 128, kernel_size=1, stride=1)
out = model(data)
# out = model(data).cuda()
print(out.size())

出现的结果是

Process finished with exit code 139 (interrupted by signal 11: SIGSEGV)

 只需要将Conv3D中的128改为64就不会出现这个问题了

代码

 

from torch.autograd import Variable
import torch

data = Variable(torch.rand(1, 256, 12, 12, 12))
model = torch.nn.Conv3d(256, 64, kernel_size=1, stride=1)
out = model(data)
# out = model(data).cuda()
print(out.size())

结果为

(1, 64, 12, 12, 12)

 

你可能感兴趣的:(pytorch)