MulaiMulai sekarang secara gratis

Mencoba berbagai nilai learning rate

Dalam latihan ini, tujuan Anda adalah menemukan learning rate yang optimal sehingga optimizer dapat menemukan titik minimum dari fungsi non-konveks \(x^{4} + x^{3} - 5x^{2}\) dalam sepuluh langkah.

Anda akan mencoba tiga nilai learning rate yang berbeda. Untuk masalah ini, cobalah nilai learning rate antara 0,001 hingga 0,1.

Anda disediakan fungsi optimize_and_plot() yang menerima learning rate sebagai argumen pertama. Fungsi ini akan menjalankan 10 langkah optimizer SGD dan menampilkan hasilnya.

Latihan ini adalah bagian dari kursus

Pengantar Deep Learning dengan PyTorch

Lihat Kursus

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# Try a first learning rate value
lr0 = ____
optimize_and_plot(lr=lr0)
Edit dan Jalankan Kode