[TF2] Support time based checkpointing #321
Description
After the conversion to TF2 we currently only support saving by an epoch count rather than saving based on time. Saving based on time can be a lot nicer in a lot of ways, and is something that we should support.