thl1246 commited on
Commit
a7a4200
Β·
verified Β·
1 Parent(s): 92b7ee9

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. README.md +80 -0
  2. os +0 -0
  3. robomimic/README.md +0 -79
README.md CHANGED
@@ -1,3 +1,83 @@
1
  ---
2
  license: mit
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
+ tags:
4
+ - robotics
5
+ - imitation-learning
6
+ - diffusion-policy
7
+ - robomimic
8
+ - manipulation
9
  ---
10
+
11
+ # ReGuide β€” Checkpoints
12
+
13
+ Pretrained **diffusion policy** and **visual dynamics model** checkpoints for
14
+ **ReGuide: From Test-Time Guidance to Self-Improving Diffusion Policies**
15
+ (Lin, Shakkottai, Kalathil, Kumar β€” Texas A&M University).
16
+
17
+ ReGuide is a self-improving framework that recycles *guided rollouts* as on-policy
18
+ recovery data to fix covariate shift in behavior-cloned diffusion policies, improving
19
+ base-policy success by **1.3–7.7Γ—** on Robomimic. See the
20
+ [project page](https://reguide-project.github.io) and paper for method details.
21
+
22
+ Checkpoints are organized by benchmark:
23
+
24
+ - [`robomimic/`](robomimic) β€” Can, Square, Transport, Tool Hang (available now)
25
+ - `libero/` β€” coming soon
26
+
27
+ ## Robomimic tasks
28
+
29
+ | Task | Demos | Notes |
30
+ |-------------|:-----:|-------|
31
+ | `can` | 15 | policy + dynamics model |
32
+ | `square` | 30 | policy + dynamics model |
33
+ | `transport` | 10 | policy + dynamics model |
34
+ | `tool_hang` | 80 | policy + dynamics model |
35
+
36
+ ## Repository layout
37
+
38
+ ```
39
+ robomimic/diffusion_policy/<task>/<variant>/
40
+ β”œβ”€β”€ checkpoints/<epoch>.ckpt # policy weights (~4.3–5.3 GB)
41
+ β”œβ”€β”€ normalizer.pth # observation/action normalizer
42
+ └── .hydra/ # full training config
43
+
44
+ robomimic/dyn_model/<task>/
45
+ β”œβ”€β”€ checkpoints/model_<epoch>.pth # visual dynamics model (~0.5–1.5 GB)
46
+ β”œβ”€β”€ normalizer.pth
47
+ └── hydra.yaml # training config
48
+ ```
49
+
50
+ The number in each checkpoint filename (e.g. `880.ckpt`) is the training epoch.
51
+
52
+ ### Diffusion policy variants
53
+
54
+ | Variant | What it is |
55
+ |-------------------------|------------|
56
+ | `base_policy` | Diffusion policy trained on the original demonstrations only. |
57
+ | `ReGuide-FS` | Retrained from scratch on demos + guided rollouts. |
58
+ | `ReGuide-FT-iteration1` | Base checkpoint fine-tuned on demos + guided rollouts (first iteration). |
59
+ | `ReGuide-FT-iteration2` | Second fine-tuning iteration on freshly collected guided rollouts. |
60
+ | `ReGuide-FS-FT` | ReGuide-FT applied on top of a ReGuide-FS policy (best on Can/Square/Transport). |
61
+
62
+ Each checkpoint ships with its full Hydra training config (`.hydra/` for policies,
63
+ `hydra.yaml` for dynamics models), which records the exact hyperparameters used. Paths
64
+ inside the configs are relative.
65
+
66
+ ## Code
67
+
68
+ Training and evaluation code: **TODO: add GitHub repo link**
69
+
70
+ ## Citation
71
+
72
+ ```bibtex
73
+ @article{lin2026reguide,
74
+ title = {ReGuide: From Test-Time Guidance to Self-Improving Diffusion Policies},
75
+ author = {Lin, Tzu-Hsiang and Shakkottai, Srinivas and Kalathil, Dileep and Kumar, P. R.},
76
+ year = {2026},
77
+ url = {https://reguide-project.github.io}
78
+ }
79
+ ```
80
+
81
+ ## License
82
+
83
+ Released under the [MIT License](https://opensource.org/license/mit).
os ADDED
File without changes
robomimic/README.md DELETED
@@ -1,79 +0,0 @@
1
- ---
2
- license: mit
3
- tags:
4
- - robotics
5
- - imitation-learning
6
- - diffusion-policy
7
- - robomimic
8
- - manipulation
9
- ---
10
-
11
- # ReGuide β€” Robomimic Checkpoints
12
-
13
- Pretrained **diffusion policy** and **visual dynamics model** checkpoints for
14
- **ReGuide: From Test-Time Guidance to Self-Improving Diffusion Policies**
15
- (Lin, Shakkottai, Kalathil, Kumar β€” Texas A&M University).
16
-
17
- ReGuide is a self-improving framework that recycles *guided rollouts* as on-policy
18
- recovery data to fix covariate shift in behavior-cloned diffusion policies, improving
19
- base-policy success by **1.3–7.7Γ—** on Robomimic. See the
20
- [project page](https://reguide-project.github.io) and paper for method details.
21
-
22
- ## Tasks
23
-
24
- | Task | Demos | Notes |
25
- |-------------|:-----:|-------|
26
- | `can` | 15 | policy + dynamics model |
27
- | `square` | 30 | policy + dynamics model |
28
- | `transport` | 10 | policy + dynamics model |
29
- | `tool_hang` | 80 | policy + dynamics model |
30
-
31
-
32
- ## Repository layout
33
-
34
- ```
35
- diffusion_policy/<task>/<variant>/
36
- β”œβ”€β”€ checkpoints/<epoch>.ckpt # policy weights (~4.3–5.3 GB)
37
- β”œβ”€β”€ normalizer.pth # observation/action normalizer
38
- └── .hydra/ # full training config
39
-
40
- dyn_model/<task>/
41
- β”œβ”€β”€ checkpoints/model_<epoch>.pth # visual dynamics model (~0.5–1.5 GB)
42
- β”œβ”€β”€ normalizer.pth
43
- └── hydra.yaml # training config
44
- ```
45
-
46
- The number in each checkpoint filename (e.g. `880.ckpt`) is the training epoch.
47
-
48
- ### Diffusion policy variants
49
-
50
- | Variant | What it is |
51
- |-------------------------|------------|
52
- | `base_policy` | Diffusion policy trained on the original demonstrations only. |
53
- | `ReGuide-FS` | Retrained from scratch on demos + guided rollouts. |
54
- | `ReGuide-FT-iteration1` | Base checkpoint fine-tuned on demos + guided rollouts (first iteration). |
55
- | `ReGuide-FT-iteration2` | Second fine-tuning iteration on freshly collected guided rollouts. |
56
- | `ReGuide-FS-FT` | ReGuide-FT applied on top of a ReGuide-FS policy (best on Can/Square/Transport). |
57
-
58
- Each checkpoint ships with its full Hydra training config (`.hydra/` for policies,
59
- `hydra.yaml` for dynamics models), which records the exact hyperparameters used. Paths
60
- inside the configs are relative.
61
-
62
- ## Code
63
-
64
- Training and evaluation code: **TODO: add GitHub repo link**
65
-
66
- ## Citation
67
-
68
- ```bibtex
69
- @article{lin2026reguide,
70
- title = {ReGuide: From Test-Time Guidance to Self-Improving Diffusion Policies},
71
- author = {Lin, Tzu-Hsiang and Shakkottai, Srinivas and Kalathil, Dileep and Kumar, P. R.},
72
- year = {2026},
73
- url = {https://reguide-project.github.io}
74
- }
75
- ```
76
-
77
- ## License
78
-
79
- Released under the [MIT License](https://opensource.org/license/mit).