文章出處
文章列表
@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要處理的所有圖片(測試階段。)
文章列表
全站熱搜