Ree1s / IDM

279 stars 20 forks source link

Missing checkpoint? #1

Closed ppriyank closed 1 year ago

ppriyank commented 1 year ago

Hey,

For training, I'm missing the checkpoint: 'best_psnr_opt.pth'. Will this be uploaded?

ppriyank commented 1 year ago

https://github.com/Ree1s/IDM/blob/main/model/model.py#L201

Adding the following fixed the issue. opt is not present at the start of training.

if self.opt['phase'] == 'train' and os.path.isfile(opt_path):