python - 当我更改优化器时，Tensorflow 内存不足

Question

当我在训练自己的 tensorflow 代码时，出现了以下问题。我第一次使用tf.train.Gradientdescentoptimizer()优化我的神经网络时，一切正常，但是在我将优化器更改为其他优化器后tf.train.adamoptimizer()，tensorflow 显示： 在分配张量 withshapt[50176, 4096] 时出现 OOM。

所以我试图减少我的batch_size，但即使我改变batch_size了batch_size = 1，它仍然无法修复。

当我运行这个程序时，我没有运行任何其他程序。

我的电脑内存是8GB，我的gpu内存是6GB。

tensorflow的版本是tensorflow-gpu1.4，操作系统是Ubuntu 16.04 LTS。

因此，如果有人遇到过同样的问题，或者有人对此问题有一些建议。非常感谢你。

我不是以英语为母语的人，如果我的描述有任何错误，我很抱歉。

标签： pythontensorflow

Answer 1

如果增加批量大小会发生什么？