| 网站首页 | 业界新闻 | 小组 | 威客 | 人才 | 下载频道 | 博客 | 代码贴 | 在线编程 | 编程论坛
欢迎加入我们,一同切磋技术
用户名:   
 
密 码:  
共有 1002 人关注过本帖
标题:基于pytorch的python程序。
只看楼主 加入收藏
abc3270560
Rank: 1
等 级:新手上路
帖 子:1
专家分:0
注 册:2019-4-18
收藏
 问题点数:0 回复次数:0 
基于pytorch的python程序。
小白最近在跑一个关于GAN的代码,但在这个代码中调用的一个程序中出现了一些问题。总的表现为超出显存。但仔细排查后发现程序每次循环的时候都把它自己定义的模型给送到GPU 了,导致显存激增,且循环完毕也不释放。想求教各路大神怎么解决这个问题。改网络结构的话,以我目前的水平是改不了。希望大神们可以帮忙!
搜索更多相关主题的帖子: python 代码 循环 释放 结构 
2019-04-18 22:07
快速回复:基于pytorch的python程序。
数据加载中...
 
   



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

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