diff options
Diffstat (limited to 'tensorflow/python/training/basic_session_run_hooks_test.py')
-rw-r--r-- | tensorflow/python/training/basic_session_run_hooks_test.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/tensorflow/python/training/basic_session_run_hooks_test.py b/tensorflow/python/training/basic_session_run_hooks_test.py index fe8a3e9062..2d469634e0 100644 --- a/tensorflow/python/training/basic_session_run_hooks_test.py +++ b/tensorflow/python/training/basic_session_run_hooks_test.py @@ -1145,7 +1145,7 @@ class SummarySaverHookTest(test.TestCase): summary_writer=self.summary_writer, summary_op=self.summary_op) - with self.test_session() as sess: + with self.cached_session() as sess: hook.begin() sess.run(variables_lib.global_variables_initializer()) mon_sess = monitored_session._HookedSession(sess, [hook]) @@ -1177,7 +1177,7 @@ class SummarySaverHookTest(test.TestCase): summary_writer=self.summary_writer, summary_op=[self.summary_op, self.summary_op2]) - with self.test_session() as sess: + with self.cached_session() as sess: hook.begin() sess.run(variables_lib.global_variables_initializer()) mon_sess = monitored_session._HookedSession(sess, [hook]) @@ -1205,7 +1205,7 @@ class SummarySaverHookTest(test.TestCase): summary_writer=self.summary_writer, summary_op=self.summary_op) - with self.test_session() as sess: + with self.cached_session() as sess: hook.begin() sess.run(variables_lib.global_variables_initializer()) mon_sess = monitored_session._HookedSession(sess, [hook]) @@ -1240,7 +1240,7 @@ class SummarySaverHookTest(test.TestCase): summary_writer=self.summary_writer, summary_op=self.summary_op) - with self.test_session() as sess: + with self.cached_session() as sess: hook.begin() sess.run(variables_lib.global_variables_initializer()) mon_sess = monitored_session._HookedSession(sess, [hook]) @@ -1388,7 +1388,7 @@ class ResourceSummarySaverHookTest(test.TestCase): summary_writer=self.summary_writer, summary_op=self.summary_op) - with self.test_session() as sess: + with self.cached_session() as sess: hook.begin() sess.run(variables_lib.global_variables_initializer()) mon_sess = monitored_session._HookedSession(sess, [hook]) |