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 merupakan bagian dari kursus
Model AI yang Dapat Diskalakan dengan PyTorch Lightning
Instruksi latihan
- Buat pengoptimal Adam menggunakan parameter model, dengan laju pembelajaran
1e-3.
Latihan interaktif langsung praktik
Cobalah latihan ini dengan melengkapi kode contoh ini.
import torch
def configure_optimizers(self):
# Create an Adam optimizer for model parameters
optimizer = ____
return optimizer