Ver Fonte

save model

丁云鹏 há 4 meses atrás
pai
commit
ebd3c814d4

+ 4 - 4
recommend-model-produce/src/main/python/models/dssm/config_ps_hdfs.yaml

@@ -13,14 +13,14 @@
 # limitations under the License.
 
 runner:
-  train_data_dir: "/dypdemo"
-  train_reader_path: "bq_reader_train"  # importlib format
+  train_data_dir: "/dyp"
+  train_reader_path: "bq_reader_train_ps"  # importlib format
   train_batch_size: 8
   model_save_path: "output_model_dssm"
   split_file_list: true
 
   reader_type: "QueueDataset"  # DataLoader / QueueDataset / RecDataset
-  pipe_command: "python bq_reader_train.py"
+  pipe_command: "python bq_reader_train_ps.py"
   thread_num: 1
   sync_mode: "async"
 
@@ -30,7 +30,7 @@ runner:
   print_interval: 1
   
   test_data_dir: "data/test"
-  infer_reader_path: "bq_reader_infer"  # importlib format
+  infer_reader_path: "bq_reader_train_ps"  # importlib format
   infer_batch_size: 1
   infer_load_path: "/app/output_model_dssm"
   infer_start_epoch: 0