提交 7e17e471 编写于 作者: M Mustafa Ispir 提交者: TensorFlower Gardener

Fixed raced condition in MonitoredSession test.

Change: 136645313
上级 9b1ae480
...@@ -2072,9 +2072,6 @@ py_test( ...@@ -2072,9 +2072,6 @@ py_test(
size = "small", size = "small",
srcs = ["training/monitored_session_test.py"], srcs = ["training/monitored_session_test.py"],
srcs_version = "PY2AND3", srcs_version = "PY2AND3",
tags = [
"notsan", # http://b/32109634
],
deps = [ deps = [
"//tensorflow:tensorflow_py", "//tensorflow:tensorflow_py",
], ],
......
...@@ -161,12 +161,12 @@ class MonitoredTrainingSessionTest(tf.test.TestCase): ...@@ -161,12 +161,12 @@ class MonitoredTrainingSessionTest(tf.test.TestCase):
logdir = _test_dir(self.get_temp_dir(), 'test_summaries') logdir = _test_dir(self.get_temp_dir(), 'test_summaries')
with tf.Graph().as_default(): with tf.Graph().as_default():
gstep = tf.contrib.framework.get_or_create_global_step() gstep = tf.contrib.framework.get_or_create_global_step()
do_step = tf.assign_add(gstep, 1) new_gstep = tf.assign_add(gstep, 1)
tf.scalar_summary('my_summary_tag', gstep * 2) tf.scalar_summary('my_summary_tag', new_gstep * 2)
with tf.train.MonitoredTrainingSession( with tf.train.MonitoredTrainingSession(
is_chief=True, checkpoint_dir=logdir) as session: is_chief=True, checkpoint_dir=logdir) as session:
for _ in range(101): # 100 is default summary writing steps for _ in range(101): # 100 is default summary writing steps
session.run(do_step) session.run(new_gstep)
summaries = testing.latest_summaries(logdir) summaries = testing.latest_summaries(logdir)
tags = [s.summary.value[0].tag for s in summaries] tags = [s.summary.value[0].tag for s in summaries]
self.assertIn('my_summary_tag', tags) self.assertIn('my_summary_tag', tags)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册