Fix bug

DancingJane · Mar 21, 2019 · ca3759a · ca3759a
1 parent 3bbb4b7
commit ca3759a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/include/caffe/layers/yolov3_detection_output_layer.hpp b/include/caffe/layers/yolov3_detection_output_layer.hpp
@@ -75,9 +75,9 @@ class Yolov3DetectionOutputLayer : public Layer<Dtype> {
 	  const vector<Blob<Dtype>*>& top);
   /// @brief Not implemented
   virtual void Backward_cpu(const vector<Blob<Dtype>*>& top,
-	  const vector<bool>& propagate_down, const vector<Blob<Dtype>*>& bottom) {return;};
+	  const vector<bool>& propagate_down, const vector<Blob<Dtype>*>& bottom);
   virtual void Backward_gpu(const vector<Blob<Dtype>*>& top,
-	  const vector<bool>& propagate_down, const vector<Blob<Dtype>*>& bottom) {return;};
+	  const vector<bool>& propagate_down, const vector<Blob<Dtype>*>& bottom);
   int side_;
   int num_class_;
   int num_;

diff --git a/src/caffe/layers/yolov3_detection_output_layer.cpp b/src/caffe/layers/yolov3_detection_output_layer.cpp
@@ -330,8 +330,17 @@ void Yolov3DetectionOutputLayer<Dtype>::Forward_cpu(
   }
 
 }
-
+template <typename Dtype>
+void Yolov3DetectionOutputLayer<Dtype>::Backward_cpu(const vector<Blob<Dtype>*>& top,
+	const vector<bool>& propagate_down, const vector<Blob<Dtype>*>& bottom) {
+	return;
+}
 #ifdef CPU_ONLY
+	template <typename Dtype>
+	void Yolov3DetectionOutputLayer<Dtype>::Backward_gpu(const vector<Blob<Dtype>*>& top,
+		const vector<bool>& propagate_down, const vector<Blob<Dtype>*>& bottom) {
+		return;
+	}
   STUB_GPU_FORWARD(Yolov3DetectionOutputLayer, Forward);
 #endif