オプティマイザーの設定
訓練ロジックが用意できたので、次はモデルのパラメータを最適化する方法を指定する必要があります。
この演習では、画像分類タスクに使用する PyTorch Lightning モジュール内の configure_optimizers メソッドを完成させます。目標は、訓練中にモデルのパラメータを更新するオプティマイザーを設定することです。これを実現するために、学習率 1e-3 の Adam オプティマイザーを使用します。
この演習はコースの一部です
PyTorch Lightning によるスケーラブルな AI モデルの構築
演習の手順
- モデルのパラメータを使って Adam オプティマイザーを作成し、学習率を
1e-3に設定しましょう。
実践的なインタラクティブ演習
このサンプルコードを完成させて、この演習に挑戦してみましょう。
import torch
def configure_optimizers(self):
# Create an Adam optimizer for model parameters
optimizer = ____
return optimizer