RuntimeError: CUDNN_STATUS_EXECUTION_FAILED

[Aryahagh]

Hi there, While I'm trying to run the model, when it reaches line 145 in scene_graph_tuning.py it returns this error. do you know What could be the reason and how I can solve it?

Thanks

Traceback (most recent call last): File "scene_graph_tuning.py", line 153, in object_result, predicate_result = model.forward_eval(im_data, im_info, ) File "/work/home/akhaghighat/3D-Scene-Graph/model/SGGenModel.py", line 92, in forward_eval features, object_rois = self.rpn(im_data, im_info) File "/usr/local/lib/python2.7/dist-packages/torch/nn/modules/module.py", line 357, in call result = self.forward(*input, **kwargs) File "./FactorizableNet/models/RPN/RPN.py", line 96, in forward features = self.features(im_data) File "/usr/local/lib/python2.7/dist-packages/torch/nn/modules/module.py", line 357, in call result = self.forward(*input, **kwargs) File "/usr/local/lib/python2.7/dist-packages/torch/nn/modules/container.py", line 67, in forward input = module(input) File "/usr/local/lib/python2.7/dist-packages/torch/nn/modules/module.py", line 357, in call result = self.forward(*input, **kwargs) File "/usr/local/lib/python2.7/dist-packages/torch/nn/modules/conv.py", line 282, in forward self.padding, self.dilation, self.groups) File "/usr/local/lib/python2.7/dist-packages/torch/nn/functional.py", line 90, in conv2d return f(input, weight, bias) RuntimeError: CUDNN_STATUS_EXECUTION_FAILED