MindSpore如何将add_node函数添加节点信息到self.node中

经验分享 Tech Blogs 模型训练-Model Training
1

1.系统环境

硬件环境(Ascend/GPU/CPU): GPU
MindSpore版本: mindspore=2.0
执行模式（PyNative/ Graph）:不限
Python版本：3.7
操作系统平台：Linux

2. 问题描述

原本需要：name，id=node_def.name，op_type，scope，output_type， topo_id就是self.node_id，而node_id是由node_def.name传入的，在.pb文件中，.name是每个节点的编号，full_name才是它的真正名字

3. 解决方案

最终，通过graph中的add_node函数添加节点信息到self.node中

def add_node(self,node):  
    node.internal_id = self._node_id_counter  
    self._node_id_counter += 1  
    self.node.append(node)  
    self.name_nodes[node.name] = node