分配张量

Question

当我尝试从keras-mplecules-master（https://github.com/maxhodak/keras-molecules运行train.py时，会出现此问题 ）。看来我的显卡没有足够的内存。有什么方法可以解决这个问题吗？谢谢。 我的显卡是GTX970，配有4GB内存。感谢。

错误讯息：

    ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[600,120,1503]

我读过一个非常类似的问题     Tensorflow Deep MNIST: Resource exhausted: OOM when allocating tensor with shape[10000,32,28,28] 但我仍然不知道应该如何更改我的代码。 我是ubuntu的新手，所以请尽可能具体。谢谢。

Answer 1

问题已经解决了。 这是与RAM使用相关的问题。 它可以通过限制batch_size来解决，因此可以同时使用较少的RAM并解决问题。