Antrenament cu precizie mixtă folosind Trainer
Vrei să simplifici interfața de antrenament pentru precizia mixtă. Ca alternativă la buclele de antrenament cu Accelerator, ai decis să folosești Trainer, deoarece modelul tău nu necesită bucle de antrenament personalizate. Configurează Trainer pentru antrenament cu precizie mixtă! Exercițiul va dura ceva timp să ruleze, datorită apelului la trainer.train().
Acest exercițiu face parte din cursul
Antrenament eficient al modelelor AI cu PyTorch
Instrucțiuni pentru exercițiu
- Activează antrenamentul cu precizie mixtă pentru
Trainer.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
training_args = TrainingArguments(
output_dir="./results",
evaluation_strategy="epoch",
# Enable mixed precision training
____=____
)
trainer = Trainer(model=model,
args=training_args,
train_dataset=dataset["train"],
eval_dataset=dataset["validation"],
compute_metrics=compute_metrics)
trainer.train()