テンソルを割り当てるときにPythonでResourceExhaustedErrorが発生する

Question

keras-mplecules-master（https://github.com/maxhodak/keras-molecules ）からtrain.pyを実行しようとすると、この問題が発生します。私のグラフィックカードには十分なメモリがないようです。これを解決する方法はありますか？ありがとうございました。 私のグラフィックカードは4GBのRAMを搭載したGTX970です。ありがとう。

エラーメッセージ：

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が少なくて済み、問題は解決されます。