최적화 도구(Optimizers)
이제 모델 학습에 사용할 수 있는 다양한 최적화 도구를 살펴보겠습니다.
train_model(optimizer, net, num_epochs)라는 사용자 정의 함수가 준비되어 있어요. 이 함수는 최적화 도구, 모델, 에포크 수를 입력으로 받아 학습 루프를 실행하고 마지막에 학습 손실을 출력합니다.
이제 train_model()을 사용해 서로 다른 최적화 도구로 짧게 몇 번 학습해 보고 결과를 비교해 볼까요?
이 연습은 강의의 일부입니다
PyTorch로 배우는 Intermediate Deep Learning
실습형 인터랙티브 연습
이 예제를 이 샘플 코드를 완성하여 풀어보세요.
import torch.optim as optim
net = Net()
# Define the SGD optimizer
optimizer = optim.____(net.parameters(), lr=0.001)
train_model(
optimizer=optimizer,
net=net,
num_epochs=10,
)