PyTorch MNIST Data Preprocess

Posted Leave a commentPosted in PyTorch

PyTorch 프레임워크를 사용해 Tensorflow때와 같이 MNIST 예제를 통해 이미지 데이터를 처리할 것이다. 라이브러리 불러오기 Batch Size 정하기 Data Loader 부르기 첫번재 iteration에서 나오는 데이터 확인 파이토치는 DataLoader를 불러 model에 넣음 데이터 시각화 자원수 줄이기 label 변수를 확인해보면 1의 값이 있다 이미지 띄우기