我更改优化器时Tensorflow内存不足

Question

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

因此，我尝试减小batch_size，但是即使我将batch_size更改为batch_size = 1，它仍然无法修复。

  

运行该程序时，我没有运行任何其他程序。

     

我的计算机的内存为8GB，而我的GPU的内存为6GB。

     

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

因此，如果遇到相同问题的任何人或对这个问题有任何建议的人。提前非常感谢您。

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

Answer 1

如果增加批量大小会怎样？