Checkpointing

Checkpointing is implemented by hand. There are likely off by one errors in recording current epoch, training iteration. This complexity will only increase once batching is implemented. A more robust format for saving checkpoints is desired (potentially there is one built-in to PyTorch?)