search

VITA-Group / EnlightenGAN

[IEEE TIP] "EnlightenGAN: Deep Light Enhancement without Paired Supervision" by Yifan Jiang, Xinyu Gong, Ding Liu, Yu Cheng, Chen Fang, Xiaohui Shen, Jianchao Yang, Pan Zhou, Zhangyang Wang
Other
890 stars 198 forks source link

Problems about trainig on custom datasets #99

Closed 1chizhang closed 2 years ago

1chizhang commented 2 years ago

When i train enlightengan on other datasets(sid datasets to rgb),i meet a problem below. pictures are 4256*2848,i'm using an 3090 gpu.Would you please help me?

1
yifanjiang19 commented 2 years ago

Directly replace the original data with yours should not occur any bugs

1chizhang commented 2 years ago

Thank you for your reply,i changed num_workers to 0,and there was no problem