search

Algolzw / image-restoration-sde

Image Restoration with Mean-Reverting Stochastic Differential Equations, ICML 2023. Winning solution of the NTIRE 2023 Image Shadow Removal Challenge.
https://algolzw.github.io/ir-sde/index.html
MIT License
558 stars 41 forks source link

关于inpainting任务超参数的问题 #46

Open shouwangzhe200 opened 1 year ago

shouwangzhe200 commented 1 year ago

您好，我尝试使用您提供的代码复现inpainting的训练，但是发现出图效果远差于官方提供的pth。请问一下inpainting的实验大概训练多机才会收敛？超参数应该怎么设置呢？

Algolzw commented 1 year ago

用官方的设置应该就可以，不过一般需要较长时间收敛（差不多40万次迭代后）。

shouwangzhe200 commented 1 year ago

很奇怪，大概1Wstep的时候loss就不怎么下降了，但是确实要20Wstep之后效果才会比较好一点