fix forward loss computation

97589713 · Mar 16, 2016 · 9e74dd7 · 9e74dd7
1 parent 5eb9499
commit 9e74dd7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/caffe/layers/sigmoid_cross_entropy_loss_layer.cpp b/src/caffe/layers/sigmoid_cross_entropy_loss_layer.cpp
@@ -51,6 +51,8 @@ void SigmoidCrossEntropyLossLayer<Dtype>::Forward_cpu(
   for (int i = 0; i < num; ++i) {
       temp_loss_pos = 0;
       temp_loss_neg = 0;
+      count_pos = 0;
+      count_neg = 0;
       for (int j = 0; j < dim; j ++) {
          if (target[i*dim+j] == 1) {
         	count_pos ++;