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