Training code for custom dataset

megvii-research / TLC

Test-time Local Converter

Other

228 stars 9 forks source link

Training code for custom dataset #9

Open cyprian opened 2 years ago

cyprian commented 2 years ago

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 commented 2 years ago

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.