I was training on my own dataset for (128x128) --> (512x512) I have attached the train logs : [train.log](https://github.com/user-attachments/files/17504451/train.log) [val.log](https://github.com/user-attachments/files/17504453/val.log) The tensorboard visualizations look like this:  And the losses & psnr aren't improving   Please help me to fix it
I was training on my own dataset for (128x128) --> (512x512)
I have attached the train logs :
train.log
val.log
The tensorboard visualizations look like this:

And the losses & psnr aren't improving


Please help me to fix it