Commit 98f103cf authored by Martino Bertoni's avatar Martino Bertoni 🌋
Browse files

remove debug prints

parent a07fba00
...@@ -52,7 +52,6 @@ def export_savedmodel(savedmodel_path, destination, ...@@ -52,7 +52,6 @@ def export_savedmodel(savedmodel_path, destination,
if tmp_path is None: if tmp_path is None:
tmp_path = tempfile.mkdtemp() tmp_path = tempfile.mkdtemp()
# save to hub module format # save to hub module format
print('SAVE export_savedmodel')
with tf.Graph().as_default(): with tf.Graph().as_default():
spec = hub.create_module_spec_from_saved_model(savedmodel_path) spec = hub.create_module_spec_from_saved_model(savedmodel_path)
module = hub.Module(spec, tags=['serve']) module = hub.Module(spec, tags=['serve'])
...@@ -60,7 +59,6 @@ def export_savedmodel(savedmodel_path, destination, ...@@ -60,7 +59,6 @@ def export_savedmodel(savedmodel_path, destination,
sess.run(tf.tables_initializer()) sess.run(tf.tables_initializer())
sess.run(tf.global_variables_initializer()) sess.run(tf.global_variables_initializer())
module.export(tmp_path, sess) module.export(tmp_path, sess)
print('DONE export_savedmodel')
if compress: if compress:
# compress the exported files to destination # compress the exported files to destination
os.system("tar -cz -f %s --owner=0 --group=0 -C %s ." % os.system("tar -cz -f %s --owner=0 --group=0 -C %s ." %
......
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