some problems

[jiaaihhy]

Since no training code is given, I write my own training program to train Restormer. However, at the beginning of the training, I could only set batchsize to 48 due to the limitation of GPUs memory. However, I found that the loss would hardly decrease when the first 10,000 to 20,000 iteration was carried out, which verified that the PSNR remained unchanged at about 26.2. Is the training relatively slow, or what is the problem? And if prob, I would like to know the upward trend of Val PSNR and the downward trend of loss during your training

[jixiang2016]

Hi, would you mind telling me the number and type of GPUS you have used in your training and the patch size and batch size? Because the memory of GPU consumed by Restormer is amazingly large when I try to train this model. Thanks so much.

[yzcv]

I am also curious about the training information about GPU used. This I think is a bottleneck of many researchers who do not have much computation resources. Hope to know about it and the training details for re-implementation. Thanks.

[swz30]

Hi,

We have released the training codes.