kaggle 深度学习训练内存不够_kaggle内存不够

作者：我家小花儿 | 2024-03-20 02:17:00

踩

kaggle内存不够

在kaggle上训练VGG网络，为了实现一个可视化算法，运用了pytorch中的hook机制。
但是在训练中出现了爆内存的现象，总结一下几种解决办法：

import gc
del variables
gc.collect()
1
2
3

handle = model.register_forward_hook(fn)
handle.remove()
1
2

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/我家小花儿/article/detail/269921