Improve serialization of model & resumption of training

[andreaskoepf]

• it should be possible to resume model training (all relevant model and optimizer parameters must be serialized)
• different model parts could be serialized individually
• model checkpoints should be written in configurable intervals (e.g. every N gradient steps)