I can tell you how I saved the model and it worked
import pickle
with open("./DUMPS/model_DUMP.txt",“wb”) as file:
_                pickle.dump(model,file)_
This saved the model, then when I started training the model instead of starting from zero I loaded the saved model:
with open(“DUMPS/model_DUMP.txt”,“rb”) as mod:
_    existing_model = pickle.load(mod)_
and loaded the state of the saved model to a already initialized model:
shared_model.load_state_dict(existing_model.state_dict())
The training continued from where I stopped the training.