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
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.rhkb.cn/news/357636.html
如若内容造成侵权/违法违规/事实不符,请联系长河编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!