e-lab/torch-toolbox

weight initialization on nn.SpatialBatchNormalization

wuhuo714 opened this issue · 1 comments

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

jhjin commented

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