MNIST 데이터셋
MNIST : handwritten digits dataset
Train set
•
(60000 samples)
Test set
•
(10000 samples)
torchvision 패키지
•
인기있는 datasets, model architectures, common image transformations 들을 제공한다
Epoch / Batch size / Iteration
•
one epoch = training set 전체가 한번 학습에 사용이 되면, 한 회 폭이 돌았다
•
batch size = 60000만장을 한번에 학습에 사용하면 좋겠지만 한계가 있다. 그래서 잘라서 사용할건데, 자르는 크기를 몇으로 할 것이냐 → 1 batch의 크기 = 100
•
number of iterations = 배치를 몇 번 학습에 사용을 했냐
ex) 1000개의 트레이닝 set이 있는데, 배치사이즈는 500 ⇒ 2 batch, 2 iterations을 학습에 사용하게 되면 한 epoch이 끝났다고 보면 된다