本文介绍了如何在CNN中计算网的FLOP的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
我想设计一个卷积神经网络,它仅占用Alexnet即可占用GPU资源.我想使用FLOP对其进行测量,但我不知道如何计算.请问有没有工具可以做到这一点? /p>
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中计算网的FLOP的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持!