diff --git a/plugins/tensorboard-plugins/tb_graph_ascend/server/app/controllers/hierarchy.py b/plugins/tensorboard-plugins/tb_graph_ascend/server/app/controllers/hierarchy.py index b5870f29885242075a8f7afc0cf44ed2be073b1e..acb60aeae93ab597cf50b4fd097969ffe5eeedc8 100644 --- a/plugins/tensorboard-plugins/tb_graph_ascend/server/app/controllers/hierarchy.py +++ b/plugins/tensorboard-plugins/tb_graph_ascend/server/app/controllers/hierarchy.py @@ -53,15 +53,11 @@ class Hierarchy: def extract_label_name(node_name, node_type): splited_subnode_name = node_name.split('.') splited_label = [] - # Module.layer1.1.relu.ReLU.forward.1 =>relu.ReLU.forward.1 - # Module.layer1.1.relu.ReLU.forward.1 =>relu.ReLU.forward.1 if node_type == MODULE: if len(splited_subnode_name) < 4: return node_name splited_label = splited_subnode_name[-4:] if not splited_subnode_name[ -4].isdigit() else splited_subnode_name[-5:] - # Module.layer1.1.ApiList.1 =>ApiList.1 - # Module.layer1.1.ApiList.0.1 =>ApiList.0.1 else: if len(splited_subnode_name) < 2: return node_name