丁云鹏 vor 4 Monaten
Ursprung
Commit
d77eb3ac37

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

@@ -36,7 +36,7 @@ runner:
   infer_start_epoch: 0
   infer_end_epoch: 1
   infer_reader_type: "QueueDataset"
-
+  
   oss_object_name: "dyp/dssm.tar.gz"
 
 hyper_parameters:

+ 1 - 1
recommend-model-produce/src/main/python/tools/static_ps_infer_v2.py

@@ -322,7 +322,7 @@ class Main(object):
             program=paddle.static.default_main_program(),
             dataset=self.reader,
             fetch_list=fetch_vars,
-            fetch_info=None,
+            fetch_info=fetch_info,
             print_period=print_step,
             debug=debug,
             fetch_handler=fetch_handler)