fix

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)

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