深度学习模型的计算量评估

推荐一个网站,可以十分方便的评估计算量,将prototxt文件复制进入即可:https://dgschwend.github.io/netscope/#/editor

如有需要也可以自己写代码,可参考:
https://blog.csdn.net/minstyrain/article/details/80348123
与https://github.com/sovrasov/flops-counter.pytorch

发布了27 篇原创文章 · 获赞 6 · 访问量 4910

猜你喜欢

转载自blog.csdn.net/weixin_42107987/article/details/104061378