fix the error of qat unit test (#33574)

PaddlePaddle · Jun 17, 2021 · 3af1629 · 3af1629
1 parent 918aeb7
commit 3af1629
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 16 deletions.
diff --git 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
@@ -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()