I think you can save most python objects with torch.save. So at the very simplest you could call torch.save on the tuple (model.state_dict(),n_input,n_hidden,n_outputs). Probably make those integers class attributes while your’re at it.  I use this all the time to save epochs elapsed with my state dicts.