Error:OOM when allocating tensor with shape[......]

問題描述: 訓練模型時,遇到: 原因: 內存不夠,OOM即ran out of memory 可以使用top命令動態監聽運行時間非常短的程序的內存使用情況:參考 啓用2個session,一個session運行top命令,另一個session運行指定的程序,top命令會輸出與這個命令相關的資源使用情況。 解決方法: 最直觀的方式就是減小batch_size或者hidden_layer中的單元數 減少
相關文章
相關標籤/搜索