Zacznij terazZacznij za darmo

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.

To ćwiczenie jest częścią kursu

Skalowalne modele AI z PyTorch Lightning

Zobacz kurs

Instrukcje do ćwiczenia

  • 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.

Interaktywne ćwiczenie praktyczne

Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.

import lightning.pytorch as pl
import torch.nn as nn

# Create the class
class ClassifierModel(____):
    # Create init method
    def __init__(self, input_dim, output_dim):
        ____
        self.classifier = nn.Linear(input_dim, output_dim)
Edytuj i uruchom kod