Add deconvolution operator, and the corresponding implement using cudnn_v3#422
Add deconvolution operator, and the corresponding implement using cudnn_v3#422antinucleon merged 12 commits intoapache:masterfrom tornadomeet:master
Conversation
tornadomeet
commented
Oct 29, 2015
- have done a simple deconvolution_shape_check;
- when using deconvolution layer, the mxnet's implemention and cudnn's implemention give the same output almostly. using the follow example: replace one line of the example/mnist/lenet.py , refer to deconv_lenet.py
|
Thanks for the great PR! Could you fix the lint? |
|
You can try make lint to see the lint result locally. |
|
fix lint now. |
|
Thanks! Another issue is could you add namespace for enum? for example https://github.com/dmlc/mxnet/blob/master/src/operator/convolution-inl.h#L24 . Because we will soon provide amalgamation version code, this part will make name conflict.. |
|
i'll add these and check now. |
|
now have added the namespace for deconvolution-inl.h and cudnn_deconvolution-inl.h. |
|
thanks for the PR! but what does deconvolution do? in my understanding it can be used to reserve a convolution layer and used for visualization. but in your lenet example, you just replace the conv layer by deconv layer for training? |
|
@tornadomeet very helpful, thanks! |
|
Last thing, could you add your name in Contributor.md? then I will merge it |
Add deconvolution operator, and the corresponding implement using cudnn_v3
|
ok, i have added my name now. |
* build: propagate USE_JEMALLOC see dmlc/MXNet.jl#418 (comment)