Train and debug your PyTorch neural networks.
Guides neural-network training with PyTorch recipes for model selection, optimization, debugging, and controlled experiments.
When to use it
Use when training or fine-tuning neural networks, investigating loss spikes or out-of-memory errors, selecting architectures, or improving GPU throughput.
It changes how the agent selects, trains, tunes, and debugs neural networks in the project.
What you provide
Declares torch version 2.0.0 or newer as a dependency.
The shown training loop uses a CUDA GPU for mixed precision and synchronization.
No additional actions listed.