问题描述
我正在尝试在 pytorch 数据加载器中使用经过训练的 ANN1 为 __getitem__
函数中的另一个 ANN2 生成扰动图像。
问题是我在执行几批后让 CUDA 内存不足。通过一些调试,我发现每次迭代时内存都在累积。
我尝试使用 del
,但只释放了非常有限的 GPU 内存。
我想知道数据加载器中是否有任何可能导致此问题的特殊机制?
顺便说一句,当我在 for 循环中仅使用 ANN1 运行数据加载器或仅使用官方数据加载器运行 ANN2 时,模拟工作正常。
提前致谢。
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)