Open
Description
It would be nice to add support for checkpointing in TrainKeras. These checkpoints should probably be collected on a regular interval + the best model.
These checkpoints should probably collect more detailed metrics, too, such as viewing sample input/outputs, etc.