文章出處

@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要處理的所有圖片(測試階段。)

文章列表


不含病毒。www.avast.com
arrow
arrow
    全站熱搜
    創作者介紹
    創作者 大師兄 的頭像
    大師兄

    IT工程師數位筆記本

    大師兄 發表在 痞客邦 留言(0) 人氣()