De init-methode onder de knie krijgen
In PyTorch Lightning is de __init__-methode het hart van de setup van je LightningModule. Hier definieer je de modellagen, parameters en initiële configuratie vóór het trainen. Door deze setup stap duidelijk te scheiden, maakt PyTorch Lightning het eenvoudiger om je projecten te onderhouden en op te schalen. In deze oefening richt je je op het initialiseren van de klasse voor je classificatiemodel.
Deze oefening maakt deel uit van de cursus
Schaalbare AI-modellen met PyTorch Lightning
Oefeninstructies
- Maak een klasse met de naam
ClassifierModeldie erft vanpl.LightningModule. - Initialiseer de bovenliggende klasse om de mogelijkheden van LightningModule te benutten.
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in te vullen.
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)