From 398c72f13431a5f944ae45dd0239df9c92910123 Mon Sep 17 00:00:00 2001
From: Jason <jjj2070@163.com>
Date: Mon, 15 Apr 2019 20:46:56 +0800
Subject: [PATCH] modify error message

---
 tensorflow2fluid/tf2fluid/tensorflow_graph.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tensorflow2fluid/tf2fluid/tensorflow_graph.py b/tensorflow2fluid/tf2fluid/tensorflow_graph.py
index 2d40fa3..fb88c84 100644
--- a/tensorflow2fluid/tf2fluid/tensorflow_graph.py
+++ b/tensorflow2fluid/tf2fluid/tensorflow_graph.py
@@ -102,8 +102,8 @@ class TensorflowGraph(Graph):
                                               self.node_map[layer.name])
                         self._make_connection(pred_node, self.node_map[pred])
                     else:
-                        raise Exception("Unsupported situation(name:[{}], \
-                            OP[{}])".format(node.layer_name, node.layer_type))
+                        raise Exception("\nUnsupported situation(name:[{}]," \
+                            "OP[{}])".format(layer.name, layer.op))
 
                 elif pred in self.node_map:
                     self._make_connection(self.node_map[pred],
-- 
GitLab