1. Nauka
    2. /
  2. Kursy
    3. /
  3. Skalowalne modele AI z PyTorch Lightning
Connected

ćwiczenie

Doskonalenie metody forward

Po zdefiniowaniu warstw w metodzie __init__ to metoda forward decyduje o tym, jak dane przez nie przepływają. W PyTorch Lightning taki podział sprawia, że kod jest przejrzysty i łatwy w utrzymaniu. Wiesz już, jak zbudować konstruktor – czas skupić się na metodzie forward i zadbać o to, by logika klasyfikacji była czytelna i gotowa do trenowania. Warstwy w __init__ są już dla ciebie zdefiniowane, więc możesz skoncentrować się wyłącznie na przepływie danych.

Biblioteki lightning.pytorch i torch.nn zostały już zaimportowane odpowiednio jako pl i nn.

Instrukcje

100 XP
  • Zaimplementuj metodę forward wewnątrz klasy ClassifierModel.
  • Zastosuj aktywację ReLU po warstwie ukrytej.