weight initialization on nn.SpatialBatchNormalization
wuhuo714 opened this issue · 1 comments
wuhuo714 commented
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