@@ -277,7 +277,8 @@ def main():
}
# 保存最终结果
- output_file = os.path.join(output_dir, f"OmniDocBench_Single_batch-{args.batch_size}.json")
+ output_file_name = Path(output_dir).name
+ output_file = os.path.join(output_dir, f"{output_file_name}.json")
final_results = {
"stats": stats,
"results": results