【caffe】epoch,[batch_size],iteration的含義


@tags caffe 概念

一個epoch表示“大層面上的一次迭代”,也就是指,(假定是訓練階段)處理完所有訓練圖片,叫一個epoch

但是每次訓練圖片可能特別多,內存/顯存塞不下,那么每個epoch內,將圖片分成一小堆一小堆的,每一小堆圖片數量相等,每一小堆就是一個batch(批次)。

因而,一個epoch內,就要處理多個batch。

batch_size表示的是,每個batch內有多少張圖片。

而一個epoch,一共需要分成多少個batch呢?這個batch的數目,就叫做train_iter(訓練階段)或者test_iter(測試階段)

總結:

train_iter * batch_size = 一個epoch要處理的所有圖片(可能因為整除的原因,不嚴格相等,讓左邊略大即可。訓練階段)

或者:
train_iter * batch_size = 一個epoch要處理的所有圖片(測試階段。)


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM