You could save the state_dict and load it for resetting the model. Have a look at the Serialization Semantics 591 to see how to do it.
Would this work for you or do you want to re-initialize it to random weights?