"mmdet3d/models/vscode:/vscode.git/clone" did not exist on "2072a9df75119b717d0128e3820b40cdf57cb3ca"
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
......@@ -139,13 +139,10 @@ class StandardEvaluable(runnable.AbstractEvaluable):
eval_fn = tf.function(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 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)
eval_iter = tf.nest.map_structure(iter, self.eval_dataset)
self.eval_begin()
self.eval_loop_fn(self.eval_iter, num_steps)
self.eval_loop_fn(eval_iter, num_steps)
return self.eval_end()
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