Padroneggiare il metodo init
In PyTorch Lightning, il metodo __init__ è il cuore della configurazione del tuo LightningModule. Qui definisci i layer del modello, i parametri e qualsiasi impostazione iniziale prima dell'addestramento. Separando chiaramente questo step di setup, PyTorch Lightning rende più semplice mantenere e scalare i tuoi progetti. In questo esercizio ti concentrerai sull'inizializzazione della classe per il tuo modello di classificazione.
Questo esercizio fa parte del corso
Modelli di AI scalabili con PyTorch Lightning
Istruzioni dell'esercizio
- Crea una classe chiamata
ClassifierModelche eredita dapl.LightningModule. - Inizializza la classe padre per sfruttare le funzionalità di LightningModule.
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
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)