Menggunakan optimizer PyTorch
Sebelumnya, Anda memperbarui bobot jaringan secara manual, sehingga mendapatkan pemahaman tentang cara kerja pelatihan di balik layar. Namun, metode ini tidak dapat diskalakan untuk jaringan dalam (deep) dengan banyak lapisan.
Untungnya, PyTorch menyediakan optimizer SGD, yang mengotomatiskan proses ini secara efisien hanya dalam beberapa baris kode. Sekarang, Anda akan melengkapi loop pelatihan dengan memperbarui bobot menggunakan optimizer PyTorch.
Sebuah neural network telah dibuat dan disediakan sebagai variabel model. Model ini digunakan untuk menjalankan forward pass dan membuat tensor prediksi pred. Tensor one-hot encoded bernama target dan fungsi loss cross entropy disimpan sebagai criterion.
torch.optim sebagai optim, dan torch.nn sebagai nn telah dimuat untuk Anda.
Latihan ini adalah bagian dari kursus
Pengantar Deep Learning dengan PyTorch
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# Create the optimizer
optimizer = ____