diff --git a/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py b/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py index fcf7a51113563667b6449f2764f4950c0150308d..f04869156f95cbc5d7b8f2cee0974658d02f4a32 100644 --- a/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py +++ b/python/paddle/fluid/contrib/slim/quantization/quantization_pass.py @@ -83,7 +83,7 @@ _op_real_in_out_name = { "swish": [["X"], ["Out"]], "dropout": [["X"], ["Out"]], "batch_norm": [["X"], ["Y"]], - "sigmoid": [["X"], ["Y"]], + "sigmoid": [["X"], ["Out"]], }