머신러닝
Epoch, Batch, and Iteration
★ ☆
2022. 6. 2. 15:24
source: https://towardsdatascience.com/epoch-vs-iterations-vs-batch-size-4dfb9c7ce9c9
Epoch vs Batch Size vs Iterations
Know your code…
towardsdatascience.com
Epoch : 전체 데이터 집합이 신경망을 forward하고 backward 한번 씩 하는 것
Batch : 데이터 셋의 부분집합. 대개의 경우에 전체 데이터 셋을 한번에 학습시킬 수 없으므로, 데이터 셋을 부분으로 나누어야 한다. 전체가 100개인 데이터를 한번에 5개씩 학습 시키면 batch size는 5이다.
Iteration: 1 epoch를 끝내기 위해서 반복되는 batch의 횟수.
전체가 100개인 데이터, batch size가 5이면 Iteration은 20이다.
전체가 2000개인 데이터, batch size가 500이면 iteration은 4이다.