開始使用免費開始

最佳化器(Optimizers)

現在來探索幾種可用來訓練模型的不同最佳化器。

我們已為你定義了一個自訂函式 train_model(optimizer, net, num_epochs)。它會接收最佳化器、模型與訓練週期數作為輸入,執行訓練迴圈,並在最後印出訓練損失。

讓我們用 train_model() 搭配不同的最佳化器做幾次短訓練,然後比較結果吧!

本練習屬於課程

Intermediate Deep Learning with PyTorch

檢視課程

動手互動練習

試著完成這個範例程式碼,體驗一下這個練習。

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,
)
編輯並執行程式碼