search

e-lab / torch-toolbox

A collection of snippets and libraries for Torch from e-Lab
https://engineering.purdue.edu/elab/
199 stars 64 forks source link

Is cudnn.SpatialConvolution working for weight initialization? #6

Closed dlmacedo closed 8 years ago

dlmacedo commented 8 years ago

Dear Friend,

I am interested in using the weight initialization lua script.

But in [https://github.com/NVIDIA/DIGITS/tree/master/examples/weight-init]() is said that cudnn.SpatialConvolution is not working. Is this true?

NOTE: at the time of writing, the torch-toolbox project does not handle all layers (most notably cudnn.SpatialConvolution) so you might need to tailor the script to your specific needs.

Thanks,

David

jhjin commented 8 years ago

@dlmacedo Thanks for your interest. Now the module is supported via the commit https://github.com/e-lab/torch-toolbox/commit/b749556653916ec5286c99e33f80b2c45dbf6c0b Please let me know if it works properly.