XPixelGroup / BasicSR

Open Source Image and Video Restoration Toolbox for Super-resolution, Denoise, Deblurring, etc. Currently, it includes EDSR, RCAN, SRResNet, SRGAN, ESRGAN, EDVR, BasicVSR, SwinIR, ECBSR, etc. Also support StyleGAN2, DFDNet.
https://basicsr.readthedocs.io/en/latest/
Apache License 2.0
6.86k stars 1.2k forks source link

Validation # PSNR: inf #136

Open Marshall-yao opened 5 years ago

Marshall-yao commented 5 years ago

Hi,xintao. I trained the ESRGAN codes with DIV2K800 training set and DIV2K100 validation set. When the training iterates to 125,000 times(epoch 121), the PSNR is inf. After that, the problem of INF appeared intermittently,such as 140000,150000 iterations. Besides, PSNR is stable at around 31.1 when it is not inf.

I think it is abnormal training state when psnr is inf. how can i recognize the noramal training state? To see l_g_pix , l_g_fea,l_g_gan,l_d_real,l_d_fake,D_real,D_fake? What is the noraml range of above values? thanks.

xinntao commented 5 years ago

It should not be Inf.

A log file for your reference. train_190201-171143.log

Marshall-yao commented 5 years ago

Thanks a lot. could you send me a train log about ESRGAN since you gave me the SRGAN log?

xinntao commented 5 years ago

@yaolugithub I cannot find an ESRGAN's log. They are very similar I think.