Training code for custom dataset

[cyprian]

Thank you for sharing your research!
Will you be able to provide training code so that we can train it on a custom dataset?

[achusky]

Thanks for your interests.
Our TLC is a test-time method that can be applied to already trained models. In other words, TLC is decoupled from the training so that you can use the training process from original papers and use TLC only at test-time. For the training code, you can follow the official implementation: HINet, MPRNet, Restormer.