只是提出问题,并没有解决
简介
最近一直在修改一个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)