I have some Tensorflow code in which I would like to be able to store checkpoints of a model and restore them for later training. So during training I perform a callback to