diff --git a/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py b/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py index 213033f883c283112a234d7f9dcb448b5680e6da..201ffc35dd397c9fc883f1ff258bac00ccc38cbf 100644 --- 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,