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 bf411e5b38efaed4c99b628d19f1021df5e0facf..3cc61ce8c580889d027ebb3185e840942d4809bb 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 @@ class TestImperativeQat(unittest.TestCase): 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 @@ class TestImperativeQat(unittest.TestCase): 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 3d2cad388d172edb76a439e6b920bc4a03c26754..1a6c9c41638db30124803c80aaf535a38f8c8475 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 @@ class TestImperativeQatChannelWise(TestImperativeQat): 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()