Lightning AI Studios: Never set up a local environment again →

← Back to blog

Get Full Control of your Training Using 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.