|
@@ -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)
|