Pelatihan presisi campuran dengan Trainer
Anda ingin menyederhanakan antarmuka pelatihan untuk pelatihan presisi campuran. Sebagai alternatif dari loop pelatihan dengan Accelerator, Anda memutuskan untuk menggunakan Trainer karena model Anda tidak memerlukan loop pelatihan kustom. Siapkan Trainer agar menggunakan pelatihan presisi campuran! Latihan ini akan memerlukan waktu untuk berjalan saat memanggil trainer.train().
Latihan ini merupakan bagian dari kursus
Pelatihan Model AI Efisien dengan PyTorch
Instruksi latihan
- Aktifkan pelatihan presisi campuran untuk
Trainer.
Latihan interaktif langsung praktik
Cobalah latihan ini dengan melengkapi kode contoh ini.
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()