infer.sh

#!/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