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

bd809033 · cc · GitHub · ae3bb16d · bd809033
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

python/paddle/fluid/contrib/slim/quantization/quantization_pass.py ...ddle/fluid/contrib/slim/quantization/quantization_pass.py +3 -0

未找到文件。
--- 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,