Add SAGE model card and paper citation
Browse files
README.md
CHANGED
|
@@ -1,3 +1,37 @@
|
|
| 1 |
---
|
| 2 |
-
license:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
+
license: mit
|
| 3 |
+
library_name: pytorch
|
| 4 |
+
tags:
|
| 5 |
+
- robotics
|
| 6 |
+
- world-model
|
| 7 |
+
- planning
|
| 8 |
+
- pusht
|
| 9 |
+
- ogbench
|
| 10 |
+
arxiv: 2607.17973
|
| 11 |
---
|
| 12 |
+
|
| 13 |
+
# SAGE Checkpoints
|
| 14 |
+
|
| 15 |
+
Official checkpoints for [SAGE: Subgoal-Conditioned Action Generation for Latent World Model Planning](https://arxiv.org/abs/2607.17973).
|
| 16 |
+
|
| 17 |
+
## Files
|
| 18 |
+
|
| 19 |
+
| File | Environment | Component |
|
| 20 |
+
| --- | --- | --- |
|
| 21 |
+
| `pusht_generator_epoch6.pt` | PushT | Subgoal generator |
|
| 22 |
+
| `pusht_action_prior_epoch3.pt` | PushT | Action prior |
|
| 23 |
+
| `cube_generator_epoch6.pt` | OGBench Cube | Subgoal generator |
|
| 24 |
+
| `cube_action_prior_epoch3.pt` | OGBench Cube | Action prior |
|
| 25 |
+
|
| 26 |
+
The accompanying code and usage instructions are available in the SAGE release package.
|
| 27 |
+
|
| 28 |
+
## Citation
|
| 29 |
+
|
| 30 |
+
```bibtex
|
| 31 |
+
@article{cheng2026sage,
|
| 32 |
+
title={SAGE: Subgoal-Conditioned Action Generation for Latent World Model Planning},
|
| 33 |
+
author={Cheng, Letian and Zhang, Qi and Wang, Qixun and Wang, Yisen},
|
| 34 |
+
journal={arXiv preprint arXiv:2607.17973},
|
| 35 |
+
year={2026}
|
| 36 |
+
}
|
| 37 |
+
```
|