在使用Pytorch訓練圖像分類model時，常常明明不應該GPU out of memory，但卻發生了
最近發現其中一項問題發生在我在做test的時候一次使用了10000張圖片輸入到model內進行預測，這似乎會使model中的batchnorm layer分配很多GPU memory
(繼續閱讀...)
文章標籤

Jcoi 發表在 痞客邦 留言(0) 人氣(166)