Skip to content

GitLab

Commit 09050ac3 authored by Le Hou's avatar Le Hou Committed by A. Unique TensorFlower
Browse files

orbit.StandardEvaluator minor update 

PiperOrigin-RevId: 329355546
parent 32309b8b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
orbit/standard_runner.py
View file @ 09050ac3
...@@ -193,7 +193,7 @@ class StandardEvaluator(runner.AbstractEvaluator, metaclass=abc.ABCMeta): ...@@ -193,7 +193,7 @@ class StandardEvaluator(runner.AbstractEvaluator, metaclass=abc.ABCMeta):
"""See base class.""" """See base class."""
outputs = self.eval_begin() # pylint: disable=assignment-from-no-return outputs = self.eval_begin() # pylint: disable=assignment-from-no-return
eval_iter = tf.nest.map_structure(iter, self._eval_dataset) eval_iter = tf.nest.map_structure(iter, self.eval_dataset)
if self._eval_loop_fn is None: if self._eval_loop_fn is None:
eval_fn = self.eval_step eval_fn = self.eval_step
if self._eval_use_tf_function: if self._eval_use_tf_function:
......
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