CommencerCommencer gratuitement

Maîtriser la méthode init

Dans PyTorch Lightning, la méthode ` __init__ est au cœur de votre configuration d'LightningModule`. C'est ici que vous définissez les couches du modèle, les paramètres et toute configuration initiale avant l'entraînement. En séparant clairement cette étape de configuration, PyTorch Lightning facilite la maintenance et la mise à l'échelle de vos projets. Dans cet exercice, vous allez vous concentrer sur l'initialisation de la classe pour votre modèle de classification.

Cet exercice fait partie du cours

Modèles d'IA évolutifs avec PyTorch Lightning

Afficher le cours

Instructions

  • Créez une classe nommée « ClassifierModel » qui hérite de « pl.LightningModule ».
  • Initialisez la classe parente afin de tirer parti des fonctionnalités de LightningModule.

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

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)
Modifier et exécuter le code