Ver Fonte

dssm train

丁云鹏 há 4 meses atrás
pai
commit
95646fec7e

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

@@ -306,7 +306,7 @@ class Main(object):
             
         # 设置输出文件路径
         output_dir = config.get("runner.inference_output_dir", "inference_results")
-        output_file = os.path.join(output_dir, f"epoch_{epoch}_results.jsonl")
+        output_file = os.path.join(output_dir, f"epoch_{epoch}_results.json")
         
         # 创建处理器实例
         fetch_handler = InferenceFetchHandler(var_dict = self.metrics, output_file = output_file)