文章标题
2016-07-27 17:17
585 查看
import mxnet as mx def mlp(): data=mx.sym.Variable('data') fc1=mx.sym.FullyConnected(data, name="fc1", num_hidden=512) act1=mx.sym.Activation(fc1, name="relu1", act_type="relu") fc2=mx.sym.FullyConnected(act1, name="fc2", num_hidden=512) act2=mx.sym.Activation(fc2, name="relu2", act_type="relu") fc3=mx.sym.FullyConnected(act2, name="fc3", num_hidden=10) mlp=mx.sym.SoftmaxOutput(fc3, name="softmax") return mlp if __name__=="__main__": num_epoch=3 batch_size=100 train_dataiter=mx.io.CSVIter(data_csv="mnist.train", data_shape=(28, 28), label_csv="label.train", label_shape=(1,), batch_size=batch_size) val_dataiter=mx.io.CSVIter(data_csv="mnist.val", data_shape=(28, 28), label_csv="label.val", label_shape=(1,), batch_size=batch_size) mlp=mlp() ###config model_args model_args = dict() ##the first parameter is the number of the batch_num model_args['lr_scheduler'] = mx.lr_scheduler.FactorScheduler(500, 0.9) model=mx.model.FeedForward( ctx=mx.gpu(0), symbol=mlp, num_epoch=5, learning_rate=0.01, momentum=0.9, wd=0.01, **model_args) ####cofig log file import logging LOG_FILE='mnist.log' logging.basicConfig(filename=LOG_FILE, level=logging.DEBUG) model.fit( X=train_dataiter, eval_data=val_dataiter, batch_end_callback=mx.callback.Speedometer(batch_size, 50),)
相关文章推荐
- Mxnet Scala Package 学习笔记 一
- mxnet中im2rec.py的小问题
- Ubuntu 15.04+GTX 1080+MXNet+CUDA+CUDNN
- 用MXnet入门实战深度学习之一:安装GPU版mxnet并跑一个MNIST手写数字识别
- 用MXnet入门实战深度学习之一:安装GPU版mxnet并跑一个MNIST手写数字识别
- 用MXnet入门实战深度学习之一:安装GPU版mxnet并跑一个MNIST手写数字识别
- 用MXnet入门实战深度学习之一:安装GPU版mxnet并跑一个MNIST手写数字识别
- 用MXnet入门实战深度学习之一:安装GPU版mxnet并跑一个MNIST手写数字识别
- 用MXnet入门实战深度学习之一:安装GPU版mxnet并跑一个MNIST手写数字识别
- 用MXnet入门实战深度学习之一:安装GPU版mxnet并跑一个MNIST手写数字识别
- 用MXnet入门实战深度学习之一:安装GPU版mxnet并跑一个MNIST手写数字识别
- 用MXnet入门实战深度学习之一:安装GPU版mxnet并跑一个MNIST手写数字识别
- 用MXnet入门实战深度学习之一:安装GPU版mxnet并跑一个MNIST手写数字识别
- 用MXnet入门实战深度学习之一:安装GPU版mxnet并跑一个MNIST手写数字识别
- 用MXnet入门实战深度学习之一:安装GPU版mxnet并跑一个MNIST手写数字识别
- 深度学习二:Neural art:用机器模仿梵高
- mxnet实战之艺术画
- csviter mxnet
- MXNet系统上ilsvrc12数据集的制作与inception-bn网络的训练
- mxnet代码解析之dependency engine