PyTorch Lightning - Automatic Batch Size Finder

Опубликовано: 21 Февраль 2026
на канале: Lightning AI
12,567
48

PyTorch Lighting is a lightweight PyTorch wrapper for high-performance AI research that reduces the boilerplate without limiting flexibility.
In this series, we are covering all the tricks PyTorch Lightning offers to supercharge your training.

In this video, we give a short intro to Lightning's flag auto_scale_batch_size.

Join our Discord to participate in the discussion:   / discord  
Lightning Website: https://lightning.ai/
GitHub: https://github.com/PyTorchLightning/p...
Follow us on Twitter:   / lightningai  
Read our docs: https://bit.ly/3w5nUhf