Traceback (most recent call last):
File “main.py”, line 46, in
main(config, args)
File “main.py”, line 19, in main
Mission.train()
File “/opt/notebooks/TextZoom/src/interfaces/super_resolution.py”, line 33, in train
model_dict = self.generator_init()
File “/opt/notebooks/TextZoom/src/interfaces/base.py”, line 149, in generator_init
model = model.to(self.device)
File “/root/.conda/envs/textZoom/lib/python3.6/site-packages/torch/nn/modules/module.py”, line 381, in to
return self._apply(convert)
File “/root/.conda/envs/textZoom/lib/python3.6/site-packages/torch/nn/modules/module.py”, line 187, in _apply
module._apply(fn)
File “/root/.conda/envs/textZoom/lib/python3.6/site-packages/torch/nn/modules/module.py”, line 187, in _apply
module._apply(fn)
File “/root/.conda/envs/textZoom/lib/python3.6/site-packages/torch/nn/modules/module.py”, line 187, in _apply
module._apply(fn)
File “/root/.conda/envs/textZoom/lib/python3.6/site-packages/torch/nn/modules/rnn.py”, line 117, in _apply
self.flatten_parameters()
File “/root/.conda/envs/textZoom/lib/python3.6/site-packages/torch/nn/modules/rnn.py”, line 113, in flatten_parameters
self.batch_first, bool(self.bidirectional))
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
I’m facing this problem. Do you have any suggestion?