rusrakhimov commited on
Commit
8de53df
·
verified ·
1 Parent(s): 47b40e9

Add 24 headline checkpoints (Qantara + LeWM, 4 envs x 3 seeds)

Browse files
README.md ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ library_name: pytorch
4
+ tags:
5
+ - jepa
6
+ - world-models
7
+ - model-based-rl
8
+ - planning
9
+ ---
10
+
11
+ # Qantara — headline checkpoints
12
+
13
+ Released checkpoints for **Qantara**, a compact (~15M-param) JEPA world model for
14
+ goal-conditioned planning from pixels and actions, and for our **LeWM** reproduction
15
+ baseline. These reproduce the LeWM-suite headline numbers (Table 1) from the ICML 2026
16
+ workshop paper.
17
+
18
+ - **Code:** https://github.com/corl-team/qantara
19
+ - **Paper:** https://arxiv.org/abs/2607.04978
20
+
21
+ ## Files
22
+
23
+ 24 checkpoints = 2 methods × 4 environments × 3 training seeds.
24
+
25
+ | Pattern | Method |
26
+ |---|---|
27
+ | `qantara-<env>-s<seed>.ckpt` | Qantara (γ=1, λ_z=3, nulldrop=0) headline |
28
+ | `lewm-<env>-s<seed>.ckpt` | LeWM reproduction baseline |
29
+
30
+ `env ∈ {pusht, tworoom, cube, reacher}`, `seed ∈ {11, 22, 33}`.
31
+
32
+ ## Loading
33
+
34
+ Each file is a full model object (`torch.save` of a `jepa.JEPA`). Clone the code repo,
35
+ then:
36
+
37
+ ```python
38
+ import torch
39
+ model = torch.load("qantara-pusht-s11.ckpt", map_location="cpu", weights_only=False)
40
+ ```
41
+
42
+ The repository's `eval.py` consumes these directly. See the repo README for the full
43
+ train → eval → figure pipeline.
44
+
45
+ ## Citation
46
+
47
+ ```bibtex
48
+ @inproceedings{qantara2026,
49
+ title = {Qantara: Bridge-Flow Training for Multi-Paradigm JEPA Control},
50
+ author = {Rakhimov, Ruslan and Bredis, George and Maksyuta, Yuriy and Gavrilov, Daniil},
51
+ booktitle = {ICML 2026 Workshop on Decision-Making from Offline Datasets to Online Adaptation: Black-Box Optimization to Reinforcement Learning},
52
+ year = {2026},
53
+ url = {https://arxiv.org/abs/2607.04978},
54
+ }
55
+ ```
lewm-cube-s11.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:af4c0fa00eba5a3f92f371384933ddc2f54a1e4567f6eedb7b457203dee28f5e
3
+ size 77610047
lewm-cube-s22.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b26955903c1b4e6a9c02d38c19accea7636af3ee65a1ea173c9564cbaa83832e
3
+ size 77610047
lewm-cube-s33.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6288d87d2e49f380d4bfd3c75e952cbfb71bcaab99edc3d505eff959d391f0b9
3
+ size 77610047
lewm-pusht-s11.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:db2a24772723e1e57851763a8bc810504340e834e523c371f2bcb430bf2a8656
3
+ size 77616022
lewm-pusht-s22.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:783ca80a4e58464c83d4c261589b3db7303b8b08e29d4fccbdba3e17f20e205b
3
+ size 77616022
lewm-pusht-s33.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:186b9cd00738618a59a9a17802dcef81cdd5ccab764fbebf2b1cfe81878be5b7
3
+ size 77616022
lewm-reacher-s11.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:83c48ad8d8ba5c1b52f63a1d0ba9f3cec892eb93f86d381b12c2400accb8d24d
3
+ size 77610470
lewm-reacher-s22.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c4fbcaf5328db17342f4a6b010a4ad29f7af6b97123e9892423ce12cfb59e74
3
+ size 77610470
lewm-reacher-s33.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6c9ba4eaa4c9083e2dfc3afdbe4ae886e55f06fc2bcf8a7085eb819d97d0d668
3
+ size 77610470
lewm-tworoom-s11.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a977a53a68101c6bf2eeb165cd29b83b3965352314dfd0dd480c8d1e19536e5d
3
+ size 77610470
lewm-tworoom-s22.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eb1bc361575ad6494f21ca8fcf480f9806743c84b2b1d7739ea89aa42cd5b428
3
+ size 77610470
lewm-tworoom-s33.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6db29ddc3bb93647c23b86a88b72456acb175bb3ba67469cd4757778fd372293
3
+ size 77610470
qantara-cube-s11.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c7209f65841f565e7735ac72d9809398b7a85796c4dc046e1ee6dc94bc4ecc8
3
+ size 91710611
qantara-cube-s22.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c6347374afdcf7f77577bd58bc9ad05b624deee0a096380292d8fca29b92b7be
3
+ size 91710611
qantara-cube-s33.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3866c57df2ca00268508a909303a048635db3391d7639bccd60f3d7b3c8b85ca
3
+ size 91710611
qantara-pusht-s11.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9c0177685b9bf15cae299ca7357ee0bf685fb8dacc122fbdce4e76757e15fb9d
3
+ size 91694043
qantara-pusht-s22.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9d1360a982be5793c6e92b387de43dac1e0b599ac4b3e20eba4a2c7aa7558a88
3
+ size 91694043
qantara-pusht-s33.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:37ab6c1539084e33974887a79dbc29617ef53a5f8ff8e4b5903440c19dcdca9b
3
+ size 91694043
qantara-reacher-s11.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3f1d43e5274c45d8f04fd50c3d1f15448b1c84361687e8fd9a463ac85708a769
3
+ size 91688571
qantara-reacher-s22.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7c54b58a56cfb94a955216ba7f244ef516f1b0594839237423d2a4270ce53741
3
+ size 91688571
qantara-reacher-s33.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:670a4333299105c989cdec412695ade5ada4419deba94274bafd3882b80750ff
3
+ size 91688571
qantara-tworoom-s11.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:530a6aee7a1510dedf20b77d65aed9f9246cccb959516123375f720436a76976
3
+ size 91688571
qantara-tworoom-s22.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4b3d816faa64be1524188323ec5e2f86c3cc25989bdb9f7e3a4b51b140ff5f81
3
+ size 91688571
qantara-tworoom-s33.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:602af94acfdf1fb46b6bca0f632fe679222eb4380837d10dd5b86a1c0dd1dc67
3
+ size 91688571