知行编程网知行编程网  2022-03-31 14:03 知行编程网 隐藏边栏 |   抢沙发  26 
文章评分 0 次,平均分 0.0

Pytorch 网络结构可视化

作者:田海山

https://zhuanlan.zhihu.com/p/66320870


安装

可以通过以下的命令进行安装

本教程基于以下的版本:

载入库

网络结构可视化

载入alexnet,draw_model函数需要传入三个参数,第一个为model,第二个参数为input_shape,第三个参数为orientation,可以选择'LR'或者'TB',分别代表左右布局与上下布局。

在notebook中,执行完上面的代码会显示如下的图,将网络的结构及各个层的name和shape进行了可视化。


Pytorch 网络结构可视化


统计网络参数

可以通过model_stats方法统计各层的参数情况。


Pytorch 网络结构可视化


考:

https://github.com/microsoft/tensorwatch

— 完 —

为您推荐

NumPy手写全部主流机器学习模型,代码超3万行

手把手带你走进卷积神经网络!

重磅!支持中文的算法可视化网站,全交互式体验

如何利用Python开发人工智能入门项目

一位博士生选择自杀,在论文中了顶会之后

本篇文章来源于: 深度学习这件小事

本文为原创文章,版权归所有,欢迎分享本文,转载请保留出处!

知行编程网
知行编程网 关注:1    粉丝:1
这个人很懒,什么都没写

发表评论

表情 格式 链接 私密 签到
扫一扫二维码分享