Save the whole model object
Is there anyway of saving a whole model object with PyTorch lightning?
e.g. I want something like this:
model = load("mypath")
prediction = model.predict(x) 


without needing access to the original Model class.
I know how to load things from a checkpoint, but that requires having access to the model class.
Is this possible? Any help would be much appreciated