Pytorch 计算模型FLOPs 和 参数量(两种方法)

项目场景：

Pytorch 框架中，计算深度神经网络的 FLOPS 和 参数量

解决方案：

方法一：
利用thop
1.安装thop

pip install thop

2.计算

@hye
      from thop import profile
      flops, params = profile(net, (input, ))

net: 需要计算的模型
input: 输入的样本

方法二：
利用torchsummaryX
1.安装torchsummaryX

pip install torchsummaryX

2.计算

@hye
      from torchsummaryX import summary
      summary(net, input)

net: 需要计算的模型
input: 输入的样本