MulaiMulai sekarang secara gratis

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

Lihat Kursus

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,
)
Edit dan Jalankan Kode