search

The-Learning-And-Vision-Atelier-LAVA / DASR

[CVPR 2021] Unsupervised Degradation Representation Learning for Blind Super-Resolution
MIT License
388 stars 50 forks source link

想问一下训练时显存占用问题 #44

Closed lbc12345 closed 2 years ago

lbc12345 commented 2 years ago

您好！非常喜欢您的工作，我自己在两张1080ti上从头train的时候，发现显存占用只有759MB和691MB，但是程序也在正常输出训练过程，请问这正常吗？

LongguangWang commented 2 years ago

Hi @lbc12345, 前100个epoch只在训练网络中的encoder部分，所以显存占用比较小；100epoch之后显存占用就会比较大了。

lbc12345 commented 2 years ago

感谢您的回答，是我没有认真看代码，谢谢您！