Commit 335331c2 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 f56dedec
......@@ -236,7 +236,8 @@ class TrainerConfig(base_config.Config):
continuous_eval_timeout: int = 60 * 60
# Train/Eval routines.
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
# Best checkpoint export.
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