Pytorch mnist

MNIST로 torch 연습

필요한 모듈 import

Cuda가 가능하면 cuda 아니면 cpu 설정

파라미터 설정

MNIST 데이터 불러오기

첫 번째 데이터 확인

미니 배치 구성

CNN Class에 학습 모델 수립

model, loss, optim 선언

Train data 학습

Test data 학습 및 accuracy 확인