关于到某一iter显存爆炸的问题

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

关于到某一iter显存爆炸的问题 #107

Closed SueZhang2000 closed 1 month ago

SueZhang2000 commented 1 month ago

作者您好！想询问一下跑到11200iter时程序会卡死的情况是什么原因呢？已经连续两次出现这种情况了，是训练集输入图像尺寸过大的原因吗？若能回复不胜感激！祝学长多中paper科研顺利！

Algolzw commented 1 month ago

你可以查看一下显存和内存使用情况呢？如果是内存使用过大，可以尝试使用更小的patch size以及把dataloader中的pin_memory设为False。

SueZhang2000 commented 1 month ago

您好！是系统内存的问题，已解决，非常感谢！