드롭아웃을 사용하면 훈련 시간이 길어지는 단점이 있지만, 모델 성능을 향상하기 위해 상당히 자주 쓰는 방법입니다.
그럼 배치 정규화와 드롭아웃에 대한 파이토치 예제를 알아보겠습니다. 여기에서 사용되는 데이터셋은 파이토치 torchvision.datasets에서 제공하는 FashionMNIST 데이터셋입니다.
먼저 필요한 라이브러리를 호출합니다.
코드 8-1 라이브러리 호출
import torch
import matplotlib.pyplot as plt
import numpy as np
import torchvision
import torchvision.transforms as transforms
import torch.nn as nn
import torch.optim as optim
예제에서 사용할 FashionMNIST 데이터셋을 내려받습니다.
코드 8-2 데이터셋 내려받기
trainset = torchvision.datasets.FashionMNIST(
root='../chap08/data/', train=True,
download=True,
transform=transforms.ToTensor())