tensorflow錯誤解決:“tensorflow.python.framework.errors_impl.ResourceExhaustedError”

tensorflow錯誤:“tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape[32,256,55,55]”

解決方法:

  • 減少批處理Batch 的大小
  • 降低全連接層的維度
  • 增加池化 層
  • 縮小輸入圖片大小

總之呢,就是顯卡內存不能滿足此次訓練,所以就是要降低所需內存要求。

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章