About the All-in-one performance about promptir HOT 2 CLOSED

Hello,
The results reported in the paper and the model checkpoint released are generated by the model trained for 120 epochs, using a batch size of 8.
The above results seem to be reported after 50 epochs. Please try to train it longer until 120 epochs. I think the performance on dehazing and denoising would improve with more training.
We had observed that batch size of 8 per GPU i.e a total batch size of 32 provided the best result.

from promptir.

Hello, The results reported in the paper and the model checkpoint released are generated by the model trained for 120 epochs, using a batch size of 8. The above results seem to be reported after 50 epochs. Please try to train it longer until 120 epochs. I think the performance on dehazing and denoising would improve with more training. We had observed that batch size of 8 per GPU i.e a total batch size of 32 provided the best result.

oh, thanks. And I also want to know how to get this visualization.

I wonder which of the three prompt generation modules you used to visualize the code？（Since there are 3 prompt components in the model）

from promptir.