From f4d7a9b9d19c5ba688708a66f8c3b1b95f158ac7 Mon Sep 17 00:00:00 2001 From: tangwei Date: Wed, 15 Apr 2020 23:37:16 +0800 Subject: [PATCH] fix bug --- fleetrec/core/factory.py | 2 +- fleetrec/examples/build_in/ctr-dnn_train.yaml | 2 +- setup.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/fleetrec/core/factory.py b/fleetrec/core/factory.py index 9a1fa3cf..3aa3197d 100644 --- a/fleetrec/core/factory.py +++ b/fleetrec/core/factory.py @@ -32,7 +32,7 @@ class TrainerFactory(object): def _build_trainer(config, yaml_path): print(envs.pretty_print_envs(envs.get_global_envs())) - train_mode = envs.get_global_env("train.strategy.mode") + train_mode = envs.get_global_env("train.trainer") if train_mode is None: train_mode = envs.get_runtime_envion("train.trainer") diff --git a/fleetrec/examples/build_in/ctr-dnn_train.yaml b/fleetrec/examples/build_in/ctr-dnn_train.yaml index a1c1eaa5..87de5fae 100644 --- a/fleetrec/examples/build_in/ctr-dnn_train.yaml +++ b/fleetrec/examples/build_in/ctr-dnn_train.yaml @@ -19,7 +19,7 @@ train: reader: mode: "dataset" batch_size: 2 - class: "fleetrec.models.ctr_dnn.data_generator" + class: "fleetrec.models.ctr_dnn.reader" train_data_path: "/root/FleetRec/fleetrec/models/ctr_dnn/data/train/" model: diff --git a/setup.py b/setup.py index e1390ba5..ca65bb3a 100644 --- a/setup.py +++ b/setup.py @@ -4,7 +4,7 @@ setup for fleet-rec. from setuptools import setup -packages = ["fleetrec", "fleetrec.models", +packages = ["fleetrec", "fleetrec.models", "fleetrec.models.ctr_dnn", "fleetrec.examples", "fleetrec.examples.user_define", "fleetrec.examples.build_in", "fleetrec.core", "fleetrec.core.engine", "fleetrec.core.metrics", "fleetrec.core.models", "fleetrec.core.trainers", "fleetrec.core.utils"] -- GitLab