diff --git a/fleetrec/core/trainers/single_trainer.py b/fleetrec/core/trainers/single_trainer.py index d606fd38d793ce1d2fa319d862c9fd6c8ec2def7..553f0a901b0bbe49080d776a9b254e53333c3d4b 100644 --- a/fleetrec/core/trainers/single_trainer.py +++ b/fleetrec/core/trainers/single_trainer.py @@ -20,8 +20,8 @@ from __future__ import print_function import logging import paddle.fluid as fluid -from fleetrec.trainer.transpiler_trainer import TranspileTrainer -from fleetrec.utils import envs +from fleetrec.core.trainers.transpiler_trainer import TranspileTrainer +from fleetrec.core.utils import envs logging.basicConfig(format="%(asctime)s - %(levelname)s - %(message)s") logger = logging.getLogger("fluid") diff --git a/fleetrec/core/trainers/transpiler_trainer.py b/fleetrec/core/trainers/transpiler_trainer.py index 06ddc63ca12146d6efda16dba90216a941702737..e8ff15b9d2967f00ceb9ddf2c642db7266541ff1 100644 --- a/fleetrec/core/trainers/transpiler_trainer.py +++ b/fleetrec/core/trainers/transpiler_trainer.py @@ -20,8 +20,8 @@ import os import paddle.fluid as fluid from paddle.fluid.incubate.fleet.parameter_server.distribute_transpiler import fleet -from fleetrec.trainer.trainer import Trainer -from fleetrec.utils import envs +from fleetrec.core.trainer import Trainer +from fleetrec.core.utils import envs class TranspileTrainer(Trainer):