LoslegenKostenlos loslegen

Die init-Methode richtig anwenden

In PyTorch Lightning ist die Methode „ __init__ “ das Herzstück deiner „ LightningModule “-Konfiguration. Hier legst du vor dem Training die Modellschichten, Parameter und alle Anfangseinstellungen fest. Durch die klare Trennung dieses Einrichtungsschritts macht PyTorch Lightning die Wartung und Skalierung deiner Projekte einfacher. In dieser Übung konzentrierst du dich auf die Initialisierung der Klasse für dein Klassifizierungsmodell.

Diese Übung ist Teil des Kurses

Skalierbare KI-Modelle mit PyTorch Lightning

Kurs anzeigen

Anleitung zur Übung

  • Erstell eine Klasse namens „ ClassifierModel ”, die von „ pl.LightningModule ” erbt.
  • Initialisiere die übergeordnete Klasse, um die Funktionen von LightningModule zu nutzen.

Interaktive Übung

Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.

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)
Code bearbeiten und ausführen