Začněte nyníZačněte zdarma

Zvládni metodu init

V PyTorch Lightning je metoda __init__ základním kamenem nastavení tvého LightningModule. Právě zde definuješ vrstvy modelu, parametry a veškerou počáteční konfiguraci před trénováním. Díky čistému oddělení tohoto inicializačního kroku PyTorch Lightning usnadňuje správu i škálování projektů. V tomto cvičení se zaměříš na inicializaci třídy pro svůj klasifikační model.

Toto cvičení je součástí kurzu

Škálovatelné modely AI s PyTorch Lightning

Zobrazit kurz

Pokyny k cvičení

  • Vytvoř třídu s názvem ClassifierModel, která dědí z pl.LightningModule.
  • Inicializuj nadřazenou třídu, aby sis zpřístupnil/a možnosti LightningModule.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

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)
Upravit a spustit kód