Fixed typo in Threshold Layer definition

diff --git a/include/caffe/neuron_layers.hpp b/include/caffe/neuron_layers.hpp
index 180ce7a..5118e8a 100644 (file)
--- a/include/caffe/neuron_layers.hpp
+++ b/include/caffe/neuron_layers.hpp
@@ -215,7 +215,7 @@ class TanHLayer : public NeuronLayer<Dtype> {
 template <typename Dtype>
 class ThresholdLayer : public NeuronLayer<Dtype> {
  public:
-  explicit DropoutLayer(const LayerParameter& param)
+  explicit ThresholdLayer(const LayerParameter& param)
       : NeuronLayer<Dtype>(param) {}
   virtual void SetUp(const vector<Blob<Dtype>*>& bottom,
       vector<Blob<Dtype>*>* top);