Allow saving / loading of state of optimizer

[JohnGiorgi]

Currently, 'loading' a model loads its architecture and its weights, which is useful for transfer learning. However, a mechanism for loading the entire model, including the state of the optimizer, would allow for the resuming of training.