Skip to content

GitLab

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

orbit.StandardEvaluator minor update 

PiperOrigin-RevId: 329355546
parent 91afecbb
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
orbit/standard_runner.py
View file @ b74b4ee1
......@@ -193,7 +193,7 @@ class StandardEvaluator(runner.AbstractEvaluator, metaclass=abc.ABCMeta):
"""See base class."""
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:
eval_fn = self.eval_step
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