Janspiry / Image-Super-Resolution-via-Iterative-Refinement

Unofficial implementation of Image Super-Resolution via Iterative Refinement by Pytorch
Apache License 2.0
3.54k stars 461 forks source link

About model convergence #133

Open yuan0713 opened 11 months ago

yuan0713 commented 11 months ago

Hi, How can I determine if SR3 model has converged? I'm training a model on FFHQ and using CelebA-HQ for validation, the results for each epoch are not consistent. Like: INFO: <epoch: 38, iter: 650,000> psnr: 2.4686e+01 INFO: <epoch: 39, iter: 660,000> psnr: 2.5092e+01 INFO: <epoch: 39, iter: 670,000> psnr: 2.4681e+01 INFO: <epoch: 40, iter: 680,000> psnr: 2.4613e+01 INFO: <epoch: 41, iter: 690,000> psnr: 2.4460e+01 INFO: <epoch: 41, iter: 700,000> psnr: 2.4871e+01 INFO: <epoch: 42, iter: 710,000> psnr: 2.5073e+01

Should I choose the best results as my final outcome?

Thanks!