시작하기무료로 시작하기

Trainer 설정하기

학습 인자를 설정했으니, 미세 조정한 모델을 사용하기 전에 이제 Trainer를 설정해야 해요.

TrainingArgumentsTrainer는 물론 이전에 사용한 model, tokenizer, training_args도 이미 로드되어 있어요.

참고: 이 연습 문제에서는 실행 속도를 높이기 위해 실제 학습이 비활성화되어 있어요.

이 연습은 강의의 일부입니다

Python으로 배우는 LLM 입문

강의 보기

연습 안내

  • Trainer() 객체를 설정하세요.
  • 앞서 정의한 학습 인자와 tokenizer를 할당하세요.
  • 모델을 학습시키세요.

실습형 인터랙티브 연습

이 예제를 이 샘플 코드를 완성하여 풀어보세요.

# Set up the trainer object
trainer = ____(
    model=model,
    # Assign the training arguments and tokenizer
    ____,
    train_dataset=tokenized_training_data,
    eval_dataset=tokenized_test_data,
    ____
)

# Train the model
____
코드 편집 및 실행