1.安装直接使用 pip
进行安装即可:
pip install torchinfo
2.导入该模块
from torchinfo import summary
3.使用模块
summary(model)#这里的model是你自己的model,可以添加参数进去
4.效果图:
第一个图片是直接打印model吗,比较杂乱
第二个图片是使用summary
1.安装直接使用 pip
进行安装即可:
pip install torchinfo
2.导入该模块
from torchinfo import summary
3.使用模块
summary(model)#这里的model是你自己的model,可以添加参数进去
4.效果图:
第一个图片是直接打印model吗,比较杂乱
第二个图片是使用summary