Well-structured training code with PyTorch Lightning.
What it does
Guides the creation of organized neural network training code with reusable data pipelines and scalable training configurations.
When to use it
Use when building, training, or deploying neural networks with PyTorch Lightning, including data pipelines and distributed training.
How to use it
It changes how the agent structures PyTorch model code, separating initialization, training, validation, testing, prediction, and optimizer configuration.
What you provide
- An existing project
What you need · 3
The executable training templates use Python; no runtime version is specified.
The model template imports lightning and subclasses its LightningModule.
The model template imports torch for neural network layers, loss functions, and optimization.
Access · 0
No additional actions listed.