ComeçarComece de graça

Dominando o método init

No PyTorch Lightning, o método ` __init__ é o coração da sua configuração LightningModule `. É onde você define as camadas do modelo, os parâmetros e qualquer configuração inicial antes do treinamento. Ao separar claramente essa etapa de configuração, o PyTorch Lightning facilita a manutenção e a escalabilidade dos seus projetos. Neste exercício, você vai se concentrar em inicializar a classe para o seu modelo de classificação.

Este exercício faz parte do curso

Modelos de IA escaláveis com PyTorch Lightning

Ver curso

Instruções do exercício

  • Crie uma classe chamada “ ClassifierModel ” que herda de “ pl.LightningModule ”.
  • Inicialize a classe pai para aproveitar os recursos do LightningModule.

Exercício interativo prático

Experimente este exercício completando este código de exemplo.

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 e executar o código