cd $PROJECT_ROOT/tools/PMPD
pip install .
cd $PROJECT_ROOT/src/code/pruning
nohup python evaluate.py --model qwen7b \
                --dataset gsm8k \
                --reward_model prm \
                --scheduler part_split \
                --prompt_type better \
                --device cuda:$3 \
                --max_steps 16384 \
                --prefill_bit $1 \
                --naive_bit $2 \
                --high_bit_steps 512 \
                --part cot \
                --do_sample True \
                --temperature 0.6 \
                --prune_path 7b_gsm8k_654_prm_20_agent \
                --output results/qwen7b_gsm8k_$1_$2.json > gsm8k_7b_$1_$2.log 