PyTorch Lightning Spells

Some useful plugins for PyTorch Lightning.

Documentation

IMPORTANT NOTICE: There is ongoing work to make this package fully compatible with PyTorch Lightning version 2.1. Submitting issues and pull requests for compatibility with this version is highly appreciated.

Support for older versions of PyTorch Lightning:

• Please install version 0.1.0 if you are working with PyTorch Lightning version 2.1.
• Please install version 0.0.5 if you are working with PyTorch Lightning version 1.3 or 1.4.

Installation

pip install pytorch-lightning-spells

or the latest in the main branch:

pip install https://github.com/veritable-tech/pytorch-lightning-spells/archive/master.zip