From d833ab3493c83a10f32f5c5012875d4aba807b56 Mon Sep 17 00:00:00 2001 From: Jonathan L Long Date: Fri, 12 Sep 2014 16:07:34 -0700 Subject: [PATCH] check that LRN's local_size is odd as the current implementation requires --- src/caffe/layers/lrn_layer.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/src/caffe/layers/lrn_layer.cpp b/src/caffe/layers/lrn_layer.cpp index 55b7673..d9e41e9 100644 --- a/src/caffe/layers/lrn_layer.cpp +++ b/src/caffe/layers/lrn_layer.cpp @@ -10,6 +10,7 @@ template void LRNLayer::LayerSetUp(const vector*>& bottom, vector*>* top) { size_ = this->layer_param_.lrn_param().local_size(); + CHECK_EQ(size_ % 2, 1) << "LRN only supports odd values for local_size"; pre_pad_ = (size_ - 1) / 2; alpha_ = this->layer_param_.lrn_param().alpha(); beta_ = this->layer_param_.lrn_param().beta(); -- 2.7.4