1. Leren
    2. /
  2. Cursussen
    3. /
  3. Intermediate Deep Learning with PyTorch
Connected

Opdracht

최적화 도구(Optimizers)

이제 모델 학습에 사용할 수 있는 다양한 최적화 도구를 살펴보겠습니다.

train_model(optimizer, net, num_epochs)라는 사용자 정의 함수가 준비되어 있어요. 이 함수는 최적화 도구, 모델, 에포크 수를 입력으로 받아 학습 루프를 실행하고 마지막에 학습 손실을 출력합니다.

이제 train_model()을 사용해 서로 다른 최적화 도구로 짧게 몇 번 학습해 보고 결과를 비교해 볼까요?

Instructies 1/3

undefined XP
  • 1
    • optimizer를 확률적 경사 하강법(Stochastic Gradient Descent)으로 정의하세요.
  • 2
    • 최적화 도구를 Root Mean Square Propagation(RMSprop)으로 정의하고, 첫 번째 인수로 모델의 파라미터를 전달하세요.
  • 3
    • 최적화 도구를 Adaptive Moments Estimation(Adam)으로 정의하고, 학습률을 0.001로 설정하세요.