Train deep learning models with PyTorch Lightning.
PyTorch Lightning is a framework that helps you organize and scale your PyTorch code and it has recently become very popular among machine learning and deep learning researchers. With ZenML's PyTorch integration, you can load, train, and deploy PyTorch Lightning models within your ZenML pipelines.
Features with ZenML
Main Features
How to use ZenML with
PyTorch Lightning
Additional Resources
Go to Github