設定最佳化器
現在我們已經有訓練邏輯，接下來需要指定如何最佳化模型參數。
在這個練習中，你要完成用於影像分類任務的 PyTorch Lightning 模組裡的 configure_optimizers 方法。你的目標是設定一個會在訓練期間更新模型參數的最佳化器。為了完成這個目標，請使用 Adam 最佳化器，學習率為 1e-3。
本練習屬於課程
Scalable AI Models with PyTorch Lightning
練習說明
- 使用模型的參數建立 Adam 最佳化器，並將學習率設為
1e-3。
動手互動練習
試著完成這個範例程式碼，體驗一下這個練習。
import torch
def configure_optimizers(self):
# Create an Adam optimizer for model parameters
optimizer = ____
return optimizer