ПочатиПочніть безкоштовно

Тонке налаштування CV: конфігурація тренера

Тепер, коли ви підготували набір даних і адаптували попередньо натреновану модель до нових класів, час налаштувати ваш тренер.

TrainingArguments і Trainer завантажено з бібліотеки transformers. Модель (model) і набір даних (dataset) завантажено так, як ви їх налаштували раніше.

Ця вправа є частиною курсу

Багатомодальні моделі з Hugging Face

Переглянути курс

Інструкції до вправи

  • Змініть швидкість навчання на 6e-5.
  • Передайте модель, тренувальні дані та тестові дані екземпляру Trainer.

Інтерактивна практична вправа

Спробуйте виконати цю вправу, доповнивши цей зразок коду.

training_args = TrainingArguments(
    output_dir="dataset_finetune",
    # Adjust the learning rate
    ____,
    gradient_accumulation_steps=4,
    num_train_epochs=3,
    push_to_hub=False
)

trainer = Trainer(
    # Provide the model and datasets
    model=____,
    args=training_args,
    data_collator=data_collator,
    train_dataset=____,
    eval_dataset=____,
    processing_class=image_processor,
    compute_metrics=compute_metrics,
)
Редагувати та запускати код