丁云鹏 1 год назад
Родитель
Сommit
fad2d6be69
1 измененных файлов с 1 добавлено и 0 удалено
  1. 1 0
      recommend-model-produce/src/main/python/models/dssm/config_ps_hdfs.yaml

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

@@ -35,6 +35,7 @@ runner:
   infer_load_path: "/app/output_model_dssm"
   infer_load_path: "/app/output_model_dssm"
   infer_start_epoch: 0
   infer_start_epoch: 0
   infer_end_epoch: 1
   infer_end_epoch: 1
+  infer_reader_type: "QueueDataset"
 
 
 hyper_parameters:
 hyper_parameters:
   optimizer:
   optimizer: