Training Issues
msx-123 opened this issue · comments
First of all, thank you for your excellent work.
Currently I have two questions.
- The training demo you gave has the training in the main_train_psnr.py file, but I don't find this main_train_psnr file in the project, how do I train it?
- I am more concerned about the training configuration and training time, how long did you train the DIV2K dataset with classic super resolution using what configuration of graphics card?
Looking forward to your reply, thanks!
If you meet I/O bottleneck, preparing the dataset as lmdb (https://github.com/cszn/KAIR/blob/master/scripts/data_preparation/create_lmdb.py) or extracting them as small patches (https://github.com/cszn/KAIR/blob/master/scripts/data_preparation/extract_subimages.py) can help a lot. The PSNR performance will the same.
If you meet I/O bottleneck, preparing the dataset as lmdb (https://github.com/cszn/KAIR/blob/master/scripts/data_preparation/create_lmdb.py) or extracting them as small patches (https://github.com/cszn/KAIR/blob/master/scripts/data_preparation/extract_subimages.py) can help a lot. The PSNR performance will the same.
how to use the lmdb file?