use DCHECK in SoftmaxLossLayer so that bounds checking has no perf pe…

…nalty without DEBUG
Sigmame · Nov 24, 2014 · 33eb36e · 33eb36e
1 parent 9443c44
commit 33eb36e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/caffe/layers/softmax_loss_layer.cpp b/src/caffe/layers/softmax_loss_layer.cpp
@@ -44,8 +44,8 @@ void SoftmaxWithLossLayer<Dtype>::Forward_cpu(
   for (int i = 0; i < num; ++i) {
     for (int j = 0; j < spatial_dim; j++) {
       const int label_value = static_cast<int>(label[i * spatial_dim + j]);
-      CHECK_GE(label_value, 0);
-      CHECK_GT(dim, label_value * spatial_dim);
+      DCHECK_GE(label_value, 0);
+      DCHECK_GT(dim, label_value * spatial_dim);
       loss -= log(std::max(prob_data[i * dim +
           label_value * spatial_dim + j],
                            Dtype(FLT_MIN)));