如何在CNN中计算网的FLOP [英] how to calculate a net's FLOPs in CNN

查看:90
本文介绍了如何在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的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!

查看全文
登录 关闭
扫码关注1秒登录
发送“验证码”获取 | 15天全站免登陆