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

ćwiczenie

Opanuj metodę init

W PyTorch Lightning metoda __init__ to serce konfiguracji twojego LightningModule. To właśnie tutaj definiujesz warstwy modelu, parametry i wszelkie ustawienia początkowe – jeszcze przed uruchomieniem treningu. Dzięki wyraźnemu oddzieleniu tego kroku konfiguracyjnego PyTorch Lightning ułatwia zarządzanie projektem i jego skalowanie. W tym ćwiczeniu skupisz się na inicjalizacji klasy dla swojego modelu klasyfikacji.

Instrukcje

100 XP
  • Utwórz klasę o nazwie ClassifierModel, która dziedziczy po pl.LightningModule.
  • Zainicjalizuj klasę nadrzędną, aby móc korzystać z możliwości LightningModule.