vla / scripts /run_train_debug.sh
anhtld's picture
Initial commit: DoVLA-CIL codebase (h=16 breakthrough)
adc02fa verified
Raw
History Blame
875 Bytes
#!/usr/bin/env bash
set -euo pipefail
ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
cd "$ROOT_DIR"
export OPENCLAUDE_MOCK="${OPENCLAUDE_MOCK:-1}"
OUT_ROOT="${DOVLA_DEBUG_ROOT:-outputs/phase5_train_debug}"
TASKS_PATH="$OUT_ROOT/tasks.jsonl"
DATASET_DIR="$OUT_ROOT/cil"
RUN_DIR="$OUT_ROOT/run"
mkdir -p "$OUT_ROOT"
python scripts/generate_tasks.py --mock --num-tasks 3 --out "$TASKS_PATH" --seed 0
python scripts/generate_cil.py \
--backend toy \
--tasks "$TASKS_PATH" \
--out "$DATASET_DIR" \
--num-states-per-task 2 \
--k 4 \
--seed 0 \
--shard-size 8 \
--inline-observations
python scripts/train_dovla.py \
--dataset "$DATASET_DIR" \
--out "$RUN_DIR" \
--epochs 1 \
--batch-groups 2 \
--records-per-group 4 \
--hidden-dim 64 \
--lr 0.001 \
--device "${DOVLA_DEVICE:-auto}" \
--seed 0
echo "debug training run: $RUN_DIR"