batch_normalization
batch_normalization copied to clipboard
Batch Normalization Layer for Caffe
Batch Normalization Layer for Caffe
This implementation of Batch Normalization is based on MVNLayer in Caffe.
To add this layer, you have to modify common_layers.hpp
, layer_factory.cpp
and caffe.proto
. See Caffe wiki
NOTE
This implementation is very basic which just performs batch normalization computation. Two pieces in the paper are still missing here:
- fixed mean & variance for inference
- per batch shuffling for thorough randomness
Update log
2016/01/18 An improved version is provided here with moving average statistics for inference.