Optimizer
Sekarang saatnya mengeksplorasi berbagai optimizer yang dapat Anda gunakan untuk melatih model Anda.
Fungsi kustom bernama train_model(optimizer, net, num_epochs) telah disiapkan untuk Anda. Fungsi ini menerima optimizer, model, dan jumlah epoch sebagai masukan, menjalankan loop pelatihan, dan mencetak loss pelatihan di akhir.
Mari gunakan train_model() untuk menjalankan beberapa pelatihan singkat dengan optimizer yang berbeda dan membandingkan hasilnya!
Latihan ini merupakan bagian dari kursus
Deep Learning Lanjutan dengan PyTorch
Latihan interaktif langsung praktik
Cobalah latihan ini dengan melengkapi kode contoh ini.
import torch.optim as optim
net = Net()
# Define the SGD optimizer
optimizer = optim.____(net.parameters(), lr=0.001)
train_model(
optimizer=optimizer,
net=net,
num_epochs=10,
)