Menjalankan Lightning Trainer
Sekarang setelah Anda mendefinisikan model, Anda siap mengonfigurasi Trainer Lightning untuk menjalankan loop pelatihan. Latihan ini membantu Anda memahami cara mengintegrasikan model, pemuat data (data loader), dan konfigurasi pelatihan—menguasai pengaturan ini penting sebelum berlanjut ke pelatihan yang lebih lanjut.
torch dan lightning.pytorch yang diimpor sebagai pl telah dimuatkan untuk Anda. Sebuah model sudah diinisialisasi menggunakan kelas yang Anda definisikan pada latihan sebelumnya.
Latihan ini adalah bagian dari kursus
Model AI yang Dapat Diskalakan dengan PyTorch Lightning
Petunjuk latihan
- Konfigurasikan
Trainerdengan 3 epoch. - Mulai proses pelatihan dengan
modelyang telah disiapkan.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
from lightning.pytorch import Trainer
# Configure the Trainer with max epochs
trainer = ____(max_epochs=____, devices=1)
# Start training with the predefined model
trainer.____(____, train_dataloader, val_dataloader)