作者:田海山
https://zhuanlan.zhihu.com/p/66320870
安装
可以通过以下的命令进行安装
本教程基于以下的版本:
载入库
网络结构可视化
载入alexnet,draw_model函数需要传入三个参数,第一个为model,第二个参数为input_shape,第三个参数为orientation,可以选择'LR'或者'TB',分别代表左右布局与上下布局。
在notebook中,执行完上面的代码会显示如下的图,将网络的结构及各个层的name和shape进行了可视化。
统计网络参数
可以通过model_stats方法统计各层的参数情况。
参考:
https://github.com/microsoft/tensorwatch
— 完 —
为您推荐
NumPy手写全部主流机器学习模型,代码超3万行
手把手带你走进卷积神经网络!
重磅!支持中文的算法可视化网站,全交互式体验
如何利用Python开发人工智能入门项目
一位博士生选择自杀,在论文中了顶会之后
本篇文章来源于: 深度学习这件小事
本文为原创文章,版权归知行编程网所有,欢迎分享本文,转载请保留出处!
内容反馈