Converting .pth (PyTorch) to .ckpt (PyTorch Lightning)

Hey there! I am looking for an elegant solution for converting a .pth (PyTorch) model file into a .ckpt (PyTorch Lightning) model file.

The solution I’m considering right now is parsing all the layers in the .pth’s state_dict and building a model which would be saved in .ckpt, but I’d prefer something less arduous.

Any Ideas?