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