mirror of
https://github.com/microsoft/SkillOpt.git
synced 2026-07-06 16:05:12 +08:00
44 lines
2.5 KiB
Bash
Executable File
44 lines
2.5 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# ──────────────────────────────────────────────────────────────────────────────
|
|
# ReflACT — SearchQA training launch script
|
|
#
|
|
# Usage:
|
|
# bash scripts/run_searchqa.sh
|
|
# bash scripts/run_searchqa.sh --data_path data/searchqa_train_2000.json
|
|
# bash scripts/run_searchqa.sh --num_epochs 2 --edit_budget 6
|
|
# ──────────────────────────────────────────────────────────────────────────────
|
|
set -euo pipefail
|
|
|
|
# ── Paths ────────────────────────────────────────────────────────────────────
|
|
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
|
|
PROJECT_ROOT="$(dirname "${SCRIPT_DIR}")"
|
|
|
|
# Ensure ReflACT is importable
|
|
export PYTHONPATH="${PROJECT_ROOT}:${PYTHONPATH:-}"
|
|
|
|
# ── Model configuration ─────────────────────────────────────────────────────
|
|
export TEACHER_DEPLOYMENT="${TEACHER_DEPLOYMENT:-gpt-5.5}"
|
|
export STUDENT_DEPLOYMENT="${STUDENT_DEPLOYMENT:-gpt-5.5}"
|
|
|
|
# ── Output directory ─────────────────────────────────────────────────────────
|
|
TIMESTAMP=$(date +%Y%m%d_%H%M%S)
|
|
DEFAULT_OUT_ROOT="${PROJECT_ROOT}/outputs/reflact_searchqa_${STUDENT_DEPLOYMENT}_${TIMESTAMP}"
|
|
|
|
# ── Run ──────────────────────────────────────────────────────────────────────
|
|
echo "============================================================"
|
|
echo " ReflACT — Reflective Agent Tuning (SearchQA)"
|
|
echo "============================================================"
|
|
echo " Teacher: ${TEACHER_DEPLOYMENT}"
|
|
echo " Student: ${STUDENT_DEPLOYMENT}"
|
|
echo "============================================================"
|
|
|
|
cd "${PROJECT_ROOT}"
|
|
|
|
python scripts/train.py \
|
|
--config configs/searchqa_default.yaml \
|
|
--out_root "${DEFAULT_OUT_ROOT}" \
|
|
"$@"
|
|
|
|
echo ""
|
|
echo "Done! Results saved to: ${DEFAULT_OUT_ROOT}"
|