From eea74c30e6f85634f6cbb47528b05f640db97155 Mon Sep 17 00:00:00 2001
From: Alireza Shafaei <alireza@shafaei.net>
Date: Tue, 12 Aug 2014 13:54:51 -0700
Subject: [PATCH] Fixed the GPU implementation of EuclideanLoss to report the
 loss to the top layer

---
 src/caffe/layers/euclidean_loss_layer.cu | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/caffe/layers/euclidean_loss_layer.cu b/src/caffe/layers/euclidean_loss_layer.cu
index dd14f19..f4dfd0b 100644
--- a/src/caffe/layers/euclidean_loss_layer.cu
+++ b/src/caffe/layers/euclidean_loss_layer.cu
@@ -19,6 +19,9 @@ Dtype EuclideanLossLayer<Dtype>::Forward_gpu(const vector<Blob<Dtype>*>& bottom,
   Dtype dot;
   caffe_gpu_dot(count, diff_.gpu_data(), diff_.gpu_data(), &dot);
   Dtype loss = dot / bottom[0]->num() / Dtype(2);
+  if (top->size() == 1) {
+    (*top)[0]->mutable_cpu_data()[0] = loss;
+  }
   return loss;
 }
 
-- 
2.7.4