search

OpenGVLab / unmasked_teacher

[ICCV2023 Oral] Unmasked Teacher: Towards Training-Efficient Video Foundation Models
https://arxiv.org/abs/2303.16058
MIT License
267 stars 13 forks source link

out of memory #19

Closed JackSparrow3 closed 8 months ago

JackSparrow3 commented 8 months ago

很棒的工作！感谢分享。我在跑单卡训练msrvtt demo时候，GPU显存随着每次迭代都在涨，然后马上爆显存，不知道作者有没有遇到过？

Andy1621 commented 8 months ago

没有遇到过呢

JackSparrow3 commented 8 months ago

没有遇到过呢 貌似是我的版本问题，在训练的时候关闭了checkpoint这个就解决了。