神經網絡佔用內存(顯存)的計算

所佔用內存 KB = 參數x4 / 1024
所佔用內存 MB = 參數x4 / 1024 / 1024

比如:某網絡權重參數量106073,那麼他佔用的內存是106073x4/1024=414.34kb。
乘4是因爲float32佔用4字節。

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