这篇教程计算pytorch构建的网络的参数，空间大小，MAdd，FLOPs等指标写得很实用，希望能帮到您。
推荐一个神器（pytorch）：torchstat
可以用来计算pytorch构建的网络的参数，空间大小，MAdd，FLOPs等指标，简单好用。
比如：我想知道alexnet的网络的一些参数。
只需要：
from torchstat import stat
import torchvision.models as models
model = model.alexnet()
stat(model, (3, 224, 224))
就能得到结果啦：
https://github.com/sovrasov/flops-counter.pytorch
深度学习模型运行的浮点次数FLOPs和训练参数程序自动获取方法
CNN 模型所需的计算力（flops）和参数（parameters）数量是怎么计算的？ |
