Skip to content

feat: arrange-3-flowers task (DOF layout fix, data conversion, training/deploy docs)Feat/arrange 3 flowers - #118

Merged
StarrickLiu merged 7 commits into
X-Square-Robot:mainfrom
yuzhuangli:feat/arrange-3-flowers
Aug 31, 2026
Merged

StarrickLiu merged 7 commits into
X-Square-Robot:mainfrom
yuzhuangli:feat/arrange-3-flowers

Conversation

@yuzhuangli

Copy link
Copy Markdown
Contributor

Summary

  • Fix LeRobot DOF layout when training with arm-only action_padding (loader, norm stats, inference).
  • Add arrange-3-flowers training guide, example configs, and RTX5090 deploy scripts.
  • Add x2robot2lerobot.py and template configs for x2robot → LeRobot v3 conversion.
  • Fix open-loop eval to strip action_padding from ground-truth actions.

Motivation

When the dataset stores extra DOFs (velocity/height/head) but training only predicts dual-arm actions with action_padding, the loader, norm stats, and serving must agree on the effective layout.

Test plan

  • python scripts/compute_norm_stats.py --train_config workspace/example/arrange_3_flowers_wrc_red.yml (with local paths)
  • Smoke train with arrange_3_flowers_wrc_red_smoke.yml
  • Start serving via bash workspace/rtx5090/run_server.sh and check /healthz
  • Run open-loop eval plot script on a checkpoint

Notes

  • Example paths use <path/to/...> placeholders per CONTRIBUTING.md.
  • Machine-specific workspace/rtx5090/local/env.sh is gitignored.

yuzhuangli and others added 7 commits August 27, 2026 21:33
Match LAYOUT_SKIP_KEYS across loader, norm stats, and inference so datasets
with extra velocity/height/head DOFs can train on arm-only layouts.

Co-authored-by: Cursor <cursoragent@cursor.com>
Add LeRobot-based arrange_3_flowers training doc, example configs, and
rtx5090 install/run scripts with env-based path layout for inference.

Co-authored-by: Cursor <cursoragent@cursor.com>
Link arrange_3_flowers training example and RTX 5090 deploy docs for newcomers.

Co-authored-by: Cursor <cursoragent@cursor.com>
Align ground-truth action dimensions with model predictions when datasets
store extra DOF padding beyond the effective arm-only dof_config.

Co-authored-by: Cursor <cursoragent@cursor.com>
Cherry-pick DOF layout fix, training/deploy docs, and open-loop eval fix
from fix/lerobot-dof-layout onto upstream/main. Add x2robot2lerobot
conversion script with template configs, sanitize internal paths to
placeholders, and stop tracking machine-specific rtx5090/local/env.sh.

Co-authored-by: Cursor <cursoragent@cursor.com>
Drop the hardcoded Alibaba OSS URL for Mellanox RDMA user-space libs so
the open-source example no longer depends on an internal artifact host.

Co-authored-by: Cursor <cursoragent@cursor.com>
Fix truncated Qwen path placeholders, use generic wallx conda default,
and point NOTES.md at env.example instead of the gitignored env.sh.

Co-authored-by: Cursor <cursoragent@cursor.com>
@StarrickLiu
StarrickLiu merged commit c0f5d67 into X-Square-Robot:main Aug 31, 2026
1 check failed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants