BaşlayınÜcretsiz başlayın

CV ince ayarı: trainer yapılandırması

Artık veri kümesini hazırlayıp önceden eğitilmiş bir modeli yeni sınıflara uyarladığına göre, sırada trainer'ını yapılandırmak var.

TrainingArguments ve Trainer, transformers kütüphanesinden yüklendi. Model (model) ve veri kümesi (dataset), daha önce yaptığın şekilde yüklendi.

Bu egzersiz, kursun bir parçasıdır

Hugging Face ile Multi-Modal Modeller

Kursa Göz Atın

Egzersiz talimatları

  • Öğrenme oranını 6e-5 olarak ayarla.
  • Trainer örneğine modeli, eğitim verilerini ve test verilerini ver.

Uygulamalı etkileşimli egzersiz

Bu egzersizi bu örnek kodu tamamlayarak deneyin.

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,
)
Kodu Düzenle ve Çalıştır