丁云鹏 5 ماه پیش
والد
کامیت
be529594bb
1فایلهای تغییر یافته به همراه2 افزوده شده و 1 حذف شده
  1. 2 1
      recommend-model-produce/src/main/python/tools/static_ps_infer_v2.py

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

@@ -260,13 +260,14 @@ class Main(object):
                 "dump_fields": config.get("runner.dump_fields")
             })
         print(paddle.static.default_main_program()._fleet_opt)
-        self.exe.infer_from_dataset(
+        results = self.exe.infer_from_dataset(
             program=paddle.static.default_main_program(),
             dataset=self.reader,
             fetch_list=fetch_vars,
             fetch_info=fetch_info,
             print_period=print_step,
             debug=debug)
+        print("results {}".format(results))
 
 
     def heter_train_loop(self, epoch):