[AI/딥러닝/Python-pytorch] CIFAR-10의 이미지 로딩 / Batch Size

↑ 0번째 이미지에 대한 정보값임. 0~49999번째 이미지(총 5만개)가 있음.

 

 

[3, 32, 32]을 [32, 32, 3]으로 바꿔줘야 위와 같이 이미지가 나타남.

 

0번째 이미지는 라벨이 6이었는데 CIFAR-10에서 6번째 인덱스는 개구리인 것을 알 수 있다.

 

 

---

Batch Size를 정해서 돌려보기

- Batch Size 개념

 

그런데 50000을 64로 나누면 나누어 떨어지지 않음. 나머지가 존재하게 되는데 이걸 버릴 것인지 함께 돌릴 것인지는 drop_last로 정함.(False가 디폴트. 버리지 않고 함께 돌림)