diff options
Diffstat (limited to 'src/caffe/layers/softmax_loss_layer.cu')
-rw-r--r-- | src/caffe/layers/softmax_loss_layer.cu | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/src/caffe/layers/softmax_loss_layer.cu b/src/caffe/layers/softmax_loss_layer.cu index 32f3e670..9ef8dd23 100644 --- a/src/caffe/layers/softmax_loss_layer.cu +++ b/src/caffe/layers/softmax_loss_layer.cu @@ -9,10 +9,9 @@ namespace caffe { template <typename Dtype> -Dtype SoftmaxWithLossLayer<Dtype>::Forward_gpu( +void SoftmaxWithLossLayer<Dtype>::Forward_gpu( const vector<Blob<Dtype>*>& bottom, vector<Blob<Dtype>*>* top) { - // The forward pass computes the softmax prob values. - return Forward_cpu(bottom, top); + Forward_cpu(bottom, top); } template <typename Dtype> |