ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[682112,156]

运行出错结果如下:



出错原因主要和自己的算法有关,在图像识别的程序中,我的filters的数量分配如下:


Filters的数量越多,同一张图片经过这些filters提取特征后占用显卡内存数量越大,经过这两层众多数量的filter卷积后,占用的内存极大。另外:


图片经过这些filters提取特征后缓存在显卡内存中,经过flaten展平,在经过一层包含1024个神经元的神经网络学习,每一批(batch)数据经过整个算法处理的内容都缓存在显卡中,可想而知内存占用非常大。最后还和batchsize有关,显卡是按块(batch)一批一批进入显卡训练,因此batch越大,占用内存越大。经过调整后的算法如下:

把filter的数量,batchsize以及dense中的神经元数量改小后算法就能跑了,运行如下:




评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值