From 1dacdd819544bd491a1c7e00349ea1acae5bc66c Mon Sep 17 00:00:00 2001 From: Yangqing Jia Date: Sun, 28 Sep 2014 16:04:53 -0700 Subject: [PATCH] add long-ignored threshold layer --- src/caffe/layer_factory.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/src/caffe/layer_factory.cpp b/src/caffe/layer_factory.cpp index b60b3b6..d0bbd3f 100644 --- a/src/caffe/layer_factory.cpp +++ b/src/caffe/layer_factory.cpp @@ -180,6 +180,7 @@ REGISTER_LAYER_CLASS(SIGMOID_CROSS_ENTROPY_LOSS, SigmoidCrossEntropyLossLayer); REGISTER_LAYER_CLASS(SLICE, SliceLayer); REGISTER_LAYER_CLASS(SOFTMAX_LOSS, SoftmaxWithLossLayer); REGISTER_LAYER_CLASS(SPLIT, SplitLayer); +REGISTER_LAYER_CLASS(THRESHOLD, ThresholdLayer); REGISTER_LAYER_CLASS(WINDOW_DATA, WindowDataLayer); } // namespace caffe -- 2.7.4