【Deep learning】使用tensorflow mnist 代码 内存崩溃

在tensorflow 官方给出的mnist代码中,测试模型准确率部分在显存小的GPU上是无法运行的,会出现out of memory的问题。

原因是, test.images的数量太大(没记错应该是10000)

test_acc=accuracy.eval(feed_dict={x: mnist.test.images, y_: mnist.test.labels, keep_prob: 1.0})
在前向计算的时候,计算需要巨大资源,你的内存无法支撑起来。


-------------------------------------------

依然是使用batch 进行测试,分成10份

for i in range(10):
    batch = mnist.test.next_batch(1000)
    accuResult = accuracy.eval(feed_dict={x:batch[0],y_:batch[1],keep_prob:1.0})
这样就解决问题了。


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值