I followed the instructions and installed the requirements in the environment.yaml but I get the following error:ย
Traceback (most recent call last):
File "run_trained_model.py", line 147, in <module>
run_trained_model(opt)
File "run_trained_model.py", line 53, in run_trained_model
output_images_path, args.f_factor_path, None, model_params["final_shape_addition"])
File "run_trained_model.py", line 68, in run_model_on_path
model_params, f_factor_path, final_shape_addition)
File "/media/cgl/MUSKETEER_DATA/frameworks/unpaired_hdr_tmo/utils/model_save_util.py", line 209, in run_model_on_single_image
fake = G_net(gray_im_log.unsqueeze(0), apply_crop=model_params["add_frame"], diffY=diffY, diffX=diffX)
File "/home/cgl/miniconda3/envs/vinker/lib/python3.6/site-packages/torch/nn/modules/module.py", line 547, in __call__
result = self.forward(*input, **kwargs)
File "/media/cgl/MUSKETEER_DATA/frameworks/unpaired_hdr_tmo/models/unet_multi_filters/Unet.py", line 86, in forward
next_x = self.inc(x)
File "/home/cgl/miniconda3/envs/vinker/lib/python3.6/site-packages/torch/nn/modules/module.py", line 547, in __call__
result = self.forward(*input, **kwargs)
File "/media/cgl/MUSKETEER_DATA/frameworks/unpaired_hdr_tmo/models/unet_multi_filters/unet_parts.py", line 202, in forward
x = self.conv(x)
File "/home/cgl/miniconda3/envs/vinker/lib/python3.6/site-packages/torch/nn/modules/module.py", line 547, in __call__
result = self.forward(*input, **kwargs)
File "/media/cgl/MUSKETEER_DATA/frameworks/unpaired_hdr_tmo/models/unet_multi_filters/unet_parts.py", line 64, in forward
x = self.conv(x)
File "/home/cgl/miniconda3/envs/vinker/lib/python3.6/site-packages/torch/nn/modules/module.py", line 547, in __call__
result = self.forward(*input, **kwargs)
File "/home/cgl/miniconda3/envs/vinker/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 343, in forward
return self.conv2d_forward(input, self.weight)
File "/home/cgl/miniconda3/envs/vinker/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 340, in conv2d_forward
self.padding, self.dilation, self.groups)
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED