ReTime / scripts /run_frameskip_plus500.sh
mmmmmcyaaa's picture
Add Florence-Bridge code and handoff documentation
977b06e verified
Raw History Blame Contribute Delete
973 Bytes
#!/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