Loading Lightning model in PyTorch
How to load a model saved in PyTorch Lightning in Vanilla PyTorch?
check this out: https://pytorch-lightning.readthedocs.io/en/latest/common/production_inference.html#id1