Home Explore Help
Sign In
algorithm
/
recommend-model
15
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Browse Source

add infer code

often 5 months ago
parent
6e709ad61d
commit
d3956bb30d
1 changed files with 1 additions and 1 deletions
Unified View Show Diff Stats
  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
View File 

@@ -33,7 +33,7 @@ runner:
 
   infer_load_path: "output_model_dssm"
 
   infer_load_path: "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:
 
     class: adam
 
     class: adam