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

Deprecated tf.Session removed in bidirectional_sequence_lstm_test.py

Browse Source
This commit is contained in:
Siju Samuel 2019-07-10 14:44:18 +05:30
parent c98c853676
commit 0a33907500
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/bidirectional_sequence_lstm_test.py
View File

@ -178,7 +178,7 @@ class BidirectionalSequenceLstmTest(test_util.TensorFlowTestCase):
x, prediction, output_class = self.buildModel(fw_lstm_layer, bw_lstm_layer, x, prediction, output_class = self.buildModel(fw_lstm_layer, bw_lstm_layer,
is_dynamic_rnn) 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
@ -240,7 +240,7 @@ class BidirectionalSequenceLstmTest(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(self.buildLstmLayer(), x, prediction, output_class = self.buildModel(self.buildLstmLayer(),
self.buildLstmLayer(), False) self.buildLstmLayer(), False)
@ -258,7 +258,7 @@ class BidirectionalSequenceLstmTest(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(self.buildLstmLayer(), x, prediction, output_class = self.buildModel(self.buildLstmLayer(),
self.buildLstmLayer(), True) self.buildLstmLayer(), True)