데이터로더를 이용하여 전처리된 데이터셋을 메모리로 불러옵니다. 단 메모리로 불러오는 시점은 train_loader, valid_loader, test_loader가 호출되는 시점입니다.
코드 7-55 데이터셋 메모리로 가져오기
batch_size = 64
train_loader = DataLoader(dataset=train_dataset,
batch_size=batch_size,
shuffle=True)
valid_loader = DataLoader(dataset=test_dataset,
batch_size=batch_size,
shuffle=True) ------ 일반적으로 검증과 테스트 용도의 데이터셋은 섞어서 사용하지 않습니다. 예제에서는 다양한 학습을 위해 True로 지정했습니다.
test_loader = DataLoader(dataset=test_dataset,
batch_size=batch_size,
shuffle=True)
필요한 변수에 대한 값을 설정합니다.
코드 7-56 변수 값 설정
batch_size = 100
n_iters = 6000
num_epochs = n_iters / (len(train_dataset) / batch_size)
num_epochs = int(num_epochs)