search

daitao / SAN

Second-order Attention Network for Single Image Super-resolution (CVPR-2019)
557 stars 105 forks source link

RuntimeError: CUDNN_STATUS_MAPPING_ERROR #43

Open blackhumor22 opened 4 years ago

blackhumor22 commented 4 years ago

Traceback (most recent call last): File "main.py", line 19, in t.train() File "/opt/data/private/SAN-master/TrainCode/trainer.py", line 51, in train sr = self.model(lr, idx_scale) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 491, in call result = self.forward(*input, kwargs) File "/opt/data/private/SAN-master/TrainCode/model/init.py", line 58, in forward return self.model(x) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 491, in call result = self.forward(*input, *kwargs) File "/opt/data/private/SAN-master/TrainCode/model/san.py", line 515, in forward x = self.sub_mean(x) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 491, in call result = self.forward(input, kwargs) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/conv.py", line 301, in forward self.padding, self.dilation, self.groups) RuntimeError: CUDNN_STATUS_MAPPING_ERROR

How can I solve it？