基于深度学习的超分辨率重建pytorch程序训练不起来

#超分辨率 #深度学习 #Transformer #pytorch #IPT
超分辨率重建问题,使用的是Transformer框架,但是好像训练不起来,PSNR显示为nan,epoch也显示只有1,请问是什么问题呢

img

这里的意思是在第1个epoch出现了PSNR为无穷大(Nan)的情况!建议对forward里面的每一步的变量的值打印看看,是哪一步的计算导致出现了无穷大(Nan)