背景
使用tensorflow训练模型时出现报错
2022-01-04 09:30:14.672423: W tensorflow/core/framework/cpu_allocator_impl.cc:81] Allocation of 1294139392 exceeds 10% of free system memory.
原因
内存不足了,不够保存模型
解决方法
1、把placeholder位置放在卷积之前,
2、把batchsize改小
3、把测试集改为批次放入预测
tip:batch_size改小一些。。。肯定运行速度就会慢很多,但只要在你能接受的范围内就行~