File size: 968 Bytes
fa8ed31 b8166f5 fa8ed31 b28a77d fa8ed31 b28a77d fa8ed31 |
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 29 30 |
#!/bin/bash
# 定义参数
MODEL_DIR="/inspire/hdd/ws-ba572160-47f8-4ca1-984e-d6bcdeb95dbb/a100-maybe/albus/ICCV_2025/CogAgent/ckpt/cogagent-9b-20241220"
PLATFORM="Mac"
MAX_LENGTH=4096
TOP_K=1
OUTPUT_IMAGE_PATH="./results/"
FORMAT_KEY="status_action_op_sensitive"
# 定义执行次数
NUM_RUNS=170 # 设置执行次数
# 循环执行 Python 脚本多次
for i in $(seq 1 $NUM_RUNS)
do
INPUT_JSON="/inspire/hdd/ws-ba572160-47f8-4ca1-984e-d6bcdeb95dbb/a100-maybe/albus/ICCV_2025/CogAgent/dataset/$i.json"
OUTPUT_JSON="/inspire/hdd/ws-ba572160-47f8-4ca1-984e-d6bcdeb95dbb/a100-maybe/albus/ICCV_2025/CogAgent/results/json/$i.json"
echo "Running script, round $i..."
python cogagent_infer.py \
--model_dir "$MODEL_DIR" \
--platform "$PLATFORM" \
--max_length $MAX_LENGTH \
--top_k $TOP_K \
--output_image_path f"$OUTPUT_IMAGE_PATH{i}" \
--format_key "$FORMAT_KEY" \
--input_json "$INPUT_JSON" \
--output_json "$OUTPUT_JSON"
done
|