推荐一个网站,可以十分方便的评估计算量,将prototxt文件复制进入即可:https://dgschwend.github.io/netscope/#/editor
如有需要也可以自己写代码,可参考:
https://blog.csdn.net/minstyrain/article/details/80348123
与https://github.com/sovrasov/flops-counter.pytorch
推荐一个网站,可以十分方便的评估计算量,将prototxt文件复制进入即可:https://dgschwend.github.io/netscope/#/editor
如有需要也可以自己写代码,可参考:
https://blog.csdn.net/minstyrain/article/details/80348123
与https://github.com/sovrasov/flops-counter.pytorch