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
918 stars 200 forks source link

A question about the training #100

Closed xinchenduobian closed 3 years ago

xinchenduobian commented 3 years ago

May I ask which directory the training data set is placed in? There are only detailed steps about the test script in the readme. Thank you very much

yifanjiang19 commented 3 years ago

../final_dataset/trainA and ../final_dataset/trainB

xinchenduobian commented 3 years ago

Thank you very much, but when I run the training script and run the command: python scripts/script.py --train, the following problems still occur. I tried to delete all the files in the pycache folder, but it didn't work. 截图_2021-08-23_11-20-11

yifanjiang19 commented 3 years ago

It seems to be a bug with yaml. I guess you can simply delete it as yaml is actually not used

xinchenduobian commented 3 years ago

Yes, just comment out the training script: #config = get_config(opt.config). Thank you very much.