Začněte nyníZačněte zdarma

Fine-tuning CV: konfigurace tréneru

Teď, když máš připravený dataset a předtrénovaný model upravený pro nové třídy, je čas nakonfigurovat trenér.

TrainingArguments a Trainer byly načteny z knihovny transformers. Model (model) a dataset (dataset) jsou načteny tak, jak jsi je dříve nastavil/a.

Toto cvičení je součástí kurzu

Multi-Modal Models with Hugging Face

Zobrazit kurz

Pokyny k cvičení

  • Nastav learning rate na 6e-5.
  • Předej instanci Trainer model, trénovací data a testovací data.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

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,
)
Upravit a spustit kód