Add clear_devices when exporting models

EricFang2014 · Apr 27, 2017 · c29f742 · c29f742
1 parent 2a03979
commit c29f742
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/dense_classifier.py b/dense_classifier.py
@@ -488,6 +488,7 @@ def inference(inputs, is_train=True):
         builder.add_meta_graph_and_variables(
             sess,
             [tag_constants.SERVING],
+            clear_devices=True,
             signature_def_map={
                 signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY:
                 model_signature,
@@ -575,7 +576,8 @@ def export_model(sess, saver, signature, model_path, model_version):
   logging.info("Export the model to {}".format(model_path))
   model_exporter = exporter.Exporter(saver)
   model_exporter.init(sess.graph.as_graph_def(),
-                      named_graph_signatures=signature)
+                      named_graph_signatures=signature,
+                      clear_devices=True)
   try:
     model_exporter.export(model_path, tf.constant(model_version), sess)
   except Exception as e:

diff --git a/sparse_classifier.py b/sparse_classifier.py
@@ -431,6 +431,7 @@ def inference(sparse_ids, sparse_values, is_train=True):
         builder.add_meta_graph_and_variables(
             sess,
             [tag_constants.SERVING],
+            clear_devices=True,
             signature_def_map={
                 signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY:
                 model_signature,
@@ -535,7 +536,8 @@ def export_model(sess, saver, signature, model_path, model_version):
   logging.info("Export the model to {}".format(model_path))
   model_exporter = exporter.Exporter(saver)
   model_exporter.init(sess.graph.as_graph_def(),
-                      named_graph_signatures=signature)
+                      named_graph_signatures=signature,
+                      clear_devices=True)
   try:
     model_exporter.export(model_path, tf.constant(model_version), sess)
   except Exception as e: