Add more information in CheckpointManager's summary line.
PiperOrigin-RevId: 307162471 Change-Id: Ifdc5e6ab7800f80b35968e6650918277f0178f85
This commit is contained in:
parent
65b4e47ece
commit
7ae5b12f47
@ -511,7 +511,7 @@ def meta_graph_filename(checkpoint_filename, meta_graph_suffix="meta"):
|
|||||||
# TODO(allenl): Allow tf.keras.Model instances in the constructor directly?
|
# TODO(allenl): Allow tf.keras.Model instances in the constructor directly?
|
||||||
@tf_export("train.CheckpointManager")
|
@tf_export("train.CheckpointManager")
|
||||||
class CheckpointManager(object):
|
class CheckpointManager(object):
|
||||||
"""Deletes old checkpoints.
|
"""Manages multiple checkpoints by keeping some and deleting unneeded ones.
|
||||||
|
|
||||||
Example usage:
|
Example usage:
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user