feat: add SO101 recording and ACT workflow with progress tracking

This commit is contained in:
ylhan
2026-09-10 16:46:58 +08:00
parent 2e33355a9c
commit cbf51facff
16 changed files with 604 additions and 7 deletions
+14
View File
@@ -0,0 +1,14 @@
#!/usr/bin/env bash
source "$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd)/block_white_box_common.sh"
if [[ "${DRY_RUN:-0}" != 1 && ! -f "$DATASET_ROOT/meta/info.json" ]]; then
echo "请先录制并检查数据:$DATASET_ROOT" >&2; exit 1
fi
if [[ "${DRY_RUN:-0}" != 1 ]]; then
python -c 'import torch; import sys; sys.exit(0 if torch.cuda.is_available() else "CUDA 不可用,请先修复 NVIDIA 驱动;已停止训练以避免回退到 CPU。")'
fi
TRAIN_OUTPUT="${TRAIN_OUTPUT:-$LEROBOT_PROJECT_DIR/outputs/train/act_block_white_box_$(date +%Y%m%d_%H%M%S)}"
run_task lerobot-train --policy.type=act --policy.device=cuda --policy.push_to_hub=false \
"--dataset.repo_id=$DATASET_ID" "--dataset.root=$DATASET_ROOT" \
"--output_dir=$TRAIN_OUTPUT" --job_name=act_block_white_box \
"--batch_size=${BATCH_SIZE:-8}" "--steps=${STEPS:-20000}" --num_workers=4 \
--save_freq=2000 --env_eval_freq=0 --seed=42 --wandb.enable=false "$@"