Upload folder using huggingface_hub
Browse files- README.md +80 -0
- os +0 -0
- 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).
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|