ComenzarEmpieza gratis

Dominar el método init

En PyTorch Lightning, el método « __init__ » es el núcleo de la configuración de « LightningModule ». Es donde se definen las capas del modelo, los parámetros y cualquier configuración inicial antes del entrenamiento. Al separar claramente este paso de configuración, PyTorch Lightning facilita el mantenimiento y la escalabilidad de tus proyectos. En este ejercicio, te centrarás en inicializar la clase para tu modelo de clasificación.

Este ejercicio forma parte del curso

Modelos de IA escalables con PyTorch Lightning

Ver curso

Instrucciones del ejercicio

  • Crea una clase llamada « ClassifierModel » que herede de « pl.LightningModule ».
  • Inicializa la clase principal para aprovechar las capacidades de LightningModule.

Ejercicio interactivo práctico

Prueba este ejercicio completando el código de muestra.

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)
Editar y ejecutar código