From aa5307ce5b0856cc2eb21186c8bc4c9276638d41 Mon Sep 17 00:00:00 2001 From: Lfc1993 <1635066070@qq.com> Date: Tue, 30 Apr 2019 10:24:44 +0800 Subject: [PATCH] fix python3 run_time_error in ops. test=develop (#17170) fix python3 run_time_error in layers.ops caused by locals() --- python/paddle/fluid/layers/ops.py | 15 ++++++--------- 1 file changed, 6 insertions(+), 9 deletions(-) diff --git a/python/paddle/fluid/layers/ops.py b/python/paddle/fluid/layers/ops.py index 03884d618e..636e83996f 100644 --- a/python/paddle/fluid/layers/ops.py +++ b/python/paddle/fluid/layers/ops.py @@ -102,10 +102,9 @@ _hard_shrink_ = generate_layer_fn('hard_shrink') def hard_shrink(x, threshold=None): - locals_var = locals().keys() + locals_var = locals() kwargs = dict() - for name in locals_var: - val = locals()[name] + for name, val in locals_var.items(): if val is not None: kwargs[name] = val return _hard_shrink_(**kwargs) @@ -124,10 +123,9 @@ _cum_sum_ = generate_layer_fn('cumsum') def cumsum(x, axis=None, exclusive=None, reverse=None): - locals_var = locals().keys() + locals_var = locals() kwargs = dict() - for name in locals_var: - val = locals()[name] + for name, val in locals_var.items(): if val is not None: kwargs[name] = val return _cum_sum_(**kwargs) @@ -146,10 +144,9 @@ _thresholded_relu_ = generate_layer_fn('thresholded_relu') def thresholded_relu(x, threshold=None): - locals_var = locals().keys() + locals_var = locals() kwargs = dict() - for name in locals_var: - val = locals()[name] + for name, val in locals_var.items(): if val is not None: kwargs[name] = val -- GitLab