berkus
/
wifi-densepose
mirror of https://github.com/ruvnet/wifi-densepose.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat: Mac Mini M4 Pro training script (7-step pipeline) 

Clone, copy data via Tailscale, train, benchmark, sync results,
publish to HuggingFace — all automated for M4 Pro hardware.

Co-Authored-By: claude-flow <ruv@ruv.net>
This commit is contained in:
ruv 2026-04-02 22:42:32 -04:00
parent ade0fe82f6
commit ccc543c0e7
1 changed files with 82 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

82
scripts/mac-mini-train.sh Normal file
View File

@ -0,0 +1,82 @@
#!/bin/bash
set -euo pipefail
echo "=== WiFi-DensePose Mac Mini M4 Pro Training Pipeline ==="
echo "Host: $(hostname) | $(sysctl -n hw.ncpu 2>/dev/null || nproc) cores | $(sysctl -n hw.memsize 2>/dev/null | awk '{printf "%.0f GB", $1/1073741824}' || free -h | awk '/Mem:/{print $2}')"
echo ""
REPO_DIR="${HOME}/Projects/wifi-densepose"
WINDOWS_HOST="100.102.238.73" # Tailscale IP of Windows machine
# Step 1: Clone or update repo
echo "[1/7] Setting up repository..."
if [ -d "$REPO_DIR/.git" ]; then
cd "$REPO_DIR" && git pull origin main
else
git clone https://github.com/ruvnet/RuView.git "$REPO_DIR"
cd "$REPO_DIR"
fi
# Step 2: Install Node.js if needed
echo "[2/7] Checking Node.js..."
if ! command -v node &>/dev/null; then
echo "Installing Node.js via Homebrew..."
brew install node
fi
echo "Node $(node --version)"
# Step 3: Copy training data from Windows via Tailscale
echo "[3/7] Copying training data from Windows machine..."
mkdir -p data/recordings
scp -o ConnectTimeout=5 "ruv@${WINDOWS_HOST}:Projects/wifi-densepose/data/recordings/pretrain-*.csi.jsonl" data/recordings/ 2>/dev/null || {
echo " Could not reach Windows machine. Checking for local data..."
if ls data/recordings/pretrain-*.csi.jsonl &>/dev/null; then
echo " Found local training data."
else
echo " ERROR: No training data found. Run collect-training-data.py on Windows first."
exit 1
fi
}
echo " Data: $(wc -l data/recordings/pretrain-*.csi.jsonl | tail -1)"
# Step 4: Run enhanced training (larger model, more epochs)
echo "[4/7] Training (enhanced config for M4 Pro)..."
time node scripts/train-ruvllm.js \
--data data/recordings/pretrain-*.csi.jsonl \
2>&1 | tee models/csi-ruvllm/training.log
# Step 5: Benchmark
echo "[5/7] Benchmarking..."
node scripts/benchmark-ruvllm.js \
--model models/csi-ruvllm \
--data data/recordings/pretrain-*.csi.jsonl \
2>&1 | tee models/csi-ruvllm/benchmark.log
# Step 6: Copy results back to Windows
echo "[6/7] Syncing results back to Windows..."
scp -r -o ConnectTimeout=5 models/csi-ruvllm/ "ruv@${WINDOWS_HOST}:Projects/wifi-densepose/models/csi-ruvllm-m4pro/" 2>/dev/null || {
echo " Could not reach Windows. Results are in: $REPO_DIR/models/csi-ruvllm/"
}
# Step 7: Publish to HuggingFace
echo "[7/7] Publishing to HuggingFace..."
if command -v gcloud &>/dev/null; then
mkdir -p dist/models
cp models/csi-ruvllm/model.safetensors dist/models/
cp models/csi-ruvllm/config.json dist/models/
cp models/csi-ruvllm/presence-head.json dist/models/
cp models/csi-ruvllm/quantized/* dist/models/ 2>/dev/null || true
cp models/csi-ruvllm/lora/* dist/models/ 2>/dev/null || true
cp models/csi-ruvllm/model.rvf.jsonl dist/models/ 2>/dev/null || true
cp models/csi-ruvllm/training-metrics.json dist/models/ 2>/dev/null || true
cp docs/huggingface/MODEL_CARD.md dist/models/README.md 2>/dev/null || true
bash scripts/publish-huggingface.sh --version v0.5.4 2>&1 || echo " HF publish skipped (check gcloud auth)"
else
echo " gcloud not installed — skipping HF publish. Run manually:"
echo " bash scripts/publish-huggingface.sh --version v0.5.4"
fi
echo ""
echo "=== Complete ==="
echo "Models: $REPO_DIR/models/csi-ruvllm/"
echo "Logs: training.log, benchmark.log"