pytorch中模型结构图的可视化

1. 使用summary输出模型结构

• 安装torchsummary：pip install torchsummary
• 使用：
  from torchsummary import summary
summary(your_model, input_size=(channels, H, W), device='cpu')
• 参考文章：使用torchsummary打印torch每层形状：https://www.cnblogs.com/yyf2019/p/11674266.html

2. 使用 torchviz 输出模型结构图

• 安装Graphviz：sudo apt-get install graphviz
• 在虚拟环境下安装： pip install graphviz和 pip install torchviz
• 使用

from torchviz import make_dot

# ...... 

x = Variable(torch