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

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

查看所有标签

猜你喜欢:

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

码农书籍

算法竞赛入门经典

刘汝佳 / 清华大学出版社 / 2009-11 / 24.00元

《算法竞赛入门经典》是一本算法竞赛的入门教材，把C/C++语言、算法和解题有机地结合在了一起，淡化理论，注重学习方法和实践技巧。全书内容分为11章，包括程序设计入门、循环结构程序设计、数组和字符串、函数和递归、基础题目选解、数据结构基础、暴力求解法、高效算法设计、动态规划初步、数学概念与方法、图论模型与算法，覆盖了算法竞赛入门所需的主要知识点，并附有大量习题。书中的代码规范、简洁、易懂，不仅能帮助......一起来看看《算法竞赛入门经典》这本书的介绍吧!

码农工具