Merge pull request apache#20 from GodBlessZhk/ts

fix bug: num_rois should be batch_images x rois_per_image
StOnEGiggity · Aug 2, 2017 · b493711 · b493711
2 parents b90b9b5 + bf0180d
commit b493711
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/operator/roi_pooling.cu b/src/operator/roi_pooling.cu
@@ -205,7 +205,7 @@ inline void ROIPoolBackwardAcc(const Tensor<gpu, 4, Dtype> &in_grad,
   Dtype *bottom_diff = in_grad.dptr_;
   Dtype *argmax_data = max_idx.dptr_;
   const int count = in_grad.shape_.Size();
-  const int num_rois = bbox.size(1);
+  const int num_rois = bbox.size(0) * bbox.size(1);
   const int channels = in_grad.size(1);
   const int height = in_grad.size(2);
   const int width = in_grad.size(3);