fix a bug

8 years ago · 141a323c34
parent 0c4446868b
commit 141a323c34
1 changed files with 1 additions and 1 deletions
--- a/paddle/operators/spp_op.cc
+++ b/paddle/operators/spp_op.cc
@ -29,7 +29,7 @@ class SppOpMaker : public framework::OpProtoAndCheckerMaker {
              "(Tensor) The output tensor of spp operator."
              "N * M."
              "M = C * H * W");
-    AddAttr<int>("pyramid_height", "int", "multi level pooling");
+    AddAttr<int>("pyramid_height", "(int), multi level pooling");
    AddComment(R"DOC(
        "Does spatial pyramid pooling on the input image by taking the max,
        etc. within regions so that the result vector of different sized