File size: 973 Bytes
977b06e | 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 | #!/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
|