matheusgadelha / MRTNet

Code for Multiresolution Tree Networks for 3D Point Cloud Processing (ECCV 2018)
44 stars 6 forks source link

RuntimeError: CuDNN error: CUDNN_STATUS_BAD_PARAM #10

Closed JUN-TSUZUKI closed 5 years ago

JUN-TSUZUKI commented 5 years ago

I appreciate your sincere contribution. I am a beginner of DeepLearning. Please tell me about the following error.

python3 mr_train_4k.py -d ../Dataset/3Dmodel/chair/AfterProcessing/ Setting up a new session... /usr/local/lib/python3.5/dist-packages/torch/nn/functional.py:1890: UserWarning: nn.functional.upsample is deprecated. Use nn.functional.interpolate instead. warnings.warn("nn.functional.upsample is deprecated. Use nn.functional.interpolate instead.") Traceback (most recent call last): File "mr_train_4k.py", line 36, in trainer.train(2000) File "/home/islab/Tsuzuki/MRTNet/tools/Trainer.py", line 354, in train out_data = self.model(in_data)####### File "/usr/local/lib/python3.5/dist-packages/torch/nn/modules/module.py", line 477, in call result = self.forward(*input, kwargs) File "/home/islab/Tsuzuki/MRTNet/models/AutoEncoder.py", line 688, in forward encoding = self.enc_forward(x)[0] File "/home/islab/Tsuzuki/MRTNet/models/AutoEncoder.py", line 643, in enc_forward enc_tensors.append(enc_op(enc_tensors[-1]))###### File "/usr/local/lib/python3.5/dist-packages/torch/nn/modules/module.py", line 477, in call result = self.forward(*input, *kwargs) File "/home/islab/Tsuzuki/MRTNet/models/AutoEncoder.py", line 170, in forward out2 = self.conv2(in2)###### File "/usr/local/lib/python3.5/dist-packages/torch/nn/modules/module.py", line 477, in call result = self.forward(input, kwargs) File "/usr/local/lib/python3.5/dist-packages/torch/nn/modules/container.py", line 91, in forward input = module(input) File "/usr/local/lib/python3.5/dist-packages/torch/nn/modules/module.py", line 477, in call result = self.forward(*input, **kwargs) File "/usr/local/lib/python3.5/dist-packages/torch/nn/modules/conv.py", line 176, in forward self.padding, self.dilation, self.groups) RuntimeError: CuDNN error: CUDNN_STATUS_BAD_PARAM

Best regards