cd $PROJECT_ROOT/tools/PMPD
pip install .
cd $PROJECT_ROOT/src/code/cot_split
nohup python split_reward.py --data_path simple_split/math_qwen7b_3_5,3.log \
                             --dataset math --reward prm \
                             --agent_path $DATA_ROOT/Qwen2.5-7B-Instruct/ \
                             --cuda_id 0 \
                             --n 20 \
                             --save_path simple_split/math_qwen7b_3_5,3_prm_20_agent.png \
                             --split_method agent
nohup python /aifs4su/hansirui/zhangzy/gpu.py --util 0.8 --gpu 0