init メソッドを使いこなす
PyTorch Lightning において、__init__ メソッドは LightningModule の設定の中核を担います。ここでモデルの層やパラメータ、学習前の初期設定を定義します。この設定ステップを明確に分離することで、PyTorch Lightning はプロジェクトの保守とスケーリングを容易にします。この演習では、分類モデル用のクラスを初期化することに取り組みます。
この演習はコースの一部です
PyTorch Lightning によるスケーラブルな AI モデルの構築
演習の手順
pl.LightningModuleを継承するClassifierModelという名前のクラスを作成してください。- 親クラスを初期化して、LightningModule の機能を活用できるようにしてください。
実践的なインタラクティブ演習
このサンプルコードを完成させて、この演習に挑戦してみましょう。
import lightning.pytorch as pl
import torch.nn as nn
# Create the class
class ClassifierModel(____):
# Create init method
def __init__(self, input_dim, output_dim):
____
self.classifier = nn.Linear(input_dim, output_dim)