Upload 38 files
Browse files- README.md +76 -1
- granular-multiview/README.md +13 -0
- granular-multiview/config.json +51 -0
- granular-multiview/model.pt +3 -0
- granular-relational/README.md +13 -0
- granular-relational/config.json +46 -0
- granular-relational/model.pt +3 -0
- manifest.json +702 -0
- pushobj-unseen-shapes/README.md +13 -0
- pushobj-unseen-shapes/config.json +36 -0
- pushobj-unseen-shapes/model.pt +3 -0
- pusht-adapted-predictor-full/README.md +11 -0
- pusht-adapted-predictor-full/config.json +73 -0
- pusht-adapted-predictor-full/model.pt +3 -0
- pusht-exact-realization-full/README.md +11 -0
- pusht-exact-realization-full/config.json +205 -0
- pusht-exact-realization-full/model.pt +3 -0
- pusht-multi-geometry/README.md +13 -0
- pusht-multi-geometry/config.json +76 -0
- pusht-multi-geometry/model.pt +3 -0
- pusht-predictive-adaptation/README.md +11 -0
- pusht-predictive-adaptation/config.json +75 -0
- pusht-predictive-adaptation/model.pt +3 -0
- pusht-relational/README.md +13 -0
- pusht-relational/config.json +26 -0
- pusht-relational/model.pt +3 -0
- pusht-visual-shifts/README.md +13 -0
- pusht-visual-shifts/config.json +36 -0
- pusht-visual-shifts/model.pt +3 -0
- reacher-relational/README.md +13 -0
- reacher-relational/config.json +23 -0
- reacher-relational/model.pt +3 -0
- reacher-temporal-transport/README.md +11 -0
- reacher-temporal-transport/config.json +24 -0
- reacher-temporal-transport/model.pt +3 -0
- reacher-world-model-full/README.md +11 -0
- reacher-world-model-full/config.json +27 -0
- reacher-world-model-full/model.pt +3 -0
README.md
CHANGED
|
@@ -1,3 +1,78 @@
|
|
| 1 |
---
|
| 2 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
+
language:
|
| 3 |
+
- en
|
| 4 |
+
tags:
|
| 5 |
+
- world-model
|
| 6 |
+
- jepa
|
| 7 |
+
- robotics
|
| 8 |
+
- decision-alignment
|
| 9 |
+
library_name: pytorch
|
| 10 |
+
datasets:
|
| 11 |
+
- Shuaijun/D-JEPA-Dataset
|
| 12 |
---
|
| 13 |
+
|
| 14 |
+
# D-JEPA checkpoints
|
| 15 |
+
|
| 16 |
+
**[Project website](https://nebulis-lab.com/D-JEPA)** ·
|
| 17 |
+
**[Model repository](https://huggingface.co/Shuaijun/D-JEPA)** ·
|
| 18 |
+
**[Decision-supervision dataset](https://huggingface.co/datasets/Shuaijun/D-JEPA-Dataset)** ·
|
| 19 |
+
**[Code](https://github.com/NEBULIS-Lab/D-JEPA)**
|
| 20 |
+
|
| 21 |
+
Task/module profiles for **D-JEPA: A Decision-Aligned Latent World Model**.
|
| 22 |
+
See each profile's `README.md` and `config.json` for architecture, provenance,
|
| 23 |
+
upstream dependencies, calibrated decisions and tensor-file checksums.
|
| 24 |
+
|
| 25 |
+
## Choose a checkpoint profile
|
| 26 |
+
|
| 27 |
+
| Task | Profile | Contents |
|
| 28 |
+
|---|---|---|
|
| 29 |
+
| PushT | [Relational alignment](./pusht-relational/README.md) | Learned decision-alignment module |
|
| 30 |
+
| PushT | [Predictive adaptation](./pusht-predictive-adaptation/README.md) | Updated parameters; requires the exact base model |
|
| 31 |
+
| PushT | [Adapted predictor](./pusht-adapted-predictor-full/README.md) | Full predictor weights |
|
| 32 |
+
| PushT | [Multi-geometry alignment](./pusht-multi-geometry/README.md) | Module using four predictive geometries |
|
| 33 |
+
| PushT | [Exact representation realization](./pusht-exact-realization-full/README.md) | Two predictors and relation module in one tensor file |
|
| 34 |
+
| Reacher | [Relational alignment](./reacher-relational/README.md) | Task-specific decision-alignment module |
|
| 35 |
+
| Reacher | [Temporal transport](./reacher-temporal-transport/README.md) | Bounded future-update module |
|
| 36 |
+
| Reacher | [Full model](./reacher-world-model-full/README.md) | Predictors, alignment and transport weights |
|
| 37 |
+
| Granular | [Relational alignment](./granular-relational/README.md) | Spatial-feature alignment module |
|
| 38 |
+
| Granular | [Multi-view alignment](./granular-multiview/README.md) | Four-view variant using the same backbone |
|
| 39 |
+
| PushObj | [Unseen-shape transfer](./pushobj-unseen-shapes/README.md) | Task-local ordinal alignment module |
|
| 40 |
+
| PushT | [Visual-shift alignment](./pusht-visual-shifts/README.md) | Corruption-trained ordinal alignment module |
|
| 41 |
+
|
| 42 |
+
These are configurations of **D-JEPA**, not 12 separate methods. Each profile
|
| 43 |
+
contains `model.pt`, `config.json` and its own card. The [manifest](./manifest.json)
|
| 44 |
+
indexes source and exported weight hashes.
|
| 45 |
+
|
| 46 |
+
## Loading and dependencies
|
| 47 |
+
|
| 48 |
+
All `model.pt` files contain tensor dictionaries and support
|
| 49 |
+
`torch.load(path, map_location="cpu", weights_only=True)`. Exact tensor values
|
| 50 |
+
from selected formal checkpoints are preserved. Packaging removes optimizer
|
| 51 |
+
state and private path metadata; exported file hashes are therefore distinct
|
| 52 |
+
from original training-file hashes.
|
| 53 |
+
|
| 54 |
+
```python
|
| 55 |
+
import torch
|
| 56 |
+
|
| 57 |
+
state = torch.load("pusht-relational/model.pt", map_location="cpu", weights_only=True)
|
| 58 |
+
```
|
| 59 |
+
|
| 60 |
+
This reads a tensor state dictionary, not an instantiated model. Use the matching
|
| 61 |
+
D-JEPA architecture and configuration; these custom modules are not loaded through
|
| 62 |
+
Transformers `AutoModel.from_pretrained`.
|
| 63 |
+
|
| 64 |
+
Profiles with `upstream_included: true` contain pretrained predictors. Other
|
| 65 |
+
profiles contain our learned modules or updated parameter subsets and require
|
| 66 |
+
their recorded predictive models/features. `pusht-exact-realization-full` is
|
| 67 |
+
one file containing two predictors and a relational operator, not a distilled
|
| 68 |
+
single-backbone student.
|
| 69 |
+
|
| 70 |
+
Reacher's physical result uses the relational selector; temporal transport is
|
| 71 |
+
also provided but its native-distance diagnostic is not a substitute result.
|
| 72 |
+
Multi-geometry's 128-start development result remains a separate protocol from
|
| 73 |
+
the independent 256-start PushT evaluation.
|
| 74 |
+
|
| 75 |
+
Local release candidate: authors have not yet selected publication licenses or
|
| 76 |
+
completed all upstream weight redistribution checks. This card grants no new
|
| 77 |
+
rights to third-party material. The author-designated repository is
|
| 78 |
+
`Shuaijun/D-JEPA`.
|
granular-multiview/README.md
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# D-JEPA: granular-multiview
|
| 2 |
+
|
| 3 |
+
- Architecture: `granular`.
|
| 4 |
+
- Tensor count: 30.
|
| 5 |
+
- Includes pretrained predictor weights: false.
|
| 6 |
+
- Original checkpoint SHA256: `e796fa3ee2fbc6f5b4abdfae99e8fa843606f923284a7fb95f59c0fc6c512c70`.
|
| 7 |
+
- Exported tensor-file SHA256: `2ee5783def21816054645f87d9cffa50900733b7218a2b9ee203ae39aec682ff`.
|
| 8 |
+
|
| 9 |
+
Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
|
| 10 |
+
|
| 11 |
+
Input width: 4. Gate: `base_minus_refined`, strict threshold `-0.010684727949480862`; quantization decimals: `None`. Candidate IDs and feature preprocessing must match the task profile.
|
| 12 |
+
|
| 13 |
+
Project license is selected by the authors before public upload.
|
granular-multiview/config.json
ADDED
|
@@ -0,0 +1,51 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"project": "D-JEPA",
|
| 3 |
+
"profile": "granular-multiview",
|
| 4 |
+
"source_sha256": "e796fa3ee2fbc6f5b4abdfae99e8fa843606f923284a7fb95f59c0fc6c512c70",
|
| 5 |
+
"weights_sha256": "2ee5783def21816054645f87d9cffa50900733b7218a2b9ee203ae39aec682ff",
|
| 6 |
+
"tensor_count": 30,
|
| 7 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 8 |
+
"architecture": "granular",
|
| 9 |
+
"model": {
|
| 10 |
+
"input_dim": 4,
|
| 11 |
+
"hidden_dim": 64,
|
| 12 |
+
"low_rank": 8,
|
| 13 |
+
"max_correction": 0.2
|
| 14 |
+
},
|
| 15 |
+
"input_dim": 4,
|
| 16 |
+
"gate_threshold": -0.010684727949480862,
|
| 17 |
+
"gate_mode": "base_minus_refined",
|
| 18 |
+
"gate_decimals": null,
|
| 19 |
+
"fusion_weights": [
|
| 20 |
+
0.0,
|
| 21 |
+
0.35,
|
| 22 |
+
0.6,
|
| 23 |
+
0.05
|
| 24 |
+
],
|
| 25 |
+
"upstream_included": false,
|
| 26 |
+
"lineage": {
|
| 27 |
+
"schema": "dajepa_granular_train_lineage_v1",
|
| 28 |
+
"method": "da_jepa_mg",
|
| 29 |
+
"dino_checkpoint_sha256": "2b1cb1db727e866c9f98136a72dcd1e55ad076379bda64b4b88273e84a23ac60",
|
| 30 |
+
"config_sha256": "4018434a9031d3367c4067b3791f8afab7a31debaf7f264945146138f181a94e",
|
| 31 |
+
"adapter_config_sha256": "0f2c0a07bf170e87776bb5c5a3485a442ecfefa61eb0b06d3e143de218b000db",
|
| 32 |
+
"train_feature_cache_sha256": "45ec06b6a5331e26f454c3e33247308fe8da950b825e8ee1d349d7d2914ad40e",
|
| 33 |
+
"calibration_feature_cache_sha256": "be8936712c9bd828921f6f714b626928dab23760d2dcf35fb7419609d294ce65",
|
| 34 |
+
"train_release_sha256": "ac21f702f23b7a01f88aeb063785d24486face6ffd59b20b07de30cea81a4aa0",
|
| 35 |
+
"calibration_release_sha256": "67cb52466c3cfd3b1c35dcc89993960a925ab114d33c4c352b179763907828f7",
|
| 36 |
+
"trainer_sha256": "9686c9d555b41ad3966811452c514a3b26af88402b1b2d0ffc45be3d8528333d",
|
| 37 |
+
"core_sha256": "d267a9cdd5e2d8947f7bcb0d7bf49a0180d643a3a345e6bad3bc41b40df7ec81"
|
| 38 |
+
},
|
| 39 |
+
"training": {
|
| 40 |
+
"seed": 20260831,
|
| 41 |
+
"updates": 1500,
|
| 42 |
+
"batch_starts": 16,
|
| 43 |
+
"learning_rate": 0.0003,
|
| 44 |
+
"weight_decay": 0.0001,
|
| 45 |
+
"max_grad_norm": 1.0,
|
| 46 |
+
"tail_k": 16,
|
| 47 |
+
"evaluate_every": 25,
|
| 48 |
+
"mg_grid_step": 0.05,
|
| 49 |
+
"supervision_count": 16
|
| 50 |
+
}
|
| 51 |
+
}
|
granular-multiview/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2ee5783def21816054645f87d9cffa50900733b7218a2b9ee203ae39aec682ff
|
| 3 |
+
size 279105
|
granular-relational/README.md
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# D-JEPA: granular-relational
|
| 2 |
+
|
| 3 |
+
- Architecture: `granular`.
|
| 4 |
+
- Tensor count: 30.
|
| 5 |
+
- Includes pretrained predictor weights: false.
|
| 6 |
+
- Original checkpoint SHA256: `8f613280f8084e8c6403e42db138f73e01b4c43b9235e62bfd36734727836895`.
|
| 7 |
+
- Exported tensor-file SHA256: `a7f2b70c1d1d853cf5dbd391f3560828f28015a72092938a978dff1a0e497504`.
|
| 8 |
+
|
| 9 |
+
Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
|
| 10 |
+
|
| 11 |
+
Input width: 2305. Gate: `base_minus_refined`, strict threshold `-0.06998290770476866`; quantization decimals: `None`. Candidate IDs and feature preprocessing must match the task profile.
|
| 12 |
+
|
| 13 |
+
Project license is selected by the authors before public upload.
|
granular-relational/config.json
ADDED
|
@@ -0,0 +1,46 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"project": "D-JEPA",
|
| 3 |
+
"profile": "granular-relational",
|
| 4 |
+
"source_sha256": "8f613280f8084e8c6403e42db138f73e01b4c43b9235e62bfd36734727836895",
|
| 5 |
+
"weights_sha256": "a7f2b70c1d1d853cf5dbd391f3560828f28015a72092938a978dff1a0e497504",
|
| 6 |
+
"tensor_count": 30,
|
| 7 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 8 |
+
"architecture": "granular",
|
| 9 |
+
"model": {
|
| 10 |
+
"input_dim": 2305,
|
| 11 |
+
"hidden_dim": 64,
|
| 12 |
+
"low_rank": 8,
|
| 13 |
+
"max_correction": 0.2
|
| 14 |
+
},
|
| 15 |
+
"input_dim": 2305,
|
| 16 |
+
"gate_threshold": -0.06998290770476866,
|
| 17 |
+
"gate_mode": "base_minus_refined",
|
| 18 |
+
"gate_decimals": null,
|
| 19 |
+
"fusion_weights": null,
|
| 20 |
+
"upstream_included": false,
|
| 21 |
+
"lineage": {
|
| 22 |
+
"schema": "dajepa_granular_train_lineage_v1",
|
| 23 |
+
"method": "da_jepa_r",
|
| 24 |
+
"dino_checkpoint_sha256": "2b1cb1db727e866c9f98136a72dcd1e55ad076379bda64b4b88273e84a23ac60",
|
| 25 |
+
"config_sha256": "4018434a9031d3367c4067b3791f8afab7a31debaf7f264945146138f181a94e",
|
| 26 |
+
"adapter_config_sha256": "0f2c0a07bf170e87776bb5c5a3485a442ecfefa61eb0b06d3e143de218b000db",
|
| 27 |
+
"train_feature_cache_sha256": "45ec06b6a5331e26f454c3e33247308fe8da950b825e8ee1d349d7d2914ad40e",
|
| 28 |
+
"calibration_feature_cache_sha256": "be8936712c9bd828921f6f714b626928dab23760d2dcf35fb7419609d294ce65",
|
| 29 |
+
"train_release_sha256": "ac21f702f23b7a01f88aeb063785d24486face6ffd59b20b07de30cea81a4aa0",
|
| 30 |
+
"calibration_release_sha256": "67cb52466c3cfd3b1c35dcc89993960a925ab114d33c4c352b179763907828f7",
|
| 31 |
+
"trainer_sha256": "9686c9d555b41ad3966811452c514a3b26af88402b1b2d0ffc45be3d8528333d",
|
| 32 |
+
"core_sha256": "d267a9cdd5e2d8947f7bcb0d7bf49a0180d643a3a345e6bad3bc41b40df7ec81"
|
| 33 |
+
},
|
| 34 |
+
"training": {
|
| 35 |
+
"seed": 20260831,
|
| 36 |
+
"updates": 1500,
|
| 37 |
+
"batch_starts": 16,
|
| 38 |
+
"learning_rate": 0.0003,
|
| 39 |
+
"weight_decay": 0.0001,
|
| 40 |
+
"max_grad_norm": 1.0,
|
| 41 |
+
"tail_k": 16,
|
| 42 |
+
"evaluate_every": 25,
|
| 43 |
+
"mg_grid_step": 0.05,
|
| 44 |
+
"supervision_count": 16
|
| 45 |
+
}
|
| 46 |
+
}
|
granular-relational/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a7f2b70c1d1d853cf5dbd391f3560828f28015a72092938a978dff1a0e497504
|
| 3 |
+
size 886593
|
manifest.json
ADDED
|
@@ -0,0 +1,702 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"profiles": [
|
| 3 |
+
{
|
| 4 |
+
"project": "D-JEPA",
|
| 5 |
+
"profile": "granular-multiview",
|
| 6 |
+
"source_sha256": "e796fa3ee2fbc6f5b4abdfae99e8fa843606f923284a7fb95f59c0fc6c512c70",
|
| 7 |
+
"weights_sha256": "2ee5783def21816054645f87d9cffa50900733b7218a2b9ee203ae39aec682ff",
|
| 8 |
+
"tensor_count": 30,
|
| 9 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 10 |
+
"architecture": "granular",
|
| 11 |
+
"model": {
|
| 12 |
+
"input_dim": 4,
|
| 13 |
+
"hidden_dim": 64,
|
| 14 |
+
"low_rank": 8,
|
| 15 |
+
"max_correction": 0.2
|
| 16 |
+
},
|
| 17 |
+
"input_dim": 4,
|
| 18 |
+
"gate_threshold": -0.010684727949480862,
|
| 19 |
+
"gate_mode": "base_minus_refined",
|
| 20 |
+
"gate_decimals": null,
|
| 21 |
+
"fusion_weights": [
|
| 22 |
+
0.0,
|
| 23 |
+
0.35,
|
| 24 |
+
0.6,
|
| 25 |
+
0.05
|
| 26 |
+
],
|
| 27 |
+
"upstream_included": false,
|
| 28 |
+
"lineage": {
|
| 29 |
+
"schema": "dajepa_granular_train_lineage_v1",
|
| 30 |
+
"method": "da_jepa_mg",
|
| 31 |
+
"dino_checkpoint_sha256": "2b1cb1db727e866c9f98136a72dcd1e55ad076379bda64b4b88273e84a23ac60",
|
| 32 |
+
"config_sha256": "4018434a9031d3367c4067b3791f8afab7a31debaf7f264945146138f181a94e",
|
| 33 |
+
"adapter_config_sha256": "0f2c0a07bf170e87776bb5c5a3485a442ecfefa61eb0b06d3e143de218b000db",
|
| 34 |
+
"train_feature_cache_sha256": "45ec06b6a5331e26f454c3e33247308fe8da950b825e8ee1d349d7d2914ad40e",
|
| 35 |
+
"calibration_feature_cache_sha256": "be8936712c9bd828921f6f714b626928dab23760d2dcf35fb7419609d294ce65",
|
| 36 |
+
"train_release_sha256": "ac21f702f23b7a01f88aeb063785d24486face6ffd59b20b07de30cea81a4aa0",
|
| 37 |
+
"calibration_release_sha256": "67cb52466c3cfd3b1c35dcc89993960a925ab114d33c4c352b179763907828f7",
|
| 38 |
+
"trainer_sha256": "9686c9d555b41ad3966811452c514a3b26af88402b1b2d0ffc45be3d8528333d",
|
| 39 |
+
"core_sha256": "d267a9cdd5e2d8947f7bcb0d7bf49a0180d643a3a345e6bad3bc41b40df7ec81"
|
| 40 |
+
},
|
| 41 |
+
"training": {
|
| 42 |
+
"seed": 20260831,
|
| 43 |
+
"updates": 1500,
|
| 44 |
+
"batch_starts": 16,
|
| 45 |
+
"learning_rate": 0.0003,
|
| 46 |
+
"weight_decay": 0.0001,
|
| 47 |
+
"max_grad_norm": 1.0,
|
| 48 |
+
"tail_k": 16,
|
| 49 |
+
"evaluate_every": 25,
|
| 50 |
+
"mg_grid_step": 0.05,
|
| 51 |
+
"supervision_count": 16
|
| 52 |
+
}
|
| 53 |
+
},
|
| 54 |
+
{
|
| 55 |
+
"project": "D-JEPA",
|
| 56 |
+
"profile": "granular-relational",
|
| 57 |
+
"source_sha256": "8f613280f8084e8c6403e42db138f73e01b4c43b9235e62bfd36734727836895",
|
| 58 |
+
"weights_sha256": "a7f2b70c1d1d853cf5dbd391f3560828f28015a72092938a978dff1a0e497504",
|
| 59 |
+
"tensor_count": 30,
|
| 60 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 61 |
+
"architecture": "granular",
|
| 62 |
+
"model": {
|
| 63 |
+
"input_dim": 2305,
|
| 64 |
+
"hidden_dim": 64,
|
| 65 |
+
"low_rank": 8,
|
| 66 |
+
"max_correction": 0.2
|
| 67 |
+
},
|
| 68 |
+
"input_dim": 2305,
|
| 69 |
+
"gate_threshold": -0.06998290770476866,
|
| 70 |
+
"gate_mode": "base_minus_refined",
|
| 71 |
+
"gate_decimals": null,
|
| 72 |
+
"fusion_weights": null,
|
| 73 |
+
"upstream_included": false,
|
| 74 |
+
"lineage": {
|
| 75 |
+
"schema": "dajepa_granular_train_lineage_v1",
|
| 76 |
+
"method": "da_jepa_r",
|
| 77 |
+
"dino_checkpoint_sha256": "2b1cb1db727e866c9f98136a72dcd1e55ad076379bda64b4b88273e84a23ac60",
|
| 78 |
+
"config_sha256": "4018434a9031d3367c4067b3791f8afab7a31debaf7f264945146138f181a94e",
|
| 79 |
+
"adapter_config_sha256": "0f2c0a07bf170e87776bb5c5a3485a442ecfefa61eb0b06d3e143de218b000db",
|
| 80 |
+
"train_feature_cache_sha256": "45ec06b6a5331e26f454c3e33247308fe8da950b825e8ee1d349d7d2914ad40e",
|
| 81 |
+
"calibration_feature_cache_sha256": "be8936712c9bd828921f6f714b626928dab23760d2dcf35fb7419609d294ce65",
|
| 82 |
+
"train_release_sha256": "ac21f702f23b7a01f88aeb063785d24486face6ffd59b20b07de30cea81a4aa0",
|
| 83 |
+
"calibration_release_sha256": "67cb52466c3cfd3b1c35dcc89993960a925ab114d33c4c352b179763907828f7",
|
| 84 |
+
"trainer_sha256": "9686c9d555b41ad3966811452c514a3b26af88402b1b2d0ffc45be3d8528333d",
|
| 85 |
+
"core_sha256": "d267a9cdd5e2d8947f7bcb0d7bf49a0180d643a3a345e6bad3bc41b40df7ec81"
|
| 86 |
+
},
|
| 87 |
+
"training": {
|
| 88 |
+
"seed": 20260831,
|
| 89 |
+
"updates": 1500,
|
| 90 |
+
"batch_starts": 16,
|
| 91 |
+
"learning_rate": 0.0003,
|
| 92 |
+
"weight_decay": 0.0001,
|
| 93 |
+
"max_grad_norm": 1.0,
|
| 94 |
+
"tail_k": 16,
|
| 95 |
+
"evaluate_every": 25,
|
| 96 |
+
"mg_grid_step": 0.05,
|
| 97 |
+
"supervision_count": 16
|
| 98 |
+
}
|
| 99 |
+
},
|
| 100 |
+
{
|
| 101 |
+
"project": "D-JEPA",
|
| 102 |
+
"profile": "pushobj-unseen-shapes",
|
| 103 |
+
"source_sha256": "941dbd680a7dd179a798e208785bb2a68e3011753eb6d938f5f44a1f7c5ff404",
|
| 104 |
+
"weights_sha256": "8797c805e77fa56017f2ba9d1699f5ffd2bdc415cb2882e01322290ff6e32024",
|
| 105 |
+
"tensor_count": 30,
|
| 106 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 107 |
+
"architecture": "set_aligner",
|
| 108 |
+
"input_dim": 3,
|
| 109 |
+
"max_correction": 0.2,
|
| 110 |
+
"fusion_alpha": 0.3,
|
| 111 |
+
"gate_threshold": 0.0,
|
| 112 |
+
"gate_mode": "refined_gap",
|
| 113 |
+
"gate_decimals": null,
|
| 114 |
+
"upstream_included": false,
|
| 115 |
+
"training": {
|
| 116 |
+
"schema": "djepa-pushobj-relational-v1",
|
| 117 |
+
"seed": 20260906,
|
| 118 |
+
"epochs": 120,
|
| 119 |
+
"learning_rate": 0.0003,
|
| 120 |
+
"batch_size": 16,
|
| 121 |
+
"input_dim": 3,
|
| 122 |
+
"max_correction": 0.2,
|
| 123 |
+
"source_model": "frozen_pushobj_release",
|
| 124 |
+
"feature_mode": "ordinal",
|
| 125 |
+
"fusion_weight_selected_on_calibration": false,
|
| 126 |
+
"base_mode": "rank_fusion",
|
| 127 |
+
"fusion_alpha": 0.3,
|
| 128 |
+
"train_completion_sha256": "855bd0c8afe39add66b23fa5b6b5b401025ca1a6ef0d314dce253b1d5694e8c3",
|
| 129 |
+
"calibration_completion_sha256": "8ca0e50c69c47f0aed75e4c07b22f1ee00f0597621fc167b9da8110801b8c4e1",
|
| 130 |
+
"torch": "2.3.0+cu121",
|
| 131 |
+
"train_code_sha256": "b890ce79e62d3787f405f14971dee1acdd1edc2e9f58b33ae9146b1ccc5e3247",
|
| 132 |
+
"model_code_sha256": "4db0e1740c89fdcbaea98cd07abd1f5d0fbdee85abb42d3791524dc6a6a8bc86"
|
| 133 |
+
},
|
| 134 |
+
"selected_epoch": 40
|
| 135 |
+
},
|
| 136 |
+
{
|
| 137 |
+
"project": "D-JEPA",
|
| 138 |
+
"profile": "pusht-adapted-predictor-full",
|
| 139 |
+
"source_sha256": "ccbd6ca30c550b60a186c5d0a1fa5f1a43b7a08b704b19037586ff3de52b0887",
|
| 140 |
+
"weights_sha256": "26cb528b6ba5b3b6e7d464eeb9a75f599ac28641617666e0638c9e51de2b9083",
|
| 141 |
+
"tensor_count": 309,
|
| 142 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 143 |
+
"architecture": "pretrained_predictor",
|
| 144 |
+
"upstream_included": true,
|
| 145 |
+
"lineage": {
|
| 146 |
+
"method": "dtail_jepa_p_v2",
|
| 147 |
+
"pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
|
| 148 |
+
"tdjepa_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
|
| 149 |
+
"refiner_sha256": "84c2b0fb93afdad9279b220ecf696642528899bb233a21ab1489c31ec7359d9f",
|
| 150 |
+
"refiner_lineage": {
|
| 151 |
+
"method": "dtail_jepa_r_v2",
|
| 152 |
+
"pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
|
| 153 |
+
"lewm_weights_sha256": "5487ac45812ded38e11472eb7084d28c81c18e85154293125dd51c278fcd72c9",
|
| 154 |
+
"tdjepa_weights_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
|
| 155 |
+
"fusion_alpha": 0.42,
|
| 156 |
+
"fit_start_count": 448,
|
| 157 |
+
"calibration_start_count": 64,
|
| 158 |
+
"best_update": 550
|
| 159 |
+
},
|
| 160 |
+
"trainable_names": [
|
| 161 |
+
"predictor.transformer.layers.5.attn.norm.weight",
|
| 162 |
+
"predictor.transformer.layers.5.attn.norm.bias",
|
| 163 |
+
"predictor.transformer.layers.5.attn.to_qkv.weight",
|
| 164 |
+
"predictor.transformer.layers.5.attn.to_out.0.weight",
|
| 165 |
+
"predictor.transformer.layers.5.attn.to_out.0.bias",
|
| 166 |
+
"predictor.transformer.layers.5.mlp.net.0.weight",
|
| 167 |
+
"predictor.transformer.layers.5.mlp.net.0.bias",
|
| 168 |
+
"predictor.transformer.layers.5.mlp.net.1.weight",
|
| 169 |
+
"predictor.transformer.layers.5.mlp.net.1.bias",
|
| 170 |
+
"predictor.transformer.layers.5.mlp.net.4.weight",
|
| 171 |
+
"predictor.transformer.layers.5.mlp.net.4.bias",
|
| 172 |
+
"predictor.transformer.layers.5.adaLN_modulation.1.weight",
|
| 173 |
+
"predictor.transformer.layers.5.adaLN_modulation.1.bias",
|
| 174 |
+
"pred_proj.net.0.weight",
|
| 175 |
+
"pred_proj.net.0.bias",
|
| 176 |
+
"pred_proj.net.1.weight",
|
| 177 |
+
"pred_proj.net.1.bias",
|
| 178 |
+
"pred_proj.net.3.weight",
|
| 179 |
+
"pred_proj.net.3.bias"
|
| 180 |
+
],
|
| 181 |
+
"best_update": 200,
|
| 182 |
+
"fit_start_count": 384,
|
| 183 |
+
"calibration_start_count": 128,
|
| 184 |
+
"fusion_alpha": 0.42,
|
| 185 |
+
"config_digest": "c65cc25c5968eb762c77ec6069170551d20119f2cfa71ddd4beab795be167007",
|
| 186 |
+
"code_sha256": {
|
| 187 |
+
"dcv/dtail_v2_plastic.py": "09f0140772ea98962fcc78a2383d2d8d309b3e55bcea71f3c1ff28af15b460a9",
|
| 188 |
+
"dcv/dtail_v2_plastic_data.py": "514feac427ce8318f5199eed209449ec87cf730b60bbf02e20e09edac9c1b115",
|
| 189 |
+
"dcv/train_dtail_v2_plastic.py": "1e3da1d9b2608dd1509efdcb5a4e248a8ea4498f921c9052909474088183c1c5",
|
| 190 |
+
"dcv/run_dtail_v2_plastic_experiment.py": "a90d5547d9cda4db9ce4fd604b4e6bbcd6bfdaae0c9119b364adfc7b42165aee",
|
| 191 |
+
"dcv/dtail_v2_refine.py": "c1339ea569efd1547a024c06f0782a51b5fd3094dd7ca19dd1c4c2fd13b538cd",
|
| 192 |
+
"dcv/dtail_v2_fusion.py": "cb8b908d75d5bf09fdfbe935759571e387cd2dc76ec4a4d9b6bb9d0c261671c7",
|
| 193 |
+
"dcv/dtail_v2_data.py": "73f1c3b1a0dcb6f9c6357ab0b0a5dc2820b8a8651234c0be0dcf2593cdf4891d",
|
| 194 |
+
"dcv/run_dtail_v2.py": "a32c14b47801e6f5b7a1b11eca3b8fb57eccb5fc33f8a5f49ccd5ca2c83a15f8",
|
| 195 |
+
"dcv/hard_suite_reader.py": "8f12d1eaa6aedf439a038e98dad48e2f9776d1abc4b8dced89dd10033cf50f4e",
|
| 196 |
+
"dcv/hard_suite_runtime_inputs.py": "80cef66da9b554a55191f6474ef062b9b8e0202eefcd73ba0038deed8c598578",
|
| 197 |
+
"closure_jepa/rollout.py": "b671a527c18f25c432b3cf47fc2bdd0aba44e0b5b878ec19fd3f2814bb908538",
|
| 198 |
+
"closure_jepa/tdjepa_pool_adapter.py": "157ba3b9500e4818607b1225648bd2f08ee367dbc995b7f5a69bceb07433d8a8",
|
| 199 |
+
"closure_jepa/tdjepa_runtime.py": "eff63651f17a867ae091da53bdb93c597fcab20c099cfbaaf5d22b46431465aa"
|
| 200 |
+
},
|
| 201 |
+
"runtime_action_inputs": {
|
| 202 |
+
"artifact_sha256": "4187ade9296ff98534b335304035ede51c9c2d488f11e3993bff7ae0f3adc143",
|
| 203 |
+
"sidecar_sha256": "a1d78eac4ac03320a16d1c7f790dafb4c8291544b0cee0bab7044e23acfff8fe",
|
| 204 |
+
"validation_report_sha256": "c880526c322936f6da3b35f2c97c3b0d7b287d24553a6ff9f01d5b1a6061ae43",
|
| 205 |
+
"completion_marker_path": "source-file:pusht_dtail_runtime_actions_v1_1.npz.complete.json"
|
| 206 |
+
}
|
| 207 |
+
}
|
| 208 |
+
},
|
| 209 |
+
{
|
| 210 |
+
"project": "D-JEPA",
|
| 211 |
+
"profile": "pusht-exact-realization-full",
|
| 212 |
+
"source_sha256": "edf28ce23a2c9cc511c63f1f8d64f93e9c156436ff1fccfc1cecbc5545d55be2",
|
| 213 |
+
"weights_sha256": "983be0921f3b1d393ce9c6b0e6acfda8272b546ec7009e93ec0a4466a91a1ba2",
|
| 214 |
+
"tensor_count": 642,
|
| 215 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 216 |
+
"architecture": "exact_realization_world_model",
|
| 217 |
+
"upstream_included": true,
|
| 218 |
+
"decision": {
|
| 219 |
+
"fusion_alpha": 0.42,
|
| 220 |
+
"gate_threshold": -0.03162526342176623,
|
| 221 |
+
"ordinal_denominator": 62,
|
| 222 |
+
"gate_comparison": "strict_gt",
|
| 223 |
+
"advantage_quantization": "decimal_half_even_4"
|
| 224 |
+
},
|
| 225 |
+
"lifting": {
|
| 226 |
+
"mode": "td_radial_ordinal_v1",
|
| 227 |
+
"radius_denominator": 64,
|
| 228 |
+
"zero_rms_threshold": 1e-12,
|
| 229 |
+
"preserved_prefix_steps": 4,
|
| 230 |
+
"native_cost_dtype": "float32"
|
| 231 |
+
},
|
| 232 |
+
"preprocessing": {
|
| 233 |
+
"image_mean": [
|
| 234 |
+
0.485,
|
| 235 |
+
0.456,
|
| 236 |
+
0.406
|
| 237 |
+
],
|
| 238 |
+
"image_std": [
|
| 239 |
+
0.229,
|
| 240 |
+
0.224,
|
| 241 |
+
0.225
|
| 242 |
+
],
|
| 243 |
+
"context_frames": 3,
|
| 244 |
+
"action_block": 5,
|
| 245 |
+
"future_blocks": 5,
|
| 246 |
+
"action_mean": [
|
| 247 |
+
-0.007812565192580223,
|
| 248 |
+
0.006860686466097832
|
| 249 |
+
],
|
| 250 |
+
"action_std": [
|
| 251 |
+
0.2084674835205078,
|
| 252 |
+
0.20674866437911987
|
| 253 |
+
]
|
| 254 |
+
},
|
| 255 |
+
"lineage": {
|
| 256 |
+
"source_sha256": {
|
| 257 |
+
"tdjepa_weights": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
|
| 258 |
+
"tdjepa_config": "c7c802b3c96d5233ead1839c10fc3a85c0c484ce10a755af119a9ee5b1575075",
|
| 259 |
+
"lewm_object": "5487ac45812ded38e11472eb7084d28c81c18e85154293125dd51c278fcd72c9",
|
| 260 |
+
"lewm_weights": "48938400ae3464c9680731287f583a9cb516f55a8ec64ea13a91be47fb15b607",
|
| 261 |
+
"lewm_config": "2564086e961e7b5c7c04dffc451091115b389a590645ff19653c64fd0bc16e09",
|
| 262 |
+
"refiner": "84c2b0fb93afdad9279b220ecf696642528899bb233a21ab1489c31ec7359d9f",
|
| 263 |
+
"refiner_config": "b1ebb7780bdb2ccfeb345b7f9ca5a3877becb760636a61d03daf2fb4d9387695",
|
| 264 |
+
"runtime_actions": "4187ade9296ff98534b335304035ede51c9c2d488f11e3993bff7ae0f3adc143",
|
| 265 |
+
"hard_suite": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba"
|
| 266 |
+
},
|
| 267 |
+
"packaging_config_sha256": "6932eedf77a4cd8d1f5cdf6e85961291ed10637bff7375c3800c8927e097e9ce",
|
| 268 |
+
"code_sha256": {
|
| 269 |
+
"dtail_wm/unified_r_model.py": "6efad8811121cbd5c218daad6e319dd967de1d989fb6bc23fbe4bb2f78f3f1c3",
|
| 270 |
+
"dtail_wm/package_unified_r.py": "13b0bd40ffe2851cddf566d87d2aadc833fed223b63974a6a751fe88f2276f58",
|
| 271 |
+
"reviewed_r/dtail_v2_data.py": "73f1c3b1a0dcb6f9c6357ab0b0a5dc2820b8a8651234c0be0dcf2593cdf4891d",
|
| 272 |
+
"reviewed_r/dtail_v2_fusion.py": "cb8b908d75d5bf09fdfbe935759571e387cd2dc76ec4a4d9b6bb9d0c261671c7",
|
| 273 |
+
"reviewed_r/dtail_v2_refine.py": "c1339ea569efd1547a024c06f0782a51b5fd3094dd7ca19dd1c4c2fd13b538cd",
|
| 274 |
+
"tdjepa/jepa.py": "7de87eabd061b9af3186e57198ff951e747168e4b5260bd94a51db9bd18ec62d",
|
| 275 |
+
"tdjepa/module.py": "f7688fea10208c5127093bd1a7525935e12d2a6df0a0ae1f069f7c4dda99252a",
|
| 276 |
+
"lewm/lewm.py": "c93fe07c57fedfa183d0a68a57d4092c2fcd53a109901027a1798d5dbae4bd6c",
|
| 277 |
+
"lewm/module.py": "3a6209baf30c11f77a5f6637282a1821fc62cea827438a97b23933caab51d8c7",
|
| 278 |
+
"stable_pretraining/backbone/utils.py": "269b18ca0a8e0cc9e1c7909bb349b4b094243acdc80a3fb07e847d63d2c3303b"
|
| 279 |
+
},
|
| 280 |
+
"runtime_source_sha256": {
|
| 281 |
+
"snapshot_manifest": "8bc16050969b571d65c190e147ab59e63c1bce2b69b38ff2d179b271bff6b4f2"
|
| 282 |
+
},
|
| 283 |
+
"source_candidate_contract": "63_unique_positive_literal_ids_expert_zero_excluded",
|
| 284 |
+
"packaging_commit": "42676e0a98249ca62ae4646a1cc5826181c36a66",
|
| 285 |
+
"state_counts": {
|
| 286 |
+
"tdjepa": 309,
|
| 287 |
+
"lewm": 303,
|
| 288 |
+
"refiner": 30
|
| 289 |
+
},
|
| 290 |
+
"lewm_state_attestation": {
|
| 291 |
+
"sorted_key_sha256": "ade2562bd5d2be228bb9f8fb2fea26fedef83d2f8a84a4c74f469acfc18fd89d",
|
| 292 |
+
"tensor_content_sha256": "f6c74a65523cb0448d620a42fa76203f22b4e19d68661518aff93eecf2e3cb77"
|
| 293 |
+
}
|
| 294 |
+
},
|
| 295 |
+
"model_architecture": {
|
| 296 |
+
"candidate_count": 63,
|
| 297 |
+
"horizon": 5,
|
| 298 |
+
"latent_dim": 192,
|
| 299 |
+
"tdjepa_config": {
|
| 300 |
+
"_target_": "jepa.JEPA",
|
| 301 |
+
"encoder": {
|
| 302 |
+
"_target_": "stable_pretraining.backbone.utils.vit_hf",
|
| 303 |
+
"size": "tiny",
|
| 304 |
+
"patch_size": 14,
|
| 305 |
+
"image_size": 224,
|
| 306 |
+
"pretrained": false,
|
| 307 |
+
"use_mask_token": false
|
| 308 |
+
},
|
| 309 |
+
"predictor": {
|
| 310 |
+
"_target_": "module.ARPredictor",
|
| 311 |
+
"num_frames": 3,
|
| 312 |
+
"input_dim": 192,
|
| 313 |
+
"hidden_dim": 192,
|
| 314 |
+
"output_dim": 192,
|
| 315 |
+
"depth": 6,
|
| 316 |
+
"heads": 16,
|
| 317 |
+
"mlp_dim": 2048,
|
| 318 |
+
"dim_head": 64,
|
| 319 |
+
"dropout": 0.1,
|
| 320 |
+
"emb_dropout": 0.0
|
| 321 |
+
},
|
| 322 |
+
"action_encoder": {
|
| 323 |
+
"_target_": "module.Embedder",
|
| 324 |
+
"input_dim": 10,
|
| 325 |
+
"emb_dim": 192
|
| 326 |
+
},
|
| 327 |
+
"projector": {
|
| 328 |
+
"_target_": "module.MLP",
|
| 329 |
+
"input_dim": 192,
|
| 330 |
+
"output_dim": 192,
|
| 331 |
+
"hidden_dim": 2048,
|
| 332 |
+
"norm_fn": {
|
| 333 |
+
"_target_": "torch.nn.BatchNorm1d",
|
| 334 |
+
"_partial_": true
|
| 335 |
+
}
|
| 336 |
+
},
|
| 337 |
+
"pred_proj": {
|
| 338 |
+
"_target_": "module.MLP",
|
| 339 |
+
"input_dim": 192,
|
| 340 |
+
"output_dim": 192,
|
| 341 |
+
"hidden_dim": 2048,
|
| 342 |
+
"norm_fn": {
|
| 343 |
+
"_target_": "torch.nn.BatchNorm1d",
|
| 344 |
+
"_partial_": true
|
| 345 |
+
}
|
| 346 |
+
},
|
| 347 |
+
"planning_cost_mode": "td_jepa",
|
| 348 |
+
"dist_head": {
|
| 349 |
+
"_target_": "module.QuasimetricHead",
|
| 350 |
+
"input_dim": 192,
|
| 351 |
+
"hidden_dim": 512,
|
| 352 |
+
"sym_dim": 128,
|
| 353 |
+
"asym_dim": 128
|
| 354 |
+
},
|
| 355 |
+
"planning_cost_mse_blend": 0.1
|
| 356 |
+
},
|
| 357 |
+
"lewm_config": {
|
| 358 |
+
"_target_": "stable_worldmodel.wm.lewm.LeWM",
|
| 359 |
+
"encoder": {
|
| 360 |
+
"_target_": "stable_pretraining.backbone.utils.vit_hf",
|
| 361 |
+
"size": "tiny",
|
| 362 |
+
"patch_size": 14,
|
| 363 |
+
"image_size": 224,
|
| 364 |
+
"pretrained": false,
|
| 365 |
+
"use_mask_token": false
|
| 366 |
+
},
|
| 367 |
+
"predictor": {
|
| 368 |
+
"_target_": "stable_worldmodel.wm.lewm.module.Predictor",
|
| 369 |
+
"num_frames": 3,
|
| 370 |
+
"input_dim": 192,
|
| 371 |
+
"hidden_dim": 192,
|
| 372 |
+
"output_dim": 192,
|
| 373 |
+
"depth": 6,
|
| 374 |
+
"heads": 16,
|
| 375 |
+
"mlp_dim": 2048,
|
| 376 |
+
"dim_head": 64,
|
| 377 |
+
"dropout": 0.1,
|
| 378 |
+
"emb_dropout": 0.0
|
| 379 |
+
},
|
| 380 |
+
"action_encoder": {
|
| 381 |
+
"_target_": "stable_worldmodel.wm.lewm.module.Embedder",
|
| 382 |
+
"input_dim": 10,
|
| 383 |
+
"emb_dim": 192
|
| 384 |
+
},
|
| 385 |
+
"projector": {
|
| 386 |
+
"_target_": "stable_worldmodel.wm.lewm.module.MLP",
|
| 387 |
+
"input_dim": 192,
|
| 388 |
+
"output_dim": 192,
|
| 389 |
+
"hidden_dim": 2048,
|
| 390 |
+
"norm_fn": {
|
| 391 |
+
"_target_": "torch.nn.BatchNorm1d",
|
| 392 |
+
"_partial_": true
|
| 393 |
+
}
|
| 394 |
+
},
|
| 395 |
+
"pred_proj": {
|
| 396 |
+
"_target_": "stable_worldmodel.wm.lewm.module.MLP",
|
| 397 |
+
"input_dim": 192,
|
| 398 |
+
"output_dim": 192,
|
| 399 |
+
"hidden_dim": 2048,
|
| 400 |
+
"norm_fn": {
|
| 401 |
+
"_target_": "torch.nn.BatchNorm1d",
|
| 402 |
+
"_partial_": true
|
| 403 |
+
}
|
| 404 |
+
}
|
| 405 |
+
},
|
| 406 |
+
"refiner_config": {
|
| 407 |
+
"hidden_dim": 64,
|
| 408 |
+
"low_rank": 8,
|
| 409 |
+
"max_correction": 0.2
|
| 410 |
+
}
|
| 411 |
+
},
|
| 412 |
+
"deployment": "One file containing two pretrained predictors and the relational operator; not single-backbone distillation."
|
| 413 |
+
},
|
| 414 |
+
{
|
| 415 |
+
"project": "D-JEPA",
|
| 416 |
+
"profile": "pusht-multi-geometry",
|
| 417 |
+
"source_sha256": "d187c0efc194e6faa39007be835f3cca820b16fd95adfb1d7283f3d44d108b56",
|
| 418 |
+
"weights_sha256": "801c92fc312b071fbc43b1e2153ae1ca25c08b1a3ae95110b38c7482e99c0fe3",
|
| 419 |
+
"tensor_count": 30,
|
| 420 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 421 |
+
"architecture": "set_aligner",
|
| 422 |
+
"input_dim": 388,
|
| 423 |
+
"max_correction": 0.2,
|
| 424 |
+
"gate_threshold": 0.027186155319213864,
|
| 425 |
+
"gate_mode": "base_minus_refined",
|
| 426 |
+
"gate_decimals": null,
|
| 427 |
+
"base_geometry": "jepa_wm",
|
| 428 |
+
"geometry_order": [
|
| 429 |
+
"lewm",
|
| 430 |
+
"temporal_distance",
|
| 431 |
+
"jepa_wm",
|
| 432 |
+
"dino_wm"
|
| 433 |
+
],
|
| 434 |
+
"upstream_included": false,
|
| 435 |
+
"lineage": {
|
| 436 |
+
"schema": "dtail_multi_geometry_lineage_v1",
|
| 437 |
+
"scope": "locked_development",
|
| 438 |
+
"config_sha256": "16a4170801d3b84cfaea19565f1d7fb80663adcb30fa6164eeec3d54da7cba1d",
|
| 439 |
+
"runner_sha256": "7dff9d912ec14c1a8dc540907626b203843ff5cb632a9fcb730d1e89cf0ebfe9",
|
| 440 |
+
"pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
|
| 441 |
+
"registry_sha256": "84b7b1044b958a0bfc85f9838b8ff82568fad5c76ac742ac29a79cedf6b87f65",
|
| 442 |
+
"dependency_sha256": {
|
| 443 |
+
"source-file:multi_geometry.py": "e3a50a62fd36b4552ae19623691b044442af36bbb1c3925d410c027884d76ac2",
|
| 444 |
+
"source-file:physical_join.py": "c742a241edab03e43107f52a0207fa669a66e0fe430ec29736789892ca510c03",
|
| 445 |
+
"source-file:aggregate.py": "95b478b8f5a7827e3c1963176245d8e229fbca3f73798b71becbce02cb6c75ad",
|
| 446 |
+
"source-file:registry.py": "e7ccd4e18f708c31267f5681f56812cc811a489a84d9c2499cd0dca9d3d7d08f",
|
| 447 |
+
"source-file:releases.py": "3ecdb263a4e87bf3e5ba317d7baf15f648dd132d97cf641cc49e728678f632f2",
|
| 448 |
+
"source-file:score_io.py": "126a7b713a8d02c8b0af69922a1073495e45a1713b2ab44f2ca683a15e438beb",
|
| 449 |
+
"source-file:route_experiment.py": "ad558c3b16209ff6b8b709a645ab134e63fc8883397f8124552e80d4f0239090",
|
| 450 |
+
"source-file:run_multi_geometry_routes.py": "7dff9d912ec14c1a8dc540907626b203843ff5cb632a9fcb730d1e89cf0ebfe9",
|
| 451 |
+
"source-file:dtail_v2_data.py": "73f1c3b1a0dcb6f9c6357ab0b0a5dc2820b8a8651234c0be0dcf2593cdf4891d",
|
| 452 |
+
"source-file:dtail_v2_refine.py": "c1339ea569efd1547a024c06f0782a51b5fd3094dd7ca19dd1c4c2fd13b538cd",
|
| 453 |
+
"source-file:run_dtail_v2.py": "a32c14b47801e6f5b7a1b11eca3b8fb57eccb5fc33f8a5f49ccd5ca2c83a15f8"
|
| 454 |
+
},
|
| 455 |
+
"train_release_sha256": "3c6770e13abfaf4e890d0e273cf530e0b54df690f46683f7395c4c9a3b5e74bc",
|
| 456 |
+
"development_release_sha256": "5b0dbee6f1693692037ef65016678adeee7cfa7b08fb5646ff86dbc95a9e0c72",
|
| 457 |
+
"external_score_sha256": {
|
| 458 |
+
"train": {
|
| 459 |
+
"jepa_wm": "0c6d7efa4abb1c822d9511cf0ef36aed0f7e5acb157c0973e344313e8c2681e2",
|
| 460 |
+
"dino_wm": "5eaff47859b272b90c424893e41f281a66902cbc45e2b7162a06bc87d5e462a7"
|
| 461 |
+
},
|
| 462 |
+
"development": {
|
| 463 |
+
"jepa_wm": "7fc522ca6ddde02a4eb673a4e469b5a30e7a3510b5452d3bcdfedf0d3bd51c83",
|
| 464 |
+
"dino_wm": "80c4fb904fa12c305294431047962ae35e8bbf4810056d1085d1d80903491588"
|
| 465 |
+
}
|
| 466 |
+
},
|
| 467 |
+
"geometry_order": [
|
| 468 |
+
"lewm",
|
| 469 |
+
"temporal_distance",
|
| 470 |
+
"jepa_wm",
|
| 471 |
+
"dino_wm"
|
| 472 |
+
],
|
| 473 |
+
"convex_weights": [
|
| 474 |
+
0.0,
|
| 475 |
+
0.0,
|
| 476 |
+
0.25,
|
| 477 |
+
0.75
|
| 478 |
+
],
|
| 479 |
+
"fusion_train_correct": 444,
|
| 480 |
+
"fit_count": 448,
|
| 481 |
+
"method": "dtail_jepa_base_multi_geometry",
|
| 482 |
+
"initial_state_sha256": "947de445ddfe14280fa74a200bf12c56d47113a191942d1e5144a06dbff8a836",
|
| 483 |
+
"best_update": 900,
|
| 484 |
+
"threshold": 0.027186155319213864,
|
| 485 |
+
"calibration_correct": 64,
|
| 486 |
+
"calibration_count": 64
|
| 487 |
+
},
|
| 488 |
+
"evaluation_scope": "development_128"
|
| 489 |
+
},
|
| 490 |
+
{
|
| 491 |
+
"project": "D-JEPA",
|
| 492 |
+
"profile": "pusht-predictive-adaptation",
|
| 493 |
+
"source_sha256": "ccbd6ca30c550b60a186c5d0a1fa5f1a43b7a08b704b19037586ff3de52b0887",
|
| 494 |
+
"weights_sha256": "b2bcb7880fb9477a7bb9aaa3c3380b66e4cd5e1767a4093f990c72766ff8e49a",
|
| 495 |
+
"tensor_count": 22,
|
| 496 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 497 |
+
"architecture": "predictor_patch",
|
| 498 |
+
"upstream_included": false,
|
| 499 |
+
"base_checkpoint_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
|
| 500 |
+
"lineage": {
|
| 501 |
+
"method": "dtail_jepa_p_v2",
|
| 502 |
+
"pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
|
| 503 |
+
"tdjepa_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
|
| 504 |
+
"refiner_sha256": "84c2b0fb93afdad9279b220ecf696642528899bb233a21ab1489c31ec7359d9f",
|
| 505 |
+
"refiner_lineage": {
|
| 506 |
+
"method": "dtail_jepa_r_v2",
|
| 507 |
+
"pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
|
| 508 |
+
"lewm_weights_sha256": "5487ac45812ded38e11472eb7084d28c81c18e85154293125dd51c278fcd72c9",
|
| 509 |
+
"tdjepa_weights_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
|
| 510 |
+
"fusion_alpha": 0.42,
|
| 511 |
+
"fit_start_count": 448,
|
| 512 |
+
"calibration_start_count": 64,
|
| 513 |
+
"best_update": 550
|
| 514 |
+
},
|
| 515 |
+
"trainable_names": [
|
| 516 |
+
"predictor.transformer.layers.5.attn.norm.weight",
|
| 517 |
+
"predictor.transformer.layers.5.attn.norm.bias",
|
| 518 |
+
"predictor.transformer.layers.5.attn.to_qkv.weight",
|
| 519 |
+
"predictor.transformer.layers.5.attn.to_out.0.weight",
|
| 520 |
+
"predictor.transformer.layers.5.attn.to_out.0.bias",
|
| 521 |
+
"predictor.transformer.layers.5.mlp.net.0.weight",
|
| 522 |
+
"predictor.transformer.layers.5.mlp.net.0.bias",
|
| 523 |
+
"predictor.transformer.layers.5.mlp.net.1.weight",
|
| 524 |
+
"predictor.transformer.layers.5.mlp.net.1.bias",
|
| 525 |
+
"predictor.transformer.layers.5.mlp.net.4.weight",
|
| 526 |
+
"predictor.transformer.layers.5.mlp.net.4.bias",
|
| 527 |
+
"predictor.transformer.layers.5.adaLN_modulation.1.weight",
|
| 528 |
+
"predictor.transformer.layers.5.adaLN_modulation.1.bias",
|
| 529 |
+
"pred_proj.net.0.weight",
|
| 530 |
+
"pred_proj.net.0.bias",
|
| 531 |
+
"pred_proj.net.1.weight",
|
| 532 |
+
"pred_proj.net.1.bias",
|
| 533 |
+
"pred_proj.net.3.weight",
|
| 534 |
+
"pred_proj.net.3.bias"
|
| 535 |
+
],
|
| 536 |
+
"best_update": 200,
|
| 537 |
+
"fit_start_count": 384,
|
| 538 |
+
"calibration_start_count": 128,
|
| 539 |
+
"fusion_alpha": 0.42,
|
| 540 |
+
"config_digest": "c65cc25c5968eb762c77ec6069170551d20119f2cfa71ddd4beab795be167007",
|
| 541 |
+
"code_sha256": {
|
| 542 |
+
"dcv/dtail_v2_plastic.py": "09f0140772ea98962fcc78a2383d2d8d309b3e55bcea71f3c1ff28af15b460a9",
|
| 543 |
+
"dcv/dtail_v2_plastic_data.py": "514feac427ce8318f5199eed209449ec87cf730b60bbf02e20e09edac9c1b115",
|
| 544 |
+
"dcv/train_dtail_v2_plastic.py": "1e3da1d9b2608dd1509efdcb5a4e248a8ea4498f921c9052909474088183c1c5",
|
| 545 |
+
"dcv/run_dtail_v2_plastic_experiment.py": "a90d5547d9cda4db9ce4fd604b4e6bbcd6bfdaae0c9119b364adfc7b42165aee",
|
| 546 |
+
"dcv/dtail_v2_refine.py": "c1339ea569efd1547a024c06f0782a51b5fd3094dd7ca19dd1c4c2fd13b538cd",
|
| 547 |
+
"dcv/dtail_v2_fusion.py": "cb8b908d75d5bf09fdfbe935759571e387cd2dc76ec4a4d9b6bb9d0c261671c7",
|
| 548 |
+
"dcv/dtail_v2_data.py": "73f1c3b1a0dcb6f9c6357ab0b0a5dc2820b8a8651234c0be0dcf2593cdf4891d",
|
| 549 |
+
"dcv/run_dtail_v2.py": "a32c14b47801e6f5b7a1b11eca3b8fb57eccb5fc33f8a5f49ccd5ca2c83a15f8",
|
| 550 |
+
"dcv/hard_suite_reader.py": "8f12d1eaa6aedf439a038e98dad48e2f9776d1abc4b8dced89dd10033cf50f4e",
|
| 551 |
+
"dcv/hard_suite_runtime_inputs.py": "80cef66da9b554a55191f6474ef062b9b8e0202eefcd73ba0038deed8c598578",
|
| 552 |
+
"closure_jepa/rollout.py": "b671a527c18f25c432b3cf47fc2bdd0aba44e0b5b878ec19fd3f2814bb908538",
|
| 553 |
+
"closure_jepa/tdjepa_pool_adapter.py": "157ba3b9500e4818607b1225648bd2f08ee367dbc995b7f5a69bceb07433d8a8",
|
| 554 |
+
"closure_jepa/tdjepa_runtime.py": "eff63651f17a867ae091da53bdb93c597fcab20c099cfbaaf5d22b46431465aa"
|
| 555 |
+
},
|
| 556 |
+
"runtime_action_inputs": {
|
| 557 |
+
"artifact_sha256": "4187ade9296ff98534b335304035ede51c9c2d488f11e3993bff7ae0f3adc143",
|
| 558 |
+
"sidecar_sha256": "a1d78eac4ac03320a16d1c7f790dafb4c8291544b0cee0bab7044e23acfff8fe",
|
| 559 |
+
"validation_report_sha256": "c880526c322936f6da3b35f2c97c3b0d7b287d24553a6ff9f01d5b1a6061ae43",
|
| 560 |
+
"completion_marker_path": "source-file:pusht_dtail_runtime_actions_v1_1.npz.complete.json"
|
| 561 |
+
}
|
| 562 |
+
},
|
| 563 |
+
"usage": "Load exact pretrained Temporal-Distance-JEPA then replace these parameters. This is a parameter patch, not a full predictor."
|
| 564 |
+
},
|
| 565 |
+
{
|
| 566 |
+
"project": "D-JEPA",
|
| 567 |
+
"profile": "pusht-relational",
|
| 568 |
+
"source_sha256": "84c2b0fb93afdad9279b220ecf696642528899bb233a21ab1489c31ec7359d9f",
|
| 569 |
+
"weights_sha256": "efea4301f605f681c7f85f934a6ecdcafa9984058f72be3be2d625bfc9e16941",
|
| 570 |
+
"tensor_count": 30,
|
| 571 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 572 |
+
"architecture": "set_aligner",
|
| 573 |
+
"input_dim": 386,
|
| 574 |
+
"max_correction": 0.2,
|
| 575 |
+
"fusion_alpha": 0.42,
|
| 576 |
+
"gate_threshold": -0.03162526342176623,
|
| 577 |
+
"gate_mode": "base_minus_refined",
|
| 578 |
+
"gate_decimals": null,
|
| 579 |
+
"upstream_included": false,
|
| 580 |
+
"lineage": {
|
| 581 |
+
"method": "dtail_jepa_r_v2",
|
| 582 |
+
"pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
|
| 583 |
+
"lewm_weights_sha256": "5487ac45812ded38e11472eb7084d28c81c18e85154293125dd51c278fcd72c9",
|
| 584 |
+
"tdjepa_weights_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
|
| 585 |
+
"fusion_alpha": 0.42,
|
| 586 |
+
"fit_start_count": 448,
|
| 587 |
+
"calibration_start_count": 64,
|
| 588 |
+
"best_update": 550
|
| 589 |
+
}
|
| 590 |
+
},
|
| 591 |
+
{
|
| 592 |
+
"project": "D-JEPA",
|
| 593 |
+
"profile": "pusht-visual-shifts",
|
| 594 |
+
"source_sha256": "c9552d2ab5d477d5c07d71f64ac66a315f3bbcf287f866c40d64e857c67bac22",
|
| 595 |
+
"weights_sha256": "cf36d6bd0af016d8adf2358d78feeda33c1bbf062eef1d0be6332ca1bb28f2e3",
|
| 596 |
+
"tensor_count": 30,
|
| 597 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 598 |
+
"architecture": "set_aligner",
|
| 599 |
+
"input_dim": 3,
|
| 600 |
+
"max_correction": 0.2,
|
| 601 |
+
"fusion_alpha": 0.5,
|
| 602 |
+
"gate_threshold": 0.01,
|
| 603 |
+
"gate_mode": "refined_gap",
|
| 604 |
+
"gate_decimals": null,
|
| 605 |
+
"upstream_included": false,
|
| 606 |
+
"training": {
|
| 607 |
+
"schema": "djepa-pushobj-relational-v1",
|
| 608 |
+
"seed": 20260906,
|
| 609 |
+
"epochs": 120,
|
| 610 |
+
"learning_rate": 0.0003,
|
| 611 |
+
"batch_size": 16,
|
| 612 |
+
"input_dim": 3,
|
| 613 |
+
"max_correction": 0.2,
|
| 614 |
+
"source_model": "frozen_pusht_visual_shift_release",
|
| 615 |
+
"feature_mode": "ordinal",
|
| 616 |
+
"fusion_weight_selected_on_calibration": true,
|
| 617 |
+
"base_mode": "rank_fusion",
|
| 618 |
+
"fusion_alpha": 0.5,
|
| 619 |
+
"train_completion_sha256": "f13d2fa6a3ff152532c1acad7ec209e389737cbbee27a7ee9fce46bc39819bb8",
|
| 620 |
+
"calibration_completion_sha256": "a8e5952e7f3a6e2000d075f9198bffa99ddae1bc94b3b027953c0431d5bb81b6",
|
| 621 |
+
"torch": "2.3.0+cu121",
|
| 622 |
+
"train_code_sha256": "b890ce79e62d3787f405f14971dee1acdd1edc2e9f58b33ae9146b1ccc5e3247",
|
| 623 |
+
"model_code_sha256": "4db0e1740c89fdcbaea98cd07abd1f5d0fbdee85abb42d3791524dc6a6a8bc86"
|
| 624 |
+
},
|
| 625 |
+
"selected_epoch": 20
|
| 626 |
+
},
|
| 627 |
+
{
|
| 628 |
+
"project": "D-JEPA",
|
| 629 |
+
"profile": "reacher-relational",
|
| 630 |
+
"source_sha256": "2bca723eaca2ef1dd7d01182b1f6f0b545aba646076b9e8ce7ba6f9f119bce05",
|
| 631 |
+
"weights_sha256": "5160c04ac103589b33820133b0cb221189931382c3e6ef3bf21d9232bed830d3",
|
| 632 |
+
"tensor_count": 30,
|
| 633 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 634 |
+
"architecture": "set_aligner",
|
| 635 |
+
"input_dim": 386,
|
| 636 |
+
"max_correction": 0.2,
|
| 637 |
+
"fusion_alpha": 0.4,
|
| 638 |
+
"gate_threshold": 0.12511829772303182,
|
| 639 |
+
"gate_mode": "base_minus_refined",
|
| 640 |
+
"gate_decimals": 8,
|
| 641 |
+
"upstream_included": false,
|
| 642 |
+
"bindings": {
|
| 643 |
+
"training_completion_sha256": "9af702ae3c9f3321863fc96c7f7dd1132748f6846cc07e289f8d28657fd7c636",
|
| 644 |
+
"full_pool_completion_sha256": "839dbbd1ace0dde4492b1d4e5a167fab12814f1330db0c00ea9d301639bdd964",
|
| 645 |
+
"capacity_gate_completion_sha256": "d6bab3c79667923f85efdb9e187f80eb20739db200ec83452efbee6edc47f6b2",
|
| 646 |
+
"b2_config_sha256": "737a766a978f16847ba8dbb240ce13f11bcda28b6d23b19f38082218563c6f73",
|
| 647 |
+
"packager_code_sha256": "4d8cbd215a0cf852a90d355be0dbeab661d6aeff17f4cc6eb6f01319446f8298"
|
| 648 |
+
}
|
| 649 |
+
},
|
| 650 |
+
{
|
| 651 |
+
"project": "D-JEPA",
|
| 652 |
+
"profile": "reacher-temporal-transport",
|
| 653 |
+
"source_sha256": "2bca723eaca2ef1dd7d01182b1f6f0b545aba646076b9e8ce7ba6f9f119bce05",
|
| 654 |
+
"weights_sha256": "cf8c57a00e891fc002f9e595e3cf89c8a52a163e42c57059919bc1cfdddd883f",
|
| 655 |
+
"tensor_count": 4,
|
| 656 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 657 |
+
"architecture": "temporal_transport",
|
| 658 |
+
"hidden_dim": 32,
|
| 659 |
+
"radius": 0.1,
|
| 660 |
+
"fusion_alpha": 0.4,
|
| 661 |
+
"gate_threshold": 0.12511829772303182,
|
| 662 |
+
"gate_mode": "base_minus_refined",
|
| 663 |
+
"gate_decimals": 8,
|
| 664 |
+
"upstream_included": false,
|
| 665 |
+
"bindings": {
|
| 666 |
+
"training_completion_sha256": "9af702ae3c9f3321863fc96c7f7dd1132748f6846cc07e289f8d28657fd7c636",
|
| 667 |
+
"full_pool_completion_sha256": "839dbbd1ace0dde4492b1d4e5a167fab12814f1330db0c00ea9d301639bdd964",
|
| 668 |
+
"capacity_gate_completion_sha256": "d6bab3c79667923f85efdb9e187f80eb20739db200ec83452efbee6edc47f6b2",
|
| 669 |
+
"b2_config_sha256": "737a766a978f16847ba8dbb240ce13f11bcda28b6d23b19f38082218563c6f73",
|
| 670 |
+
"packager_code_sha256": "4d8cbd215a0cf852a90d355be0dbeab661d6aeff17f4cc6eb6f01319446f8298"
|
| 671 |
+
},
|
| 672 |
+
"physical_selector": "relational score; transported distance is a distinct diagnostic"
|
| 673 |
+
},
|
| 674 |
+
{
|
| 675 |
+
"project": "D-JEPA",
|
| 676 |
+
"profile": "reacher-world-model-full",
|
| 677 |
+
"source_sha256": "2bca723eaca2ef1dd7d01182b1f6f0b545aba646076b9e8ce7ba6f9f119bce05",
|
| 678 |
+
"weights_sha256": "f7331bf87bf4c8543fa8797072e660266e55d9a9354702120cec1faadd76609a",
|
| 679 |
+
"tensor_count": 646,
|
| 680 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 681 |
+
"architecture": "reacher_world_model",
|
| 682 |
+
"fusion_alpha": 0.4,
|
| 683 |
+
"gate_threshold": 0.12511829772303182,
|
| 684 |
+
"gate_mode": "base_minus_refined",
|
| 685 |
+
"gate_decimals": 8,
|
| 686 |
+
"upstream_included": true,
|
| 687 |
+
"bindings": {
|
| 688 |
+
"training_completion_sha256": "9af702ae3c9f3321863fc96c7f7dd1132748f6846cc07e289f8d28657fd7c636",
|
| 689 |
+
"full_pool_completion_sha256": "839dbbd1ace0dde4492b1d4e5a167fab12814f1330db0c00ea9d301639bdd964",
|
| 690 |
+
"capacity_gate_completion_sha256": "d6bab3c79667923f85efdb9e187f80eb20739db200ec83452efbee6edc47f6b2",
|
| 691 |
+
"b2_config_sha256": "737a766a978f16847ba8dbb240ce13f11bcda28b6d23b19f38082218563c6f73",
|
| 692 |
+
"packager_code_sha256": "4d8cbd215a0cf852a90d355be0dbeab661d6aeff17f4cc6eb6f01319446f8298"
|
| 693 |
+
},
|
| 694 |
+
"state_groups": [
|
| 695 |
+
"tdjepa",
|
| 696 |
+
"lewm",
|
| 697 |
+
"adapters"
|
| 698 |
+
],
|
| 699 |
+
"physical_selector": "relational score"
|
| 700 |
+
}
|
| 701 |
+
]
|
| 702 |
+
}
|
pushobj-unseen-shapes/README.md
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# D-JEPA: pushobj-unseen-shapes
|
| 2 |
+
|
| 3 |
+
- Architecture: `set_aligner`.
|
| 4 |
+
- Tensor count: 30.
|
| 5 |
+
- Includes pretrained predictor weights: false.
|
| 6 |
+
- Original checkpoint SHA256: `941dbd680a7dd179a798e208785bb2a68e3011753eb6d938f5f44a1f7c5ff404`.
|
| 7 |
+
- Exported tensor-file SHA256: `8797c805e77fa56017f2ba9d1699f5ffd2bdc415cb2882e01322290ff6e32024`.
|
| 8 |
+
|
| 9 |
+
Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
|
| 10 |
+
|
| 11 |
+
Input width: 3. Gate: `refined_gap`, strict threshold `0.0`; quantization decimals: `None`. Candidate IDs and feature preprocessing must match the task profile.
|
| 12 |
+
|
| 13 |
+
Project license is selected by the authors before public upload.
|
pushobj-unseen-shapes/config.json
ADDED
|
@@ -0,0 +1,36 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"project": "D-JEPA",
|
| 3 |
+
"profile": "pushobj-unseen-shapes",
|
| 4 |
+
"source_sha256": "941dbd680a7dd179a798e208785bb2a68e3011753eb6d938f5f44a1f7c5ff404",
|
| 5 |
+
"weights_sha256": "8797c805e77fa56017f2ba9d1699f5ffd2bdc415cb2882e01322290ff6e32024",
|
| 6 |
+
"tensor_count": 30,
|
| 7 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 8 |
+
"architecture": "set_aligner",
|
| 9 |
+
"input_dim": 3,
|
| 10 |
+
"max_correction": 0.2,
|
| 11 |
+
"fusion_alpha": 0.3,
|
| 12 |
+
"gate_threshold": 0.0,
|
| 13 |
+
"gate_mode": "refined_gap",
|
| 14 |
+
"gate_decimals": null,
|
| 15 |
+
"upstream_included": false,
|
| 16 |
+
"training": {
|
| 17 |
+
"schema": "djepa-pushobj-relational-v1",
|
| 18 |
+
"seed": 20260906,
|
| 19 |
+
"epochs": 120,
|
| 20 |
+
"learning_rate": 0.0003,
|
| 21 |
+
"batch_size": 16,
|
| 22 |
+
"input_dim": 3,
|
| 23 |
+
"max_correction": 0.2,
|
| 24 |
+
"source_model": "frozen_pushobj_release",
|
| 25 |
+
"feature_mode": "ordinal",
|
| 26 |
+
"fusion_weight_selected_on_calibration": false,
|
| 27 |
+
"base_mode": "rank_fusion",
|
| 28 |
+
"fusion_alpha": 0.3,
|
| 29 |
+
"train_completion_sha256": "855bd0c8afe39add66b23fa5b6b5b401025ca1a6ef0d314dce253b1d5694e8c3",
|
| 30 |
+
"calibration_completion_sha256": "8ca0e50c69c47f0aed75e4c07b22f1ee00f0597621fc167b9da8110801b8c4e1",
|
| 31 |
+
"torch": "2.3.0+cu121",
|
| 32 |
+
"train_code_sha256": "b890ce79e62d3787f405f14971dee1acdd1edc2e9f58b33ae9146b1ccc5e3247",
|
| 33 |
+
"model_code_sha256": "4db0e1740c89fdcbaea98cd07abd1f5d0fbdee85abb42d3791524dc6a6a8bc86"
|
| 34 |
+
},
|
| 35 |
+
"selected_epoch": 40
|
| 36 |
+
}
|
pushobj-unseen-shapes/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8797c805e77fa56017f2ba9d1699f5ffd2bdc415cb2882e01322290ff6e32024
|
| 3 |
+
size 279233
|
pusht-adapted-predictor-full/README.md
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# D-JEPA: pusht-adapted-predictor-full
|
| 2 |
+
|
| 3 |
+
- Architecture: `pretrained_predictor`.
|
| 4 |
+
- Tensor count: 309.
|
| 5 |
+
- Includes pretrained predictor weights: true.
|
| 6 |
+
- Original checkpoint SHA256: `ccbd6ca30c550b60a186c5d0a1fa5f1a43b7a08b704b19037586ff3de52b0887`.
|
| 7 |
+
- Exported tensor-file SHA256: `26cb528b6ba5b3b6e7d464eeb9a75f599ac28641617666e0638c9e51de2b9083`.
|
| 8 |
+
|
| 9 |
+
Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
|
| 10 |
+
|
| 11 |
+
Publication hold: upstream weight redistribution terms must be confirmed before upload.
|
pusht-adapted-predictor-full/config.json
ADDED
|
@@ -0,0 +1,73 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"project": "D-JEPA",
|
| 3 |
+
"profile": "pusht-adapted-predictor-full",
|
| 4 |
+
"source_sha256": "ccbd6ca30c550b60a186c5d0a1fa5f1a43b7a08b704b19037586ff3de52b0887",
|
| 5 |
+
"weights_sha256": "26cb528b6ba5b3b6e7d464eeb9a75f599ac28641617666e0638c9e51de2b9083",
|
| 6 |
+
"tensor_count": 309,
|
| 7 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 8 |
+
"architecture": "pretrained_predictor",
|
| 9 |
+
"upstream_included": true,
|
| 10 |
+
"lineage": {
|
| 11 |
+
"method": "dtail_jepa_p_v2",
|
| 12 |
+
"pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
|
| 13 |
+
"tdjepa_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
|
| 14 |
+
"refiner_sha256": "84c2b0fb93afdad9279b220ecf696642528899bb233a21ab1489c31ec7359d9f",
|
| 15 |
+
"refiner_lineage": {
|
| 16 |
+
"method": "dtail_jepa_r_v2",
|
| 17 |
+
"pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
|
| 18 |
+
"lewm_weights_sha256": "5487ac45812ded38e11472eb7084d28c81c18e85154293125dd51c278fcd72c9",
|
| 19 |
+
"tdjepa_weights_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
|
| 20 |
+
"fusion_alpha": 0.42,
|
| 21 |
+
"fit_start_count": 448,
|
| 22 |
+
"calibration_start_count": 64,
|
| 23 |
+
"best_update": 550
|
| 24 |
+
},
|
| 25 |
+
"trainable_names": [
|
| 26 |
+
"predictor.transformer.layers.5.attn.norm.weight",
|
| 27 |
+
"predictor.transformer.layers.5.attn.norm.bias",
|
| 28 |
+
"predictor.transformer.layers.5.attn.to_qkv.weight",
|
| 29 |
+
"predictor.transformer.layers.5.attn.to_out.0.weight",
|
| 30 |
+
"predictor.transformer.layers.5.attn.to_out.0.bias",
|
| 31 |
+
"predictor.transformer.layers.5.mlp.net.0.weight",
|
| 32 |
+
"predictor.transformer.layers.5.mlp.net.0.bias",
|
| 33 |
+
"predictor.transformer.layers.5.mlp.net.1.weight",
|
| 34 |
+
"predictor.transformer.layers.5.mlp.net.1.bias",
|
| 35 |
+
"predictor.transformer.layers.5.mlp.net.4.weight",
|
| 36 |
+
"predictor.transformer.layers.5.mlp.net.4.bias",
|
| 37 |
+
"predictor.transformer.layers.5.adaLN_modulation.1.weight",
|
| 38 |
+
"predictor.transformer.layers.5.adaLN_modulation.1.bias",
|
| 39 |
+
"pred_proj.net.0.weight",
|
| 40 |
+
"pred_proj.net.0.bias",
|
| 41 |
+
"pred_proj.net.1.weight",
|
| 42 |
+
"pred_proj.net.1.bias",
|
| 43 |
+
"pred_proj.net.3.weight",
|
| 44 |
+
"pred_proj.net.3.bias"
|
| 45 |
+
],
|
| 46 |
+
"best_update": 200,
|
| 47 |
+
"fit_start_count": 384,
|
| 48 |
+
"calibration_start_count": 128,
|
| 49 |
+
"fusion_alpha": 0.42,
|
| 50 |
+
"config_digest": "c65cc25c5968eb762c77ec6069170551d20119f2cfa71ddd4beab795be167007",
|
| 51 |
+
"code_sha256": {
|
| 52 |
+
"dcv/dtail_v2_plastic.py": "09f0140772ea98962fcc78a2383d2d8d309b3e55bcea71f3c1ff28af15b460a9",
|
| 53 |
+
"dcv/dtail_v2_plastic_data.py": "514feac427ce8318f5199eed209449ec87cf730b60bbf02e20e09edac9c1b115",
|
| 54 |
+
"dcv/train_dtail_v2_plastic.py": "1e3da1d9b2608dd1509efdcb5a4e248a8ea4498f921c9052909474088183c1c5",
|
| 55 |
+
"dcv/run_dtail_v2_plastic_experiment.py": "a90d5547d9cda4db9ce4fd604b4e6bbcd6bfdaae0c9119b364adfc7b42165aee",
|
| 56 |
+
"dcv/dtail_v2_refine.py": "c1339ea569efd1547a024c06f0782a51b5fd3094dd7ca19dd1c4c2fd13b538cd",
|
| 57 |
+
"dcv/dtail_v2_fusion.py": "cb8b908d75d5bf09fdfbe935759571e387cd2dc76ec4a4d9b6bb9d0c261671c7",
|
| 58 |
+
"dcv/dtail_v2_data.py": "73f1c3b1a0dcb6f9c6357ab0b0a5dc2820b8a8651234c0be0dcf2593cdf4891d",
|
| 59 |
+
"dcv/run_dtail_v2.py": "a32c14b47801e6f5b7a1b11eca3b8fb57eccb5fc33f8a5f49ccd5ca2c83a15f8",
|
| 60 |
+
"dcv/hard_suite_reader.py": "8f12d1eaa6aedf439a038e98dad48e2f9776d1abc4b8dced89dd10033cf50f4e",
|
| 61 |
+
"dcv/hard_suite_runtime_inputs.py": "80cef66da9b554a55191f6474ef062b9b8e0202eefcd73ba0038deed8c598578",
|
| 62 |
+
"closure_jepa/rollout.py": "b671a527c18f25c432b3cf47fc2bdd0aba44e0b5b878ec19fd3f2814bb908538",
|
| 63 |
+
"closure_jepa/tdjepa_pool_adapter.py": "157ba3b9500e4818607b1225648bd2f08ee367dbc995b7f5a69bceb07433d8a8",
|
| 64 |
+
"closure_jepa/tdjepa_runtime.py": "eff63651f17a867ae091da53bdb93c597fcab20c099cfbaaf5d22b46431465aa"
|
| 65 |
+
},
|
| 66 |
+
"runtime_action_inputs": {
|
| 67 |
+
"artifact_sha256": "4187ade9296ff98534b335304035ede51c9c2d488f11e3993bff7ae0f3adc143",
|
| 68 |
+
"sidecar_sha256": "a1d78eac4ac03320a16d1c7f790dafb4c8291544b0cee0bab7044e23acfff8fe",
|
| 69 |
+
"validation_report_sha256": "c880526c322936f6da3b35f2c97c3b0d7b287d24553a6ff9f01d5b1a6061ae43",
|
| 70 |
+
"completion_marker_path": "source-file:pusht_dtail_runtime_actions_v1_1.npz.complete.json"
|
| 71 |
+
}
|
| 72 |
+
}
|
| 73 |
+
}
|
pusht-adapted-predictor-full/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:26cb528b6ba5b3b6e7d464eeb9a75f599ac28641617666e0638c9e51de2b9083
|
| 3 |
+
size 73174815
|
pusht-exact-realization-full/README.md
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# D-JEPA: pusht-exact-realization-full
|
| 2 |
+
|
| 3 |
+
- Architecture: `exact_realization_world_model`.
|
| 4 |
+
- Tensor count: 642.
|
| 5 |
+
- Includes pretrained predictor weights: true.
|
| 6 |
+
- Original checkpoint SHA256: `edf28ce23a2c9cc511c63f1f8d64f93e9c156436ff1fccfc1cecbc5545d55be2`.
|
| 7 |
+
- Exported tensor-file SHA256: `983be0921f3b1d393ce9c6b0e6acfda8272b546ec7009e93ec0a4466a91a1ba2`.
|
| 8 |
+
|
| 9 |
+
Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
|
| 10 |
+
|
| 11 |
+
Publication hold: upstream weight redistribution terms must be confirmed before upload.
|
pusht-exact-realization-full/config.json
ADDED
|
@@ -0,0 +1,205 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"project": "D-JEPA",
|
| 3 |
+
"profile": "pusht-exact-realization-full",
|
| 4 |
+
"source_sha256": "edf28ce23a2c9cc511c63f1f8d64f93e9c156436ff1fccfc1cecbc5545d55be2",
|
| 5 |
+
"weights_sha256": "983be0921f3b1d393ce9c6b0e6acfda8272b546ec7009e93ec0a4466a91a1ba2",
|
| 6 |
+
"tensor_count": 642,
|
| 7 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 8 |
+
"architecture": "exact_realization_world_model",
|
| 9 |
+
"upstream_included": true,
|
| 10 |
+
"decision": {
|
| 11 |
+
"fusion_alpha": 0.42,
|
| 12 |
+
"gate_threshold": -0.03162526342176623,
|
| 13 |
+
"ordinal_denominator": 62,
|
| 14 |
+
"gate_comparison": "strict_gt",
|
| 15 |
+
"advantage_quantization": "decimal_half_even_4"
|
| 16 |
+
},
|
| 17 |
+
"lifting": {
|
| 18 |
+
"mode": "td_radial_ordinal_v1",
|
| 19 |
+
"radius_denominator": 64,
|
| 20 |
+
"zero_rms_threshold": 1e-12,
|
| 21 |
+
"preserved_prefix_steps": 4,
|
| 22 |
+
"native_cost_dtype": "float32"
|
| 23 |
+
},
|
| 24 |
+
"preprocessing": {
|
| 25 |
+
"image_mean": [
|
| 26 |
+
0.485,
|
| 27 |
+
0.456,
|
| 28 |
+
0.406
|
| 29 |
+
],
|
| 30 |
+
"image_std": [
|
| 31 |
+
0.229,
|
| 32 |
+
0.224,
|
| 33 |
+
0.225
|
| 34 |
+
],
|
| 35 |
+
"context_frames": 3,
|
| 36 |
+
"action_block": 5,
|
| 37 |
+
"future_blocks": 5,
|
| 38 |
+
"action_mean": [
|
| 39 |
+
-0.007812565192580223,
|
| 40 |
+
0.006860686466097832
|
| 41 |
+
],
|
| 42 |
+
"action_std": [
|
| 43 |
+
0.2084674835205078,
|
| 44 |
+
0.20674866437911987
|
| 45 |
+
]
|
| 46 |
+
},
|
| 47 |
+
"lineage": {
|
| 48 |
+
"source_sha256": {
|
| 49 |
+
"tdjepa_weights": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
|
| 50 |
+
"tdjepa_config": "c7c802b3c96d5233ead1839c10fc3a85c0c484ce10a755af119a9ee5b1575075",
|
| 51 |
+
"lewm_object": "5487ac45812ded38e11472eb7084d28c81c18e85154293125dd51c278fcd72c9",
|
| 52 |
+
"lewm_weights": "48938400ae3464c9680731287f583a9cb516f55a8ec64ea13a91be47fb15b607",
|
| 53 |
+
"lewm_config": "2564086e961e7b5c7c04dffc451091115b389a590645ff19653c64fd0bc16e09",
|
| 54 |
+
"refiner": "84c2b0fb93afdad9279b220ecf696642528899bb233a21ab1489c31ec7359d9f",
|
| 55 |
+
"refiner_config": "b1ebb7780bdb2ccfeb345b7f9ca5a3877becb760636a61d03daf2fb4d9387695",
|
| 56 |
+
"runtime_actions": "4187ade9296ff98534b335304035ede51c9c2d488f11e3993bff7ae0f3adc143",
|
| 57 |
+
"hard_suite": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba"
|
| 58 |
+
},
|
| 59 |
+
"packaging_config_sha256": "6932eedf77a4cd8d1f5cdf6e85961291ed10637bff7375c3800c8927e097e9ce",
|
| 60 |
+
"code_sha256": {
|
| 61 |
+
"dtail_wm/unified_r_model.py": "6efad8811121cbd5c218daad6e319dd967de1d989fb6bc23fbe4bb2f78f3f1c3",
|
| 62 |
+
"dtail_wm/package_unified_r.py": "13b0bd40ffe2851cddf566d87d2aadc833fed223b63974a6a751fe88f2276f58",
|
| 63 |
+
"reviewed_r/dtail_v2_data.py": "73f1c3b1a0dcb6f9c6357ab0b0a5dc2820b8a8651234c0be0dcf2593cdf4891d",
|
| 64 |
+
"reviewed_r/dtail_v2_fusion.py": "cb8b908d75d5bf09fdfbe935759571e387cd2dc76ec4a4d9b6bb9d0c261671c7",
|
| 65 |
+
"reviewed_r/dtail_v2_refine.py": "c1339ea569efd1547a024c06f0782a51b5fd3094dd7ca19dd1c4c2fd13b538cd",
|
| 66 |
+
"tdjepa/jepa.py": "7de87eabd061b9af3186e57198ff951e747168e4b5260bd94a51db9bd18ec62d",
|
| 67 |
+
"tdjepa/module.py": "f7688fea10208c5127093bd1a7525935e12d2a6df0a0ae1f069f7c4dda99252a",
|
| 68 |
+
"lewm/lewm.py": "c93fe07c57fedfa183d0a68a57d4092c2fcd53a109901027a1798d5dbae4bd6c",
|
| 69 |
+
"lewm/module.py": "3a6209baf30c11f77a5f6637282a1821fc62cea827438a97b23933caab51d8c7",
|
| 70 |
+
"stable_pretraining/backbone/utils.py": "269b18ca0a8e0cc9e1c7909bb349b4b094243acdc80a3fb07e847d63d2c3303b"
|
| 71 |
+
},
|
| 72 |
+
"runtime_source_sha256": {
|
| 73 |
+
"snapshot_manifest": "8bc16050969b571d65c190e147ab59e63c1bce2b69b38ff2d179b271bff6b4f2"
|
| 74 |
+
},
|
| 75 |
+
"source_candidate_contract": "63_unique_positive_literal_ids_expert_zero_excluded",
|
| 76 |
+
"packaging_commit": "42676e0a98249ca62ae4646a1cc5826181c36a66",
|
| 77 |
+
"state_counts": {
|
| 78 |
+
"tdjepa": 309,
|
| 79 |
+
"lewm": 303,
|
| 80 |
+
"refiner": 30
|
| 81 |
+
},
|
| 82 |
+
"lewm_state_attestation": {
|
| 83 |
+
"sorted_key_sha256": "ade2562bd5d2be228bb9f8fb2fea26fedef83d2f8a84a4c74f469acfc18fd89d",
|
| 84 |
+
"tensor_content_sha256": "f6c74a65523cb0448d620a42fa76203f22b4e19d68661518aff93eecf2e3cb77"
|
| 85 |
+
}
|
| 86 |
+
},
|
| 87 |
+
"model_architecture": {
|
| 88 |
+
"candidate_count": 63,
|
| 89 |
+
"horizon": 5,
|
| 90 |
+
"latent_dim": 192,
|
| 91 |
+
"tdjepa_config": {
|
| 92 |
+
"_target_": "jepa.JEPA",
|
| 93 |
+
"encoder": {
|
| 94 |
+
"_target_": "stable_pretraining.backbone.utils.vit_hf",
|
| 95 |
+
"size": "tiny",
|
| 96 |
+
"patch_size": 14,
|
| 97 |
+
"image_size": 224,
|
| 98 |
+
"pretrained": false,
|
| 99 |
+
"use_mask_token": false
|
| 100 |
+
},
|
| 101 |
+
"predictor": {
|
| 102 |
+
"_target_": "module.ARPredictor",
|
| 103 |
+
"num_frames": 3,
|
| 104 |
+
"input_dim": 192,
|
| 105 |
+
"hidden_dim": 192,
|
| 106 |
+
"output_dim": 192,
|
| 107 |
+
"depth": 6,
|
| 108 |
+
"heads": 16,
|
| 109 |
+
"mlp_dim": 2048,
|
| 110 |
+
"dim_head": 64,
|
| 111 |
+
"dropout": 0.1,
|
| 112 |
+
"emb_dropout": 0.0
|
| 113 |
+
},
|
| 114 |
+
"action_encoder": {
|
| 115 |
+
"_target_": "module.Embedder",
|
| 116 |
+
"input_dim": 10,
|
| 117 |
+
"emb_dim": 192
|
| 118 |
+
},
|
| 119 |
+
"projector": {
|
| 120 |
+
"_target_": "module.MLP",
|
| 121 |
+
"input_dim": 192,
|
| 122 |
+
"output_dim": 192,
|
| 123 |
+
"hidden_dim": 2048,
|
| 124 |
+
"norm_fn": {
|
| 125 |
+
"_target_": "torch.nn.BatchNorm1d",
|
| 126 |
+
"_partial_": true
|
| 127 |
+
}
|
| 128 |
+
},
|
| 129 |
+
"pred_proj": {
|
| 130 |
+
"_target_": "module.MLP",
|
| 131 |
+
"input_dim": 192,
|
| 132 |
+
"output_dim": 192,
|
| 133 |
+
"hidden_dim": 2048,
|
| 134 |
+
"norm_fn": {
|
| 135 |
+
"_target_": "torch.nn.BatchNorm1d",
|
| 136 |
+
"_partial_": true
|
| 137 |
+
}
|
| 138 |
+
},
|
| 139 |
+
"planning_cost_mode": "td_jepa",
|
| 140 |
+
"dist_head": {
|
| 141 |
+
"_target_": "module.QuasimetricHead",
|
| 142 |
+
"input_dim": 192,
|
| 143 |
+
"hidden_dim": 512,
|
| 144 |
+
"sym_dim": 128,
|
| 145 |
+
"asym_dim": 128
|
| 146 |
+
},
|
| 147 |
+
"planning_cost_mse_blend": 0.1
|
| 148 |
+
},
|
| 149 |
+
"lewm_config": {
|
| 150 |
+
"_target_": "stable_worldmodel.wm.lewm.LeWM",
|
| 151 |
+
"encoder": {
|
| 152 |
+
"_target_": "stable_pretraining.backbone.utils.vit_hf",
|
| 153 |
+
"size": "tiny",
|
| 154 |
+
"patch_size": 14,
|
| 155 |
+
"image_size": 224,
|
| 156 |
+
"pretrained": false,
|
| 157 |
+
"use_mask_token": false
|
| 158 |
+
},
|
| 159 |
+
"predictor": {
|
| 160 |
+
"_target_": "stable_worldmodel.wm.lewm.module.Predictor",
|
| 161 |
+
"num_frames": 3,
|
| 162 |
+
"input_dim": 192,
|
| 163 |
+
"hidden_dim": 192,
|
| 164 |
+
"output_dim": 192,
|
| 165 |
+
"depth": 6,
|
| 166 |
+
"heads": 16,
|
| 167 |
+
"mlp_dim": 2048,
|
| 168 |
+
"dim_head": 64,
|
| 169 |
+
"dropout": 0.1,
|
| 170 |
+
"emb_dropout": 0.0
|
| 171 |
+
},
|
| 172 |
+
"action_encoder": {
|
| 173 |
+
"_target_": "stable_worldmodel.wm.lewm.module.Embedder",
|
| 174 |
+
"input_dim": 10,
|
| 175 |
+
"emb_dim": 192
|
| 176 |
+
},
|
| 177 |
+
"projector": {
|
| 178 |
+
"_target_": "stable_worldmodel.wm.lewm.module.MLP",
|
| 179 |
+
"input_dim": 192,
|
| 180 |
+
"output_dim": 192,
|
| 181 |
+
"hidden_dim": 2048,
|
| 182 |
+
"norm_fn": {
|
| 183 |
+
"_target_": "torch.nn.BatchNorm1d",
|
| 184 |
+
"_partial_": true
|
| 185 |
+
}
|
| 186 |
+
},
|
| 187 |
+
"pred_proj": {
|
| 188 |
+
"_target_": "stable_worldmodel.wm.lewm.module.MLP",
|
| 189 |
+
"input_dim": 192,
|
| 190 |
+
"output_dim": 192,
|
| 191 |
+
"hidden_dim": 2048,
|
| 192 |
+
"norm_fn": {
|
| 193 |
+
"_target_": "torch.nn.BatchNorm1d",
|
| 194 |
+
"_partial_": true
|
| 195 |
+
}
|
| 196 |
+
}
|
| 197 |
+
},
|
| 198 |
+
"refiner_config": {
|
| 199 |
+
"hidden_dim": 64,
|
| 200 |
+
"low_rank": 8,
|
| 201 |
+
"max_correction": 0.2
|
| 202 |
+
}
|
| 203 |
+
},
|
| 204 |
+
"deployment": "One file containing two pretrained predictors and the relational operator; not single-backbone distillation."
|
| 205 |
+
}
|
pusht-exact-realization-full/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:983be0921f3b1d393ce9c6b0e6acfda8272b546ec7009e93ec0a4466a91a1ba2
|
| 3 |
+
size 145803819
|
pusht-multi-geometry/README.md
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# D-JEPA: pusht-multi-geometry
|
| 2 |
+
|
| 3 |
+
- Architecture: `set_aligner`.
|
| 4 |
+
- Tensor count: 30.
|
| 5 |
+
- Includes pretrained predictor weights: false.
|
| 6 |
+
- Original checkpoint SHA256: `d187c0efc194e6faa39007be835f3cca820b16fd95adfb1d7283f3d44d108b56`.
|
| 7 |
+
- Exported tensor-file SHA256: `801c92fc312b071fbc43b1e2153ae1ca25c08b1a3ae95110b38c7482e99c0fe3`.
|
| 8 |
+
|
| 9 |
+
Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
|
| 10 |
+
|
| 11 |
+
Input width: 388. Gate: `base_minus_refined`, strict threshold `0.027186155319213864`; quantization decimals: `None`. Candidate IDs and feature preprocessing must match the task profile.
|
| 12 |
+
|
| 13 |
+
Project license is selected by the authors before public upload.
|
pusht-multi-geometry/config.json
ADDED
|
@@ -0,0 +1,76 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"project": "D-JEPA",
|
| 3 |
+
"profile": "pusht-multi-geometry",
|
| 4 |
+
"source_sha256": "d187c0efc194e6faa39007be835f3cca820b16fd95adfb1d7283f3d44d108b56",
|
| 5 |
+
"weights_sha256": "801c92fc312b071fbc43b1e2153ae1ca25c08b1a3ae95110b38c7482e99c0fe3",
|
| 6 |
+
"tensor_count": 30,
|
| 7 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 8 |
+
"architecture": "set_aligner",
|
| 9 |
+
"input_dim": 388,
|
| 10 |
+
"max_correction": 0.2,
|
| 11 |
+
"gate_threshold": 0.027186155319213864,
|
| 12 |
+
"gate_mode": "base_minus_refined",
|
| 13 |
+
"gate_decimals": null,
|
| 14 |
+
"base_geometry": "jepa_wm",
|
| 15 |
+
"geometry_order": [
|
| 16 |
+
"lewm",
|
| 17 |
+
"temporal_distance",
|
| 18 |
+
"jepa_wm",
|
| 19 |
+
"dino_wm"
|
| 20 |
+
],
|
| 21 |
+
"upstream_included": false,
|
| 22 |
+
"lineage": {
|
| 23 |
+
"schema": "dtail_multi_geometry_lineage_v1",
|
| 24 |
+
"scope": "locked_development",
|
| 25 |
+
"config_sha256": "16a4170801d3b84cfaea19565f1d7fb80663adcb30fa6164eeec3d54da7cba1d",
|
| 26 |
+
"runner_sha256": "7dff9d912ec14c1a8dc540907626b203843ff5cb632a9fcb730d1e89cf0ebfe9",
|
| 27 |
+
"pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
|
| 28 |
+
"registry_sha256": "84b7b1044b958a0bfc85f9838b8ff82568fad5c76ac742ac29a79cedf6b87f65",
|
| 29 |
+
"dependency_sha256": {
|
| 30 |
+
"source-file:multi_geometry.py": "e3a50a62fd36b4552ae19623691b044442af36bbb1c3925d410c027884d76ac2",
|
| 31 |
+
"source-file:physical_join.py": "c742a241edab03e43107f52a0207fa669a66e0fe430ec29736789892ca510c03",
|
| 32 |
+
"source-file:aggregate.py": "95b478b8f5a7827e3c1963176245d8e229fbca3f73798b71becbce02cb6c75ad",
|
| 33 |
+
"source-file:registry.py": "e7ccd4e18f708c31267f5681f56812cc811a489a84d9c2499cd0dca9d3d7d08f",
|
| 34 |
+
"source-file:releases.py": "3ecdb263a4e87bf3e5ba317d7baf15f648dd132d97cf641cc49e728678f632f2",
|
| 35 |
+
"source-file:score_io.py": "126a7b713a8d02c8b0af69922a1073495e45a1713b2ab44f2ca683a15e438beb",
|
| 36 |
+
"source-file:route_experiment.py": "ad558c3b16209ff6b8b709a645ab134e63fc8883397f8124552e80d4f0239090",
|
| 37 |
+
"source-file:run_multi_geometry_routes.py": "7dff9d912ec14c1a8dc540907626b203843ff5cb632a9fcb730d1e89cf0ebfe9",
|
| 38 |
+
"source-file:dtail_v2_data.py": "73f1c3b1a0dcb6f9c6357ab0b0a5dc2820b8a8651234c0be0dcf2593cdf4891d",
|
| 39 |
+
"source-file:dtail_v2_refine.py": "c1339ea569efd1547a024c06f0782a51b5fd3094dd7ca19dd1c4c2fd13b538cd",
|
| 40 |
+
"source-file:run_dtail_v2.py": "a32c14b47801e6f5b7a1b11eca3b8fb57eccb5fc33f8a5f49ccd5ca2c83a15f8"
|
| 41 |
+
},
|
| 42 |
+
"train_release_sha256": "3c6770e13abfaf4e890d0e273cf530e0b54df690f46683f7395c4c9a3b5e74bc",
|
| 43 |
+
"development_release_sha256": "5b0dbee6f1693692037ef65016678adeee7cfa7b08fb5646ff86dbc95a9e0c72",
|
| 44 |
+
"external_score_sha256": {
|
| 45 |
+
"train": {
|
| 46 |
+
"jepa_wm": "0c6d7efa4abb1c822d9511cf0ef36aed0f7e5acb157c0973e344313e8c2681e2",
|
| 47 |
+
"dino_wm": "5eaff47859b272b90c424893e41f281a66902cbc45e2b7162a06bc87d5e462a7"
|
| 48 |
+
},
|
| 49 |
+
"development": {
|
| 50 |
+
"jepa_wm": "7fc522ca6ddde02a4eb673a4e469b5a30e7a3510b5452d3bcdfedf0d3bd51c83",
|
| 51 |
+
"dino_wm": "80c4fb904fa12c305294431047962ae35e8bbf4810056d1085d1d80903491588"
|
| 52 |
+
}
|
| 53 |
+
},
|
| 54 |
+
"geometry_order": [
|
| 55 |
+
"lewm",
|
| 56 |
+
"temporal_distance",
|
| 57 |
+
"jepa_wm",
|
| 58 |
+
"dino_wm"
|
| 59 |
+
],
|
| 60 |
+
"convex_weights": [
|
| 61 |
+
0.0,
|
| 62 |
+
0.0,
|
| 63 |
+
0.25,
|
| 64 |
+
0.75
|
| 65 |
+
],
|
| 66 |
+
"fusion_train_correct": 444,
|
| 67 |
+
"fit_count": 448,
|
| 68 |
+
"method": "dtail_jepa_base_multi_geometry",
|
| 69 |
+
"initial_state_sha256": "947de445ddfe14280fa74a200bf12c56d47113a191942d1e5144a06dbff8a836",
|
| 70 |
+
"best_update": 900,
|
| 71 |
+
"threshold": 0.027186155319213864,
|
| 72 |
+
"calibration_correct": 64,
|
| 73 |
+
"calibration_count": 64
|
| 74 |
+
},
|
| 75 |
+
"evaluation_scope": "development_128"
|
| 76 |
+
}
|
pusht-multi-geometry/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:801c92fc312b071fbc43b1e2153ae1ca25c08b1a3ae95110b38c7482e99c0fe3
|
| 3 |
+
size 377793
|
pusht-predictive-adaptation/README.md
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# D-JEPA: pusht-predictive-adaptation
|
| 2 |
+
|
| 3 |
+
- Architecture: `predictor_patch`.
|
| 4 |
+
- Tensor count: 22.
|
| 5 |
+
- Includes pretrained predictor weights: false.
|
| 6 |
+
- Original checkpoint SHA256: `ccbd6ca30c550b60a186c5d0a1fa5f1a43b7a08b704b19037586ff3de52b0887`.
|
| 7 |
+
- Exported tensor-file SHA256: `b2bcb7880fb9477a7bb9aaa3c3380b66e4cd5e1767a4093f990c72766ff8e49a`.
|
| 8 |
+
|
| 9 |
+
Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
|
| 10 |
+
|
| 11 |
+
Project license is selected by the authors before public upload.
|
pusht-predictive-adaptation/config.json
ADDED
|
@@ -0,0 +1,75 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"project": "D-JEPA",
|
| 3 |
+
"profile": "pusht-predictive-adaptation",
|
| 4 |
+
"source_sha256": "ccbd6ca30c550b60a186c5d0a1fa5f1a43b7a08b704b19037586ff3de52b0887",
|
| 5 |
+
"weights_sha256": "b2bcb7880fb9477a7bb9aaa3c3380b66e4cd5e1767a4093f990c72766ff8e49a",
|
| 6 |
+
"tensor_count": 22,
|
| 7 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 8 |
+
"architecture": "predictor_patch",
|
| 9 |
+
"upstream_included": false,
|
| 10 |
+
"base_checkpoint_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
|
| 11 |
+
"lineage": {
|
| 12 |
+
"method": "dtail_jepa_p_v2",
|
| 13 |
+
"pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
|
| 14 |
+
"tdjepa_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
|
| 15 |
+
"refiner_sha256": "84c2b0fb93afdad9279b220ecf696642528899bb233a21ab1489c31ec7359d9f",
|
| 16 |
+
"refiner_lineage": {
|
| 17 |
+
"method": "dtail_jepa_r_v2",
|
| 18 |
+
"pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
|
| 19 |
+
"lewm_weights_sha256": "5487ac45812ded38e11472eb7084d28c81c18e85154293125dd51c278fcd72c9",
|
| 20 |
+
"tdjepa_weights_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
|
| 21 |
+
"fusion_alpha": 0.42,
|
| 22 |
+
"fit_start_count": 448,
|
| 23 |
+
"calibration_start_count": 64,
|
| 24 |
+
"best_update": 550
|
| 25 |
+
},
|
| 26 |
+
"trainable_names": [
|
| 27 |
+
"predictor.transformer.layers.5.attn.norm.weight",
|
| 28 |
+
"predictor.transformer.layers.5.attn.norm.bias",
|
| 29 |
+
"predictor.transformer.layers.5.attn.to_qkv.weight",
|
| 30 |
+
"predictor.transformer.layers.5.attn.to_out.0.weight",
|
| 31 |
+
"predictor.transformer.layers.5.attn.to_out.0.bias",
|
| 32 |
+
"predictor.transformer.layers.5.mlp.net.0.weight",
|
| 33 |
+
"predictor.transformer.layers.5.mlp.net.0.bias",
|
| 34 |
+
"predictor.transformer.layers.5.mlp.net.1.weight",
|
| 35 |
+
"predictor.transformer.layers.5.mlp.net.1.bias",
|
| 36 |
+
"predictor.transformer.layers.5.mlp.net.4.weight",
|
| 37 |
+
"predictor.transformer.layers.5.mlp.net.4.bias",
|
| 38 |
+
"predictor.transformer.layers.5.adaLN_modulation.1.weight",
|
| 39 |
+
"predictor.transformer.layers.5.adaLN_modulation.1.bias",
|
| 40 |
+
"pred_proj.net.0.weight",
|
| 41 |
+
"pred_proj.net.0.bias",
|
| 42 |
+
"pred_proj.net.1.weight",
|
| 43 |
+
"pred_proj.net.1.bias",
|
| 44 |
+
"pred_proj.net.3.weight",
|
| 45 |
+
"pred_proj.net.3.bias"
|
| 46 |
+
],
|
| 47 |
+
"best_update": 200,
|
| 48 |
+
"fit_start_count": 384,
|
| 49 |
+
"calibration_start_count": 128,
|
| 50 |
+
"fusion_alpha": 0.42,
|
| 51 |
+
"config_digest": "c65cc25c5968eb762c77ec6069170551d20119f2cfa71ddd4beab795be167007",
|
| 52 |
+
"code_sha256": {
|
| 53 |
+
"dcv/dtail_v2_plastic.py": "09f0140772ea98962fcc78a2383d2d8d309b3e55bcea71f3c1ff28af15b460a9",
|
| 54 |
+
"dcv/dtail_v2_plastic_data.py": "514feac427ce8318f5199eed209449ec87cf730b60bbf02e20e09edac9c1b115",
|
| 55 |
+
"dcv/train_dtail_v2_plastic.py": "1e3da1d9b2608dd1509efdcb5a4e248a8ea4498f921c9052909474088183c1c5",
|
| 56 |
+
"dcv/run_dtail_v2_plastic_experiment.py": "a90d5547d9cda4db9ce4fd604b4e6bbcd6bfdaae0c9119b364adfc7b42165aee",
|
| 57 |
+
"dcv/dtail_v2_refine.py": "c1339ea569efd1547a024c06f0782a51b5fd3094dd7ca19dd1c4c2fd13b538cd",
|
| 58 |
+
"dcv/dtail_v2_fusion.py": "cb8b908d75d5bf09fdfbe935759571e387cd2dc76ec4a4d9b6bb9d0c261671c7",
|
| 59 |
+
"dcv/dtail_v2_data.py": "73f1c3b1a0dcb6f9c6357ab0b0a5dc2820b8a8651234c0be0dcf2593cdf4891d",
|
| 60 |
+
"dcv/run_dtail_v2.py": "a32c14b47801e6f5b7a1b11eca3b8fb57eccb5fc33f8a5f49ccd5ca2c83a15f8",
|
| 61 |
+
"dcv/hard_suite_reader.py": "8f12d1eaa6aedf439a038e98dad48e2f9776d1abc4b8dced89dd10033cf50f4e",
|
| 62 |
+
"dcv/hard_suite_runtime_inputs.py": "80cef66da9b554a55191f6474ef062b9b8e0202eefcd73ba0038deed8c598578",
|
| 63 |
+
"closure_jepa/rollout.py": "b671a527c18f25c432b3cf47fc2bdd0aba44e0b5b878ec19fd3f2814bb908538",
|
| 64 |
+
"closure_jepa/tdjepa_pool_adapter.py": "157ba3b9500e4818607b1225648bd2f08ee367dbc995b7f5a69bceb07433d8a8",
|
| 65 |
+
"closure_jepa/tdjepa_runtime.py": "eff63651f17a867ae091da53bdb93c597fcab20c099cfbaaf5d22b46431465aa"
|
| 66 |
+
},
|
| 67 |
+
"runtime_action_inputs": {
|
| 68 |
+
"artifact_sha256": "4187ade9296ff98534b335304035ede51c9c2d488f11e3993bff7ae0f3adc143",
|
| 69 |
+
"sidecar_sha256": "a1d78eac4ac03320a16d1c7f790dafb4c8291544b0cee0bab7044e23acfff8fe",
|
| 70 |
+
"validation_report_sha256": "c880526c322936f6da3b35f2c97c3b0d7b287d24553a6ff9f01d5b1a6061ae43",
|
| 71 |
+
"completion_marker_path": "source-file:pusht_dtail_runtime_actions_v1_1.npz.complete.json"
|
| 72 |
+
}
|
| 73 |
+
},
|
| 74 |
+
"usage": "Load exact pretrained Temporal-Distance-JEPA then replace these parameters. This is a parameter patch, not a full predictor."
|
| 75 |
+
}
|
pusht-predictive-adaptation/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b2bcb7880fb9477a7bb9aaa3c3380b66e4cd5e1767a4093f990c72766ff8e49a
|
| 3 |
+
size 10387305
|
pusht-relational/README.md
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# D-JEPA: pusht-relational
|
| 2 |
+
|
| 3 |
+
- Architecture: `set_aligner`.
|
| 4 |
+
- Tensor count: 30.
|
| 5 |
+
- Includes pretrained predictor weights: false.
|
| 6 |
+
- Original checkpoint SHA256: `84c2b0fb93afdad9279b220ecf696642528899bb233a21ab1489c31ec7359d9f`.
|
| 7 |
+
- Exported tensor-file SHA256: `efea4301f605f681c7f85f934a6ecdcafa9984058f72be3be2d625bfc9e16941`.
|
| 8 |
+
|
| 9 |
+
Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
|
| 10 |
+
|
| 11 |
+
Input width: 386. Gate: `base_minus_refined`, strict threshold `-0.03162526342176623`; quantization decimals: `None`. Candidate IDs and feature preprocessing must match the task profile.
|
| 12 |
+
|
| 13 |
+
Project license is selected by the authors before public upload.
|
pusht-relational/config.json
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"project": "D-JEPA",
|
| 3 |
+
"profile": "pusht-relational",
|
| 4 |
+
"source_sha256": "84c2b0fb93afdad9279b220ecf696642528899bb233a21ab1489c31ec7359d9f",
|
| 5 |
+
"weights_sha256": "efea4301f605f681c7f85f934a6ecdcafa9984058f72be3be2d625bfc9e16941",
|
| 6 |
+
"tensor_count": 30,
|
| 7 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 8 |
+
"architecture": "set_aligner",
|
| 9 |
+
"input_dim": 386,
|
| 10 |
+
"max_correction": 0.2,
|
| 11 |
+
"fusion_alpha": 0.42,
|
| 12 |
+
"gate_threshold": -0.03162526342176623,
|
| 13 |
+
"gate_mode": "base_minus_refined",
|
| 14 |
+
"gate_decimals": null,
|
| 15 |
+
"upstream_included": false,
|
| 16 |
+
"lineage": {
|
| 17 |
+
"method": "dtail_jepa_r_v2",
|
| 18 |
+
"pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
|
| 19 |
+
"lewm_weights_sha256": "5487ac45812ded38e11472eb7084d28c81c18e85154293125dd51c278fcd72c9",
|
| 20 |
+
"tdjepa_weights_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
|
| 21 |
+
"fusion_alpha": 0.42,
|
| 22 |
+
"fit_start_count": 448,
|
| 23 |
+
"calibration_start_count": 64,
|
| 24 |
+
"best_update": 550
|
| 25 |
+
}
|
| 26 |
+
}
|
pusht-relational/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:efea4301f605f681c7f85f934a6ecdcafa9984058f72be3be2d625bfc9e16941
|
| 3 |
+
size 377281
|
pusht-visual-shifts/README.md
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# D-JEPA: pusht-visual-shifts
|
| 2 |
+
|
| 3 |
+
- Architecture: `set_aligner`.
|
| 4 |
+
- Tensor count: 30.
|
| 5 |
+
- Includes pretrained predictor weights: false.
|
| 6 |
+
- Original checkpoint SHA256: `c9552d2ab5d477d5c07d71f64ac66a315f3bbcf287f866c40d64e857c67bac22`.
|
| 7 |
+
- Exported tensor-file SHA256: `cf36d6bd0af016d8adf2358d78feeda33c1bbf062eef1d0be6332ca1bb28f2e3`.
|
| 8 |
+
|
| 9 |
+
Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
|
| 10 |
+
|
| 11 |
+
Input width: 3. Gate: `refined_gap`, strict threshold `0.01`; quantization decimals: `None`. Candidate IDs and feature preprocessing must match the task profile.
|
| 12 |
+
|
| 13 |
+
Project license is selected by the authors before public upload.
|
pusht-visual-shifts/config.json
ADDED
|
@@ -0,0 +1,36 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"project": "D-JEPA",
|
| 3 |
+
"profile": "pusht-visual-shifts",
|
| 4 |
+
"source_sha256": "c9552d2ab5d477d5c07d71f64ac66a315f3bbcf287f866c40d64e857c67bac22",
|
| 5 |
+
"weights_sha256": "cf36d6bd0af016d8adf2358d78feeda33c1bbf062eef1d0be6332ca1bb28f2e3",
|
| 6 |
+
"tensor_count": 30,
|
| 7 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 8 |
+
"architecture": "set_aligner",
|
| 9 |
+
"input_dim": 3,
|
| 10 |
+
"max_correction": 0.2,
|
| 11 |
+
"fusion_alpha": 0.5,
|
| 12 |
+
"gate_threshold": 0.01,
|
| 13 |
+
"gate_mode": "refined_gap",
|
| 14 |
+
"gate_decimals": null,
|
| 15 |
+
"upstream_included": false,
|
| 16 |
+
"training": {
|
| 17 |
+
"schema": "djepa-pushobj-relational-v1",
|
| 18 |
+
"seed": 20260906,
|
| 19 |
+
"epochs": 120,
|
| 20 |
+
"learning_rate": 0.0003,
|
| 21 |
+
"batch_size": 16,
|
| 22 |
+
"input_dim": 3,
|
| 23 |
+
"max_correction": 0.2,
|
| 24 |
+
"source_model": "frozen_pusht_visual_shift_release",
|
| 25 |
+
"feature_mode": "ordinal",
|
| 26 |
+
"fusion_weight_selected_on_calibration": true,
|
| 27 |
+
"base_mode": "rank_fusion",
|
| 28 |
+
"fusion_alpha": 0.5,
|
| 29 |
+
"train_completion_sha256": "f13d2fa6a3ff152532c1acad7ec209e389737cbbee27a7ee9fce46bc39819bb8",
|
| 30 |
+
"calibration_completion_sha256": "a8e5952e7f3a6e2000d075f9198bffa99ddae1bc94b3b027953c0431d5bb81b6",
|
| 31 |
+
"torch": "2.3.0+cu121",
|
| 32 |
+
"train_code_sha256": "b890ce79e62d3787f405f14971dee1acdd1edc2e9f58b33ae9146b1ccc5e3247",
|
| 33 |
+
"model_code_sha256": "4db0e1740c89fdcbaea98cd07abd1f5d0fbdee85abb42d3791524dc6a6a8bc86"
|
| 34 |
+
},
|
| 35 |
+
"selected_epoch": 20
|
| 36 |
+
}
|
pusht-visual-shifts/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cf36d6bd0af016d8adf2358d78feeda33c1bbf062eef1d0be6332ca1bb28f2e3
|
| 3 |
+
size 279233
|
reacher-relational/README.md
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# D-JEPA: reacher-relational
|
| 2 |
+
|
| 3 |
+
- Architecture: `set_aligner`.
|
| 4 |
+
- Tensor count: 30.
|
| 5 |
+
- Includes pretrained predictor weights: false.
|
| 6 |
+
- Original checkpoint SHA256: `2bca723eaca2ef1dd7d01182b1f6f0b545aba646076b9e8ce7ba6f9f119bce05`.
|
| 7 |
+
- Exported tensor-file SHA256: `5160c04ac103589b33820133b0cb221189931382c3e6ef3bf21d9232bed830d3`.
|
| 8 |
+
|
| 9 |
+
Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
|
| 10 |
+
|
| 11 |
+
Input width: 386. Gate: `base_minus_refined`, strict threshold `0.12511829772303182`; quantization decimals: `8`. Candidate IDs and feature preprocessing must match the task profile.
|
| 12 |
+
|
| 13 |
+
Project license is selected by the authors before public upload.
|
reacher-relational/config.json
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"project": "D-JEPA",
|
| 3 |
+
"profile": "reacher-relational",
|
| 4 |
+
"source_sha256": "2bca723eaca2ef1dd7d01182b1f6f0b545aba646076b9e8ce7ba6f9f119bce05",
|
| 5 |
+
"weights_sha256": "5160c04ac103589b33820133b0cb221189931382c3e6ef3bf21d9232bed830d3",
|
| 6 |
+
"tensor_count": 30,
|
| 7 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 8 |
+
"architecture": "set_aligner",
|
| 9 |
+
"input_dim": 386,
|
| 10 |
+
"max_correction": 0.2,
|
| 11 |
+
"fusion_alpha": 0.4,
|
| 12 |
+
"gate_threshold": 0.12511829772303182,
|
| 13 |
+
"gate_mode": "base_minus_refined",
|
| 14 |
+
"gate_decimals": 8,
|
| 15 |
+
"upstream_included": false,
|
| 16 |
+
"bindings": {
|
| 17 |
+
"training_completion_sha256": "9af702ae3c9f3321863fc96c7f7dd1132748f6846cc07e289f8d28657fd7c636",
|
| 18 |
+
"full_pool_completion_sha256": "839dbbd1ace0dde4492b1d4e5a167fab12814f1330db0c00ea9d301639bdd964",
|
| 19 |
+
"capacity_gate_completion_sha256": "d6bab3c79667923f85efdb9e187f80eb20739db200ec83452efbee6edc47f6b2",
|
| 20 |
+
"b2_config_sha256": "737a766a978f16847ba8dbb240ce13f11bcda28b6d23b19f38082218563c6f73",
|
| 21 |
+
"packager_code_sha256": "4d8cbd215a0cf852a90d355be0dbeab661d6aeff17f4cc6eb6f01319446f8298"
|
| 22 |
+
}
|
| 23 |
+
}
|
reacher-relational/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5160c04ac103589b33820133b0cb221189931382c3e6ef3bf21d9232bed830d3
|
| 3 |
+
size 377281
|
reacher-temporal-transport/README.md
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# D-JEPA: reacher-temporal-transport
|
| 2 |
+
|
| 3 |
+
- Architecture: `temporal_transport`.
|
| 4 |
+
- Tensor count: 4.
|
| 5 |
+
- Includes pretrained predictor weights: false.
|
| 6 |
+
- Original checkpoint SHA256: `2bca723eaca2ef1dd7d01182b1f6f0b545aba646076b9e8ce7ba6f9f119bce05`.
|
| 7 |
+
- Exported tensor-file SHA256: `cf8c57a00e891fc002f9e595e3cf89c8a52a163e42c57059919bc1cfdddd883f`.
|
| 8 |
+
|
| 9 |
+
Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
|
| 10 |
+
|
| 11 |
+
Project license is selected by the authors before public upload.
|
reacher-temporal-transport/config.json
ADDED
|
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"project": "D-JEPA",
|
| 3 |
+
"profile": "reacher-temporal-transport",
|
| 4 |
+
"source_sha256": "2bca723eaca2ef1dd7d01182b1f6f0b545aba646076b9e8ce7ba6f9f119bce05",
|
| 5 |
+
"weights_sha256": "cf8c57a00e891fc002f9e595e3cf89c8a52a163e42c57059919bc1cfdddd883f",
|
| 6 |
+
"tensor_count": 4,
|
| 7 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 8 |
+
"architecture": "temporal_transport",
|
| 9 |
+
"hidden_dim": 32,
|
| 10 |
+
"radius": 0.1,
|
| 11 |
+
"fusion_alpha": 0.4,
|
| 12 |
+
"gate_threshold": 0.12511829772303182,
|
| 13 |
+
"gate_mode": "base_minus_refined",
|
| 14 |
+
"gate_decimals": 8,
|
| 15 |
+
"upstream_included": false,
|
| 16 |
+
"bindings": {
|
| 17 |
+
"training_completion_sha256": "9af702ae3c9f3321863fc96c7f7dd1132748f6846cc07e289f8d28657fd7c636",
|
| 18 |
+
"full_pool_completion_sha256": "839dbbd1ace0dde4492b1d4e5a167fab12814f1330db0c00ea9d301639bdd964",
|
| 19 |
+
"capacity_gate_completion_sha256": "d6bab3c79667923f85efdb9e187f80eb20739db200ec83452efbee6edc47f6b2",
|
| 20 |
+
"b2_config_sha256": "737a766a978f16847ba8dbb240ce13f11bcda28b6d23b19f38082218563c6f73",
|
| 21 |
+
"packager_code_sha256": "4d8cbd215a0cf852a90d355be0dbeab661d6aeff17f4cc6eb6f01319446f8298"
|
| 22 |
+
},
|
| 23 |
+
"physical_selector": "relational score; transported distance is a distinct diagnostic"
|
| 24 |
+
}
|
reacher-temporal-transport/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cf8c57a00e891fc002f9e595e3cf89c8a52a163e42c57059919bc1cfdddd883f
|
| 3 |
+
size 3081
|
reacher-world-model-full/README.md
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# D-JEPA: reacher-world-model-full
|
| 2 |
+
|
| 3 |
+
- Architecture: `reacher_world_model`.
|
| 4 |
+
- Tensor count: 646.
|
| 5 |
+
- Includes pretrained predictor weights: true.
|
| 6 |
+
- Original checkpoint SHA256: `2bca723eaca2ef1dd7d01182b1f6f0b545aba646076b9e8ce7ba6f9f119bce05`.
|
| 7 |
+
- Exported tensor-file SHA256: `f7331bf87bf4c8543fa8797072e660266e55d9a9354702120cec1faadd76609a`.
|
| 8 |
+
|
| 9 |
+
Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
|
| 10 |
+
|
| 11 |
+
Publication hold: upstream weight redistribution terms must be confirmed before upload.
|
reacher-world-model-full/config.json
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"project": "D-JEPA",
|
| 3 |
+
"profile": "reacher-world-model-full",
|
| 4 |
+
"source_sha256": "2bca723eaca2ef1dd7d01182b1f6f0b545aba646076b9e8ce7ba6f9f119bce05",
|
| 5 |
+
"weights_sha256": "f7331bf87bf4c8543fa8797072e660266e55d9a9354702120cec1faadd76609a",
|
| 6 |
+
"tensor_count": 646,
|
| 7 |
+
"public_status": "local_release_candidate_license_not_yet_selected",
|
| 8 |
+
"architecture": "reacher_world_model",
|
| 9 |
+
"fusion_alpha": 0.4,
|
| 10 |
+
"gate_threshold": 0.12511829772303182,
|
| 11 |
+
"gate_mode": "base_minus_refined",
|
| 12 |
+
"gate_decimals": 8,
|
| 13 |
+
"upstream_included": true,
|
| 14 |
+
"bindings": {
|
| 15 |
+
"training_completion_sha256": "9af702ae3c9f3321863fc96c7f7dd1132748f6846cc07e289f8d28657fd7c636",
|
| 16 |
+
"full_pool_completion_sha256": "839dbbd1ace0dde4492b1d4e5a167fab12814f1330db0c00ea9d301639bdd964",
|
| 17 |
+
"capacity_gate_completion_sha256": "d6bab3c79667923f85efdb9e187f80eb20739db200ec83452efbee6edc47f6b2",
|
| 18 |
+
"b2_config_sha256": "737a766a978f16847ba8dbb240ce13f11bcda28b6d23b19f38082218563c6f73",
|
| 19 |
+
"packager_code_sha256": "4d8cbd215a0cf852a90d355be0dbeab661d6aeff17f4cc6eb6f01319446f8298"
|
| 20 |
+
},
|
| 21 |
+
"state_groups": [
|
| 22 |
+
"tdjepa",
|
| 23 |
+
"lewm",
|
| 24 |
+
"adapters"
|
| 25 |
+
],
|
| 26 |
+
"physical_selector": "relational score"
|
| 27 |
+
}
|
reacher-world-model-full/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f7331bf87bf4c8543fa8797072e660266e55d9a9354702120cec1faadd76609a
|
| 3 |
+
size 145806043
|