the result is a little distorted

[Winnie202]

Thanks for sharing your code. The results I get after training have obvious signs of fragmentation. Do you have any improvement suggest?training 60 epochs, but indicators such as psnr did not get much better in the later period , result like this: