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
583 stars 43 forks source link

卡在enumerate(train_loader) #35

Open lkangelos opened 1 year ago

lkangelos commented 1 year ago

debug的时候enumerate(train_loader)就需要好久,终端运行,直接卡着不动了。。。

Algolzw commented 1 year ago

你好卡在哪个界面你可以提供吗。我们代码是不会直接输出日志的,所有的日志会保存在log文件中

lkangelos commented 1 year ago

你好卡在哪个界面你可以提供吗。我们代码是不会直接输出日志的,所有的日志会保存在log文件中

好的,谢谢回复,发现代码主要时间都在花在enumerate(train_loader)这里。请问有办法改善吗?

Algolzw commented 1 year ago

也许是训练图像太大了导致加载速度慢,你可以预先裁剪一下图像,参考代码:https://github.com/megvii-research/DCLS-SR/blob/master/codes/scripts/extract_subimgs_single.py