PanXiongAdam
commented
4 years ago Hello! there is the same problem in my computer, Did you solve it? If you solve it, Can you tell me how to do it, please? Thank you very much！
Open zeroXscorpion7 opened 4 years ago
PanXiongAdam
commented
4 years ago Hello! there is the same problem in my computer, Did you solve it? If you solve it, Can you tell me how to do it, please? Thank you very much！
File "train.py", line 54, in
model.forward(data)
File "/home/aaronsung/pyTorch/DepthAwareCNN-master/models/Deeplab.py", line 94, in forward
self.segpred = self.model(self.image,self.depth)
File "/home/aaronsung/.conda/envs/pytorch/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, kwargs)
File "/home/aaronsung/pyTorch/DepthAwareCNN-master/models/Deeplab.py", line 23, in forward
output = self.Scale(x,depth) # for original scale
File "/home/aaronsung/.conda/envs/pytorch/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, *kwargs)
File "/home/aaronsung/pyTorch/DepthAwareCNN-master/models/VGG_Deeplab.py", line 430, in forward
x,depth = self.features(x,depth)
File "/home/aaronsung/.conda/envs/pytorch/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(input, kwargs)
File "/home/aaronsung/pyTorch/DepthAwareCNN-master/models/VGG_Deeplab.py", line 228, in forward
x = self.conv1_1(x,self.conv1_1_depthconvweight depth)
File "/home/aaronsung/.conda/envs/pytorch/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(input, kwargs)
File "/home/aaronsung/pyTorch/DepthAwareCNN-master/models/VGG_Deeplab.py", line 77, in forward
x = module(x,depth)
File "/home/aaronsung/.conda/envs/pytorch/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, *kwargs)
File "/home/aaronsung/pyTorch/DepthAwareCNN-master/models/ops/depthconv/modules/depthconv.py", line 48, in forward
self.padding, self.dilation)
File "/home/aaronsung/pyTorch/DepthAwareCNN-master/models/ops/depthconv/functions/depthconv.py", line 25, in depth_conv
return f(input, depth, weight, bias)
File "/home/aaronsung/pyTorch/DepthAwareCNN-master/models/ops/depthconv/functions/depthconv.py", line 65, in forward
self.padding[1], self.padding[0], self.dilation[1], self.dilation[0])
File "/home/aaronsung/.conda/envs/pytorch/lib/python3.6/site-packages/torch/utils/ffi/init.py", line 202, in safe_call
result = torch._C._safe_call(args, kwargs)
torch.FatalError: invalid argument 5: 4D weight tensor (nOutputPlane,nInputPlane,kH,kW) expected, but got: (null) at /home/aaronsung/pyTorch/DepthAwareCNN-master/models/ops/depthconv/src/depthconv_cuda.c:15
pytorch=0.4.1,CUDA=10.0,ubuntu=18.04 How could I slove it?