使用pytorchviz画出好看的Pytorch模型网络结构
在Pytorch中可以使用tensorboardX画出网络结构图,但总觉得不太好看。pytorchviz 似乎是一个不错的选择。
使用方法:
from torchviz import make_dot
from torch.autograd import Variablenet = Model()
x = Variable(torch.randn(3,32,32))vis_graph = make_dot(net(x), params=dict(net.named_parameters()))
vis_graph.directory = "Net_Structure"# 设置可视化文件的导出路径
vis_graph.format = "png"# 如果不加这一句,生成的则是pdf
vis_graph.view()
如果单纯地
pip install torchviz
,会报以下错误:graphviz.backend.ExecutableNotFound: failed to execute ['dot', '-Tpdf', '-O', 'Digraph.gv'], make sure the Graphviz executables are on your systems' PATH正确的方法:
- 下载package并解压到本地,例如
D:/test
. 我下载的是:这个 - Add
D:\test\Graphviz\bin
to User path - Add
D:\test\Graphviz\bin\dot.exe
to System Path - 重启计算机
https://zhuanlan.zhihu.com/p/220403674?utm_source=wechat_session https://stackoverflow.com/questions/35064304/runtimeerror-make-sure-the-graphviz-executables-are-on-your-systems-path-aft
推荐阅读
- 由浅入深理解AOP
- 【译】20个更有效地使用谷歌搜索的技巧
- mybatisplus如何在xml的连表查询中使用queryWrapper
- MybatisPlus|MybatisPlus LambdaQueryWrapper使用int默认值的坑及解决
- MybatisPlus使用queryWrapper如何实现复杂查询
- iOS中的Block
- Linux下面如何查看tomcat已经使用多少线程
- 使用composer自动加载类文件
- android|android studio中ndk的使用
- 使用协程爬取网页,计算网页数据大小