CLTRAY commited on
Commit
f896212
·
verified ·
1 Parent(s): 0d47f33

Add SAGE model card and paper citation

Browse files
Files changed (1) hide show
  1. README.md +35 -1
README.md CHANGED
@@ -1,3 +1,37 @@
1
  ---
2
- license: apache-2.0
 
 
 
 
 
 
 
 
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
+ ```