I input the following command to train my data, but have some error.
CUDA_VISIBLE_DEVICE=3 python train.py --dataroot blurred_sharp/ --learn_residual --resize_or_crop crop --fineSize 265 --no_html
error:
Traceback (most recent call last):
File "train.py", line 59, in
train(opt, data_loader, model, visualizer)
File "train.py", line 22, in train
model.optimize_parameters()
File "/home/liaochunxiu/DeblurGAN-master/models/conditional_gan_model.py", line 104, in optimize_parameters
self.forward()
File "/home/liaochunxiu/DeblurGAN-master/models/conditional_gan_model.py", line 76, in forward
self.fake_B = self.netG.forward(self.real_A)
File "/home/liaochunxiu/DeblurGAN-master/models/networks.py", line 146, in forward
output = self.model(input)
File "/usr/local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 357, in call
result = self.forward(*input, *kwargs)
File "/usr/local/lib/python3.6/site-packages/torch/nn/modules/container.py", line 67, in forward
input = module(input)
File "/usr/local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 357, in call
result = self.forward(input, **kwargs)
File "/usr/local/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 282, in forward
self.padding, self.dilation, self.groups)
File "/usr/local/lib/python3.6/site-packages/torch/nn/functional.py", line 90, in conv2d
return f(input, weight, bias)
RuntimeError: CUDNN_STATUS_INTERNAL_ERROR
I input the following command to train my data, but have some error. CUDA_VISIBLE_DEVICE=3 python train.py --dataroot blurred_sharp/ --learn_residual --resize_or_crop crop --fineSize 265 --no_html
error: Traceback (most recent call last): File "train.py", line 59, in
train(opt, data_loader, model, visualizer)
File "train.py", line 22, in train
model.optimize_parameters()
File "/home/liaochunxiu/DeblurGAN-master/models/conditional_gan_model.py", line 104, in optimize_parameters
self.forward()
File "/home/liaochunxiu/DeblurGAN-master/models/conditional_gan_model.py", line 76, in forward
self.fake_B = self.netG.forward(self.real_A)
File "/home/liaochunxiu/DeblurGAN-master/models/networks.py", line 146, in forward
output = self.model(input)
File "/usr/local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 357, in call
result = self.forward(*input, *kwargs)
File "/usr/local/lib/python3.6/site-packages/torch/nn/modules/container.py", line 67, in forward
input = module(input)
File "/usr/local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 357, in call
result = self.forward(input, **kwargs)
File "/usr/local/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 282, in forward
self.padding, self.dilation, self.groups)
File "/usr/local/lib/python3.6/site-packages/torch/nn/functional.py", line 90, in conv2d
return f(input, weight, bias)
RuntimeError: CUDNN_STATUS_INTERNAL_ERROR
Can you tell me how to do for this? Thanks~