Improve model card: Add pipeline tag, links, license, and usage info

#1
by nielsr HF Staff - opened
Files changed (1) hide show
  1. README.md +77 -3
README.md CHANGED
@@ -1,3 +1,77 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ pipeline_tag: robotics
4
+ ---
5
+
6
+ # Much Ado About Noising: Dispelling the Myths of Generative Robotic Control
7
+
8
+ This repository contains the code for the paper [Much Ado About Noising: Dispelling the Myths of Generative Robotic Control](https://huggingface.co/papers/2512.01809). It provides a PyTorch-based framework for behavior cloning with flow matching and related generative models, incorporating best practices from diffusion model training. The research investigates the factors underlying the success of generative control policies (GCPs) in robotics, suggesting their advantage stems from iterative computation with intermediate supervision and stochasticity, rather than multimodal action distributions or complex observation-to-action mappings.
9
+
10
+ - **Project Page:** https://simchowitzlabpublic.github.io/much-ado-about-noising-project/
11
+ - **Code Repository:** https://github.com/simchowitzlabpublic/much-ado-about-noising
12
+
13
+ ## Features
14
+
15
+ - 🧩 **Clean & Modular**: Composable components for losses, samplers, networks, and encoders
16
+ - ⚡ **Fast**: Optimized with torch.compile and CUDA graphs for maximum throughput
17
+ - 📊 **Best Practices**: EMA, warmup scheduling, auto-resume, and proven training techniques
18
+ - 🎯 **Diverse Algorithms**: Support for flow matching, consistency models, shortcut models, and regression
19
+ - 🤖 **Robot-Ready**: Pre-configured for Robomimic, Kitchen, and PushT tasks
20
+
21
+ ## Installation
22
+
23
+ ```bash
24
+ uv sync
25
+ # install robomimic dependencies
26
+ uv sync --extras robomimic
27
+ # install kitchen dependencies
28
+ uv sync --extras kitchen
29
+ ```
30
+
31
+ ## Quick Start
32
+
33
+ ### Training
34
+
35
+ ```bash
36
+ # Train Robomimic (state observations)
37
+ uv run examples/train_robomimic.py \
38
+ task=lift_ph_state \
39
+ network=chiunet \
40
+ optimization.loss_type=flow \
41
+ log.wandb_mode=online
42
+
43
+ # Train Robomimic (image observations)
44
+ uv run examples/train_robomimic.py \
45
+ task=lift_ph_image \
46
+ network=chiunet \
47
+ optimization.batch_size=256
48
+
49
+ # Train Kitchen
50
+ uv run examples/train_kitchen.py task=kitchen_state
51
+
52
+ # Train PushT
53
+ uv run examples/train_pusht.py task=pusht_state
54
+ ```
55
+
56
+ ### Evaluation
57
+
58
+ ```bash
59
+ # Evaluate trained model
60
+ uv run examples/train_robomimic.py \
61
+ mode=eval \
62
+ optimization.model_path="/path/to/checkpoint.pt"
63
+ ```
64
+
65
+ ## Citation
66
+
67
+ If you find our work helpful, please cite the paper:
68
+
69
+ ```bibtex
70
+ @article{anon2025much,
71
+ title={Much Ado About Noising: Dispelling the Myths of Generative Robotic Control},
72
+ author={Anonymous},
73
+ journal={arXiv preprint arXiv:2512.01809},
74
+ year={2025},
75
+ url={https://arxiv.org/abs/2512.01809}
76
+ }
77
+ ```