一、torchinfo工具
可以生成下面这种结构图
使用方法为:
就这三行!
当然,前提是安装torchinfo
pip install torchinfo
二、NN-SVG
http://alexlenail.me/NN-SVG/LeNet.html
很容易做出下面的图来
可以生成下面这种结构图
使用方法为:
就这三行!
当然,前提是安装torchinfo
pip install torchinfo
http://alexlenail.me/NN-SVG/LeNet.html
很容易做出下面的图来