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

weight initialization on nn.SpatialBatchNormalization #19

Open wuhuo714 opened 7 years ago

wuhuo714 commented 7 years ago

Hi, can weight-init.lua initialize the weights and bias of the layer nn.SpatialBatchNormalization ?

jhjin commented 7 years ago

it does not support nn.SpatialBatchNormalization. You can try the similar way as other supported function but with the reset function overloading https://github.com/torch/nn/blob/master/BatchNormalization.lua#L65