🙈Pythorch稲妻って何?🙉!!

🤔 Pythorch稲妻とは?

Pythorch稲妻も前回コメントしたignitと似ているので、比較的オープンソースライブラリです.igniterはpytorchの公式ライブラリですが、稲妻には韓国語の資料がもっとあるようです.ignitのコアがEngineであるように、稲妻のコアはTrainerとlightningmoduleである.

⚡Lightning module

flightモジュールは、ほとんどの学習アルゴリズムで定義されているクラスです.ライティングモジュールを実装するには、ライティングモジュールクラスを継承する必要があります.

class MyClass(pl.LightningModule):
	def __init__(self):
        super().__init__()
        self.model = 모델
    
    def forward(self, x):
        pass

    def training_step(self, batch, batch_idx):
        pass

    def validation_step(self, batch, batch_idx):
        pass

    def test_step(self, batch, batch_idx):
        pass
    
    def configure_optimizers(self):
        pass	

LightningModule自体はpytorchのnn.moduleを継承したクラスなので、nn.moduleで利用できるものはすべて使えます.

👤 Trainer

Trainerは基本的にオプティマイザstep、backward、logging、分散学習などの部分です.だからプレイヤーは直接使用を修正するのではなく、実現した部分をTrainerから使用します.

まとめ

ユーザーのコードスタイルが似ている利点は非常に大きい.いずれにしても、キエフで他の開発者のコードを見たとき、「この部分はどこですか?」「なにこれ?」このような問題を提起した私にとって、ignitや稲妻のようなオープンソースコードをアクティブにしたいと思っています.私が行ったプロジェクトに稲妻を適用します.