PyTorch LightningΒ is a very lightweight wrapper on PyTorch which is more like a coding standard than a framework. If you refactor your PyTorch code into the Lightning format you get the bells and whistles of top research teams without all the work. The format allows you to get rid of a ton of boilerplate code and makes it highly reproducible.

https://medium.com/@_willfalcon/how-to-refactor-your-pytorch-code-to-get-these-42-benefits-of-pytorch-lighting-6fdd0dc97538

Was this post helpful?