关于高版本torchvision+cuda无法复现出文中结果

[lolccsdz]

您的模型结果很优秀，所以想用作baseline。但是出现了一个问题，在使用torch==1.8.1+cu111 torchvision==0.9.1+cu111可以复现出文中的结果，当使用较高版本torch==2.1.1 torchvision==0.16.1+cuda118时（其他包无变化），PSNR从开始训练（epoch=0）到最后测试结果始终降了2个点（下面是高版本cuda训练测试结果的图片）。所以想请教探讨一下可能导致问题出现的原因？十分感谢。