search

IGITUGraz / WeatherDiffusion

Code for "Restoring Vision in Adverse Weather Conditions with Patch-Based Denoising Diffusion Models" [TPAMI 2023]
MIT License
286 stars 25 forks source link

The training crashes #24

Open OwXiaoM opened 1 year ago

OwXiaoM commented 1 year ago

Dear authors, thank you for sharing your codes for the community, I have a question about the model training, is it nomarl that the model crashes during the training, as I tried to reproduct the model on derain and desnow datasets, but during certain iters, as 800000, the visual results shows like pure noise or pure black. Should I wait more iters or is there any error in my reproduction? Thank you very much image image

OwXiaoM commented 1 year ago

I've verified that the black patch shows when the loss is NAN

Kushpatel49 commented 1 year ago

hey there can you help me to setup this code locally as i am not able to find datasets and .txt files?

yongliang-wu commented 11 months ago

hey there can you help me to setup this code locally as i am not able to find datasets and .txt files?

you can find the .txt files from #10