标题:基于pytorch的python程序。
只看楼主
abc3270560
Rank: 1
等 级:新手上路
帖 子:1
专家分:0
注 册:2019-4-18
 问题点数:0 回复次数:0 
基于pytorch的python程序。
小白最近在跑一个关于GAN的代码,但在这个代码中调用的一个程序中出现了一些问题。总的表现为超出显存。但仔细排查后发现程序每次循环的时候都把它自己定义的模型给送到GPU 了,导致显存激增,且循环完毕也不释放。想求教各路大神怎么解决这个问题。改网络结构的话,以我目前的水平是改不了。希望大神们可以帮忙!
搜索更多相关主题的帖子: python 代码 循环 释放 结构 
2019-04-18 22:07



参与讨论请移步原网站贴子:https://bbs.bccn.net/thread-494243-1-1.html




关于我们 | 广告合作 | 编程中国 | 清除Cookies | TOP | 手机版

编程中国 版权所有,并保留所有权利。
Powered by Discuz, Processed in 0.110278 second(s), 8 queries.
Copyright©2004-2024, BCCN.NET, All Rights Reserved