environment/run_peract2_13_rollouts.sh

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

REPO_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
source "${REPO_ROOT}/environment/runtime_env_vars.sh"

CHECKPOINT="${1:-${REPO_ROOT}/artifacts/outputs/r3d/proxy_interaction_r3d_stage2_clip_seed13/checkpoint_best.pt}"
OUTPUT_ROOT="${2:-${REPO_ROOT}/artifacts/outputs/r3d}"

echo "[peract2] checkpoint=${CHECKPOINT}"
echo "[peract2] output_root=${OUTPUT_ROOT}"

/workspace/.tools/micromamba/bin/micromamba run -r /workspace/.micromamba -p /workspace/envs/rlbench \
  bash -lc "cd /workspace/reveal_vla_bimanual && PYTHONPATH=/workspace/reveal_vla_bimanual python -m eval.run_peract2_launch_smoke --output-dir ${OUTPUT_ROOT}/peract2_13_launch_smoke --headless"

/workspace/.tools/micromamba/bin/micromamba run -r /workspace/.micromamba -p /workspace/envs/rlbench \
  bash -lc "cd /workspace/reveal_vla_bimanual && PYTHONPATH=/workspace/reveal_vla_bimanual python -m eval.run_rlbench_rollout_eval --checkpoint ${CHECKPOINT} --output-dir ${OUTPUT_ROOT}/rlbench_open_drawer_r3d_rollout --tasks open_drawer --episodes-per-task 1 --episode-length 5 --resolution 224 --device cuda --chunk-commit-steps 4 --headless --plan --allow-unsupervised-planning"

echo "[peract2] done"