Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

[MXNET-1258]fix unittest for ROIAlign Operator #13609

Merged
merged 12 commits into from
Feb 5, 2019
Prev Previous commit
Next Next commit
merge master
  • Loading branch information
wkcn committed Jan 15, 2019
commit 521eb5c58de91f77060c4196e6a3a81fefde92cf
6 changes: 3 additions & 3 deletions tests/python/unittest/test_operator.py
Original file line number Diff line number Diff line change
Expand Up @@ -6978,9 +6978,9 @@ def test_roi_align_value(sampling_ratio=0):
output.backward(dy)
real_output, [dx, drois] = roialign_forward_backward(data.asnumpy(), rois.asnumpy(), pooled_size,
spatial_scale, sampling_ratio, dy.asnumpy())
assert_almost_equal(output.asnumpy(), real_output, atol=1e-5)
assert_almost_equal(data.grad.asnumpy(), dx, atol=1e-5)
assert_almost_equal(rois.grad.asnumpy(), drois, atol=1e-5)
assert_almost_equal(output.asnumpy(), real_output, atol=1e-3)
assert_almost_equal(data.grad.asnumpy(), dx, atol=1e-3)
assert_almost_equal(rois.grad.asnumpy(), drois, atol=1e-3)

# modified from test_roipooling()
def test_roi_align_autograd(sampling_ratio=0):
Expand Down