Thank you for the great work!!
Here is my question: I would like to train the Restormer on my own dataset with "Deraining" setup.
However, after about 50000 iterations, the loss is not descending at all, and val psnr almost didn't change since the start of the training.
I've confirmed that the data flow is correct, where input and output image are matched.
Is this because the Restormer needs a longer time to warm up? Or am I doing anything wrong?
By the way, I am using 2 GPU for training, so I changed the yaml file and train.sh accordingly to fit 2 GPU.
I'm looking forward to your reply. Thank you in advance!