pytorch网络结果的可视化

最近想在pytorch下显示网络结构，试了几种方案都不成功，今天试一下Graphviz

参考博客连接https://www.cnblogs.com/linfangnan/p/13210536.html

一.安装Graphviz

下载的有点慢

不管用，还是得手动添加到path中

测试是否安装成功

好了，第一步完成

二.graphviz库安装

 

cmd输入下列命令

pip install git+https://github.com/szagoruyko/pytorchviz

三、接下来我使用这个博客https://blog.csdn.net/qq_18293213/article/details/79047742的方法，将make_dot函数copy到自定义网络的后面，然后加上如下语句即可

 

x = torch.rand(64,1,28,28)
model = BaseNet()
y = model(x)
g = make_dot(y)
g.view()

网络结构图如图所示，是一个PDF

个人感觉不如Keras的summary好看，期待Pytorch会有新的API来更好的可视化网络吧！欢迎大家多交流！