fix

d1c428da · WangXi · sandyhouse · f32f84a4 · d1c428da
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/distributed/fleet/meta_optimizers/sharding/offload_helper.py ...tributed/fleet/meta_optimizers/sharding/offload_helper.py +1 -1

未找到文件。
--- a/python/paddle/distributed/fleet/meta_optimizers/sharding/offload_helper.py
+++ b/python/paddle/distributed/fleet/meta_optimizers/sharding/offload_helper.py
@@ -79,7 +79,7 @@ class OffloadHelper(object):
                # {Moment1Out = [''], Moment2Out = [''], ParamOut = ['']} =
                # adam(inputs={Moment1 = [''], Moment2 = [''], Param = ['']})
                vars_name.append(op.desc.input("Moment1")[0])
-                vars_name.append(op.desc.input("Moment1")[0])
+                vars_name.append(op.desc.input("Moment2")[0])
            elif op.type == 'momentum':
                pass
            elif op.type == 'lars':