Skip to content

GitLab

Commit 98074f7a authored by Ruoxin Sang's avatar Ruoxin Sang Committed by A. Unique TensorFlower
Browse files

Remove workaround to make eval iterator as a class member. 

PiperOrigin-RevId: 305378415
parent 2ff0a3fe
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 5 deletions
official/staging/training/standard_runnable.py
View file @ 98074f7a
...@@ -139,13 +139,10 @@ class StandardEvaluable(runnable.AbstractEvaluable): ...@@ -139,13 +139,10 @@ class StandardEvaluable(runnable.AbstractEvaluable):
eval_fn = tf.function(eval_fn) eval_fn = tf.function(eval_fn)
self.eval_loop_fn = utils.create_loop_fn(eval_fn) self.eval_loop_fn = utils.create_loop_fn(eval_fn)
# TODO(b/147718615): When async RPC is enabled in eager runtime, we make eval_iter = tf.nest.map_structure(iter, self.eval_dataset)
# eval iterator as a class member so it doesn't get destroyed when out of
# the function scope.
self.eval_iter = tf.nest.map_structure(iter, self.eval_dataset)
self.eval_begin() self.eval_begin()
self.eval_loop_fn(self.eval_iter, num_steps) self.eval_loop_fn(eval_iter, num_steps)
return self.eval_end() return self.eval_end()
def eval_begin(self): def eval_begin(self):
......
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