experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Deprecated tf.Session removed in unidirectional_sequence_lstm_test.py

Browse Source
This commit is contained in:
Siju Samuel 2019-07-10 14:43:52 +05:30
parent 8b9c822ac4
commit 6cdf3802a9
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/lite/experimental/examples/lstm/unidirectional_sequence_lstm_test.py
View File

@ -162,7 +162,7 @@ class UnidirectionalSequenceLstmTest(test_util.TensorFlowTestCase):
tf.reset_default_graph() tf.reset_default_graph()
x, prediction, output_class = self.buildModel(lstm_layer, is_dynamic_rnn) x, prediction, output_class = self.buildModel(lstm_layer, is_dynamic_rnn)
new_sess = tf.Session(config=CONFIG) new_sess = tf.compat.v1.Session(config=CONFIG)
saver = tf.train.Saver() saver = tf.train.Saver()
saver.restore(new_sess, model_dir) saver.restore(new_sess, model_dir)
return x, prediction, output_class, new_sess return x, prediction, output_class, new_sess
@ -224,7 +224,7 @@ class UnidirectionalSequenceLstmTest(test_util.TensorFlowTestCase):
return result return result
def testStaticRnnMultiRnnCell(self): def testStaticRnnMultiRnnCell(self):
sess = tf.Session(config=CONFIG) sess = tf.compat.v1.Session(config=CONFIG)
x, prediction, output_class = self.buildModel( x, prediction, output_class = self.buildModel(
self.buildLstmLayer(), is_dynamic_rnn=False) self.buildLstmLayer(), is_dynamic_rnn=False)
@ -242,7 +242,7 @@ class UnidirectionalSequenceLstmTest(test_util.TensorFlowTestCase):
@test_util.enable_control_flow_v2 @test_util.enable_control_flow_v2
def testDynamicRnnMultiRnnCell(self): def testDynamicRnnMultiRnnCell(self):
sess = tf.Session(config=CONFIG) sess = tf.compat.v1.Session(config=CONFIG)
x, prediction, output_class = self.buildModel( x, prediction, output_class = self.buildModel(
self.buildLstmLayer(), is_dynamic_rnn=True) self.buildLstmLayer(), is_dynamic_rnn=True)