From 5d9d02db5f02a95b2e69344d135ae05c78d7988c Mon Sep 17 00:00:00 2001 From: wjj19950828 Date: Mon, 15 Aug 2022 12:10:40 +0800 Subject: [PATCH] fixed bug --- tests/onnx/test_auto_scan_averagepool_10.py | 9 +++++---- tests/onnx/test_auto_scan_averagepool_7.py | 2 +- 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/tests/onnx/test_auto_scan_averagepool_10.py b/tests/onnx/test_auto_scan_averagepool_10.py index bfe35e0..1e569a2 100644 --- a/tests/onnx/test_auto_scan_averagepool_10.py +++ b/tests/onnx/test_auto_scan_averagepool_10.py @@ -30,8 +30,9 @@ class TestAveragePoolConvert(OPConvertAutoScanTest): input_shape = draw( st.lists( st.integers( - min_value=10, max_value=20), min_size=4, max_size=4)) - + min_value=20, max_value=30), min_size=4, max_size=4)) + # BS = 1 + input_shape[0] = 1 kernel_size = draw( st.lists( st.integers( @@ -40,7 +41,7 @@ class TestAveragePoolConvert(OPConvertAutoScanTest): strides = draw( st.lists( st.integers( - min_value=1, max_value=5), min_size=2, max_size=2)) + min_value=1, max_value=2), min_size=2, max_size=2)) if draw(st.booleans()): auto_pad = "NOTSET" @@ -96,7 +97,7 @@ class TestAveragePoolConvert(OPConvertAutoScanTest): return (config, attrs) def test(self): - self.run_and_statis(max_examples=30) + self.run_and_statis(max_examples=300) if __name__ == "__main__": diff --git a/tests/onnx/test_auto_scan_averagepool_7.py b/tests/onnx/test_auto_scan_averagepool_7.py index e0a68cf..666800b 100644 --- a/tests/onnx/test_auto_scan_averagepool_7.py +++ b/tests/onnx/test_auto_scan_averagepool_7.py @@ -40,7 +40,7 @@ class TestAveragePoolConvert(OPConvertAutoScanTest): strides = draw( st.lists( st.integers( - min_value=1, max_value=5), min_size=2, max_size=2)) + min_value=1, max_value=2), min_size=2, max_size=2)) if draw(st.booleans()): auto_pad = "NOTSET" -- GitLab