From a01c46f2be2151d5d0a5c6039d50e1ed4349aa06 Mon Sep 17 00:00:00 2001 From: Omar Aflak Date: Thu, 1 Feb 2018 07:55:49 -0800 Subject: [PATCH] removed unused parameter from export_model() The model passed to the export method wasn't actually used --- tensorflow_model/mnist_convnet_keras.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tensorflow_model/mnist_convnet_keras.py b/tensorflow_model/mnist_convnet_keras.py index 8d1fdfb..57887b0 100644 --- a/tensorflow_model/mnist_convnet_keras.py +++ b/tensorflow_model/mnist_convnet_keras.py @@ -74,11 +74,11 @@ def train(model, x_train, y_train, x_test, y_test): validation_data=(x_test, y_test)) -def export_model(saver, model, input_node_names, output_node_name): +def export_model(input_node_names, output_node_name): tf.train.write_graph(K.get_session().graph_def, 'out', \ MODEL_NAME + '_graph.pbtxt') - saver.save(K.get_session(), 'out/' + MODEL_NAME + '.chkp') + tf.train.Saver().save(K.get_session(), 'out/' + MODEL_NAME + '.chkp') freeze_graph.freeze_graph('out/' + MODEL_NAME + '_graph.pbtxt', None, \ False, 'out/' + MODEL_NAME + '.chkp', output_node_name, \ @@ -109,7 +109,7 @@ def main(): train(model, x_train, y_train, x_test, y_test) - export_model(tf.train.Saver(), model, ["conv2d_1_input"], "dense_2/Softmax") + export_model(["conv2d_1_input"], "dense_2/Softmax") if __name__ == '__main__':