WaitXian's picture
Upload liqi files
0ab2566 verified
Raw History Blame Contribute Delete
838 Bytes
"""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)