weight initialization on nn.SpatialBatchNormalization

[wuhuo714]

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

[jhjin]

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