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

Metoda forward k dokonalosti

Po nastavení vrstev v metodě __init__ určuje metoda forward, jak data těmito vrstvami proudí. V PyTorch Lightning toto oddělení zajišťuje přehlednost a snadnou údržbu kódu. Strukturu konstruktoru už znáš – teď se soustřeď na forward pass a zajisti, aby byla klasifikační logika jasná a připravená pro trénování. Vrstvy v __init__ jsou už definovány za tebe, takže se můžeš plně věnovat průchodu dat sítí.

lightning.pytorch a torch.nn jsou již naimportovány jako pl a nn.

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

Škálovatelné modely AI s PyTorch Lightning

Zobrazit kurz

Pokyny k cvičení

  • Implementuj metodu forward uvnitř třídy ClassifierModel.
  • Za skrytou vrstvou aplikuj ReLU aktivaci.

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

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

class ClassifierModel(pl.LightningModule):
  
    def __init__(self, input_dim, hidden_dim, output_dim):
        super().__init__()
        self.hidden = nn.Linear(input_dim, hidden_dim)
        self.output = nn.Linear(hidden_dim, output_dim)
        
    # Define forward method
    def ____(self, ____):
        # Complete the forward pass
        x = self.hidden(x)
        x = ____(x)
        x = self.output(x)
        return x
Upravit a spustit kód