Fix error message typo in SoftmaxWithLossLayer
authortabe <tabe@preferred.jp>
Wed, 26 Feb 2014 02:00:56 +0000 (11:00 +0900)
committerEvan Shelhamer <shelhamer@imaginarynumber.net>
Wed, 26 Feb 2014 20:37:21 +0000 (12:37 -0800)
src/caffe/layers/softmax_loss_layer.cu

index 3a001c0..9bb2313 100644 (file)
@@ -15,7 +15,7 @@ namespace caffe {
 template <typename Dtype>
 void SoftmaxWithLossLayer<Dtype>::SetUp(const vector<Blob<Dtype>*>& bottom,
       vector<Blob<Dtype>*>* top) {
-  CHECK_EQ(bottom.size(), 2) << "SoftmaxLoss Layer takes a single blob as input.";
+  CHECK_EQ(bottom.size(), 2) << "SoftmaxLoss Layer takes two blobs as input.";
   CHECK_EQ(top->size(), 0) << "SoftmaxLoss Layer takes no blob as output.";
   softmax_bottom_vec_.clear();
   softmax_bottom_vec_.push_back(bottom[0]);