dssm train

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

@@ -316,13 +316,13 @@ class Main(object):
         output_file = os.path.join(output_dir, f"epoch_{epoch}_results.jsonl")
         
         # 创建处理器实例
-        fetch_handler = FetchHandler(var_dict=fetch_vars, period_secs=1)
+        fetch_handler = FetchHandler(var_dict=self.metrics, period_secs=1)
         print(paddle.static.default_main_program()._fleet_opt)
         self.exe.infer_from_dataset(
             program=paddle.static.default_main_program(),
             dataset=self.reader,
             fetch_list=fetch_vars,
-            fetch_info=fetch_info,
+            fetch_info=None,
             print_period=print_step,
             debug=debug,
             fetch_handler=fetch_handler)