up

PaddlePaddle · Jun 30, 2021 · 8d85cff · 8d85cff · paddle-bot-old · Jun 30, 2021
1 parent 496dbbf
commit 8d85cff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/paddle/fluid/contrib/slim/quantization/imperative/qat.py b/python/paddle/fluid/contrib/slim/quantization/imperative/qat.py
@@ -516,7 +516,7 @@ def _gather_output_scale():
                 out_scale = utils.load_variable_data(scope, out_scale_name)
                 out_scale = utils.fp_numpy_to_naive(out_scale)
 
-                if previous_op is not None:
+                if previous_op.type != "feed":
                     argname, index = utils._get_output_name_index(previous_op,
                                                                   in_var_name)
                     previous_op._set_attr(argname + str(index) + "_threshold",