38 lines
1.1 KiB
Bash
Executable File
38 lines
1.1 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# 补生 Video-MME 6 类缺失题型（180 题）
|
|
# 背景：原 360 题中 6 类 task_type 使用了错误的自创类型，
|
|
# 已归档为 archived_wrong_task_types.tar.gz。
|
|
# 本脚本只生成缺失的 6 类，生成完成后手动合并到 accepted_questions.json。
|
|
|
|
set -euo pipefail
|
|
|
|
cd "$(dirname "$0")/.."
|
|
|
|
STORE_DIR="${STORE_DIR:-store}"
|
|
CONFIG="${CONFIG:-config/question_gen_180_补.yaml}"
|
|
DB_PATH="${DB_PATH:-logs/question_gen_补6类.db}"
|
|
|
|
export PYTHONUNBUFFERED=1
|
|
export HF_HUB_OFFLINE=1
|
|
export TRANSFORMERS_OFFLINE=1
|
|
|
|
# shellcheck source=../.env
|
|
source .env
|
|
|
|
[ "${MODE:-}" = "mock" ] && export LLM_MOCK=1 VLM_MOCK=1
|
|
|
|
PYTHON="$(conda run -n Video-Tree-TRM which python)"
|
|
|
|
"${PYTHON}" tools/generate_questions.py generate-v2 \
|
|
--store-dir "$STORE_DIR" \
|
|
--config "$CONFIG" \
|
|
--db-path "$DB_PATH" \
|
|
--task-types \
|
|
"Attribute Perception" \
|
|
"Counting Problem" \
|
|
"Information Synopsis" \
|
|
"OCR Problems" \
|
|
"Spatial Perception" \
|
|
"Temporal Perception" \
|
|
${SEED:+--seed "$SEED"}
|