MXNET-Scala 实现估算卷积神经网络浮点数运算量

栏目: Scala · 发布时间: 6年前

内容简介：Implementation of the estimation of model size and flop counts for convolutional neural networks with MXNET-Scala.For now, the estimation of flops only consider Layers: Convolution, Deconvolution, FullyConnected, Pooling, relu

MXNET-Scala Useful Tools

Implementation of the estimation of model size and flop counts for convolutional neural networks with MXNET-Scala.

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

For now, the estimation of flops only consider Layers: Convolution, Deconvolution, FullyConnected, Pooling, relu

Building

Tested on Ubuntu 14.04

Requirements

sbt 0.13
Mxnet

steps

1, compile Mxnet with CUDA, then compile the scala-pkg;

cd Mxnet-Scala/UsefulTools
mkdir lib

3, copy your compiled mxnet-full_2.11-linux-x86_64-gpu-1.3.1-SNAPSHOT.jar into lib folder;

4, run sbt, compile the project

Running

run cal_flops.sh under scripts folder

caffenet
flops: 723.0072 MFLOPS
model size: 232.56387 MB

squeezenet1-0
flops: 861.60394 MFLOPS
model size: 4.7623596 MB

resnet-101
flops: 7818.2407 MFLOPS
model size: 170.28586 MB

resnext-101-64x4d
flops: 15491.882 MFLOPS
model size: 319.13058 MB

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持码农网

查看所有标签

猜你喜欢:

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

码农书籍

Google 广告优化与工具

宫鑫 / 电子工业出版社 / 2010-7 / 60.00元

《Google 广告优化与工具》全面地阐述了Google AdWords这个高效广告投放平台的各方面内容，包括Google广告的渠道组成、质量得分、关键词和广告语策略、后期跟踪机制以及各种辅助工具的使用等。引导读者一步步建立一个强大的Google搜索引擎营销投放策略。本书案例丰富、知识面广且层次清晰，适用于各个层次的搜索引擎营销优化人员，不论是电商企业的领导人、工作在账户优化第一线的具体操作者、营......一起来看看《Google 广告优化与工具》这本书的介绍吧!

码农工具