Commit d2195d91 authored by remi cresson's avatar remi cresson

ENH: don't delete the SavedModel directory

parent bd756e6f
......@@ -114,13 +114,9 @@ def CreateSavedModel(sess, inputs, outputs, directory):
directory: the output path for the SavedModel
"""
directory += "/SavedModel"
print("Create a SavedModel in " + directory)
# Delete the directory if it already exists
if os.path.exists(directory):
shutil.rmtree(directory)
# Get graph
graph = tf.get_default_graph()
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment