Add 24 headline checkpoints (Qantara + LeWM, 4 envs x 3 seeds)
Browse files- README.md +55 -0
- lewm-cube-s11.ckpt +3 -0
- lewm-cube-s22.ckpt +3 -0
- lewm-cube-s33.ckpt +3 -0
- lewm-pusht-s11.ckpt +3 -0
- lewm-pusht-s22.ckpt +3 -0
- lewm-pusht-s33.ckpt +3 -0
- lewm-reacher-s11.ckpt +3 -0
- lewm-reacher-s22.ckpt +3 -0
- lewm-reacher-s33.ckpt +3 -0
- lewm-tworoom-s11.ckpt +3 -0
- lewm-tworoom-s22.ckpt +3 -0
- lewm-tworoom-s33.ckpt +3 -0
- qantara-cube-s11.ckpt +3 -0
- qantara-cube-s22.ckpt +3 -0
- qantara-cube-s33.ckpt +3 -0
- qantara-pusht-s11.ckpt +3 -0
- qantara-pusht-s22.ckpt +3 -0
- qantara-pusht-s33.ckpt +3 -0
- qantara-reacher-s11.ckpt +3 -0
- qantara-reacher-s22.ckpt +3 -0
- qantara-reacher-s33.ckpt +3 -0
- qantara-tworoom-s11.ckpt +3 -0
- qantara-tworoom-s22.ckpt +3 -0
- qantara-tworoom-s33.ckpt +3 -0
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
|