Ubuntu系统下,训练Tensorflow - GAN网络资源耗尽ResourceExhaustedError1的问题

1、tensorflow训练网络模型出现资源耗尽的问题

输入的训练代码为:

$ python main.py --phase=train --data_dir=./data/CK+ --snapshot=5 --n_label=8 --image_size=256 --batch_size=16 --epoch=100

其中,主要的问题在输入训练图片的大小。此处为256*256*3。训练到%17时,


突然出现的资源耗尽问题提示如下。

2、解决办法

在网上查了一下。据说这是一个很玄的问题,遇上了是你的“运气”。

(1)我用了网上的一些方法,没有解决。一气之下,想把服务器关掉,以为是服务器资源不够,关机释放一下。但考虑到师姐还在算着东西,就只有把自己的电脑重启了一次。发现,再运行就ok了。不知道是不是运气好。

(2)科学的方法为把模型最简化。最后我将迭代的次数减少到20,图片的大小缩小到64*64*3,保存模型的间隔增大为20。

$ python main.py --phase=train --data_dir=./data/CK+ --snapshot=20 --n_label=8 --image_size=64 --batch_size=16 --epoch=120
最后模型得以顺利训练。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值