实现估算 MXNET 卷积神经网络浮点数运算量 Python 版本

栏目: 数据库 · 发布时间: 5年前

内容简介：Implementation of the estimation of model size and flop counts for convolutional neural networks.ref:The estimation of flops only consider layers: Convolution, Deconvolution, FullyConnected, Pooling, relu

MXNET-Python Tool For Calculate Flops And Model Size

Implementation of the estimation of model size and flop counts for convolutional neural networks.

ref: https://github.com/albanie/convnet-burden

The estimation of flops only consider layers: Convolution, Deconvolution, FullyConnected, Pooling, relu

Scala Version

Running

python calculateFlops.py -s symbols/caffenet-symbol.json -ds  data,1,3,224,224 -ls prob_label,1,1000

('flops: ', '723.007176', ' MFLOPS')
('model size: ', '232.563873291', ' MB')


python calculateFlops.py -s symbols/squeezenet_v1.0-symbol.json -ds  data,1,3,224,224 -ls prob_label,1,1000

('flops: ', '861.603864', ' MFLOPS')
('model size: ', '4.76235961914', ' MB')


python calculateFlops.py -s symbols/resnet-101-symbol.json -ds  data,1,3,224,224 -ls softmax_label,1,1000 

('flops: ', '7818.240488', ' MFLOPS')
('model size: ', '169.912773132', ' MB')

python calculateFlops.py -s symbols/resnext-101-64x4d-symbol.json -ds data,1,3,224,224 -ls softmax_label,1,1000

('flops: ', '15491.88196', ' MFLOPS')
('model size: ', '318.356620789', ' MB')


python calculateFlops.py -s symbols/fcn8s-symbol.json -ds  data,1,3,384,384 -ls softmax_label,1,21,384,384

('flops: ', '120420.573296', ' MFLOPS')
('model size: ', '513.037715912', ' MB')


python calculateFlops.py -s symbols/fcn32s-symbol.json -ds data,1,3,384,384 -ls softmax_label,1,21,384,384

('flops: ', '120265.786832', ' MFLOPS')
('model size: ', '519.382160187', ' MB')

以上就是本文的全部内容，希望本文的内容对大家的学习或者工作能带来一定的帮助，也希望大家多多支持码农网

查看所有标签

猜你喜欢:

本站部分资源来源于网络，本站转载出于传递更多信息之目的，版权归原作者或者来源机构所有，如转载稿涉及版权问题，请联系我们。

码农书籍

阿里巴巴Java开发手册

杨冠宝 / 电子工业出版社 / 2018-1 / 35

《阿里巴巴Java开发手册》的愿景是码出高效，码出质量。它结合作者的开发经验和架构历程，提炼阿里巴巴集团技术团队的集体编程经验和软件设计智慧，浓缩成为立体的编程规范和最佳实践。众所周知，现代软件行业的高速发展对开发者的综合素质要求越来越高，因为不仅是编程相关的知识点，其他维度的知识点也会影响软件的最终交付质量，比如，数据库的表结构和索引设计缺陷可能带来软件的架构缺陷或性能风险；单元测试的失位导致集......一起来看看《阿里巴巴Java开发手册》这本书的介绍吧!

码农工具