1. 학습
    2. /
  2. 강의
    3. /
  3. Hugging Face로 배우는 멀티모달 모델
Connected

연습 문제

CV 파인튜닝: Trainer 구성

이제 데이터셋을 준비하고 사전 학습된 모델을 새 클래스에 맞게 조정했으니, trainer를 구성할 차례예요.

TrainingArguments와 Trainer는 transformers 라이브러리에서 불러왔어요. 모델(model)과 데이터셋(dataset)도 이전에 설정한 대로 로드되어 있어요.

지침

100 XP
  • 학습률을 6e-5로 조정하세요.
  • Trainer 인스턴스에 모델, 학습 데이터, 테스트 데이터를 전달하세요.