1.安装直接使用 pip 进行安装即可：

 pip install torchinfo

2.导入该模块

from torchinfo import summary

3.使用模块

summary(model)#这里的model是你自己的model，可以添加参数进去

4.效果图：

第一个图片是直接打印model吗，比较杂乱

第二个图片是使用summary