mindspore保存模型的中间图,并利用dot将图结构可视化

以动态模型混淆的网络为例。
https://www.mindspore.cn/mindarmour/docs/zh-CN/master/dynamic_obfuscation_protection.html
加载网络并设置password后
添加代码:

ms.context.set_context(save_graphs=True,save_graphs_path="./saved_graphs/",save_graph_dot=True)
ms.export(net,input_tensor,file_name="obf_net",file_format="MINDIR")

mindspore保存模型的中间图,并利用dot将图结构可视化_第1张图片
把这个名字特殊的dot文件导出为pdf

!dot -Tpdf saved_graphs/ModelDigraph_0062.dot -o my_net.pdf

mindspore保存模型的中间图,并利用dot将图结构可视化_第2张图片
这样就可以看到可视化后的网络结构了。

你可能感兴趣的:(MindSpore,mindspore,python,可视化)