pytorch之网络参数统计 torchstat & torchsummary

参考 :

https://blog.csdn.net/weixin_45292794/article/details/108227437

 

https://blog.csdn.net/jzwong/article/details/110954725

 

 

 

=======================================================

 

 

 

 

两个pytorch的网络参数统计工具：

 

 

1.  torchstat

pip install torchstat

 

 

 

 

2.  torchsummary

两种安装方式，测试发现均可使用：

pip install torchsummary

或

pip install torch-summary

 

 

 

 

 

 

 

 

 

 

 

===============================================

 

 

 

 

 

使用方式：

1.  torchstat

from torchstat import stat
import torchvision.models as models
model = models.resnet152()
stat(model, (3, 224, 224))

 

 

 

 

 

 

 

 

 

 

 

 

2.  torchsummary

import torchvision.models as models
from torchsummary import summary

model = models.resnet152()
summary(model.cuda(),input_size=(3,32,32),batch_size=-1)

 

如果安装方式：

pip install torchsummary

 

那么打印结果没有层次感：

 

 

 

 

 

 

如果安装方式：

pip install torch-summary

 

那么打印结果有层次感：

 

 

 

 

 

 

使用起来还是   pip install torch-summary  显示结果简洁清爽，不过功能强大上还是  pip install torchstat   更胜一筹。

 

 

 

 

建议配合使用：

torch-summary只能看到网络结构和参数数量：

 

 

 

 

 

 

 

torchstat 可以看到更多的信息，如参数的内存大小：

 

 

 

 

 关于计算指标：

FLOPS、MAdds、MACC指标：

 https://blog.csdn.net/sunlanchang/article/details/103666342

 

 

========================================================