I have been trying to run the demo script and I am getting the following error. Can you help out here? I am somewhat inexperienced with pytorch.
Traceback (most recent call last):
File "demo.py", line 79, in <module>
out = model(im, None, mean_class_data=cls_data)
File "/home/samagra14/miniconda3/envs/FSviewpoint/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in __call__
result = self.forward(*input, **kwargs)
File "/home/samagra14/6d_pose/FewShotViewpoint/auxiliary/model.py", line 103, in forward
img_feature = self.img_encoder(im) # B*C
File "/home/samagra14/miniconda3/envs/FSviewpoint/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in __call__
result = self.forward(*input, **kwargs)
File "./auxiliary/resnet.py", line 137, in forward
x = self.conv1(x)
File "/home/samagra14/miniconda3/envs/FSviewpoint/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in __call__
result = self.forward(*input, **kwargs)
File "/home/samagra14/miniconda3/envs/FSviewpoint/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 320, in forward
self.padding, self.dilation, self.groups)
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED