PyTorch Lightning DocumentationΒΆ
PyTorch Lightning is the deep learning framework for professional AI researchers and machine learning engineers who need maximal flexibility without sacrificing performance at scale. Lightning organizes PyTorch code to remove boilerplate and unlock Trainer features like multi-GPU, mixed-precision, gradient accumulation, and more.
pip install lightning
import lightning as L
from lightning.pytorch import Trainer
trainer = Trainer(max_epochs=100, accelerator="auto")
trainer.fit(model, train_dataloader, val_dataloader)
Getting Started
User Guide
Reference