Mengonfigurasi pengoptimal
Sekarang setelah kita memiliki logika pelatihan, kita perlu menentukan cara mengoptimalkan parameter model.
Dalam latihan ini, Anda akan melengkapi metode configure_optimizers di dalam modul PyTorch Lightning yang digunakan untuk tugas klasifikasi gambar. Tujuan Anda adalah menyiapkan pengoptimal yang akan memperbarui parameter model selama pelatihan. Untuk menyelesaikannya, Anda akan menggunakan pengoptimal Adam dengan laju pembelajaran 1e-3.
Latihan ini adalah bagian dari kursus
Model AI yang Dapat Diskalakan dengan PyTorch Lightning
Petunjuk latihan
- Buat pengoptimal Adam menggunakan parameter model, dengan laju pembelajaran
1e-3.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
import torch
def configure_optimizers(self):
# Create an Adam optimizer for model parameters
optimizer = ____
return optimizer