зеркало из https://github.com/microsoft/caffe.git
fix softmax loss layer bug; all tests pass
This commit is contained in:
Родитель
8a3f0c225b
Коммит
ed23b68906
|
@ -16,9 +16,7 @@ template <typename Dtype>
|
||||||
Dtype SoftmaxWithLossLayer<Dtype>::Forward_gpu(
|
Dtype SoftmaxWithLossLayer<Dtype>::Forward_gpu(
|
||||||
const vector<Blob<Dtype>*>& bottom, vector<Blob<Dtype>*>* top) {
|
const vector<Blob<Dtype>*>& bottom, vector<Blob<Dtype>*>* top) {
|
||||||
// The forward pass computes the softmax prob values.
|
// The forward pass computes the softmax prob values.
|
||||||
softmax_bottom_vec_[0] = bottom[0];
|
return Forward_cpu(bottom, top);
|
||||||
softmax_layer_->Forward(softmax_bottom_vec_, &softmax_top_vec_);
|
|
||||||
return Dtype(0);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
template <typename Dtype>
|
template <typename Dtype>
|
||||||
|
|
Загрузка…
Ссылка в новой задаче