diff --git a/python/paddle/fluid/incubate/fleet/parameter_server/ir/pserver_pass.py b/python/paddle/fluid/incubate/fleet/parameter_server/ir/pserver_pass.py
index e3bda62fc4afd0963651cbf8d2b05e770ccba1f5..a60c4e149f582e4f364910611d18cda5fbca4f07 100644
--- 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')