From e8a467dadc584084bd1c7fc05e452d797c257459 Mon Sep 17 00:00:00 2001 From: mapingshuo Date: Thu, 24 Sep 2020 17:39:31 +0800 Subject: [PATCH] enable zero import --- .../paddle/distributed/fleet/meta_optimizers/zero_optimizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/distributed/fleet/meta_optimizers/zero_optimizer.py b/python/paddle/distributed/fleet/meta_optimizers/zero_optimizer.py index 3f9222577cf..4d6d9c577e5 100644 --- a/python/paddle/distributed/fleet/meta_optimizers/zero_optimizer.py +++ b/python/paddle/distributed/fleet/meta_optimizers/zero_optimizer.py @@ -16,7 +16,7 @@ from .common import OpRole, OP_ROLE_KEY, OP_ROLE_VAR_KEY, CollectiveHelper from .common import is_update_op, is_loss_grad_op, is_backward_op, is_optimizer_op from .meta_optimizer_base import MetaOptimizerBase from paddle.fluid import unique_name, core -from zero.decorator import decorate as amp_decorate +from paddle.distributed.fleet.meta_optimizers.zero.decorator import decorate as amp_decorate import paddle.fluid as fluid import math -- GitLab