From 83cf6758bd312963a27bb5f8132415a4b3ec7534 Mon Sep 17 00:00:00 2001 From: Guanghua Yu <742925032@qq.com> Date: Thu, 8 Sep 2022 10:25:10 +0800 Subject: [PATCH] fix ptq UT error (#45846) --- .../slim/tests/test_post_training_quantization_mobilenetv1.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/python/paddle/fluid/contrib/slim/tests/test_post_training_quantization_mobilenetv1.py b/python/paddle/fluid/contrib/slim/tests/test_post_training_quantization_mobilenetv1.py index 499367d339c..00a4e2c2aa4 100644 --- a/python/paddle/fluid/contrib/slim/tests/test_post_training_quantization_mobilenetv1.py +++ b/python/paddle/fluid/contrib/slim/tests/test_post_training_quantization_mobilenetv1.py @@ -242,7 +242,6 @@ class TestPostTrainingQuantization(unittest.TestCase): model_path, quantizable_op_type, batch_size, - batch_nums, algo="KL", round_type="round", is_full_quantize=False, @@ -268,7 +267,6 @@ class TestPostTrainingQuantization(unittest.TestCase): batch_size=batch_size, batch_nums=batch_nums, algo=algo, - batch_nums=batch_nums, quantizable_op_type=quantizable_op_type, round_type=round_type, is_full_quantize=is_full_quantize, -- GitLab