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使用<

你可能感兴趣的:(深度学习宝典)