torchsummary和torch-summary的区别
torchsummary和torch_summary的区别
- 同
- 异
- 安装
- 功能
- torch-summary 输出
- torchsummary 输出
一开始torchsummary 和 torch-summary 傻傻分不清楚。写一篇简单介绍一下吧
推荐使用 torch-summary
同
基础用法都是一样的。
from torchsummary import summarymodel = Model()summary(model,(3,224,224),device='cpu')
异
安装
区别仅仅是一个横线
pip install torchsummary
pip install torch-summary
功能
torch-summary 可以看成torchsummary的加强版。支持更多功能。比如自带输出模型计算量、每层之间有层次感等等
torch-summary 输出
torchsummary 输出
本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！