Trainer ile karma hassasiyetli eğitim
Karma hassasiyetli eğitim için eğitim arayüzünü basitleştirmek istiyorsun. Accelerator ile eğitim döngülerine alternatif olarak, modelinin özel eğitim döngülerine ihtiyaç duymaması nedeniyle Trainer kullanmaya karar verdin. Trainer'ı karma hassasiyetli eğitim kullanacak şekilde ayarla! trainer.train() çağrısı nedeniyle bu egzersizin çalışması biraz zaman alacaktır.
Bu egzersiz, kursun bir parçasıdır
PyTorch ile Verimli AI Model Eğitimi
Egzersiz talimatları
Traineriçin karma hassasiyetli eğitimi etkinleştir.
Uygulamalı etkileşimli egzersiz
Bu egzersizi bu örnek kodu tamamlayarak deneyin.
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()