#!/usr/bin/env bash
set -euo pipefail

if [[ $# -ne 2 ]]; then
  echo "usage: $0 <run-id> <frameskip-cache.json>" >&2
  exit 64
fi

run_id=$1
cache=$2
[[ -f "$cache" ]] || { echo "cache not found: $cache" >&2; exit 66; }
project="${OPTR_PROJECT_ROOT:-$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)}"
python="${OPTR_PYTHON:-python}"
export FRAMESKIP_CACHE="$cache"
export FRAMESKIP_START_GLOBAL_DRAW_ID="${FRAMESKIP_START_GLOBAL_DRAW_ID:-1280000}"
export PYTHONPATH="${project}:${project}/third_party/frameskip:${PYTHONPATH:-}"
export WANDB_MODE="${WANDB_MODE:-offline}"
export PYTHONUNBUFFERED=1

cd "$project"
"$python" -m accelerate.commands.launch \
  --config_file configs/deepspeed_florence_2gpu.yaml \
  --num_processes 2 \
  external_baselines/frameskip_adapted/train_frameskip_continuation.py \
  --config_yaml configs/florence_bridge_20k_plus500.yaml \
  --run_id "$run_id" \
  --run_root_dir "$OPTR_RUN_ROOT" \
  --datasets.vla_data.temporal_proposal_path=null