This is an automated email from the ASF dual-hosted git repository.
wuwei pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 725ae75 Fix flaky NMS test by making sure scores are unique (#9140)
725ae75 is described below
commit 725ae75af4997ff3a5107cc82d64609773de23a0
Author: masahi <[email protected]>
AuthorDate: Thu Sep 30 02:30:13 2021 +0900
Fix flaky NMS test by making sure scores are unique (#9140)
---
tests/python/frontend/pytorch/test_forward.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/tests/python/frontend/pytorch/test_forward.py
b/tests/python/frontend/pytorch/test_forward.py
index c27469e..9238acd 100644
--- a/tests/python/frontend/pytorch/test_forward.py
+++ b/tests/python/frontend/pytorch/test_forward.py
@@ -1963,8 +1963,9 @@ def test_forward_nms():
boxes = torch.rand(num_boxes, box_len, dtype=torch.float) * 0.5
boxes[:, 2] += boxes[:, 0]
boxes[:, 3] += boxes[:, 1]
- scores = torch.from_numpy(np.random.uniform(-1, 1,
size=(num_boxes,)).astype(np.float32))
- return boxes, scores
+ scores = np.linspace(0, 1, num=num_boxes).astype("float32")
+ np.random.shuffle(scores)
+ return boxes, torch.from_numpy(scores)
targets = ["llvm", "cuda"]
