Kan Wu created MXNET-403:
----------------------------
Summary: fix gradient number and document for box_iou operator
Key: MXNET-403
URL: https://issues.apache.org/jira/browse/MXNET-403
Project: Apache MXNet
Issue Type: Bug
Reporter: Kan Wu
{color:#24292e}Hi.{color}
{color:#24292e} There is a bug in mxnet.symbol.contrib.box_iou.{color}
{color:#24292e} When trying to get the gradient of box_iou, it will cause that
{color}{{mxnet.base.MXNetError: [12:31:16] src/pass/gradient.cc:171: Check
failed: (*rit)->inputs.size() == input_grads.size() (2 vs. 1) Gradient function
not returning enough gradient}}{color:#24292e}.{color}
{color:#24292e} I fix the number of gradient and the document in box_iou
operator.{color}
Testing Code:
{color:#d73a49}import{color} mxnet {color:#d73a49}as{color} mx
{color:#d73a49}from{color} mxnet {color:#d73a49}import{color} autograd
x {color:#d73a49}={color}
mx.nd.array([[{color:#005cc5}0.5{color},{color:#005cc5}0.5{color},{color:#005cc5}1.0{color},{color:#005cc5}1.0{color}],
[{color:#005cc5}0.0{color},{color:#005cc5}0.0{color},{color:#005cc5}0.5{color},{color:#005cc5}0.5{color}]])
y {color:#d73a49}={color}
mx.nd.array([[{color:#005cc5}0.25{color},{color:#005cc5}0.25{color},{color:#005cc5}0.75{color},{color:#005cc5}0.75{color}]])
x.attach_grad()
y.attach_grad()
{color:#d73a49}with{color} autograd.record():
out {color:#d73a49}={color} mx.nd.contrib.box_iou(x,y)
{color:#005cc5}print{color} (out)
out.backward()
--
This message was sent by Atlassian JIRA
(v7.6.3#76005)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
