fix softmax loss layer bug; all tests pass

This commit is contained in:
Jeff Donahue 2014-03-13 22:44:41 -07:00
Родитель 8a3f0c225b
Коммит ed23b68906
1 изменённых файлов: 1 добавлений и 3 удалений

Просмотреть файл

@ -16,9 +16,7 @@ template <typename Dtype>
Dtype SoftmaxWithLossLayer<Dtype>::Forward_gpu(
const vector<Blob<Dtype>*>& bottom, vector<Blob<Dtype>*>* top) {
// The forward pass computes the softmax prob values.
softmax_bottom_vec_[0] = bottom[0];
softmax_layer_->Forward(softmax_bottom_vec_, &softmax_top_vec_);
return Dtype(0);
return Forward_cpu(bottom, top);
}
template <typename Dtype>