MindInsight计算图可视设计

MindInsight计算图可视设计

特性背景

计算图可视的功能,主要协助开发者在下面这些场景中使用。

  • 开发者在编写深度学习神经网络的代码时,可以使用计算图的功能查看神经网络中算子的数据流走向,以及模型结构。
  • 计算图还可以方便开发者查看指定节点的输入和输出节点,以及所查找的节点的属性信息。
  • 开发者在调试网络时,可以通过可视化的计算图,轻易跟踪数据,包括数据维度、类型的变更等。

总体设计

概念设计

 

后端设计

后端的类图如下,主要分为Graph基类和Node两个类,其中MsGraph是继承了Graph基类,用于解析MindSpore ANF的计算图文件。Node类聚合成一张图,与Graph为聚合关系。

 

前端设计

数据将通过Web UI进行绘图并展示,前端采用d3-graph-viz 3.x插件进行绘图辅助。


图1:辅助功能

如图1所示,提供选择文件、查询节点、查看节点信息、节点输入输出等辅助功能。

图2:计算图主体展示

计算图中,根据斜线(/)对节点的名称划分层次,并逐层展示,参考图2计算图主体展示。双击一个作用域节点后,将会展示它的子节点。

接口设计

计算图中,主要有文件接口和RESTful API接口,其中文件接口为summary.proto文件,是MindInsight和MindSpore进行数据对接的接口。 RESTful API接口是MindInsight前后端进行数据交互的接口。

文件接口设计

MindSpore与MindInsight之间的数据交互,采用protobuf定义数据格式。 summary.proto文件为总入口,计算图的消息对象定义为 GraphProto。GraphProto的详细定义可以参考anf_ir.proto文件

人工智能芯片与自动驾驶
原文地址:https://www.cnblogs.com/wujianming-110117/p/14316463.html