All posts

Get Full Control of your Training Using Callbacks

Lightning has a technique to add arbitrary self-contained programs to your training which are called callbacks.

Lightning Callbacks

Lightning 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 Lightning offers to supercharge your machine learning training.

Lightning has a technique to add arbitrary self-contained programs to your training which are called callbacks. Callbacks is a general idea from computer science where during the flow of your code execution you can plug into random places of that execution and execute arbitrary code. This means you have full control over any point in the training system and can design self-contained programs that encapsulate a full set of functionalities on its own.

Watch the video to learn more about this technique and the advantages of using callbacks.

Interested in building with Lightning?

Tell us what you need and we’ll reach out.

  • See a demo on workloads like yours
  • Get GPU pricing and capacity
  • Review security and compliance

Trusted by 450,000+ devs and AI teams

NVIDIACursorCiscoRunwaySupercellstability.aiBlack Forest Labs

You can also email us at [email protected]