项目场景：

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: 输入的样本

标签：torchsummaryX,模型,thop,Pytorch,计算,pip,input,net,FLOPs
来源： https://blog.csdn.net/qq_42530347/article/details/115659828