解决TensorFlow-gpu版本训练模型时内存不够的问题

解决TensorFlow-gpu版本训练模型时内存不够的问题

在训练模型的过程中发现自己的内存不够,选择了降低图片精度的方式,但是这样会降低训练的准确率的结果;
辗转多次方法,看了许多大佬的博客与方案后,发现了我们可以通过降低batch_size,增加每个epoch的steps的方法,从而达到了不用压缩图片的目的,因为我们的显卡一次装不下这么多的东西,那么减小batch_size,直到显卡可以一批次装下这么多东西即可。

你可能感兴趣的:(机器学习,深度学习)