I think the easiest way would be to just save and load the complete state_dict and implement a function that resets the specific tensor, e.g.:
model = Net()
model.load_state_dict(torch.load(PATH))
model.reset()