diff --git a/python/paddle/fluid/contrib/slim/tests/test_imperative_qat.py b/python/paddle/fluid/contrib/slim/tests/test_imperative_qat.py index bf411e5b38efa..3cc61ce8c5808 100644 --- a/python/paddle/fluid/contrib/slim/tests/test_imperative_qat.py +++ b/python/paddle/fluid/contrib/slim/tests/test_imperative_qat.py @@ -64,11 +64,11 @@ def tearDownClass(cls): print("Failed to delete {} due to {}".format(cls.root_path, str(e))) def set_vars(self): - self.weight_quantize_type = None - self.activation_quantize_type = None + self.weight_quantize_type = 'abs_max' + self.activation_quantize_type = 'moving_average_abs_max' print('weight_quantize_type', self.weight_quantize_type) - def run_qat_save(self): + def test_qat(self): self.set_vars() imperative_qat = ImperativeQuantAware( @@ -200,15 +200,5 @@ def run_qat_save(self): msg='Failed to save the inference quantized model.') -class TestImperativeQatAbsMax(TestImperativeQat): - def set_vars(self): - self.weight_quantize_type = 'abs_max' - self.activation_quantize_type = 'moving_average_abs_max' - print('weight_quantize_type', self.weight_quantize_type) - - def test_qat(self): - self.run_qat_save() - - if __name__ == '__main__': unittest.main() diff --git a/python/paddle/fluid/contrib/slim/tests/test_imperative_qat_channelwise.py b/python/paddle/fluid/contrib/slim/tests/test_imperative_qat_channelwise.py index 3d2cad388d172..1a6c9c41638db 100644 --- a/python/paddle/fluid/contrib/slim/tests/test_imperative_qat_channelwise.py +++ b/python/paddle/fluid/contrib/slim/tests/test_imperative_qat_channelwise.py @@ -43,9 +43,6 @@ def set_vars(self): self.activation_quantize_type = 'moving_average_abs_max' print('weight_quantize_type', self.weight_quantize_type) - def test_qat(self): - self.run_qat_save() - if __name__ == '__main__': unittest.main()