LoslegenKostenlos starten

Lebenslauf optimieren: Trainer-Einstellungen anpassen

Nachdem du den Datensatz vorbereitet und ein vortrainiertes Modell an die neuen Klassen angepasst hast, kannst du jetzt deinen Trainer konfigurieren.

Die Bibliotheken „ TrainingArguments “ und „ Trainer “ wurden aus der Bibliothek „ transformers “ geladen. Das Modell (model) und der Datensatz (dataset) wurden so geladen, wie du sie vorher eingerichtet hast.

Diese Übung ist Teil des Kurses

<Kurs>Multimodale Modelle mit Hugging Face</Kurs>
Kurs ansehen

Übungsanweisungen

  • Passe die Lernrate auf „ 6e-5 “ an.
  • Stell das Modell, die Trainingsdaten und die Testdaten für die Instanz „ Trainer “ bereit.

Interaktive praktische Übung

Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.

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,
)
Code bearbeiten und ausführen