bubbliiiing / srgan-pytorch

这是一个srgan-keras的源码,可以用于训练自己的模型。
MIT License
54 stars 10 forks source link

训练时loss值为nan #6

Open LPC1616 opened 1 year ago

LPC1616 commented 1 year ago

Epoch 175/200: 52%|█████▏ | 112/216 [00:36<00:29, 3.55it/s, D_loss=nan, G_PSNR=nan, G_SSIM=nan, G_loss=nan, lr=5.56e-6] 这是为什么呢?怎么解决这个问题?

LPC1616 commented 1 year ago

训练的是自己的数据集

bubbliiiing commented 1 year ago

只是跑飞了吧。已经有的结果应该正常预测