Skip to content
GitLab
Projects
Groups
Snippets
Help
Loading...
Help
What's new
10
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Open sidebar
Lozac'h Loic
otbtf
Commits
d2195d91
Commit
d2195d91
authored
Sep 25, 2018
by
remi cresson
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
ENH: don't delete the SavedModel directory
parent
bd756e6f
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
5 deletions
+1
-5
python/tricks.py
python/tricks.py
+1
-5
No files found.
python/tricks.py
View file @
d2195d91
...
...
@@ -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
()
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment