The recommended way to serialize the model is to store and load the state_dict and to recreate the model object as explained here 1.
Based on the error message I guess that loading the complete model fails.