登录社区云,与社区用户共同成长
邀请您加入社区
在上一篇提到,为了解决这个问题我无意间解决了另一个问题,但发现那个问题并不是根本原因,于是我又查阅了一些资料,经过一番尝试后找到了真正原因。 首先网上有一些解决方法,例如:
但是经过尝试,无论添加与否,显存还是在不断增加 最终原因是由于我在训练时将一个中间结果不断地存进了一个列表里,并始终没有清空,最终导致了显存爆炸的问题
最终注释掉这一行成功解决问题,在循环中尽量少用append方法,如果要保存中间结果,利用np.save等方法储存较好。
有“AI”的1024 = 2048,欢迎大家加入2048 AI社区
更多推荐
孙宇晨问AI要不要给5000万美元:AI的答案为什么不能只看结论
轻量开源版 IDEA 来了!
springboot高校心理咨询管理系统07798-计算机课程设计、毕业设计
扫一扫分享内容
为遵守国家网络实名制规定,未绑定将限制内容发布与互动
所有评论(0)