MindInsight计算图可视设计
MindInsight計算圖可視設計
特性背景
計算圖可視的功能,主要協(xié)助開發(fā)者在下面這些場景中使用。
? 開發(fā)者在編寫深度學習神經(jīng)網(wǎng)絡的代碼時,可以使用計算圖的功能查看神經(jīng)網(wǎng)絡中算子的數(shù)據(jù)流走向,以及模型結構。
? 計算圖還可以方便開發(fā)者查看指定節(jié)點的輸入和輸出節(jié)點,以及所查找的節(jié)點的屬性信息。
? 開發(fā)者在調(diào)試網(wǎng)絡時,可以通過可視化的計算圖,輕易跟蹤數(shù)據(jù),包括數(shù)據(jù)維度、類型的變更等。
總體設計
概念設計
后端設計
后端的類圖如下,主要分為Graph基類和Node兩個類,其中MsGraph是繼承了Graph基類,用于解析MindSpore ANF的計算圖文件。Node類聚合成一張圖,與Graph為聚合關系。
前端設計
數(shù)據(jù)將通過Web UI進行繪圖并展示,前端采用d3-graph-viz 3.x插件進行繪圖輔助。
圖1:輔助功能
如圖1所示,提供選擇文件、查詢節(jié)點、查看節(jié)點信息、節(jié)點輸入輸出等輔助功能。
圖2:計算圖主體展示
計算圖中,根據(jù)斜線(/)對節(jié)點的名稱劃分層次,并逐層展示,參考圖2計算圖主體展示。雙擊一個作用域節(jié)點后,將會展示它的子節(jié)點。
接口設計
計算圖中,主要有文件接口和RESTful API接口,其中文件接口為summary.proto文件,是MindInsight和MindSpore進行數(shù)據(jù)對接的接口。 RESTful API接口是MindInsight前后端進行數(shù)據(jù)交互的接口。
文件接口設計
MindSpore與MindInsight之間的數(shù)據(jù)交互,采用protobuf定義數(shù)據(jù)格式。 summary.proto文件為總?cè)肟?#xff0c;計算圖的消息對象定義為 GraphProto。GraphProto的詳細定義可以參考anf_ir.proto文件。
總結
以上是生活随笔為你收集整理的MindInsight计算图可视设计的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: MindInsight训练可视整体设计介
- 下一篇: MindInsight张量可视设计介绍