Implement an easy to use interface for checkpointing experiments, part 1.
* Added CheckpointManager API. * Added FileCehckpointManager implementation. * Updated an experiment to: * Use the new CheckpointManager objects. * Separate evaluation from the metrics consumer/hook. * Not store metrics in the checkpoint. TODO: Update the remaining experiments to use the checkpoint manager. TODO: Remove checkpoint utils. PiperOrigin-RevId: 274620556
Showing
- tensorflow_federated/python/research/utils/BUILD 14 additions, 0 deletionstensorflow_federated/python/research/utils/BUILD
- tensorflow_federated/python/research/utils/checkpoint_manager.py 158 additions, 0 deletions...low_federated/python/research/utils/checkpoint_manager.py
- tensorflow_federated/python/research/utils/checkpoint_manager_test.py 158 additions, 0 deletions...ederated/python/research/utils/checkpoint_manager_test.py
Please register or sign in to comment