소스 검색

add infer code

often 5 달 전
부모
커밋
d3956bb30d
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      recommend-model-produce/src/main/python/models/dssm/config_ps.yaml

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

@@ -33,7 +33,7 @@ runner:
   infer_load_path: "output_model_dssm"
   infer_start_epoch: 0
   infer_end_epoch: 1
-
+  infer_reader_type: "QueueDataset"
 hyper_parameters:
   optimizer:
     class: adam