Internal change

PiperOrigin-RevId: 317958778

official/modeling/hyperparams/config_definitions.py

@@ -178,14 +178,17 @@ class TrainerConfig(base_config.Config):
       checkpoints, if set to None, continuous eval will wait indefinetely.
   """
   optimizer_config: OptimizationConfig = OptimizationConfig()
-  train_tf_while_loop: bool = True
-  train_tf_function: bool = True
-  eval_tf_function: bool = True
+  train_steps: int = 0
+  validation_steps: Optional[int] = None
+  validation_interval: int = 100
   steps_per_loop: int = 1000
   summary_interval: int = 1000
   checkpoint_interval: int = 1000
   max_to_keep: int = 5
   continuous_eval_timeout: Optional[int] = None
+  train_tf_while_loop: bool = True
+  train_tf_function: bool = True
+  eval_tf_function: bool = True
 
 
 @dataclasses.dataclass
@@ -201,9 +204,6 @@ class ExperimentConfig(base_config.Config):
   task: TaskConfig = TaskConfig()
   trainer: TrainerConfig = TrainerConfig()
   runtime: RuntimeConfig = RuntimeConfig()
-  train_steps: int = 0
-  validation_steps: Optional[int] = None
-  validation_interval: int = 100
 
 
 _REGISTERED_CONFIGS = {}