Yeah, I see. Storing the model directly is not recommended, as you would need to create the same file and folder structure to load the model.
model.load_state_dict won’t work here, as checkpoint['model'] should contain the model.
Do you get errors using?
model = checkpoint['model']

Based on your description it seems you don’t have access to the model definition?