计算算法复杂度FLOPs

搜遍全网,流传的在网上的能方便计算FLOPs的有3个小工具。

  1. THOP: PyTorch-OpCounter
  2. torchstat
  3. ptflops
    一个不行的话多试两个。我的程序用2、3报错,最终用的1

你可能感兴趣的:(算法,pytorch,深度学习)