Commit d2195d91 authored by remi cresson's avatar remi cresson
Browse files

ENH: don't delete the SavedModel directory

parent bd756e6f
......@@ -114,12 +114,8 @@ 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):
# 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