Configura Trainer

Il tuo modello sostituirà frasi complesse con parafrasi più semplici e accessibili per migliorare la leggibilità delle traduzioni. Ad esempio, può semplificare la frase originale "The inclement weather conditions precipitated the postponement of the outdoor event" con "The bad weather caused the outdoor event to be delayed." Costruisci il Trainer per preparare l’addestramento del tuo servizio di traduzione! L’esercizio richiederà un po’ di tempo per l’esecuzione con la chiamata a trainer.train().

Alcuni dati sono già stati caricati:

model è un modello Transformer
dataset contiene l’insieme di dati MRPC di parafrasi di frasi
La funzione compute_metrics restituisce accuracy e F1 score
Hai definito training_args in un esercizio precedente

Questo esercizio fa parte del corso

Efficient AI Model Training with PyTorch

Visualizza corso

Istruzioni dell'esercizio

Passa model alla classe Trainer().
Inserisci gli argomenti di training nella classe Trainer().
Passa una funzione per calcolare le metriche alla classe Trainer().
Stampa il dispositivo scelto da trainer.

esercizio interattivo pratico

Prova questo esercizio completando questo codice di esempio.

trainer = Trainer(
    # Pass in the model
    model=____,
    # Input the training arguments
    args=____,
    train_dataset=dataset["train"],
    eval_dataset=dataset["validation"],
    # Pass in a function to compute metrics
    compute_metrics=____,
)

trainer.train()

# Print the device that the trainer chooses
print(trainer.args.____)

Modifica ed esegui il codice