Commit a94e2737 authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Internal change

PiperOrigin-RevId: 328980837
parent e8698f12
...@@ -193,7 +193,8 @@ class TrainerConfig(base_config.Config): ...@@ -193,7 +193,8 @@ class TrainerConfig(base_config.Config):
max_to_keep: max checkpoints to keep. max_to_keep: max checkpoints to keep.
continuous_eval_timeout: maximum number of seconds to wait between continuous_eval_timeout: maximum number of seconds to wait between
checkpoints, if set to None, continuous eval will wait indefinitely. This checkpoints, if set to None, continuous eval will wait indefinitely. This
is only used continuous_train_and_eval and continuous_eval modes. is only used continuous_train_and_eval and continuous_eval modes. Default
value is 24 hrs.
train_steps: number of train steps. train_steps: number of train steps.
validation_steps: number of eval steps. If `None`, the entire eval dataset validation_steps: number of eval steps. If `None`, the entire eval dataset
is used. is used.
...@@ -222,7 +223,7 @@ class TrainerConfig(base_config.Config): ...@@ -222,7 +223,7 @@ class TrainerConfig(base_config.Config):
checkpoint_interval: int = 1000 checkpoint_interval: int = 1000
# Checkpoint manager. # Checkpoint manager.
max_to_keep: int = 5 max_to_keep: int = 5
continuous_eval_timeout: Optional[int] = None continuous_eval_timeout: int = 24 * 60 * 60
# Train/Eval routines. # Train/Eval routines.
train_steps: int = 0 train_steps: int = 0
validation_steps: Optional[int] = None validation_steps: Optional[int] = None
......
Markdown is supported
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