Add TWIN preprocessing and norm-stats helper scripts
Browse files
openpi/scripts/launch_preprocess_twin_detached.sh
ADDED
|
@@ -0,0 +1,29 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/usr/bin/env bash
|
| 2 |
+
set -euo pipefail
|
| 3 |
+
|
| 4 |
+
SCRIPT_DIR="$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd)"
|
| 5 |
+
OPENPI_ROOT="${OPENPI_ROOT:-$(cd -- "$SCRIPT_DIR/.." && pwd)}"
|
| 6 |
+
|
| 7 |
+
ROOT="${ROOT:-$HOME/pi05prep-work}"
|
| 8 |
+
LOGDIR="${LOGDIR:-$ROOT/logs}"
|
| 9 |
+
TIMESTAMP="$(date -u +%Y%m%dT%H%M%SZ)"
|
| 10 |
+
LOGFILE="$LOGDIR/preprocess_${TIMESTAMP}.log"
|
| 11 |
+
PIDFILE="$LOGDIR/preprocess_${TIMESTAMP}.pid"
|
| 12 |
+
SESSION_FILE="$LOGDIR/preprocess_${TIMESTAMP}.session"
|
| 13 |
+
LATEST_LINK="$LOGDIR/preprocess_latest.log"
|
| 14 |
+
LATEST_PID_LINK="$LOGDIR/preprocess_latest.pid"
|
| 15 |
+
LATEST_SESSION_LINK="$LOGDIR/preprocess_latest.session"
|
| 16 |
+
SESSION_NAME="pi05prep_${TIMESTAMP}"
|
| 17 |
+
RUN_SCRIPT_PATH="${RUN_SCRIPT_PATH:-$SCRIPT_DIR/run_preprocess_twin.sh}"
|
| 18 |
+
|
| 19 |
+
mkdir -p "$LOGDIR"
|
| 20 |
+
|
| 21 |
+
tmux new-session -d -s "$SESSION_NAME" "cd '$ROOT' && unset LEROBOT_HOME && export SKIP_PREPARE=1 && export OPENPI_ROOT='${OPENPI_ROOT}' && export HF_HOME='${HF_HOME:-$ROOT/hf-home}' && export HF_HUB_CACHE='${HF_HUB_CACHE:-${HF_HOME:-$ROOT/hf-home}/hub}' && export HF_DATASETS_CACHE='${HF_DATASETS_CACHE:-${HF_HOME:-$ROOT/hf-home}/datasets}' && export HF_LEROBOT_HOME='${HF_LEROBOT_HOME:-${HF_HOME:-$ROOT/hf-home}/lerobot}' && bash '$RUN_SCRIPT_PATH' >> '$LOGFILE' 2>&1"
|
| 22 |
+
PID="$(tmux list-panes -t "$SESSION_NAME" -F '#{pane_pid}' | head -n 1)"
|
| 23 |
+
echo "$PID" >"$PIDFILE"
|
| 24 |
+
echo "$SESSION_NAME" >"$SESSION_FILE"
|
| 25 |
+
ln -sfn "$LOGFILE" "$LATEST_LINK"
|
| 26 |
+
ln -sfn "$PIDFILE" "$LATEST_PID_LINK"
|
| 27 |
+
ln -sfn "$SESSION_FILE" "$LATEST_SESSION_LINK"
|
| 28 |
+
|
| 29 |
+
printf 'session=%s\npid=%s\nlog=%s\npidfile=%s\nsession_file=%s\n' "$SESSION_NAME" "$PID" "$LOGFILE" "$PIDFILE" "$SESSION_FILE"
|