euclidean layer update
[jacinto-ai/caffe-jacinto.git] / src / caffe / layer_factory.hpp
index 8453fd51980cff30e99ff2e6ea813c2c686d853f..e9f8dbb4aa76ceee22dcdcfc0db29b6c20e2950d 100644 (file)
@@ -25,6 +25,8 @@ Layer<Dtype>* GetLayer(const LayerParameter& param) {
     return new DataLayer<Dtype>(param);
   } else if (type == "dropout") {
     return new DropoutLayer<Dtype>(param);
+  } else if (type == "euclidean_loss") {
+    return new EuclideanLossLayer<Dtype>(param);
   } else if (type == "im2col") {
     return new Im2colLayer<Dtype>(param);
   } else if (type == "innerproduct") {