pytorch2caffe
pytorch2caffe copied to clipboard
"TypeError: argument 0 is not a Variable", torch version:0.2.0
torch version:0.2.0 since I met the error "convert KeyError: 'MkldnnConvolutionBackward' " , I reinstall my torch to version 0.2.0. But I get another error as below:
output_var = model(input_var) File "/usr/local/python3.7.5/lib/python3.7/site-packages/torch/nn/modules/module.py", line 224, in call result = self.forward(*input, **kwargs) File "/usr/local/python3.7.5/lib/python3.7/site-packages/torchvision/models/resnet.py", line 139, in forward x = self.conv1(x) File "/usr/local/python3.7.5/lib/python3.7/site-packages/torch/nn/modules/module.py", line 224, in call result = self.forward(*input, **kwargs) File "/usr/local/python3.7.5/lib/python3.7/site-packages/torch/nn/modules/conv.py", line 254, in forward self.padding, self.dilation, self.groups) File "/usr/local/python3.7.5/lib/python3.7/site-packages/torch/nn/functional.py", line 52, in conv2d return f(input, weight, bias) TypeError: argument 0 is not a Variable