fix the typo

apache · Dec 27, 2018 · 85f0a65 · 85f0a65
1 parent 7340f14
commit 85f0a65
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/operator/contrib/roi_align.cc b/src/operator/contrib/roi_align.cc
@@ -563,9 +563,9 @@ He, Kaiming, et al. "Mask R-CNN." ICCV, 2017
   // out: [num_rois, c, pooled_h, pooled_w]
   out_shape->clear();
   if (param.position_sensitive) {
+    CHECK_EQ(dshape[1] % (param.pooled_size[0]*param.pooled_size[1]), 0) <<
+      "Input channels shuold be divided by pooled_size[0]*pooled_size[1]";
     out_shape->push_back(
-         CHECK_EQ(dshape[1] % (param.pooled_size[0]*param.pooled_size[1]), 0) <<
-           "Input channels shuold be divided by pooled_size[0]*pooled_size[1]";
          Shape4(bshape[0], dshape[1]/param.pooled_size[0]/param.pooled_size[1],
                 param.pooled_size[0], param.pooled_size[1]));
   } else {