bug fix

7ed2914b · seiriosPlus · 7bb4a4e8 · 7ed2914b
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/fluid/incubate/fleet/parameter_server/ir/pserver_pass.py .../fluid/incubate/fleet/parameter_server/ir/pserver_pass.py +1 -1

未找到文件。
--- a/python/paddle/fluid/incubate/fleet/parameter_server/ir/pserver_pass.py
+++ b/python/paddle/fluid/incubate/fleet/parameter_server/ir/pserver_pass.py
@@ -682,7 +682,7 @@ def large_scale_sparse_pass(program, main_program, config, is_startup=False):
            beta2_pow_o = main_program.global_block().vars[op.output(
                "Beta2PowOut")[0]]

-            beta1 = op.attr('shape')
+            beta1 = op.attr('beta1')
            beta2 = op.attr('beta2')
            epsilon = op.attr('epsilon')