Some useful plugins for PyTorch Lightning.
IMPORTANT NOTICE: There is ongoing work to make this package fully compatible with PyTorch Lightning version 1.5+ and torchmetric version 0.8+. Please install version 0.0.5 if you are working with PyTorch Lightning version 1.3 or 1.4.
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