1. 학습
    2. /
  2. 강의
    3. /
  3. PyTorch로 AI 모델 효율적으로 학습시키기
Connected

연습 문제

Trainer와 AdamW 활용

Transformer 모델을 사용해 언어 번역을 단순화하는 훈련을 시작하려고 합니다. 첫 번째 단계로, AdamW 옵티마이저를 기준점으로 삼고 Trainer 인터페이스를 활용해 빠르게 설정해 보겠습니다. AdamW optimizer를 사용하도록 Trainer를 구성하세요.

AdamW는 torch.optim에서 미리 임포트되어 있습니다. 다음 훈련 객체들도 미리 로드되어 있습니다: model, training_args, train_dataset, validation_dataset, compute_metrics.

지침

100 XP
  • model의 파라미터를 AdamW optimizer에 전달하세요.
  • optimizer를 Trainer에 전달하세요.