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

ćwiczenie

Wprowadzenie do LightningModule

Czas zbudować swój pierwszy LightningModule! W tym ćwiczeniu praktycznym skonfigurujesz podstawową strukturę procesu klasyfikacji. Zdefiniujesz warstwę liniową, przeprowadzisz dane przez metodę forward i obliczysz stratę w kroku treningowym. Ta przejrzysta struktura daje solidne podstawy do dalszego eksperymentowania z modelami.

Biblioteki torch oraz lightning.pytorch, zaimportowana jako pl, zostały już wczytane.

Instrukcje

100 XP
  • Zdefiniuj klasę LightModel, która dziedziczy po pl.LightningModule.
  • Zdefiniuj warstwę liniową przekształcającą dane wejściowe, przyjmując 16 cech wejściowych i 10 klas wyjściowych.