Use a pure PyTorch training loop

Enable manual optimization

Gain control of the training loop with manual optimization and LightningModule methods.

advanced

Use a Raw PyTorch Loop

Migrate complex PyTorch projects to Lightning and push bleeding-edge research with the raw PyTorch loop.

advanced

Make a custom Lightning Loop

Conduct bleeding-edge research like meta-learning and RL with a custom Loop.

expert