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

Change the default validation steps in the config to -1.

PiperOrigin-RevId: 336112806
parent 2d62cf36
...@@ -236,7 +236,8 @@ class TrainerConfig(base_config.Config): ...@@ -236,7 +236,8 @@ class TrainerConfig(base_config.Config):
continuous_eval_timeout: int = 60 * 60 continuous_eval_timeout: int = 60 * 60
# Train/Eval routines. # Train/Eval routines.
train_steps: int = 0 train_steps: int = 0
validation_steps: Optional[int] = None # Sets validation steps to be -1 to evaluate the entire dataset.
validation_steps: int = -1
validation_interval: int = 1000 validation_interval: int = 1000
# Best checkpoint export. # Best checkpoint export.
best_checkpoint_export_subdir: str = "" best_checkpoint_export_subdir: str = ""
......
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