Skip to content
Snippets Groups Projects
Commit 48d5e486 authored by Michael Reneer's avatar Michael Reneer Committed by Zachary Garrett
Browse files

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
parent c68952c4
No related branches found
No related tags found
Hide whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment