From a96118ecd6b735fb11d195abb7701dacee3fee38 Mon Sep 17 00:00:00 2001 From: chengmo Date: Wed, 6 May 2020 23:29:45 +0800 Subject: [PATCH] fix --- fleet_rec/core/trainers/tdm_cluster_trainer.py | 8 ++++---- models/recall/tdm/config.yaml | 1 - 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/fleet_rec/core/trainers/tdm_cluster_trainer.py b/fleet_rec/core/trainers/tdm_cluster_trainer.py index 336a0581..ba7bd72b 100644 --- a/fleet_rec/core/trainers/tdm_cluster_trainer.py +++ b/fleet_rec/core/trainers/tdm_cluster_trainer.py @@ -50,13 +50,13 @@ class TDMClusterTrainer(TranspileTrainer): namespace = "train.startup" load_tree = envs.get_global_env( - "cluster.load_tree", True, namespace) + "tree.load_tree", True, namespace) self.tree_layer_path = envs.get_global_env( - "cluster.tree_layer_path", "", namespace) + "tree.tree_layer_path", "", namespace) self.tree_travel_path = envs.get_global_env( - "cluster.tree_travel_path", "", namespace) + "tree.tree_travel_path", "", namespace) self.tree_info_path = envs.get_global_env( - "cluster.tree_info_path", "", namespace) + "tree.tree_info_path", "", namespace) save_init_model = envs.get_global_env( "cluster.save_init_model", False, namespace) diff --git a/models/recall/tdm/config.yaml b/models/recall/tdm/config.yaml index edcd3a93..0e575804 100644 --- a/models/recall/tdm/config.yaml +++ b/models/recall/tdm/config.yaml @@ -54,7 +54,6 @@ train: tree_travel_path: "{workspace}/tree/travel_list.npy" tree_info_path: "{workspace}/tree/tree_info.npy" tree_emb_path: "{workspace}/tree/tree_emb.npy" - single: load_persistables: False persistables_model_path: "" -- GitLab