Catch export exception

tobegit3hub · tobegit3hub · commit 3801fe35b5a4 · 2017-02-06T19:28:43.000+08:00
diff --git a/dense_classifier.py b/dense_classifier.py
@@ -498,7 +498,10 @@ def export_model(sess, saver, signature, model_path, model_version):
   model_exporter = exporter.Exporter(saver)
   model_exporter.init(sess.graph.as_graph_def(),
                       named_graph_signatures=signature)
-  model_exporter.export(model_path, tf.constant(model_version), sess)
+  try:
+    model_exporter.export(model_path, tf.constant(model_version), sess)
+  except Exception as e:
+    logging.error("Fail to export model, exception: {}".format(e))
 
 
 if __name__ == "__main__":
diff --git a/sparse_classifier.py b/sparse_classifier.py
@@ -481,7 +481,10 @@ def export_model(sess, saver, signature, model_path, model_version):
   model_exporter = exporter.Exporter(saver)
   model_exporter.init(sess.graph.as_graph_def(),
                       named_graph_signatures=signature)
-  model_exporter.export(model_path, tf.constant(model_version), sess)
+  try:
+    model_exporter.export(model_path, tf.constant(model_version), sess)
+  except Exception as e:
+    logging.error("Fail to export model, exception: {}".format(e))
 
 
 if __name__ == "__main__":
