File size: 875 Bytes
adc02fa | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 | #!/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"
|