This is an automated email from the ASF dual-hosted git repository.

anirudh2290 pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git


The following commit(s) were added to refs/heads/master by this push:
     new af55104  Fix the operator API documentation (#12942)
af55104 is described below

commit af551045a4a26d8d12d7c2b23edc2cba3c8163e9
Author: Lin Yuan <[email protected]>
AuthorDate: Wed Oct 24 11:05:44 2018 -0700

    Fix the operator API documentation (#12942)
    
    * Fix the operator API documentation
    
    * update message
---
 src/operator/contrib/multibox_target-inl.h | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/operator/contrib/multibox_target-inl.h 
b/src/operator/contrib/multibox_target-inl.h
index f5a3b11..daf870a 100644
--- a/src/operator/contrib/multibox_target-inl.h
+++ b/src/operator/contrib/multibox_target-inl.h
@@ -218,15 +218,15 @@ class MultiBoxTargetProp : public OperatorProperty {
     CHECK_EQ(in_shape->size(), 3) << "Input: [anchor, label, clsPred]";
     TShape ashape = in_shape->at(mboxtarget_enum::kAnchor);
     CHECK_EQ(ashape.ndim(), 3) << "Anchor should be batch shared N*4 tensor";
-    CHECK_EQ(ashape[0], 1) << "Anchors are shared across batches, first dim=1";
-    CHECK_GT(ashape[1], 0) << "Number boxes should > 0";
-    CHECK_EQ(ashape[2], 4) << "Box dimension should be 4: 
[xmin-ymin-xmax-ymax]";
+    CHECK_EQ(ashape[0], 1) << "Anchors are shared across batches, first 
dimension should be 1";
+    CHECK_GT(ashape[1], 0) << "Number boxes should be greater than 0";
+    CHECK_EQ(ashape[2], 4) << "Box dimension should be 4: [xmin, ymin, xmax, 
ymax]";
     TShape lshape = in_shape->at(mboxtarget_enum::kLabel);
-    CHECK_EQ(lshape.ndim(), 3) << "Label should be [batch-num_labels-(>=5)] 
tensor";
-    CHECK_GT(lshape[1], 0) << "Padded label should > 0";
-    CHECK_GE(lshape[2], 5) << "Label width must >=5";
+    CHECK_EQ(lshape.ndim(), 3) << "Label should be [batch, num_labels, 
label_width] tensor";
+    CHECK_GT(lshape[1], 0) << "Padded label should be greater than 0";
+    CHECK_GE(lshape[2], 5) << "Label width should be greater than or equal to 
5";
     TShape pshape = in_shape->at(mboxtarget_enum::kClsPred);
-    CHECK_EQ(pshape.ndim(), 3) << "Prediction: 
[nbatch-num_classes-num_anchors]";
+    CHECK_EQ(pshape.ndim(), 3) << "Prediction: [batch, num_classes, 
num_anchors]";
     CHECK_EQ(pshape[2], ashape[1]) << "Number of anchors mismatch";
     TShape loc_shape = Shape2(lshape[0], ashape.Size());  // batch - (num_box 
* 4)
     TShape lm_shape = loc_shape;

Reply via email to