Como usar o LightningModule
O “ LightningModule ” é o ponto central da abordagem do PyTorch Lightning para estruturar projetos de deep learning. Ele centraliza a arquitetura do modelo, as passagens para frente e a lógica de treinamento, deixando a orquestração dos loops de treinamento e a configuração do dispositivo para o Trainer.
Qual afirmação descreve melhor o papel principal do “ LightningModule ” no PyTorch Lightning?
Este exercicio faz parte do curso
Modelos de IA escaláveis com PyTorch Lightning
exercicio interativo prático
Transforme teoria em prática com um dos nossos exercicio interativos
Iniciar exercicio