모델에 적용될 변수 값 및 옵티마이저와 손실 함수를 설정합니다.
코드 7-59 옵티마이저와 손실 함수 설정
input_dim = 28
hidden_dim = 128
layer_dim = 1
output_dim = 10
model = GRUModel(input_dim, hidden_dim, layer_dim, output_dim)
if torch.cuda.is_available():
model.cuda()
criterion = nn.CrossEntropyLoss()
learning_rate = 0.1
optimizer = torch.optim.SGD(model.parameters(), lr=learning_rate)
훈련 데이터셋을 이용하여 모델을 학습시키고 검증 데이터셋을 이용하여 모델 성능을 측정합니다. 이 부분 역시 LSTM 셀과 동일하므로 설명은 생략합니다.