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

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