Menyiapkan Trainer

Model Anda akan mengganti frasa kompleks dengan parafrasa yang lebih sederhana dan mudah diakses untuk meningkatkan keterbacaan terjemahan. Misalnya, model dapat menyederhanakan frasa asli "The inclement weather conditions precipitated the postponement of the outdoor event" menjadi "The bad weather caused the outdoor event to be delayed." Bangun Trainer untuk menyiapkan pelatihan layanan penerjemahan bahasa Anda! Latihan ini akan memerlukan waktu untuk berjalan saat memanggil trainer.train().

Beberapa data telah dimuat sebelumnya:

model adalah model Transformer
dataset berisi himpunan data MRPC berupa parafrasa kalimat
Fungsi compute_metrics mengembalikan akurasi dan skor F1
Anda telah mendefinisikan training_args pada latihan sebelumnya

Latihan ini merupakan bagian dari kursus

Pelatihan Model AI Efisien dengan PyTorch

Lihat Kursus

Instruksi latihan

Berikan model ke kelas Trainer().
Masukkan argumen pelatihan ke dalam kelas Trainer().
Berikan sebuah fungsi untuk menghitung metrik ke kelas Trainer().
Cetak perangkat yang dipilih oleh trainer.

Latihan interaktif langsung praktik

Cobalah latihan ini dengan melengkapi kode contoh ini.

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.____)

Edit dan Jalankan Kode