From ced8c6c495f2f9a435c6d8f737d1d00d49d3fc77 Mon Sep 17 00:00:00 2001 From: momo609 <963372609.qq.com> Date: Mon, 20 Mar 2023 19:05:25 +0800 Subject: [PATCH] update ops.md --- tests/test_ops/test_roi_align.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_ops/test_roi_align.py b/tests/test_ops/test_roi_align.py index 01c183902a5..14b43e5abe7 100644 --- a/tests/test_ops/test_roi_align.py +++ b/tests/test_ops/test_roi_align.py @@ -120,5 +120,5 @@ def _test_roialign_allclose(device, dtype): def test_roialign(device, dtype): # check double only if dtype is torch.double: - _test_roialign_gradcheck(device=device, dtype=dtype) + _test_roialign_gradcheck(device=device, dtype=dtype) _test_roialign_allclose(device=device, dtype=dtype)