Puzzled about the error in finetune.py

zychen-ustc / PSD-Principled-Synthetic-to-Real-Dehazing-Guided-by-Physical-Priors

Zeyuan Chen, Yangchao Wang, Yang Yang and Dong Liu. "PSD: Principled Synthetic-to-Real Dehazing Guided by Physical Priors". IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2021.

MIT License

111 stars 18 forks source link

Puzzled about the error in finetune.py #6

Closed hello-trouble closed 2 years ago

hello-trouble commented 2 years ago

Thank you for your excellent job about the real image dehazing. I am very insterested in this research. However, I meet an error when I train the network with finetune.py. The error is as follows: "RuntimeError: CUDA error: invalid configuration argument" It happened in the line 47 of the energy_functions.py,
energy_dc_loss = loss_f(unlabel_haze, T)---->weights = torch.matmul(img_patches - mean_patches, var_fac)

Good holiday and looking forward to your help.

hello-trouble commented 2 years ago

So sorry to bother you, it has been solved now . Thanks again for your novel paper and project.