pytorch之网络参数统计 torchstat & torchsummary

两个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指标：