Pytorch 计算参数量与计算量Flops

获取网络模型的每一层参数量与计算量（Flops）———Pytorch

分析网络的参数量与计算量；

  1、ptflops
            安装： pip install --upgrade git+https://github.com/sovrasov/flops-counter.pytorch.git

           使用：具体使用参照 https://github.com/sovrasov/flops-counter.pytorch，在第三节我也贴上了我使用代码，有具体解释与注意事项，可看后面

           评价：博主就是用的这个，他可以直接看每一层的参数量与计算量，每一层的参数计算量占模型所有参数的百分比，且有每一层的卷积步长、核大小、输入输出通道数量等，并且也会输出总量；推荐

     2、thop
           安装：pip3 install thop    或者  pip3 install --upgrade git+https://github.com/Lyken17/pytorch-OpCounter.git

           使用：同样也可参照官方：https://github.com/Lyken17/pytorch-OpCounter，也可以看我使用的代码

          评价：也是很多人用的，可以自定义网络层计算，可看官方使用方法，也可以输出参数的总量与计算量，但是博主没发现如何输出每一层的计算量参数量，但是看到了借口，改源码可能需要编译，但是没时间就没弄了，以后有时间再研究；

三、使用实例
     1、ptflops使用<

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/106619126