pytorch-lightning

PyTorch Lightning helps organize PyTorch code and decouple the science code from the engineering code. It’s more of a style-guide than a framework. By organizing PyTorch code under a LightningModule, Lightning makes things like TPU, multi-GPU and 16-bit precision training (40+ other features) trivial.

For more information, please see:

• Comet + Pytorch Lightning
• pytorch_lightning.loggers.CometLogger