本文主要是介绍PyTorch训练模型,内存泄露问题解决,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
引言
- 最近采用PyTorch训练文本检测模型,遇到一个内存泄露问题,眼睁睁地看着内存一点点被占满,触发系统保护机制,被kill掉
可能出现问题地方
loss
求和未加item()
num_workers
过大- 大量使用list转tensor
最终解决方案
这篇关于PyTorch训练模型,内存泄露问题解决的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!