diff --git a/src/caffe/layers/batch_norm_layer.cpp b/src/caffe/layers/batch_norm_layer.cpp index c6a1d5b1b2c..fd1c74906fb 100644 --- a/src/caffe/layers/batch_norm_layer.cpp +++ b/src/caffe/layers/batch_norm_layer.cpp @@ -51,7 +51,7 @@ void BatchNormLayer::LayerSetUp(const vector*>& bottom, template void BatchNormLayer::Reshape(const vector*>& bottom, const vector*>& top) { - if (bottom[0]->num_axes() >= 1) + if (bottom[0]->num_axes() >= 2) CHECK_EQ(bottom[0]->shape(1), channels_); top[0]->ReshapeLike(*bottom[0]);