dssm train

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

@@ -301,8 +301,8 @@ class Main(object):
         output_file = os.path.join(output_dir, f"epoch_{epoch}_results.jsonl")
         
         # 创建处理器实例
-        fetch_handler = InferenceFetchHandler(output_file)
-        fetch_handler.set_var_dict(self.metrics)
+        fetch_handler = InferenceFetchHandler(var_dict = self.metrics, output_file =output_file)
+        # fetch_handler.set_var_dict(self.metrics)
 
         print(paddle.static.default_main_program()._fleet_opt)
         self.exe.infer_from_dataset(