이 예제는 케라스에 내장되어 제공하는 MNIST 데이터셋을 사용합니다.
MNIST 데이터셋을 내려받은 후 텐서로 변경하고 데이터로더에 전달합니다.
코드 13-2 MNIST 데이터셋을 내려받아 전처리
transform = transforms.Compose([transforms.ToTensor()]) ------ transforms.ToTensor( )는 이미지를 파이토치의 텐서 형태로 변형시켜 줍니다.
train_dataset = datasets.MNIST(
root="../chap13/data", train=True, transform=transform, download=True) ------ MNIST를 내려받은 후 텐서 형태로 변형
test_dataset = datasets.MNIST(
root="../chap13/data", train=False, transform=transform, download=True)
train_loader = DataLoader(
train_dataset, batch_size=128, shuffle=True, num_workers=4, pin_memory=False) ------ ①
test_loader = DataLoader(
test_dataset, batch_size=32, shuffle=False, num_workers=4)