데이터셋을 평균과 표준편차를 기준으로 정규화하고 텐서로 변경하는 전처리를 진행합니다.
코드 7-53 데이터 전처리
mnist_transform = transforms.Compose([
transforms.ToTensor(),
transforms.Normalize((0.5,), (1.0,))
])
GRU 셀에서도 MNIST 데이터셋을 사용합니다. MNIST 데이터셋을 내려받은 후 전처리를 적용합니다.
코드 7-54 데이터셋 내러받기 및 전처리 적용
from torchvision.datasets import MNIST
download_root = '../chap07/MNIST_DATASET'
train_dataset = MNIST(download_root, transform=mnist_transform, train=True, download=True)
valid_dataset = MNIST(download_root, transform=mnist_transform, train=False, download=True)
test_dataset = MNIST(download_root, transform=mnist_transform, train=False, download=True)