ResourceExhaustedError - Cupoy

2020/07/18 下午 01:09

電腦視覺深度學習討論版

Alice

觀看數：17

回答數：2

收藏數：0

cvdl

cvdl-d41

老師您好：

在跑Day41_train_yolov3_Sample.ipynb，training的時候，出現了以下錯誤

請問是記憶體不夠的關係嗎？

謝謝

回答列表

• 2020/07/21 下午 10:46
  Jeffrey
  贊同數：0
  不贊同數：0
  留言數：0

  可以先移除 logging, 若是OK, 一般就是空間不足

• 2020/10/01 下午 07:46
  Patrick Ruan
  贊同數：0
  不贊同數：0
  留言數：0

  可以嘗試把 batch size 調小，來解決 OOM (out of memory)

  同學從學 sdg 到 mini batch，除了了解 mini batch 對 gradient descent 的神效之外，也可以了解一下，記憶體資源不大時，減小 mini-batch 是很有幫助的。當然時間就會有所損失了。

  
  

  我過去玩 yolo 時，batch size 調到小到自己都好笑才跑起來，沒有記憶體的窘境