Add activation_type in AddQuantDequantPass to be compatible with paddleslim, test=develop (#23221)

5 years ago · bd80903333
parent ae3bb16d06
commit bd80903333
1 changed files with 3 additions and 0 deletions
--- a/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py
+++ b/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py
@ -1247,6 +1247,9 @@ class AddQuantDequantPass(object):
        "leaky_relu", "tanh", "swish"
    ]

+    # To be compatible with PaddleSlim, not remove _activation_type for now
+    _activation_type = ["relu", "relu6", "leaky_relu", "tanh", "swish"]
+
    def __init__(self,
                 scope=None,
                 place=None,