ITFormer / scripts /run_full_train_qformer_4gpu.sh
a12354's picture
Add files using upload-large-folder tool
aabdb98 verified
Raw
History Blame Contribute Delete
692 Bytes
#!/bin/bash
set -Eeuo pipefail
# Q-Former adapter baseline for comparison with run_full_train_4gpu.sh.
# Reuses the same Stage A TS encoder checkpoint by default and only changes
# the Stage B adapter from ITFormer to a plain Q-Former.
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
PROJECT_DIR="$(cd "${SCRIPT_DIR}/.." && pwd)"
export RUN_ID="${RUN_ID:-$(date +%Y%m%d_%H%M%S)}"
export ADAPTER_TYPE="${ADAPTER_TYPE:-qformer}"
export RUN_STAGE_A="${RUN_STAGE_A:-auto}"
export SFT_OUTPUT_DIR="${SFT_OUTPUT_DIR:-save/sft_qformer_qwen2.5_3B_${RUN_ID}}"
export LOG_FILE="${LOG_FILE:-logs/full_train_qformer_4gpu_${RUN_ID}.log}"
bash "${PROJECT_DIR}/scripts/run_full_train_4gpu.sh"