- 博客(5)
- 收藏
- 关注
原创 mxnet如何打印symbol输出维度
arg_name = fc1.list_arguments()out_name = fc1.list_outputs()arg_shape, out_shape, _ = fc1.infer_shape(data=(1,3,112,112))print({'input' : dict(zip(arg_name, arg_shape)),'output' : dict(zip(out_name...
2018-08-29 10:33:52 2963
原创 mxnet(gluon)学习之路-自动求导
简介mxnet 提供了自动求导的方法,相比于使用caffe需要自己写反向传播,这可以更加节约我们的时间。求导例如: y = 4 * x^2 x = [[1,2; 3,4]] dy/dx = 8x = [[8,16;24,32]] 使用mxnet自动求导: x = mx.nd.array([[1,2], [3,4]]) x.attach_grad() ...
2018-08-05 16:14:54 1475
原创 mxnet(gluon)学习之路-mnist训练
训练流程import mxnet as mxfrom mxnet.gluon import loss as gloss, nnimport mxnet.gluon as gluonfrom mxnet import autogradimport mxnet.ndarray as ndimport numpy as npimport mxnet.metricclass LeNet...
2018-08-05 00:51:45 1484
原创 mxnet(gluon)学习之路-使用HybridBlock构建网络
简介mxnet gluno接口提供 Sequential, HybridSequential 通过 add 串联的形式将构建网络,同时也提供HybridBlock通过继承的方式来构建网络,那么他们之间有什么区别呢? 1. Sequential构建的是动态图,即命令式编程形式,这种形式可以很方便的debug。例如:net = nn.Sequential()net.add( ...
2018-08-04 23:49:33 5465
原创 mxnet(gluon)学习之路-网络可视化
如何可视化网络神经网络的可视化是十分有用的,对于之前学习caffe的同学来说可以使用 netscope来进行可视化,而对于mxnet来说我们可以使用mxnet自身的函数完成可视化。import mxnet as mxfrom mxnet import gluonnum_hidden = 64net = gluon.nn.HybridSequential()with net.nam...
2018-08-04 23:06:03 1312
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人