本文介绍了如何在 CNN 中计算网络的 FLOPs的处理方法，对大家解决问题具有一定的参考价值，需要的朋友们下面随着小编来一起学习吧！
问题描述
我想设计一个占用GPU资源不超过Alexnet的卷积神经网络.我想用FLOPs来测量它，但我不知道如何计算它.请问有什么工具可以做到吗?
I want to design a convolutional neural network which occupy GPU resource no more than Alexnet.I want to use FLOPs to measure it but I don't know how to calculate it.Is there any tools to do it,please?
推荐答案
有关在线工具，请参阅 http://dgschwend.github.io/netscope/#/editor .对于 alexnet，请参见 http://dgschwend.github.io/netscope/#/preset/alexnet .这支持最广泛的已知层.对于自定义图层，您必须自己计算.
For online tool see http://dgschwend.github.io/netscope/#/editor . For alexnet see http://dgschwend.github.io/netscope/#/preset/alexnet . This supports most wide known layers. For custom layers you will have to calculate yourself.
这篇关于如何在 CNN 中计算网络的 FLOPs的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！