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 exercício faz parte do curso
Modelos de IA escaláveis com PyTorch Lightning
Exercício interativo prático
Transforme a teoria em ação com um de nossos exercícios interativos
