请问MXNet Gluon如何求出网络的计算量(乘法和加法次数), 即论文中使用的GFLOPS?

连接打不开?

有兴趣pr一下吗?

国内好像打不开gist

不好意思,最近太忙了,你们自己参考一下吧。。几个月前我实现这个遇到的主要问题有两个:

  1. FLOPs的定义在不同情况下不太一样(也就是这个概念不太统一,计算方式不只有一种),尤其是绝大多数论文的计算方法和NVIDIA论文介绍(似乎正确的)的方法结果大相径庭 (大概差两倍)参考。

  2. 就是在前面提到的issue中讨论的到底计算哪些层的FLOPs确实很让人烦恼,而且不管计算多少这个结果都不会是绝对准确的,可以肯定的是conv和dense是占绝大多数的,个人建议只计算conv,dense和bn。

你好,感謝你在gist的那份FLOPs計算的code。

我在使用的時候發現關於group convolution的計算跟先前用其他方式計算的結果不一樣,
後來發現是group會被除到兩次,
因為input channel是參考weight而來的,
但是weight shape[1]本身已經被除過一次group了,後面算FLOPs的時候應該就不用再除group了。