clear sigmoid top vec at initialization
authorJeff Donahue <jeff.donahue@gmail.com>
Wed, 16 Apr 2014 15:13:20 +0000 (08:13 -0700)
committerJeff Donahue <jeff.donahue@gmail.com>
Wed, 16 Apr 2014 15:13:20 +0000 (08:13 -0700)
src/caffe/layers/sigmoid_cross_entropy_loss_layer.cpp

index 7e3af42..f2186d6 100644 (file)
@@ -21,6 +21,7 @@ void SigmoidCrossEntropyLossLayer<Dtype>::SetUp(
       "SigmoidCrossEntropyLoss Layer takes no blob as output.";
   sigmoid_bottom_vec_.clear();
   sigmoid_bottom_vec_.push_back(bottom[0]);
+  sigmoid_top_vec_.clear();
   sigmoid_top_vec_.push_back(sigmoid_output_.get());
   sigmoid_layer_->SetUp(sigmoid_bottom_vec_, &sigmoid_top_vec_);
 }