PyTorch Lightning – Automatic Batch Size Finder

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.

To learn more about Lightning, please visit the official website:
👉 Read our docs:
👉 Github:
👉 Run on Grid:

Get social:
👉 Join our slack:
👉 Twitter:

Source of this PyTorch Lightning Video

AI video(s) you might be interested in …