提交 06a44f67 编写于 作者: T tangwei

add cluster run, rename fleet-rec to paddle-rec

上级 1c8486fc
......@@ -3,6 +3,6 @@ output
.idea/
build/
dist/
fleetrec.egg-info/
paddlerec.egg-info/
*~
*.pyc
......@@ -20,9 +20,9 @@ import sys
import os
import copy
from fleetrec.core.engine.engine import Engine
from fleetrec.core.factory import TrainerFactory
from fleetrec.core.utils import envs
from paddlerec.core.engine.engine import Engine
from paddlerec.core.factory import TrainerFactory
from paddlerec.core.utils import envs
class ClusterEngine(Engine):
......
......@@ -19,8 +19,8 @@ import sys
import os
import copy
from fleetrec.core.engine.engine import Engine
from fleetrec.core.utils import envs
from paddlerec.core.engine.engine import Engine
from paddlerec.core.utils import envs
class LocalClusterEngine(Engine):
def start_procs(self):
......@@ -47,7 +47,7 @@ class LocalClusterEngine(Engine):
user_endpoints_ips = [x.split(":")[0] for x in user_endpoints.split(",")]
user_endpoints_port = [x.split(":")[1] for x in user_endpoints.split(",")]
factory = "fleetrec.core.factory"
factory = "paddlerec.core.factory"
cmd = [sys.executable, "-u", "-m", factory, self.trainer]
for i in range(server_num):
......
......@@ -19,7 +19,7 @@ import sys
import os
import copy
from fleetrec.core.engine.engine import Engine
from paddlerec.core.engine.engine import Engine
class LocalMPIEngine(Engine):
......@@ -33,7 +33,7 @@ class LocalMPIEngine(Engine):
procs = []
log_fns = []
factory = "fleetrec.core.factory"
factory = "paddlerec.core.factory"
cmd = "mpirun -npernode 2 -timestamp-output -tag-output".split(" ")
cmd.extend([sys.executable, "-u", "-m", factory, self.trainer])
......
......@@ -17,7 +17,7 @@ import sys
import yaml
from fleetrec.core.utils import envs
from paddlerec.core.utils import envs
trainer_abs = os.path.join(os.path.dirname(
os.path.abspath(__file__)), "trainers")
......@@ -71,7 +71,7 @@ class TrainerFactory(object):
with open(config, 'r') as rb:
_config = yaml.load(rb.read(), Loader=yaml.FullLoader)
else:
raise ValueError("fleetrec's config only support yaml")
raise ValueError("paddlerec's config only support yaml")
envs.set_global_envs(_config)
envs.update_workspace()
......
......@@ -15,7 +15,7 @@
import math
import numpy as np
import paddle.fluid as fluid
from fleetrec.core.metric import Metric
from paddlerec.core.metric import Metric
class AUCMetric(Metric):
......
......@@ -2,7 +2,7 @@ import abc
import paddle.fluid as fluid
from fleetrec.core.utils import envs
from paddlerec.core.utils import envs
class Model(object):
......
......@@ -3,8 +3,8 @@ import copy
import paddle.fluid as fluid
from paddle.fluid.incubate.fleet.parameter_server.pslib import fleet
from fleetrec.core.model import Model
from fleetrec.core.utils import table
from paddlerec.core.model import Model
from paddlerec.core.utils import table
def create(config):
......
......@@ -13,7 +13,7 @@
# limitations under the License.
import paddle.fluid as fluid
from fleetrec.core.layer import Layer
from paddlerec.core.layer import Layer
class EmbeddingInputLayer(Layer):
......
......@@ -19,7 +19,7 @@ import os
import paddle.fluid.incubate.data_generator as dg
import yaml
from fleetrec.core.utils import envs
from paddlerec.core.utils import envs
class Reader(dg.MultiSlotDataGenerator):
......
......@@ -20,7 +20,7 @@ import time
import yaml
from paddle import fluid
from fleetrec.core.utils import envs
from paddlerec.core.utils import envs
class Trainer(object):
......
......@@ -25,8 +25,8 @@ from paddle.fluid.incubate.fleet.parameter_server.distribute_transpiler.distribu
from paddle.fluid.incubate.fleet.base.role_maker import PaddleCloudRoleMaker
from paddle.fluid.incubate.fleet.base.role_maker import MPISymetricRoleMaker
from fleetrec.core.utils import envs
from fleetrec.core.trainers.transpiler_trainer import TranspileTrainer
from paddlerec.core.utils import envs
from paddlerec.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 fleetrec.core.utils import envs
from fleetrec.core.trainer import Trainer
from paddlerec.core.utils import envs
from paddlerec.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 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
from paddlerec.core.utils import fs as fs
from paddlerec.core.utils import util as util
from paddlerec.core.metrics.auc_metrics import AUCMetric
from paddlerec.core.modules.modul import build as model_basic
from paddlerec.core.utils import dataset
from paddlerec.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 fleetrec.core.trainers.transpiler_trainer import TranspileTrainer
from fleetrec.core.utils import envs
from paddlerec.core.trainers.transpiler_trainer import TranspileTrainer
from paddlerec.core.utils import envs
import numpy as np
logging.basicConfig(format="%(asctime)s - %(levelname)s - %(message)s")
......
......@@ -25,8 +25,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 fleetrec.core.utils import envs
from fleetrec.core.trainers.cluster_trainer import ClusterTrainer
from paddlerec.core.utils import envs
from paddlerec.core.trainers.cluster_trainer import ClusterTrainer
logging.basicConfig(format="%(asctime)s - %(levelname)s - %(message)s")
......
......@@ -21,9 +21,9 @@ from __future__ import print_function
import logging
import paddle.fluid as fluid
from fleetrec.core.trainers.transpiler_trainer import TranspileTrainer
from fleetrec.core.trainers.single_trainer import SingleTrainer
from fleetrec.core.utils import envs
from paddlerec.core.trainers.transpiler_trainer import TranspileTrainer
from paddlerec.core.trainers.single_trainer import SingleTrainer
from paddlerec.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 fleetrec.core.trainer import Trainer
from fleetrec.core.utils import envs
from fleetrec.core.utils import dataloader_instance
from paddlerec.core.trainer import Trainer
from paddlerec.core.utils import envs
from paddlerec.core.utils import dataloader_instance
class TranspileTrainer(Trainer):
......@@ -81,7 +81,7 @@ class TranspileTrainer(Trainer):
pipe_cmd = "python {} {} {} {}".format(
reader, reader_class, state, self._config_yaml)
if train_data_path.startswith("fleetrec::"):
if train_data_path.startswith("paddlerec::"):
package_base = envs.get_runtime_environ("PACKAGE_BASE")
assert package_base is not None
train_data_path = os.path.join(
......
......@@ -16,9 +16,9 @@ from __future__ import print_function
import os
import sys
from fleetrec.core.utils.envs import lazy_instance_by_fliename
from fleetrec.core.utils.envs import get_global_env
from fleetrec.core.utils.envs import get_runtime_environ
from paddlerec.core.utils.envs import lazy_instance_by_fliename
from paddlerec.core.utils.envs import get_global_env
from paddlerec.core.utils.envs import get_runtime_environ
def dataloader(readerclass, train, yaml_file):
......@@ -31,7 +31,7 @@ def dataloader(readerclass, train, yaml_file):
namespace = "evaluate.reader"
data_path = get_global_env("test_data_path", None, namespace)
if data_path.startswith("fleetrec::"):
if data_path.startswith("paddlerec::"):
package_base = get_runtime_environ("PACKAGE_BASE")
assert package_base is not None
data_path = os.path.join(package_base, data_path.split("::")[1])
......
......@@ -18,8 +18,8 @@ import datetime
import paddle.fluid as fluid
from fleetrec.core.utils import fs as fs
from fleetrec.core.utils import util as util
from paddlerec.core.utils import fs as fs
from paddlerec.core.utils import util as util
class Dataset(object):
......
......@@ -14,7 +14,7 @@
from __future__ import print_function
import sys
from fleetrec.core.utils.envs import lazy_instance_by_fliename
from paddlerec.core.utils.envs import lazy_instance_by_fliename
if len(sys.argv) != 4:
raise ValueError("reader only accept 3 argument: 1. reader_class 2.train/evaluate 3.yaml_abs_path")
......
......@@ -93,9 +93,9 @@ def update_workspace():
return
# is fleet inner models
if workspace.startswith("fleetrec."):
if workspace.startswith("paddlerec."):
fleet_package = get_runtime_environ("PACKAGE_BASE")
workspace_dir = workspace.split("fleetrec.")[1].replace(".", "/")
workspace_dir = workspace.split("paddlerec.")[1].replace(".", "/")
path = os.path.join(fleet_package, workspace_dir)
else:
path = workspace
......@@ -127,7 +127,7 @@ def pretty_print_envs(envs, header=None):
if header:
draws += h_format.format(header[0], header[1])
else:
draws += h_format.format("fleetrec Global Envs", "Value")
draws += h_format.format("paddlerec Global Envs", "Value")
draws += line + "\n"
......
......@@ -17,7 +17,7 @@ import time
import datetime
from paddle import fluid
from fleetrec.core.utils import fs as fs
from paddlerec.core.utils import fs as fs
def save_program_proto(path, program=None):
......
import argparse
if __name__ == "__main__":
parser = argparse.ArgumentParser(description='fleet-rec check')
parser.add_argument("--model", type=str)
parser.add_argument("--engine", type=str)
print("coming soon")
......@@ -18,7 +18,7 @@ train:
strategy: "async"
epochs: 10
workspace: "fleetrec.models.rank.tagspace"
workspace: "paddlerec.models.rank.tagspace"
reader:
batch_size: 5
......
import paddle.fluid as fluid
import math
from fleetrec.core.utils import envs
from fleetrec.core.model import Model as ModelBase
from paddlerec.core.utils import envs
from paddlerec.core.model import Model as ModelBase
import paddle.fluid as fluid
import paddle.fluid.layers.nn as nn
......
......@@ -10,8 +10,8 @@ import paddle
import csv
import io
from fleetrec.core.reader import Reader
from fleetrec.core.utils import envs
from paddlerec.core.reader import Reader
from paddlerec.core.utils import envs
class TrainReader(Reader):
def init(self):
......
......@@ -18,7 +18,7 @@ train:
strategy: "async"
epochs: 10
workspace: "fleetrec.models.rank.text_classification"
workspace: "paddlerec.models.rank.text_classification"
reader:
batch_size: 5
......
import paddle.fluid as fluid
import math
from fleetrec.core.utils import envs
from fleetrec.core.model import Model as ModelBase
from paddlerec.core.utils import envs
from paddlerec.core.model import Model as ModelBase
import paddle.fluid as fluid
import paddle.fluid.layers.nn as nn
......
......@@ -10,8 +10,8 @@ import paddle
import csv
import io
from fleetrec.core.reader import Reader
from fleetrec.core.utils import envs
from paddlerec.core.reader import Reader
from paddlerec.core.utils import envs
class TrainReader(Reader):
def init(self):
......
......@@ -23,7 +23,7 @@ train:
strategy: "async"
epochs: 4
workspace: "fleetrec.models.match.dssm"
workspace: "paddlerec.models.match.dssm"
reader:
batch_size: 4
......
......@@ -15,8 +15,8 @@
import math
import paddle.fluid as fluid
from fleetrec.core.utils import envs
from fleetrec.core.model import Model as ModelBase
from paddlerec.core.utils import envs
from paddlerec.core.model import Model as ModelBase
class Model(ModelBase):
......
......@@ -13,8 +13,8 @@
# limitations under the License.
from __future__ import print_function
from fleetrec.core.reader import Reader
from fleetrec.core.utils import envs
from paddlerec.core.reader import Reader
from paddlerec.core.utils import envs
class EvaluateReader(Reader):
......
......@@ -13,8 +13,8 @@
# limitations under the License.
from __future__ import print_function
from fleetrec.core.reader import Reader
from fleetrec.core.utils import envs
from paddlerec.core.reader import Reader
from paddlerec.core.utils import envs
class TrainReader(Reader):
......
......@@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.
evaluate:
workspace: "fleetrec.models.recall.multiview-simnet"
workspace: "paddlerec.models.recall.multiview-simnet"
reader:
batch_size: 2
class: "{workspace}/evaluate_reader.py"
......@@ -24,7 +24,7 @@ train:
strategy: "async"
epochs: 2
workspace: "fleetrec.models.recall.multiview-simnet"
workspace: "paddlerec.models.recall.multiview-simnet"
reader:
batch_size: 2
......
......@@ -15,8 +15,8 @@ import numpy as np
import io
import copy
import random
from fleetrec.core.reader import Reader
from fleetrec.core.utils import envs
from paddlerec.core.reader import Reader
from paddlerec.core.utils import envs
class EvaluateReader(Reader):
......
......@@ -19,8 +19,8 @@ import paddle.fluid.layers as layers
import paddle.fluid.layers.tensor as tensor
import paddle.fluid.layers.control_flow as cf
from fleetrec.core.utils import envs
from fleetrec.core.model import Model as ModelBase
from paddlerec.core.utils import envs
from paddlerec.core.model import Model as ModelBase
class BowEncoder(object):
""" bow-encoder """
......
......@@ -15,8 +15,8 @@ import numpy as np
import io
import copy
import random
from fleetrec.core.reader import Reader
from fleetrec.core.utils import envs
from paddlerec.core.reader import Reader
from paddlerec.core.utils import envs
class TrainReader(Reader):
......
......@@ -18,7 +18,7 @@ train:
strategy: "async"
epochs: 3
workspace: "fleetrec.models.multitask.esmm"
workspace: "paddlerec.models.multitask.esmm"
reader:
batch_size: 2
......
......@@ -13,8 +13,8 @@
# limitations under the License.
from __future__ import print_function
from fleetrec.core.reader import Reader
from fleetrec.core.utils import envs
from paddlerec.core.reader import Reader
from paddlerec.core.utils import envs
from collections import defaultdict
import numpy as np
......
......@@ -15,8 +15,8 @@
import math
import paddle.fluid as fluid
from fleetrec.core.utils import envs
from fleetrec.core.model import Model as ModelBase
from paddlerec.core.utils import envs
from paddlerec.core.model import Model as ModelBase
import numpy as np
......
......@@ -13,8 +13,8 @@
# limitations under the License.
from __future__ import print_function
from fleetrec.core.reader import Reader
from fleetrec.core.utils import envs
from paddlerec.core.reader import Reader
from paddlerec.core.utils import envs
import numpy as np
......
......@@ -18,7 +18,7 @@ train:
strategy: "async"
epochs: 3
workspace: "fleetrec.models.multitask.mmoe"
workspace: "paddlerec.models.multitask.mmoe"
reader:
batch_size: 2
......
......@@ -15,8 +15,8 @@
import math
import paddle.fluid as fluid
from fleetrec.core.utils import envs
from fleetrec.core.model import Model as ModelBase
from paddlerec.core.utils import envs
from paddlerec.core.model import Model as ModelBase
class Model(ModelBase):
......
......@@ -13,8 +13,8 @@
# limitations under the License.
from __future__ import print_function
from fleetrec.core.reader import Reader
from fleetrec.core.utils import envs
from paddlerec.core.reader import Reader
from paddlerec.core.utils import envs
import numpy as np
......
......@@ -18,7 +18,7 @@ train:
strategy: "async"
epochs: 3
workspace: "fleetrec.models.multitask.share-bottom"
workspace: "paddlerec.models.multitask.share-bottom"
reader:
batch_size: 2
......
......@@ -15,8 +15,8 @@
import math
import paddle.fluid as fluid
from fleetrec.core.utils import envs
from fleetrec.core.model import Model as ModelBase
from paddlerec.core.utils import envs
from paddlerec.core.model import Model as ModelBase
class Model(ModelBase):
......
......@@ -13,8 +13,8 @@
# limitations under the License.
from __future__ import print_function
from fleetrec.core.reader import Reader
from fleetrec.core.utils import envs
from paddlerec.core.reader import Reader
from paddlerec.core.utils import envs
class TrainReader(Reader):
......
......@@ -18,7 +18,7 @@ train:
strategy: "async"
epochs: 10
workspace: "fleetrec.models.rank.dcn"
workspace: "paddlerec.models.rank.dcn"
reader:
batch_size: 2
......
......@@ -15,8 +15,8 @@ from __future__ import print_function
import math
import sys
from fleetrec.core.reader import Reader
from fleetrec.core.utils import envs
from paddlerec.core.reader import Reader
from paddlerec.core.utils import envs
try:
import cPickle as pickle
except ImportError:
......
......@@ -6,7 +6,7 @@ LOCAL_PATH = os.path.dirname(os.path.abspath(__file__))
TOOLS_PATH = os.path.join(LOCAL_PATH, "..", "..", "tools")
sys.path.append(TOOLS_PATH)
from fleetrec.tools.tools import download_file_and_uncompress
from paddlerec.tools.tools import download_file_and_uncompress
if __name__ == '__main__':
trainfile = 'train.txt'
......
import paddle.fluid as fluid
import math
from fleetrec.core.utils import envs
from fleetrec.core.model import Model as ModelBase
from paddlerec.core.utils import envs
from paddlerec.core.model import Model as ModelBase
from collections import OrderedDict
class Model(ModelBase):
......
......@@ -18,7 +18,7 @@ train:
strategy: "async"
epochs: 10
workspace: "fleetrec.models.rank.deepfm"
workspace: "paddlerec.models.rank.deepfm"
reader:
batch_size: 2
......
......@@ -13,8 +13,8 @@
# limitations under the License.
from __future__ import print_function
from fleetrec.core.reader import Reader
from fleetrec.core.utils import envs
from paddlerec.core.reader import Reader
from paddlerec.core.utils import envs
try:
import cPickle as pickle
except ImportError:
......
......@@ -6,7 +6,7 @@ LOCAL_PATH = os.path.dirname(os.path.abspath(__file__))
TOOLS_PATH = os.path.join(LOCAL_PATH, "..", "..", "tools")
sys.path.append(TOOLS_PATH)
from fleetrec.tools.tools import download_file_and_uncompress, download_file
from paddlerec.tools.tools import download_file_and_uncompress, download_file
if __name__ == '__main__':
url = "https://s3-eu-west-1.amazonaws.com/kaggle-display-advertising-challenge-dataset/dac.tar.gz"
......
import paddle.fluid as fluid
import math
from fleetrec.core.utils import envs
from fleetrec.core.model import Model as ModelBase
from paddlerec.core.utils import envs
from paddlerec.core.model import Model as ModelBase
class Model(ModelBase):
......
......@@ -18,7 +18,7 @@ train:
strategy: "async"
epochs: 10
workspace: "fleetrec.models.rank.din"
workspace: "paddlerec.models.rank.din"
reader:
batch_size: 2
......
import paddle.fluid as fluid
import math
from fleetrec.core.utils import envs
from fleetrec.core.model import Model as ModelBase
from paddlerec.core.utils import envs
from paddlerec.core.model import Model as ModelBase
class Model(ModelBase):
......
......@@ -13,8 +13,8 @@
# limitations under the License.
from __future__ import print_function
from fleetrec.core.reader import Reader
from fleetrec.core.utils import envs
from paddlerec.core.reader import Reader
from paddlerec.core.utils import envs
import numpy as np
import os
import random
......
......@@ -13,7 +13,7 @@
# limitations under the License.
engine:
workspace: "fleetrec.models.rank.dnn"
workspace: "paddlerec.models.rank.dnn"
backend: "MPI"
hdfs:
......@@ -24,7 +24,7 @@ engine:
package:
build_script: "{workspace}/package.sh"
python: "/home/tangwei/fleet_rec_env/cpython-2.7.11-ucs4"
paddlerec: "/home/tangwei/fleet_rec_env/FleetRec"
paddlerec: "/home/tangwei/fleet_rec_env/PaddleRec"
submit:
hpc: "/home/tangwei/Plines/client/smart_client_khan/"
......
......@@ -18,7 +18,7 @@ train:
strategy: "async"
epochs: 10
workspace: "fleetrec.models.rank.dnn"
workspace: "paddlerec.models.rank.dnn"
reader:
batch_size: 2
......
......@@ -15,8 +15,8 @@
import math
import paddle.fluid as fluid
from fleetrec.core.utils import envs
from fleetrec.core.model import Model as ModelBase
from paddlerec.core.utils import envs
from paddlerec.core.model import Model as ModelBase
class Model(ModelBase):
......
......@@ -56,7 +56,7 @@ function env_prepare() {
python -c "print('heheda')"
mpirun -npernode 1 python/bin/python -m pip uninstall -y fleet-rec
mpirun -npernode 1 python/bin/python -m pip uninstall -y paddle-rec
mpirun -npernode 1 python/bin/python -m pip install whl/fleet_rec-0.0.2-py2-none-any.whl --index-url=http://pip.baidu.com/pypi/simple --trusted-host pip.baidu.com
check_error
}
......@@ -64,7 +64,7 @@ function env_prepare() {
function run() {
echo "run"
g_run_stage="run"
mpirun -npernode 2 -timestamp-output -tag-output -machinefile ${PBS_NODEFILE} python/bin/python -u -m fleetrec.run -m fleetrec.models.rank.dnn --engine cluster --role worker
mpirun -npernode 2 -timestamp-output -tag-output -machinefile ${PBS_NODEFILE} python/bin/python -u -m paddlerec.run -m paddlerec.models.rank.dnn --engine cluster --role worker
}
function main() {
......
......@@ -18,7 +18,7 @@ train:
strategy: "async"
epochs: 10
workspace: "fleetrec.models.rank.wide_deep"
workspace: "paddlerec.models.rank.wide_deep"
reader:
batch_size: 2
......
import paddle.fluid as fluid
import math
from fleetrec.core.utils import envs
from fleetrec.core.model import Model as ModelBase
from paddlerec.core.utils import envs
from paddlerec.core.model import Model as ModelBase
class Model(ModelBase):
......
......@@ -13,8 +13,8 @@
# limitations under the License.
from __future__ import print_function
from fleetrec.core.reader import Reader
from fleetrec.core.utils import envs
from paddlerec.core.reader import Reader
from paddlerec.core.utils import envs
try:
import cPickle as pickle
except ImportError:
......
......@@ -18,7 +18,7 @@ train:
strategy: "async"
epochs: 10
workspace: "fleetrec.models.rank.xdeepfm"
workspace: "paddlerec.models.rank.xdeepfm"
reader:
batch_size: 2
......
......@@ -13,8 +13,8 @@
# limitations under the License.
from __future__ import print_function
from fleetrec.core.reader import Reader
from fleetrec.core.utils import envs
from paddlerec.core.reader import Reader
from paddlerec.core.utils import envs
try:
import cPickle as pickle
except ImportError:
......
......@@ -6,7 +6,7 @@ LOCAL_PATH = os.path.dirname(os.path.abspath(__file__))
TOOLS_PATH = os.path.join(LOCAL_PATH, "..", "..", "tools")
sys.path.append(TOOLS_PATH)
from fleetrec.tools.tools import download_file_and_uncompress, download_file
from paddlerec.tools.tools import download_file_and_uncompress, download_file
if __name__ == '__main__':
url_train = "https://paddlerec.bj.bcebos.com/xdeepfm%2Ftr"
......
import paddle.fluid as fluid
import math
from fleetrec.core.utils import envs
from fleetrec.core.model import Model as ModelBase
from paddlerec.core.utils import envs
from paddlerec.core.model import Model as ModelBase
class Model(ModelBase):
......
......@@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.
evaluate:
workspace: "fleetrec.models.recall.gnn"
workspace: "paddlerec.models.recall.gnn"
reader:
batch_size: 50
class: "{workspace}/evaluate_reader.py"
......@@ -24,7 +24,7 @@ train:
strategy: "async"
epochs: 2
workspace: "fleetrec.models.recall.gnn"
workspace: "paddlerec.models.recall.gnn"
reader:
batch_size: 100
......
......@@ -15,8 +15,8 @@ import numpy as np
import io
import copy
import random
from fleetrec.core.reader import Reader
from fleetrec.core.utils import envs
from paddlerec.core.reader import Reader
from paddlerec.core.utils import envs
class EvaluateReader(Reader):
......
......@@ -17,8 +17,8 @@ import math
import paddle.fluid as fluid
import paddle.fluid.layers as layers
from fleetrec.core.utils import envs
from fleetrec.core.model import Model as ModelBase
from paddlerec.core.utils import envs
from paddlerec.core.model import Model as ModelBase
class Model(ModelBase):
......
......@@ -15,8 +15,8 @@ import numpy as np
import io
import copy
import random
from fleetrec.core.reader import Reader
from fleetrec.core.utils import envs
from paddlerec.core.reader import Reader
from paddlerec.core.utils import envs
class TrainReader(Reader):
......
......@@ -18,7 +18,7 @@ train:
strategy: "async"
epochs: 3
workspace: "fleetrec.models.recall.gru4rec"
workspace: "paddlerec.models.recall.gru4rec"
reader:
batch_size: 5
......
......@@ -15,8 +15,8 @@
import math
import paddle.fluid as fluid
from fleetrec.core.utils import envs
from fleetrec.core.model import Model as ModelBase
from paddlerec.core.utils import envs
from paddlerec.core.model import Model as ModelBase
class Model(ModelBase):
......
......@@ -13,8 +13,8 @@
# limitations under the License.
from __future__ import print_function
from fleetrec.core.reader import Reader
from fleetrec.core.utils import envs
from paddlerec.core.reader import Reader
from paddlerec.core.utils import envs
class TrainReader(Reader):
......
......@@ -35,7 +35,7 @@ sh data_process.sh
```
### 训练
```shell
python -m fleetrec.run -m config.yaml -d cpu -e single
python -m paddlerec.run -m config.yaml -d cpu -e single
```
### 预测
......
......@@ -18,7 +18,7 @@ train:
strategy: "async"
epochs: 3
workspace: "fleetrec.models.recall.ssr"
workspace: "paddlerec.models.recall.ssr"
reader:
batch_size: 5
......
......@@ -15,8 +15,8 @@
import math
import paddle.fluid as fluid
from fleetrec.core.utils import envs
from fleetrec.core.model import Model as ModelBase
from paddlerec.core.utils import envs
from paddlerec.core.model import Model as ModelBase
import paddle.fluid.layers.tensor as tensor
import paddle.fluid.layers.io as io
import paddle.fluid.layers.control_flow as cf
......
......@@ -13,8 +13,8 @@
# limitations under the License.
from __future__ import print_function
from fleetrec.core.reader import Reader
from fleetrec.core.utils import envs
from paddlerec.core.reader import Reader
from paddlerec.core.utils import envs
import random
......
......@@ -18,7 +18,7 @@ train:
strategy: "async"
epochs: 2
workspace: "fleetrec.models.recall.tdm"
workspace: "paddlerec.models.recall.tdm"
reader:
batch_size: 32
......@@ -73,7 +73,7 @@ train:
save_last: True
evaluate:
workspace: "fleetrec.models.recall.tdm"
workspace: "paddlerec.models.recall.tdm"
reader:
batch_size: 1
class: "{workspace}/tdm_evaluate_reader.py"
......
......@@ -17,8 +17,8 @@
import paddle.fluid as fluid
import math
from fleetrec.core.utils import envs
from fleetrec.core.model import Model as ModelBase
from paddlerec.core.utils import envs
from paddlerec.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.
先完成此消息的编辑!
想要评论请 注册