-
Notifications
You must be signed in to change notification settings - Fork 5.6k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Refine prior_box_op #8151
Refine prior_box_op #8151
Conversation
paddle/operators/prior_box_op.cc
Outdated
.SetDefault(0.0); | ||
.SetDefault(0.0) | ||
.AddCustomChecker([](const float& step_w) { | ||
PADDLE_ENFORCE_GT(step_w, 0.0, "step_h should be larger than 0."); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
step_h should be -> step_w should be
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Done
c317825
to
d7a371c
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM.
44268e0
to
98089c4
Compare
98089c4
to
f367ad6
Compare
The checking of attribution should be placed in
OpMaker
.The
InferShape
ofxxOp
and theCompute
ofxxKernel
will be called every iteration, but the attribution is immutable. So it just needs to be checked once.