最佳化器(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,
)