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 adalah bagian dari kursus
Deep Learning Lanjutan dengan PyTorch
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
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,
)