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
580 stars 42 forks source link

迭代问题 #80

Open wjkbigface opened 7 months ago

wjkbigface commented 7 months ago

你好，我想我想问一下迭代5000次之后就不会继续训练吗，我训练的迭代了5000次之后，就一直没动，看GPU使用率还是在190%左右，这是为什么呢

wjkbigface commented 7 months ago

说错了90%左右

Algolzw commented 7 months ago

可能是在测试中，如果你测试集比较多的话就会很慢。建议建一个小的验证集（比如10-20个图像）用于验证哈。

wjkbigface commented 7 months ago

测试集170多张图片，大概一个小时还没有反应，也应该是测试集太多的问题吗

Algolzw commented 7 months ago

多大的图像呢，你可以在validation时print一些内容，比如图像name，path之类的看下是否正在验证中。