[ML] Epoch, Iteration, Batch Size 개념

1. Batch Size

• 모델 학습 중 parameter를 업데이트할 때 사용할 데이터의 개수
• Batch size만큼의 데이터를 활용해 loss function을 계산한 후 parameter update

2. Iteration (step)

• 전체 데이터셋에 대한 Batch의 개수

3. Epoch

• 전체 데이터셋을 학습한 횟수
• Training Data를 모두 사용했을 때의 횟수

4. 정리

• Dataset의 개수 : 2000
• Batch Size : 200
• Iteration : 10
  
  
  
  
  
  
  

Reference

https://wikidocs.net/55580
https://heytech.tistory.com/386