提交 e8a467da 编写于 作者: M mapingshuo

enable zero import

上级 ef11cec9
...@@ -16,7 +16,7 @@ from .common import OpRole, OP_ROLE_KEY, OP_ROLE_VAR_KEY, CollectiveHelper ...@@ -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 .common import is_update_op, is_loss_grad_op, is_backward_op, is_optimizer_op
from .meta_optimizer_base import MetaOptimizerBase from .meta_optimizer_base import MetaOptimizerBase
from paddle.fluid import unique_name, core 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 paddle.fluid as fluid
import math import math
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册