fix bug

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")

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:

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"]