推荐一个非常好用的可视化工具,torchviz/graphviz
第一步,下载
首先要下载Graphviz, [官网地址],我是在windows下,所以直接下载windows版本的zip就可以,下载后找个地方解压就行(http://www.graphviz.org/download/)
第二步,配置Graphviz的系统环境变量
在高级系统设置—>系统属性—>高级—>环境变量—>系统变量—>Path中添加Graphviz/bin的路径,我是D:\YXC\Study\Graphviz\Graphviz\bin
验证是否配置成功:
cmd控制台输入 dot -version
,如果显示下面就没问题了!
第三步,python环境中install
pip install torchviz
pip install graphviz
最后一步,重启编译器
用法
from torchviz import make_dot
import XXX
net = your_net()
net_input = XXX(定义一个与网络输入维度相同的变量)
out = net(net_input)
vis_graph =make_dot(out,params=dict(net.named_parameters()))
vis_graph.view()
好了,到这里就可以直观的看到网络结构了
问题1
graphviz.backend.ExecutableNotFound: failed to execute [‘dot’,
‘-Tpdf’, ‘-O’, ‘Digraph.gv’], make sure the Graphviz executables are
on your systems’ PATH
这个有两个原因,一种是没有把下载的Graph/bin添加到环境变量中,另一种是没有重启编译器,按照上面的步骤来的话就没有问题了。