摘要: 这是tensorflow 一个经常性错误,错误的原因在于:显卡内存不够。 解决方法就是降低显卡的使用内存,途径有以下几种措施: 1 减少Batch 的大小 2 分析错误的位置,在哪一层出现显卡不够,比如在全连接层出现的,则降低全连接层的维度,把2048改成1042啥的 3 增加pool 层,降低整个 阅读全文
posted @ 2018-06-21 20:34 图米大侠 阅读(18883) 评论(1) 推荐(1) 编辑