From d1c428dafe3b8c2b43853cb33ba9fe0a23e6d196 Mon Sep 17 00:00:00 2001 From: WangXi Date: Thu, 11 Mar 2021 12:47:29 +0800 Subject: [PATCH] fix --- .../fleet/meta_optimizers/sharding/offload_helper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/distributed/fleet/meta_optimizers/sharding/offload_helper.py b/python/paddle/distributed/fleet/meta_optimizers/sharding/offload_helper.py index 4fd087bbd50..8197f4368f2 100644 --- 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': -- GitLab