Seznámení s LightningModule
Čeká tě tvorba tvého prvního LightningModule! V tomto praktickém cvičení sestavíš základní strukturu klasifikačního workflow. Definuješ lineární vrstvu, protáhneš data metodou forward a vypočítáš ztrátu v trénovacím kroku. Tahle přehledná struktura ti dá pevný základ pro experimenty s vlastními modely.
Knihovny torch a lightning.pytorch, importovaná jako pl, jsou už předem načtené.
Toto cvičení je součástí kurzu
Škálovatelné modely AI s PyTorch Lightning
Pokyny k cvičení
- Definuj třídu
LightModel, která dědí zpl.LightningModule. - Definuj lineární vrstvu pro transformaci vstupu – předpokládej 16 vstupních příznaků a 10 výstupních tříd.
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
# Define the model class
class LightModel(____):
# Define a linear layer to transform your input
def __init__(self):
super().__init__()
self.layer = ____
def forward(self, x):
return self.layer(x)
def training_step(self, batch, batch_idx):
x, y = batch
logits = self(x)
loss = torch.nn.functional.cross_entropy(logits, y)
return loss