丁云鹏 5 months ago
parent
commit
3b389cec8a
1 changed files with 1 additions and 1 deletions
  1. 1 1
      recommend-model-produce/src/main/python/models/dnn/config.yaml

+ 1 - 1
recommend-model-produce/src/main/python/models/dnn/config.yaml

@@ -36,7 +36,7 @@ runner:
   # model_init_path: "output_model_dnn/2" # init model
   # model_init_path: "output_model_dnn/2" # init model
   use_inference: True
   use_inference: True
   save_inference_feed_varnames: ["1"]
   save_inference_feed_varnames: ["1"]
-  save_inference_fetch_varnames: ["1", "label"]
+  save_inference_fetch_varnames: ["1","2","3","4","5","6","7","8","9","10","11","12","13","14","15","16","17","18","19","20","21","22","23","24","25","26","dense_input","label", "SparseFeatFactors"]
 
 
   # distribute_config
   # distribute_config
   sync_mode: "async"
   sync_mode: "async"