内容简介：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;
2,
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
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持 码农网
本站部分资源来源于网络，本站转载出于传递更多信息之目的，版权归原作者或者来源机构所有，如转载稿涉及版权问题，请联系我们。
构建之法（第二版）
邹欣 / 人民邮电出版社 / 2015-7 / 59
软件工程牵涉的范围很广, 同时也是一般院校的同学反映比较空洞乏味的课程。 但是软件工程的技术对于投身IT 产业的学生来说是非常重要的。作者邹欣有长达20年的一线软件开发经验，他利用业余时间在数所高校进行了长达6年的软件工程教学实践，总结出了在16周的时间内让同学们通过 “做中学 (Learning By Doing)” 掌握实用的软件工程技术的教学计划，并得到高校师生的积极反馈。在此基础上，作者对......一起来看看 《构建之法（第二版）》 这本书的介绍吧!
