fix test_max_pool2d

2020-12-25 13:38:22 +08:00 · 2020-12-25 13:38:22 +08:00 · ced499215d
parent 49ac9be019
commit ced499215d
1 changed files with 1 additions and 1 deletions
--- a/python/jittor/test/test_arg_pool_op.py
+++ b/python/jittor/test/test_arg_pool_op.py
@ -173,7 +173,7 @@ class TestArgPoolOp(unittest.TestCase):
        check(jt_model, torch_model, shape, False)
        print('finish')

-    def test_avg_pool2d(self):
+    def test_max_pool2d(self):
        from torch.nn.functional import max_pool2d
        arr = np.random.random((2, 16, 33, 33))
        jt_model = j_max_pool2d(jt.array(arr), 3, 1, 1, ceil_mode=True)