caffe內存需要如何計算？

Question

當你想訓練一個網，你會得到像日誌信息：

Memory required for data: 493376512 

你如何解釋多少？它是以字節爲單位的嗎？

Answer 1

要回答你的直接問題，圖層的內存使用量以字節爲單位給出，而不是以位表示。總體而言，所報告的內存是可在初始化時計算的所有層所需的內存。參數和緩衝區空間是根據需要分配的，因此將它們彙總並報告是沒有用的 - 由於序列化重用，其中一些可能會佔用較少的總內存。

正如一個答案所說，確定最大內存使用量的最有效方法是運行正向通過並獲取其高水位標記。

Answer 2

見shelhamer的評論： https://github.com/BVLC/caffe/issues/2387#issuecomment-97910200

這是在層的輸出感 所有頂級斑點或「數據」所需的內存。它排除了圖層中的差異，參數和任何中間斑點。 @jeffdonahue你還記得 這個號碼的動機嗎？

而且longjon：

由於層分別負責分配參數 和緩衝液，也沒有辦法，一般知道的 所需的內存總量，除了通過運行一個向前傳球。