提交 d2522c61 编写于 作者: T tangwei

rename

上级 28d374d2
......@@ -3,6 +3,6 @@ output
.idea/
build/
dist/
fleet_rec.egg-info/
fleetrec.egg-info/
*~
*.pyc
......@@ -30,7 +30,7 @@ train:
mode: "dataset"
batch_size: 2
class: "fleetrec.models.rank.criteo_reader"
train_data_path: "fleetrec/models/rank/dnn/data/train"
train_data_path: "models/rank/dnn/data/train"
model:
models: "fleetrec.models.rank.dnn.model"
......
......@@ -14,8 +14,8 @@
import paddle.fluid as fluid
from fleet_rec.core.trainers.transpiler_trainer import TranspileTrainer
from fleet_rec.core.utils import envs
from fleetrec.core.trainers.transpiler_trainer import TranspileTrainer
from fleetrec.core.utils import envs
class UserDefineTrainer(TranspileTrainer):
......
......@@ -19,7 +19,7 @@ import sys
import os
import copy
from fleet_rec.core.engine.engine import Engine
from fleetrec.core.engine.engine import Engine
class LocalClusterEngine(Engine):
......
......@@ -19,7 +19,7 @@ import sys
import os
import copy
from fleet_rec.core.engine.engine import Engine
from fleetrec.core.engine.engine import Engine
class LocalMPIEngine(Engine):
......
......@@ -17,7 +17,7 @@ import sys
import yaml
from fleet_rec.core.utils import envs
from fleetrec.core.utils import envs
trainer_abs = os.path.join(os.path.dirname(os.path.abspath(__file__)), "trainers")
trainers = {}
......
......@@ -15,7 +15,7 @@
import math
import numpy as np
import paddle.fluid as fluid
from fleet_rec.core.metric import Metric
from fleetrec.core.metric import Metric
class AUCMetric(Metric):
......
import abc
from fleet_rec.core.utils import envs
from fleetrec.core.utils import envs
class Model(object):
"""R
......
......@@ -3,8 +3,8 @@ import copy
import paddle.fluid as fluid
from paddle.fluid.incubate.fleet.parameter_server.pslib import fleet
from fleet_rec.core.model import Model
from fleet_rec.core.utils import table
from fleetrec.core.model import Model
from fleetrec.core.utils import table
def create(config):
......
......@@ -13,7 +13,7 @@
# limitations under the License.
import paddle.fluid as fluid
from fleet_rec.core.layer import Layer
from fleetrec.core.layer import Layer
class EmbeddingInputLayer(Layer):
......
......@@ -19,7 +19,7 @@ import os
import paddle.fluid.incubate.data_generator as dg
import yaml
from fleet_rec.core.utils import envs
from fleetrec.core.utils import envs
class Reader(dg.MultiSlotDataGenerator):
......
......@@ -20,7 +20,7 @@ import time
import yaml
from paddle import fluid
from fleet_rec.core.utils import envs
from fleetrec.core.utils import envs
class Trainer(object):
......
......@@ -23,8 +23,8 @@ from paddle.fluid.incubate.fleet.parameter_server.distribute_transpiler import f
from paddle.fluid.incubate.fleet.parameter_server.distribute_transpiler.distributed_strategy import StrategyFactory
from paddle.fluid.incubate.fleet.base.role_maker import PaddleCloudRoleMaker
from fleet_rec.core.utils import envs
from fleet_rec.core.trainers.transpiler_trainer import TranspileTrainer
from fleetrec.core.utils import envs
from fleetrec.core.trainers.transpiler_trainer import TranspileTrainer
class ClusterTrainer(TranspileTrainer):
......
......@@ -18,8 +18,8 @@ import paddle.fluid as fluid
from paddle.fluid.incubate.fleet.parameter_server.pslib import fleet
from paddle.fluid.incubate.fleet.base.role_maker import MPISymetricRoleMaker
from fleet_rec.core.utils import envs
from fleet_rec.core.trainer import Trainer
from fleetrec.core.utils import envs
from fleetrec.core.trainer import Trainer
class CtrPaddleTrainer(Trainer):
......
......@@ -24,12 +24,12 @@ from paddle.fluid.incubate.fleet.parameter_server.pslib import fleet
from paddle.fluid.incubate.fleet.base.role_maker import GeneralRoleMaker
from fleet_rec.core.utils import fs as fs
from fleet_rec.core.utils import util as util
from fleet_rec.core.metrics.auc_metrics import AUCMetric
from fleet_rec.core.modules.modul import build as model_basic
from fleet_rec.core.utils import dataset
from fleet_rec.core.trainer import Trainer
from fleetrec.core.utils import fs as fs
from fleetrec.core.utils import util as util
from fleetrec.core.metrics.auc_metrics import AUCMetric
from fleetrec.core.modules.modul import build as model_basic
from fleetrec.core.utils import dataset
from fleetrec.core.trainer import Trainer
def wroker_numric_opt(value, env, opt):
......
......@@ -20,8 +20,8 @@ from __future__ import print_function
import logging
import paddle.fluid as fluid
from fleet_rec.core.trainers.transpiler_trainer import TranspileTrainer
from fleet_rec.core.utils import envs
from fleetrec.core.trainers.transpiler_trainer import TranspileTrainer
from fleetrec.core.utils import envs
import numpy as np
logging.basicConfig(format="%(asctime)s - %(levelname)s - %(message)s")
......
......@@ -20,9 +20,9 @@ import os
import paddle.fluid as fluid
from paddle.fluid.incubate.fleet.parameter_server.distribute_transpiler import fleet
from fleet_rec.core.trainer import Trainer
from fleet_rec.core.utils import envs
from fleet_rec.core.utils import dataloader_instance
from fleetrec.core.trainer import Trainer
from fleetrec.core.utils import envs
from fleetrec.core.utils import dataloader_instance
class TranspileTrainer(Trainer):
......
......@@ -16,8 +16,8 @@ from __future__ import print_function
import os
import sys
from fleet_rec.core.utils.envs import lazy_instance
from fleet_rec.core.utils.envs import get_global_env
from fleetrec.core.utils.envs import lazy_instance
from fleetrec.core.utils.envs import get_global_env
def dataloader(readerclass, train, yaml_file):
......
......@@ -18,8 +18,8 @@ import datetime
import paddle.fluid as fluid
from fleet_rec.core.utils import fs as fs
from fleet_rec.core.utils import util as util
from fleetrec.core.utils import fs as fs
from fleetrec.core.utils import util as util
class Dataset(object):
......
......@@ -14,7 +14,7 @@
from __future__ import print_function
import sys
from fleet_rec.core.utils.envs import lazy_instance
from fleetrec.core.utils.envs import lazy_instance
if len(sys.argv) != 4:
raise ValueError("reader only accept 3 argument: 1. reader_class 2.train/evaluate 3.yaml_abs_path")
......
......@@ -17,7 +17,7 @@ import time
import datetime
from paddle import fluid
from fleet_rec.core.utils import fs as fs
from fleetrec.core.utils import fs as fs
def save_program_proto(path, program=None):
......
......@@ -3,9 +3,9 @@ import os
import subprocess
import yaml
from fleet_rec.core.factory import TrainerFactory
from fleet_rec.core.utils import envs
from fleet_rec.core.utils import util
from fleetrec.core.factory import TrainerFactory
from fleetrec.core.utils import envs
from fleetrec.core.utils import util
engines = {}
device = ["CPU", "GPU"]
......@@ -123,7 +123,7 @@ def cluster_mpi_engine(args):
def local_cluster_engine(args):
print("launch cluster engine with cluster to run model: {}".format(args.model))
from fleet_rec.core.engine.local_cluster_engine import LocalClusterEngine
from fleetrec.core.engine.local_cluster_engine import LocalClusterEngine
cluster_envs = {}
cluster_envs["server_num"] = 1
......@@ -148,7 +148,7 @@ def local_cluster_engine(args):
def local_mpi_engine(args):
print("launch cluster engine with cluster to run model: {}".format(args.model))
from fleet_rec.core.engine.local_mpi_engine import LocalMPIEngine
from fleetrec.core.engine.local_mpi_engine import LocalMPIEngine
print("use 1X1 MPI ClusterTraining at localhost to run model: {}".format(args.model))
......
......@@ -13,8 +13,8 @@
# limitations under the License.
from __future__ import print_function
from fleet_rec.core.reader import Reader
from fleet_rec.core.utils import envs
from fleetrec.core.reader import Reader
from fleetrec.core.utils import envs
class TrainReader(Reader):
......
......@@ -15,8 +15,8 @@
import math
import paddle.fluid as fluid
from fleet_rec.core.utils import envs
from fleet_rec.core.model import Model as ModelBase
from fleetrec.core.utils import envs
from fleetrec.core.model import Model as ModelBase
class Model(ModelBase):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册