File size: 838 Bytes
0ab2566 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 | """CoT embedding 本地评测入口。
用法：在 cot_eval/eval_config.py 中修改参数后，直接运行
python cot_eval/eval_local.py
单卡直接跑；多卡也可以用
accelerate launch --num_processes 4 cot_eval/eval_local.py
python -m accelerate.commands.launch --num_processes 4 cot_eval/eval_local.py
python -m pip install jsonlines accelerate peft qwen_vl_utils
"""
import os
import sys
current_file_path = os.path.dirname(os.path.abspath(__file__))
module_path = os.path.join(current_file_path, "../")
sys.path.append(module_path)
from cot_eval.eval_config import EvalConfig
from cot_eval.eval_tuwen_qwen3vl_cot import run_eval
if __name__ == "__main__":
args = EvalConfig()
print("eval config:")
for key, value in vars(args).items():
print(f" {key}: {value}")
run_eval(args)
|