Skip to content

GitLab

You need to sign in or sign up before continuing.
Commit 72284a6c authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 350687522
parent 234dc350
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
official/core/base_trainer.py
View file @ 72284a6c
...@@ -161,7 +161,8 @@ class Trainer(orbit.StandardTrainer, orbit.StandardEvaluator): ...@@ -161,7 +161,8 @@ class Trainer(orbit.StandardTrainer, orbit.StandardEvaluator):
self, self,
eval_dataset, eval_dataset,
options=orbit.StandardEvaluatorOptions( options=orbit.StandardEvaluatorOptions(
use_tf_function=config.trainer.eval_tf_function)) use_tf_function=config.trainer.eval_tf_function,
use_tf_while_loop=config.trainer.eval_tf_while_loop))
def _validate_params(self, config): def _validate_params(self, config):
r"""Validates if the configuration object passed to the Trainer. r"""Validates if the configuration object passed to the Trainer.
......
official/core/config_definitions.py
View file @ 72284a6c
...@@ -192,6 +192,7 @@ class TrainerConfig(base_config.Config): ...@@ -192,6 +192,7 @@ class TrainerConfig(base_config.Config):
train_tf_while_loop: bool = True train_tf_while_loop: bool = True
train_tf_function: bool = True train_tf_function: bool = True
eval_tf_function: bool = True eval_tf_function: bool = True
eval_tf_while_loop: bool = False
allow_tpu_summary: bool = False allow_tpu_summary: bool = False
# Trainer intervals. # Trainer intervals.
steps_per_loop: int = 1000 steps_per_loop: int = 1000
......
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