Add sanitized world-model and HGD research checkpoints
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- .gitattributes +6 -0
- LICENSES.md +17 -0
- README.md +87 -0
- dino_wm/pusht/hydra.yaml +96 -0
- dino_wm/pusht/model_latest.pth +3 -0
- dino_wm/runtime.json +103 -0
- dino_wm/wall_single/hydra.yaml +96 -0
- dino_wm/wall_single/model_latest.pth +3 -0
- dreamerv3/cheetah_run/ckpt/20260901T093748F019112/agent.pkl +3 -0
- dreamerv3/cheetah_run/ckpt/20260901T093748F019112/done +0 -0
- dreamerv3/cheetah_run/ckpt/20260901T093748F019112/step.pkl +3 -0
- dreamerv3/cheetah_run/config.yaml +185 -0
- dreamerv3/reacher_easy/ckpt/20260830T195435F325934/agent.pkl +3 -0
- dreamerv3/reacher_easy/ckpt/20260830T195435F325934/done +0 -0
- dreamerv3/reacher_easy/ckpt/20260830T195435F325934/step.pkl +3 -0
- dreamerv3/reacher_easy/config.yaml +185 -0
- dreamerv3/runtime.json +126 -0
- le-wm/cube/config.json +49 -0
- le-wm/cube/weights.pt +3 -0
- le-wm/defenses/cube/paper_hgd/hgd_paper.best.pt +3 -0
- le-wm/defenses/cube/paper_hgd/hgd_paper.pt +3 -0
- le-wm/defenses/cube/residual_hgd/hgd_eps16.best.pt +3 -0
- le-wm/defenses/cube/residual_hgd/hgd_eps16.pt +3 -0
- le-wm/pusht/config.json +49 -0
- le-wm/pusht/weights.pt +3 -0
- le-wm/reacher/config.json +49 -0
- le-wm/reacher/weights.pt +3 -0
- le-wm/runtime.json +190 -0
- le-wm/tworoom/config.json +49 -0
- le-wm/tworoom/weights.pt +3 -0
- licenses/dino_wm.txt +21 -0
- licenses/dreamerv3.txt +19 -0
- licenses/le-wm.txt +21 -0
- licenses/planet.txt +202 -0
- licenses/tdmpc2.txt +21 -0
- manifest.json +481 -0
- planet/cartpole_swingup/attack_export.json +10 -0
- planet/cartpole_swingup/attack_weights.npz +3 -0
- planet/cartpole_swingup/config.yaml +388 -0
- planet/cartpole_swingup/tensorflow/checkpoint +2 -0
- planet/cartpole_swingup/tensorflow/model.ckpt-4504500.data-00000-of-00001 +3 -0
- planet/cartpole_swingup/tensorflow/model.ckpt-4504500.index +0 -0
- planet/cartpole_swingup/tensorflow/model.ckpt-4504500.meta +3 -0
- planet/cartpole_swingup/tf_reference.npz +3 -0
- planet/cheetah_run/attack_export.json +10 -0
- planet/cheetah_run/attack_weights.npz +3 -0
- planet/cheetah_run/config.yaml +388 -0
- planet/cheetah_run/tensorflow/checkpoint +2 -0
- planet/cheetah_run/tensorflow/model.ckpt-4504500.data-00000-of-00001 +3 -0
- planet/cheetah_run/tensorflow/model.ckpt-4504500.index +0 -0
.gitattributes
CHANGED
|
@@ -33,3 +33,9 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
planet/cartpole_swingup/tensorflow/model.ckpt-4504500.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
|
| 37 |
+
planet/cartpole_swingup/tensorflow/model.ckpt-4504500.meta filter=lfs diff=lfs merge=lfs -text
|
| 38 |
+
planet/cheetah_run/tensorflow/model.ckpt-4504500.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
|
| 39 |
+
planet/cheetah_run/tensorflow/model.ckpt-4504500.meta filter=lfs diff=lfs merge=lfs -text
|
| 40 |
+
planet/reacher_easy/tensorflow/model.ckpt-4504500.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
|
| 41 |
+
planet/reacher_easy/tensorflow/model.ckpt-4504500.meta filter=lfs diff=lfs merge=lfs -text
|
LICENSES.md
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Attribution and license notices
|
| 2 |
+
|
| 3 |
+
The `licenses/` directory contains the license notices shipped with the five
|
| 4 |
+
source code repositories used to produce or load these checkpoints:
|
| 5 |
+
|
| 6 |
+
| Project | Source notice |
|
| 7 |
+
|---|---|
|
| 8 |
+
| LeWM | [MIT notice](licenses/le-wm.txt) |
|
| 9 |
+
| DINO-WM | [MIT notice](licenses/dino_wm.txt) |
|
| 10 |
+
| PlaNet | [Apache 2.0 notice](licenses/planet.txt) |
|
| 11 |
+
| TD-MPC2 | [MIT notice](licenses/tdmpc2.txt) |
|
| 12 |
+
| DreamerV3 | [Source license](licenses/dreamerv3.txt) |
|
| 13 |
+
|
| 14 |
+
These source notices are preserved for attribution. They do not create a new
|
| 15 |
+
blanket license for this collection or replace any terms accompanying upstream
|
| 16 |
+
weights and datasets. Consult the linked upstream artifact pages for those terms.
|
| 17 |
+
No new collection-wide license is assigned in this preparation.
|
README.md
ADDED
|
@@ -0,0 +1,87 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
language:
|
| 3 |
+
- en
|
| 4 |
+
tags:
|
| 5 |
+
- reinforcement-learning
|
| 6 |
+
- world-model
|
| 7 |
+
- robustness
|
| 8 |
+
- adversarial-evaluation
|
| 9 |
+
- checkpoint
|
| 10 |
+
---
|
| 11 |
+
|
| 12 |
+
# World-model attack research checkpoints
|
| 13 |
+
|
| 14 |
+
Frozen checkpoints and supporting configurations for the research project
|
| 15 |
+
**Misranking Imagined Futures: Test-Time Attacks on Frozen World-Model Planners**.
|
| 16 |
+
This collection preserves the available checkpoints used by the local research
|
| 17 |
+
implementations, including upstream pretrained models and locally trained models.
|
| 18 |
+
|
| 19 |
+
| Directory | Included artifacts |
|
| 20 |
+
|---|---|
|
| 21 |
+
| `le-wm/` | Cube, Push-T, Reacher, and Two-Rooms model weights and architecture configurations; residual HGD and paper HGD defense checkpoints |
|
| 22 |
+
| `dino_wm/` | Released Push-T and Wall/Two-Rooms model checkpoints and training configurations |
|
| 23 |
+
| `planet/` | Cartpole Swingup, Reacher Easy, and Cheetah Run native TensorFlow checkpoints, frozen PyTorch inference exports, and TensorFlow parity fixtures |
|
| 24 |
+
| `tdmpc2/` | The four RGB checkpoints selected by the research reference manifests: Finger Spin at 1,000,000 steps; Cup Catch and Cheetah Run at 1,500,000; Cartpole Balance at 250,000 |
|
| 25 |
+
| `dreamerv3/` | Cheetah Run and Reacher Easy training configurations and immutable checkpoint directories |
|
| 26 |
+
|
| 27 |
+
These checkpoints have different observation preprocessing, architectures, task
|
| 28 |
+
interfaces, and Python dependencies. Use the corresponding research model loader;
|
| 29 |
+
they are not interchangeable Transformers models. The PlaNet PyTorch exports
|
| 30 |
+
contain 32 inference tensors and require the research implementation of PlaNet.
|
| 31 |
+
The DINO-WM checkpoints also require their DINOv2 encoder dependency and native
|
| 32 |
+
task data. LeWM and defense evaluations require their datasets and simulators.
|
| 33 |
+
|
| 34 |
+
## Download and integrity
|
| 35 |
+
|
| 36 |
+
Download the repository using the Hugging Face CLI:
|
| 37 |
+
|
| 38 |
+
```bash
|
| 39 |
+
hf download mohendy/world-model-attack-checkpoints --local-dir ./checkpoints
|
| 40 |
+
python ./checkpoints/verify_download.py
|
| 41 |
+
```
|
| 42 |
+
|
| 43 |
+
If the repository is private, authenticate with `hf auth login` first. For a
|
| 44 |
+
durable experiment record, pass `--revision` with the repository commit SHA
|
| 45 |
+
returned by the upload, instead of relying on the moving `main` branch. A partial
|
| 46 |
+
download can be checked with `python verify_download.py --model planet` from the
|
| 47 |
+
downloaded directory.
|
| 48 |
+
|
| 49 |
+
`manifest.json` records the SHA-256 and size of every checkpoint/configuration,
|
| 50 |
+
plus its original relative location in the research workspace. Its
|
| 51 |
+
`source_relative_path` field provides the destination mapping for the code
|
| 52 |
+
release's artifact layout. Binary model checkpoints preserve their source bytes.
|
| 53 |
+
Configuration output paths are sanitized and relative. HGD checkpoints are
|
| 54 |
+
re-exported with sanitized metadata; their `state_dict` tensors were verified to
|
| 55 |
+
be exactly equal to the originals. Both original and released hashes are recorded.
|
| 56 |
+
|
| 57 |
+
Copy or link artifacts into the code release using that mapping, and set explicit
|
| 58 |
+
checkpoint/data paths for your machine. Run directories for Dreamer must include
|
| 59 |
+
their `config.yaml` and chosen `ckpt/<checkpoint-id>/` directory. Select the
|
| 60 |
+
immutable directory explicitly instead of a training run's `latest` pointer.
|
| 61 |
+
|
| 62 |
+
Model-specific `runtime.json` files record the installed Python and package
|
| 63 |
+
versions from the research environment. They are provenance records, not a
|
| 64 |
+
guarantee that all packages resolve unchanged on another platform. The code
|
| 65 |
+
release maintains environment installation instructions and validation results.
|
| 66 |
+
|
| 67 |
+
## Scope and provenance
|
| 68 |
+
|
| 69 |
+
Datasets, experiment logs, videos, and per-episode adversarial perturbations are
|
| 70 |
+
not included. Checkpoint availability alone does not reproduce the paper tables:
|
| 71 |
+
the task datasets, episode identities, seed schedules, preprocessing, planner
|
| 72 |
+
configuration, and matched clean reference results must also agree.
|
| 73 |
+
|
| 74 |
+
Upstream projects and weight sources:
|
| 75 |
+
|
| 76 |
+
- [LeWM code](https://github.com/lucas-maes/le-wm) and
|
| 77 |
+
[checkpoint collection](https://huggingface.co/collections/quentinll/lewm).
|
| 78 |
+
- [DINO-WM code](https://github.com/gaoyuezhou/dino_wm) and its
|
| 79 |
+
[released artifacts](https://osf.io/bmw48/).
|
| 80 |
+
- [PlaNet code](https://github.com/google-research/planet); the local inference
|
| 81 |
+
exports record source artifact revision
|
| 82 |
+
`b26a7fe027c018f62b3178c8e126371fbdee33e5` in `attack_export.json`.
|
| 83 |
+
- [TD-MPC2 code](https://github.com/nicklashansen/tdmpc2).
|
| 84 |
+
- [DreamerV3 code](https://github.com/danijar/dreamerv3).
|
| 85 |
+
|
| 86 |
+
See [LICENSES.md](LICENSES.md) for the accompanying upstream notices. Original
|
| 87 |
+
upstream work and local research additions should be credited separately.
|
dino_wm/pusht/hydra.yaml
ADDED
|
@@ -0,0 +1,96 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
ckpt_base_path: ./checkpoints
|
| 2 |
+
training:
|
| 3 |
+
seed: 0
|
| 4 |
+
epochs: 100
|
| 5 |
+
batch_size: 32
|
| 6 |
+
save_every_x_epoch: 1
|
| 7 |
+
reconstruct_every_x_batch: 500
|
| 8 |
+
num_reconstruct_samples: 6
|
| 9 |
+
encoder_lr: 1.0e-06
|
| 10 |
+
decoder_lr: 0.0003
|
| 11 |
+
predictor_lr: 0.0005
|
| 12 |
+
action_encoder_lr: 0.0005
|
| 13 |
+
img_size: 224
|
| 14 |
+
frameskip: 5
|
| 15 |
+
concat_dim: 1
|
| 16 |
+
normalize_action: true
|
| 17 |
+
action_emb_dim: 10
|
| 18 |
+
num_action_repeat: 1
|
| 19 |
+
proprio_emb_dim: 10
|
| 20 |
+
num_proprio_repeat: 1
|
| 21 |
+
num_hist: 3
|
| 22 |
+
num_pred: 1
|
| 23 |
+
has_predictor: true
|
| 24 |
+
has_decoder: true
|
| 25 |
+
state_based: false
|
| 26 |
+
model:
|
| 27 |
+
_target_: models.visual_world_model.VWorldModel
|
| 28 |
+
image_size: 224
|
| 29 |
+
num_hist: 3
|
| 30 |
+
num_pred: 1
|
| 31 |
+
train_encoder: false
|
| 32 |
+
train_predictor: true
|
| 33 |
+
train_decoder: true
|
| 34 |
+
debug: false
|
| 35 |
+
plan_settings:
|
| 36 |
+
plan_cfg_path: conf/plan.yaml
|
| 37 |
+
planner:
|
| 38 |
+
- cem
|
| 39 |
+
goal_source:
|
| 40 |
+
- dset
|
| 41 |
+
goal_H:
|
| 42 |
+
- 5
|
| 43 |
+
alpha:
|
| 44 |
+
- 0.1
|
| 45 |
+
- 1
|
| 46 |
+
env:
|
| 47 |
+
name: pusht
|
| 48 |
+
args: []
|
| 49 |
+
kwargs:
|
| 50 |
+
with_velocity: true
|
| 51 |
+
with_target: true
|
| 52 |
+
dataset:
|
| 53 |
+
_target_: datasets.pusht_dset.load_pusht_slice_train_val
|
| 54 |
+
with_velocity: true
|
| 55 |
+
n_rollout: null
|
| 56 |
+
normalize_action: true
|
| 57 |
+
data_path: ${oc.env:DATASET_DIR}/pusht_noise
|
| 58 |
+
split_ratio: 0.9
|
| 59 |
+
transform:
|
| 60 |
+
_target_: datasets.img_transforms.default_transform
|
| 61 |
+
img_size: 224
|
| 62 |
+
decoder_path: null
|
| 63 |
+
num_workers: 16
|
| 64 |
+
encoder:
|
| 65 |
+
_target_: models.dino.DinoV2Encoder
|
| 66 |
+
name: dinov2_vits14
|
| 67 |
+
feature_key: x_norm_patchtokens
|
| 68 |
+
action_encoder:
|
| 69 |
+
_target_: models.proprio.ProprioceptiveEmbedding
|
| 70 |
+
num_frames: 1
|
| 71 |
+
tubelet_size: 1
|
| 72 |
+
use_3d_pos: false
|
| 73 |
+
proprio_encoder:
|
| 74 |
+
_target_: models.proprio.ProprioceptiveEmbedding
|
| 75 |
+
num_frames: 1
|
| 76 |
+
tubelet_size: 1
|
| 77 |
+
use_3d_pos: false
|
| 78 |
+
decoder:
|
| 79 |
+
_target_: models.vqvae.VQVAE
|
| 80 |
+
channel: 384
|
| 81 |
+
n_embed: 2048
|
| 82 |
+
n_res_block: 4
|
| 83 |
+
n_res_channel: 128
|
| 84 |
+
quantize: false
|
| 85 |
+
predictor:
|
| 86 |
+
_target_: models.vit.ViTPredictor
|
| 87 |
+
depth: 6
|
| 88 |
+
heads: 16
|
| 89 |
+
mlp_dim: 2048
|
| 90 |
+
dropout: 0.1
|
| 91 |
+
emb_dropout: 0
|
| 92 |
+
pool: mean
|
| 93 |
+
saved_folder: ./checkpoints
|
| 94 |
+
effective_batch_size: 32
|
| 95 |
+
gpu_batch_size: 1
|
| 96 |
+
wandb_run_id: null
|
dino_wm/pusht/model_latest.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e909f0cec958fc0b49a79f2e85730ae6b5f83dee61a224705f91883eb3bb74c5
|
| 3 |
+
size 377929744
|
dino_wm/runtime.json
ADDED
|
@@ -0,0 +1,103 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"python": "3.10.14",
|
| 3 |
+
"platform": "Linux x86_64",
|
| 4 |
+
"packages": {
|
| 5 |
+
"hydra-submitit-launcher": "1.2.0",
|
| 6 |
+
"submitit": "1.5.1",
|
| 7 |
+
"typing-extensions": "4.15.0",
|
| 8 |
+
"cloudpickle": "2.1.0",
|
| 9 |
+
"hydra-core": "1.2.0",
|
| 10 |
+
"packaging": "24.0",
|
| 11 |
+
"antlr4-python3-runtime": "4.9.3",
|
| 12 |
+
"omegaconf": "2.3.0",
|
| 13 |
+
"pyyaml": "6.0.1",
|
| 14 |
+
"accelerate": "0.26.1",
|
| 15 |
+
"safetensors": "0.4.3",
|
| 16 |
+
"huggingface-hub": "0.36.2",
|
| 17 |
+
"tqdm": "4.66.4",
|
| 18 |
+
"requests": "2.32.3",
|
| 19 |
+
"certifi": "2024.6.2",
|
| 20 |
+
"urllib3": "2.2.1",
|
| 21 |
+
"idna": "3.7",
|
| 22 |
+
"charset-normalizer": "3.3.2",
|
| 23 |
+
"hf-xet": "1.2.0",
|
| 24 |
+
"fsspec": "2024.5.0",
|
| 25 |
+
"filelock": "3.14.0",
|
| 26 |
+
"torch": "2.7.0+cu128",
|
| 27 |
+
"triton": "3.3.0",
|
| 28 |
+
"setuptools": "70.0.0",
|
| 29 |
+
"nvidia-cufile-cu12": "1.13.0.11",
|
| 30 |
+
"nvidia-nvjitlink-cu12": "12.8.61",
|
| 31 |
+
"nvidia-nvtx-cu12": "12.8.55",
|
| 32 |
+
"nvidia-nccl-cu12": "2.26.2",
|
| 33 |
+
"nvidia-cusparselt-cu12": "0.6.3",
|
| 34 |
+
"nvidia-cusparse-cu12": "12.5.7.53",
|
| 35 |
+
"nvidia-cusolver-cu12": "11.7.2.55",
|
| 36 |
+
"nvidia-cublas-cu12": "12.8.3.14",
|
| 37 |
+
"nvidia-curand-cu12": "10.3.9.55",
|
| 38 |
+
"nvidia-cufft-cu12": "11.3.3.41",
|
| 39 |
+
"nvidia-cudnn-cu12": "9.7.1.26",
|
| 40 |
+
"nvidia-cuda-cupti-cu12": "12.8.57",
|
| 41 |
+
"nvidia-cuda-runtime-cu12": "12.8.57",
|
| 42 |
+
"nvidia-cuda-nvrtc-cu12": "12.8.61",
|
| 43 |
+
"jinja2": "3.1.4",
|
| 44 |
+
"markupsafe": "2.1.5",
|
| 45 |
+
"networkx": "3.2.1",
|
| 46 |
+
"sympy": "1.14.0",
|
| 47 |
+
"mpmath": "1.3.0",
|
| 48 |
+
"psutil": "7.2.2",
|
| 49 |
+
"numpy": "1.26.4",
|
| 50 |
+
"dill": "0.4.1",
|
| 51 |
+
"scikit-image": "0.19.3",
|
| 52 |
+
"pywavelets": "1.6.0",
|
| 53 |
+
"tifffile": "2024.5.22",
|
| 54 |
+
"imageio": "2.34.1",
|
| 55 |
+
"pillow": "10.3.0",
|
| 56 |
+
"scipy": "1.13.1",
|
| 57 |
+
"numba": "0.59.1",
|
| 58 |
+
"llvmlite": "0.42.0",
|
| 59 |
+
"decord": "0.6.0",
|
| 60 |
+
"h5py": "3.11.0",
|
| 61 |
+
"pygame": "2.5.2",
|
| 62 |
+
"pymunk": "6.8.0",
|
| 63 |
+
"cffi": "1.16.0",
|
| 64 |
+
"pycparser": "2.22",
|
| 65 |
+
"mujoco": "3.2.7",
|
| 66 |
+
"pyopengl": "3.1.7",
|
| 67 |
+
"glfw": "2.7.0",
|
| 68 |
+
"etils": "1.5.2",
|
| 69 |
+
"zipp": "3.19.1",
|
| 70 |
+
"importlib-resources": "6.4.0",
|
| 71 |
+
"absl-py": "2.1.0",
|
| 72 |
+
"dm-control": "1.0.27",
|
| 73 |
+
"pyparsing": "3.1.2",
|
| 74 |
+
"protobuf": "6.33.6",
|
| 75 |
+
"lxml": "5.3.1",
|
| 76 |
+
"labmaze": "1.0.6",
|
| 77 |
+
"dm-tree": "0.1.8",
|
| 78 |
+
"dm-env": "1.6",
|
| 79 |
+
"gym": "0.23.1",
|
| 80 |
+
"gym-notices": "0.0.8",
|
| 81 |
+
"wandb": "0.25.1",
|
| 82 |
+
"sentry-sdk": "2.3.1",
|
| 83 |
+
"pydantic": "2.12.5",
|
| 84 |
+
"typing-inspection": "0.4.2",
|
| 85 |
+
"pydantic-core": "2.41.5",
|
| 86 |
+
"annotated-types": "0.7.0",
|
| 87 |
+
"platformdirs": "4.2.2",
|
| 88 |
+
"gitpython": "3.1.43",
|
| 89 |
+
"gitdb": "4.0.11",
|
| 90 |
+
"smmap": "5.0.1",
|
| 91 |
+
"click": "8.1.7",
|
| 92 |
+
"einops": "0.4.1",
|
| 93 |
+
"matplotlib": "3.5.3",
|
| 94 |
+
"python-dateutil": "2.9.0.post0",
|
| 95 |
+
"six": "1.16.0",
|
| 96 |
+
"kiwisolver": "1.4.5",
|
| 97 |
+
"fonttools": "4.53.0",
|
| 98 |
+
"cycler": "0.12.1",
|
| 99 |
+
"imageio-ffmpeg": "0.4.9",
|
| 100 |
+
"torchvision": "0.22.0+cu128"
|
| 101 |
+
},
|
| 102 |
+
"metadata_conflicts": []
|
| 103 |
+
}
|
dino_wm/wall_single/hydra.yaml
ADDED
|
@@ -0,0 +1,96 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
ckpt_base_path: ./checkpoints
|
| 2 |
+
training:
|
| 3 |
+
seed: 0
|
| 4 |
+
epochs: 1000
|
| 5 |
+
batch_size: 128
|
| 6 |
+
save_every_x_epoch: 5
|
| 7 |
+
reconstruct_every_x_batch: 500
|
| 8 |
+
num_reconstruct_samples: 6
|
| 9 |
+
encoder_lr: 1.0e-06
|
| 10 |
+
decoder_lr: 0.0003
|
| 11 |
+
predictor_lr: 0.0005
|
| 12 |
+
action_encoder_lr: 0.0005
|
| 13 |
+
img_size: 224
|
| 14 |
+
frameskip: 5
|
| 15 |
+
concat_dim: 1
|
| 16 |
+
normalize_action: true
|
| 17 |
+
action_emb_dim: 10
|
| 18 |
+
num_action_repeat: 1
|
| 19 |
+
proprio_emb_dim: 10
|
| 20 |
+
num_proprio_repeat: 1
|
| 21 |
+
num_hist: 1
|
| 22 |
+
num_pred: 1
|
| 23 |
+
has_predictor: true
|
| 24 |
+
has_decoder: true
|
| 25 |
+
state_based: false
|
| 26 |
+
model:
|
| 27 |
+
_target_: models.visual_world_model.VWorldModel
|
| 28 |
+
image_size: 224
|
| 29 |
+
num_hist: 1
|
| 30 |
+
num_pred: 1
|
| 31 |
+
train_encoder: false
|
| 32 |
+
train_predictor: true
|
| 33 |
+
train_decoder: true
|
| 34 |
+
debug: true
|
| 35 |
+
plan_settings:
|
| 36 |
+
plan_cfg_path: conf/plan.yaml
|
| 37 |
+
planner:
|
| 38 |
+
- gd
|
| 39 |
+
- cem
|
| 40 |
+
goal_source:
|
| 41 |
+
- dset
|
| 42 |
+
- random_state
|
| 43 |
+
goal_H:
|
| 44 |
+
- 5
|
| 45 |
+
alpha:
|
| 46 |
+
- 0.1
|
| 47 |
+
- 1
|
| 48 |
+
env:
|
| 49 |
+
name: wall
|
| 50 |
+
args: []
|
| 51 |
+
kwargs: {}
|
| 52 |
+
load_dir: ''
|
| 53 |
+
dataset:
|
| 54 |
+
_target_: datasets.wall_dset.load_wall_slice_train_val
|
| 55 |
+
n_rollout: null
|
| 56 |
+
normalize_action: true
|
| 57 |
+
data_path: ${oc.env:DATASET_DIR}/wall_single
|
| 58 |
+
split_ratio: 0.9
|
| 59 |
+
split_mode: random
|
| 60 |
+
transform:
|
| 61 |
+
_target_: datasets.img_transforms.default_transform
|
| 62 |
+
img_size: 224
|
| 63 |
+
decoder_path: null
|
| 64 |
+
linear_probe_model_path: null
|
| 65 |
+
num_workers: 0
|
| 66 |
+
encoder:
|
| 67 |
+
_target_: models.dino.DinoV2Encoder
|
| 68 |
+
name: dinov2_vits14
|
| 69 |
+
feature_key: x_norm_patchtokens
|
| 70 |
+
action_encoder:
|
| 71 |
+
_target_: models.proprio.ProprioceptiveEmbedding
|
| 72 |
+
num_frames: 1
|
| 73 |
+
tubelet_size: 1
|
| 74 |
+
use_3d_pos: false
|
| 75 |
+
proprio_encoder:
|
| 76 |
+
_target_: models.proprio.ProprioceptiveEmbedding
|
| 77 |
+
num_frames: 1
|
| 78 |
+
tubelet_size: 1
|
| 79 |
+
use_3d_pos: false
|
| 80 |
+
decoder:
|
| 81 |
+
_target_: models.vqvae.VQVAE
|
| 82 |
+
n_embed: 2048
|
| 83 |
+
n_res_block: 4
|
| 84 |
+
n_res_channel: 128
|
| 85 |
+
quantize: false
|
| 86 |
+
predictor:
|
| 87 |
+
_target_: models.vit.ViTPredictor
|
| 88 |
+
depth: 6
|
| 89 |
+
heads: 16
|
| 90 |
+
mlp_dim: 2048
|
| 91 |
+
dropout: 0.1
|
| 92 |
+
emb_dropout: 0.1
|
| 93 |
+
pool: mean
|
| 94 |
+
saved_folder: ./checkpoints
|
| 95 |
+
effective_batch_size: 128
|
| 96 |
+
gpu_batch_size: 4
|
dino_wm/wall_single/model_latest.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8441971becdae934fe08de5b163398390a32f6fe1fb0a8df113290e2468e142b
|
| 3 |
+
size 368656057
|
dreamerv3/cheetah_run/ckpt/20260901T093748F019112/agent.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1d94c5824d8c77ba8191ccd5207409986a08bbf7e04ae5d55a9d468c7d158af3
|
| 3 |
+
size 2042806149
|
dreamerv3/cheetah_run/ckpt/20260901T093748F019112/done
ADDED
|
File without changes
|
dreamerv3/cheetah_run/ckpt/20260901T093748F019112/step.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:69901686be1d13680e0c0ddbfd0a0d2f2a70bb0ae46a2e636bf78bf5bc45cce2
|
| 3 |
+
size 17
|
dreamerv3/cheetah_run/config.yaml
ADDED
|
@@ -0,0 +1,185 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
agent:
|
| 2 |
+
ac_grads: false
|
| 3 |
+
advnorm: {impl: none, limit: 1e-08, rate: 0.01}
|
| 4 |
+
conhead: {act: silu, layers: 1, norm: rms, output: binary, outscale: 1.0,
|
| 5 |
+
units: 1024, winit: trunc_normal_in}
|
| 6 |
+
contdisc: true
|
| 7 |
+
dec:
|
| 8 |
+
simple:
|
| 9 |
+
act: silu
|
| 10 |
+
bspace: 8
|
| 11 |
+
depth: 64
|
| 12 |
+
kernel: 5
|
| 13 |
+
layers: 3
|
| 14 |
+
mults: [2, 3, 4, 4]
|
| 15 |
+
norm: rms
|
| 16 |
+
outer: false
|
| 17 |
+
outscale: 1.0
|
| 18 |
+
strided: false
|
| 19 |
+
units: 1024
|
| 20 |
+
winit: trunc_normal_in
|
| 21 |
+
typ: simple
|
| 22 |
+
dyn:
|
| 23 |
+
rssm: {absolute: false, act: silu, blocks: 8, classes: 64, deter: 8192,
|
| 24 |
+
dynlayers: 1, free_nats: 1.0, hidden: 1024, imglayers: 2, norm: rms,
|
| 25 |
+
obslayers: 1, outscale: 1.0, stoch: 32, unimix: 0.01, winit:
|
| 26 |
+
trunc_normal_in}
|
| 27 |
+
typ: rssm
|
| 28 |
+
enc:
|
| 29 |
+
simple:
|
| 30 |
+
act: silu
|
| 31 |
+
depth: 64
|
| 32 |
+
kernel: 5
|
| 33 |
+
layers: 3
|
| 34 |
+
mults: [2, 3, 4, 4]
|
| 35 |
+
norm: rms
|
| 36 |
+
outer: false
|
| 37 |
+
strided: false
|
| 38 |
+
symlog: true
|
| 39 |
+
units: 1024
|
| 40 |
+
winit: trunc_normal_in
|
| 41 |
+
typ: simple
|
| 42 |
+
horizon: 333
|
| 43 |
+
imag_last: 0
|
| 44 |
+
imag_length: 15
|
| 45 |
+
imag_loss: {actent: 0.0003, lam: 0.95, slowreg: 1.0, slowtar: false}
|
| 46 |
+
loss_scales: {con: 1.0, dyn: 1.0, policy: 1.0, rec: 1.0, rep: 0.1, repval: 0.3,
|
| 47 |
+
rew: 1.0, value: 1.0}
|
| 48 |
+
opt: {agc: 0.3, anneal: 0, beta1: 0.9, beta2: 0.999, eps: 1e-20, lr: 4e-05,
|
| 49 |
+
momentum: true, schedule: const, warmup: 1000, wd: 0.0}
|
| 50 |
+
policy: {act: silu, layers: 3, maxstd: 1.0, minstd: 0.1, norm: rms, outscale:
|
| 51 |
+
0.01, unimix: 0.01, units: 1024, winit: trunc_normal_in}
|
| 52 |
+
policy_dist_cont: bounded_normal
|
| 53 |
+
policy_dist_disc: categorical
|
| 54 |
+
repl_loss: {lam: 0.95, slowreg: 1.0, slowtar: false}
|
| 55 |
+
report: true
|
| 56 |
+
report_gradnorms: false
|
| 57 |
+
repval_grad: true
|
| 58 |
+
repval_loss: true
|
| 59 |
+
retnorm: {debias: false, impl: perc, limit: 1.0, perchi: 95.0, perclo: 5.0,
|
| 60 |
+
rate: 0.01}
|
| 61 |
+
reward_grad: true
|
| 62 |
+
rewhead: {act: silu, bins: 255, layers: 1, norm: rms, output: symexp_twohot,
|
| 63 |
+
outscale: 0.0, units: 1024, winit: trunc_normal_in}
|
| 64 |
+
slowvalue: {every: 1, rate: 0.02}
|
| 65 |
+
valnorm: {impl: none, limit: 1e-08, rate: 0.01}
|
| 66 |
+
value: {act: silu, bins: 255, layers: 3, norm: rms, output: symexp_twohot,
|
| 67 |
+
outscale: 0.0, units: 1024, winit: trunc_normal_in}
|
| 68 |
+
batch_length: 64
|
| 69 |
+
batch_size: 16
|
| 70 |
+
clock_addr: ''
|
| 71 |
+
clock_port: ''
|
| 72 |
+
consec_report: 1
|
| 73 |
+
consec_train: 1
|
| 74 |
+
env:
|
| 75 |
+
atari:
|
| 76 |
+
actions: all
|
| 77 |
+
aggregate: max
|
| 78 |
+
autostart: false
|
| 79 |
+
clip_reward: false
|
| 80 |
+
gray: true
|
| 81 |
+
lives: unused
|
| 82 |
+
noops: 30
|
| 83 |
+
pooling: 2
|
| 84 |
+
repeat: 4
|
| 85 |
+
resize: pillow
|
| 86 |
+
size: [96, 96]
|
| 87 |
+
sticky: true
|
| 88 |
+
atari100k:
|
| 89 |
+
actions: needed
|
| 90 |
+
autostart: false
|
| 91 |
+
clip_reward: false
|
| 92 |
+
gray: false
|
| 93 |
+
lives: unused
|
| 94 |
+
noops: 30
|
| 95 |
+
repeat: 4
|
| 96 |
+
resize: pillow
|
| 97 |
+
size: [64, 64]
|
| 98 |
+
sticky: false
|
| 99 |
+
crafter:
|
| 100 |
+
logs: false
|
| 101 |
+
size: [64, 64]
|
| 102 |
+
dmc:
|
| 103 |
+
camera: -1
|
| 104 |
+
image: true
|
| 105 |
+
proprio: false
|
| 106 |
+
repeat: 1
|
| 107 |
+
size: [64, 64]
|
| 108 |
+
dmlab:
|
| 109 |
+
episodic: true
|
| 110 |
+
repeat: 4
|
| 111 |
+
size: [64, 64]
|
| 112 |
+
use_seed: true
|
| 113 |
+
loconav:
|
| 114 |
+
camera: -1
|
| 115 |
+
repeat: 1
|
| 116 |
+
size: [64, 64]
|
| 117 |
+
minecraft:
|
| 118 |
+
break_speed: 100.0
|
| 119 |
+
length: 36000
|
| 120 |
+
logs: false
|
| 121 |
+
size: [64, 64]
|
| 122 |
+
procgen:
|
| 123 |
+
resize: pillow
|
| 124 |
+
size: [96, 96]
|
| 125 |
+
errfile: false
|
| 126 |
+
ipv6: false
|
| 127 |
+
jax:
|
| 128 |
+
compute_dtype: bfloat16
|
| 129 |
+
coordinator_address: ''
|
| 130 |
+
debug: false
|
| 131 |
+
enable_policy: true
|
| 132 |
+
expect_devices: 0
|
| 133 |
+
jit: true
|
| 134 |
+
mock_devices: 0
|
| 135 |
+
platform: cuda
|
| 136 |
+
policy_devices: [0]
|
| 137 |
+
prealloc: true
|
| 138 |
+
train_devices: [0]
|
| 139 |
+
logdir: ./logdir/dreamer/cheetah_run/slurm-23188
|
| 140 |
+
logger:
|
| 141 |
+
filter: score|length|fps|ratio|train/loss/|train/rand/
|
| 142 |
+
fps: 15
|
| 143 |
+
outputs: [jsonl, scope]
|
| 144 |
+
timer: true
|
| 145 |
+
user: ''
|
| 146 |
+
method: name
|
| 147 |
+
random_agent: false
|
| 148 |
+
replay:
|
| 149 |
+
chunksize: 1024
|
| 150 |
+
fracs: {priority: 0.0, recency: 0.0, uniform: 1.0}
|
| 151 |
+
online: true
|
| 152 |
+
prio: {exponent: 0.8, initial: inf, maxfrac: 0.5, zero_on_sample: true}
|
| 153 |
+
priosignal: model
|
| 154 |
+
recexp: 1.0
|
| 155 |
+
size: 5000000.0
|
| 156 |
+
replay_context: 1
|
| 157 |
+
replica: 0
|
| 158 |
+
replicas: 1
|
| 159 |
+
report_length: 32
|
| 160 |
+
run:
|
| 161 |
+
actor_addr: localhost:{auto}
|
| 162 |
+
actor_batch: -1
|
| 163 |
+
actor_threads: 1
|
| 164 |
+
agent_process: false
|
| 165 |
+
debug: true
|
| 166 |
+
duration: 0
|
| 167 |
+
envs: 16
|
| 168 |
+
episode_timeout: 180
|
| 169 |
+
eval_envs: 4
|
| 170 |
+
eval_eps: 1
|
| 171 |
+
from_checkpoint: ''
|
| 172 |
+
log_every: 120
|
| 173 |
+
logger_addr: localhost:{auto}
|
| 174 |
+
remote_envs: false
|
| 175 |
+
remote_replay: false
|
| 176 |
+
replay_addr: localhost:{auto}
|
| 177 |
+
report_batches: 1
|
| 178 |
+
report_every: 300
|
| 179 |
+
save_every: 900
|
| 180 |
+
steps: 1100000.0
|
| 181 |
+
train_ratio: 256.0
|
| 182 |
+
usage: {gc: false, gputil: false, malloc: false, nvsmi: true, psutil: true}
|
| 183 |
+
script: train
|
| 184 |
+
seed: 0
|
| 185 |
+
task: dmc_cheetah_run
|
dreamerv3/reacher_easy/ckpt/20260830T195435F325934/agent.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a1fb60436f02fcfa6ba1fbd0b5b5078142cb6f6c34b13c42e4a5967ef047b394
|
| 3 |
+
size 2042658588
|
dreamerv3/reacher_easy/ckpt/20260830T195435F325934/done
ADDED
|
File without changes
|
dreamerv3/reacher_easy/ckpt/20260830T195435F325934/step.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7e5c5685c2d39badee13ea957052bbe8a3f71e577e2369e55151ad875842cb2f
|
| 3 |
+
size 17
|
dreamerv3/reacher_easy/config.yaml
ADDED
|
@@ -0,0 +1,185 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
agent:
|
| 2 |
+
ac_grads: false
|
| 3 |
+
advnorm: {impl: none, limit: 1e-08, rate: 0.01}
|
| 4 |
+
conhead: {act: silu, layers: 1, norm: rms, output: binary, outscale: 1.0,
|
| 5 |
+
units: 1024, winit: trunc_normal_in}
|
| 6 |
+
contdisc: true
|
| 7 |
+
dec:
|
| 8 |
+
simple:
|
| 9 |
+
act: silu
|
| 10 |
+
bspace: 8
|
| 11 |
+
depth: 64
|
| 12 |
+
kernel: 5
|
| 13 |
+
layers: 3
|
| 14 |
+
mults: [2, 3, 4, 4]
|
| 15 |
+
norm: rms
|
| 16 |
+
outer: false
|
| 17 |
+
outscale: 1.0
|
| 18 |
+
strided: false
|
| 19 |
+
units: 1024
|
| 20 |
+
winit: trunc_normal_in
|
| 21 |
+
typ: simple
|
| 22 |
+
dyn:
|
| 23 |
+
rssm: {absolute: false, act: silu, blocks: 8, classes: 64, deter: 8192,
|
| 24 |
+
dynlayers: 1, free_nats: 1.0, hidden: 1024, imglayers: 2, norm: rms,
|
| 25 |
+
obslayers: 1, outscale: 1.0, stoch: 32, unimix: 0.01, winit:
|
| 26 |
+
trunc_normal_in}
|
| 27 |
+
typ: rssm
|
| 28 |
+
enc:
|
| 29 |
+
simple:
|
| 30 |
+
act: silu
|
| 31 |
+
depth: 64
|
| 32 |
+
kernel: 5
|
| 33 |
+
layers: 3
|
| 34 |
+
mults: [2, 3, 4, 4]
|
| 35 |
+
norm: rms
|
| 36 |
+
outer: false
|
| 37 |
+
strided: false
|
| 38 |
+
symlog: true
|
| 39 |
+
units: 1024
|
| 40 |
+
winit: trunc_normal_in
|
| 41 |
+
typ: simple
|
| 42 |
+
horizon: 333
|
| 43 |
+
imag_last: 0
|
| 44 |
+
imag_length: 15
|
| 45 |
+
imag_loss: {actent: 0.0003, lam: 0.95, slowreg: 1.0, slowtar: false}
|
| 46 |
+
loss_scales: {con: 1.0, dyn: 1.0, policy: 1.0, rec: 1.0, rep: 0.1, repval: 0.3,
|
| 47 |
+
rew: 1.0, value: 1.0}
|
| 48 |
+
opt: {agc: 0.3, anneal: 0, beta1: 0.9, beta2: 0.999, eps: 1e-20, lr: 4e-05,
|
| 49 |
+
momentum: true, schedule: const, warmup: 1000, wd: 0.0}
|
| 50 |
+
policy: {act: silu, layers: 3, maxstd: 1.0, minstd: 0.1, norm: rms, outscale:
|
| 51 |
+
0.01, unimix: 0.01, units: 1024, winit: trunc_normal_in}
|
| 52 |
+
policy_dist_cont: bounded_normal
|
| 53 |
+
policy_dist_disc: categorical
|
| 54 |
+
repl_loss: {lam: 0.95, slowreg: 1.0, slowtar: false}
|
| 55 |
+
report: true
|
| 56 |
+
report_gradnorms: false
|
| 57 |
+
repval_grad: true
|
| 58 |
+
repval_loss: true
|
| 59 |
+
retnorm: {debias: false, impl: perc, limit: 1.0, perchi: 95.0, perclo: 5.0,
|
| 60 |
+
rate: 0.01}
|
| 61 |
+
reward_grad: true
|
| 62 |
+
rewhead: {act: silu, bins: 255, layers: 1, norm: rms, output: symexp_twohot,
|
| 63 |
+
outscale: 0.0, units: 1024, winit: trunc_normal_in}
|
| 64 |
+
slowvalue: {every: 1, rate: 0.02}
|
| 65 |
+
valnorm: {impl: none, limit: 1e-08, rate: 0.01}
|
| 66 |
+
value: {act: silu, bins: 255, layers: 3, norm: rms, output: symexp_twohot,
|
| 67 |
+
outscale: 0.0, units: 1024, winit: trunc_normal_in}
|
| 68 |
+
batch_length: 64
|
| 69 |
+
batch_size: 16
|
| 70 |
+
clock_addr: ''
|
| 71 |
+
clock_port: ''
|
| 72 |
+
consec_report: 1
|
| 73 |
+
consec_train: 1
|
| 74 |
+
env:
|
| 75 |
+
atari:
|
| 76 |
+
actions: all
|
| 77 |
+
aggregate: max
|
| 78 |
+
autostart: false
|
| 79 |
+
clip_reward: false
|
| 80 |
+
gray: true
|
| 81 |
+
lives: unused
|
| 82 |
+
noops: 30
|
| 83 |
+
pooling: 2
|
| 84 |
+
repeat: 4
|
| 85 |
+
resize: pillow
|
| 86 |
+
size: [96, 96]
|
| 87 |
+
sticky: true
|
| 88 |
+
atari100k:
|
| 89 |
+
actions: needed
|
| 90 |
+
autostart: false
|
| 91 |
+
clip_reward: false
|
| 92 |
+
gray: false
|
| 93 |
+
lives: unused
|
| 94 |
+
noops: 30
|
| 95 |
+
repeat: 4
|
| 96 |
+
resize: pillow
|
| 97 |
+
size: [64, 64]
|
| 98 |
+
sticky: false
|
| 99 |
+
crafter:
|
| 100 |
+
logs: false
|
| 101 |
+
size: [64, 64]
|
| 102 |
+
dmc:
|
| 103 |
+
camera: -1
|
| 104 |
+
image: true
|
| 105 |
+
proprio: false
|
| 106 |
+
repeat: 1
|
| 107 |
+
size: [64, 64]
|
| 108 |
+
dmlab:
|
| 109 |
+
episodic: true
|
| 110 |
+
repeat: 4
|
| 111 |
+
size: [64, 64]
|
| 112 |
+
use_seed: true
|
| 113 |
+
loconav:
|
| 114 |
+
camera: -1
|
| 115 |
+
repeat: 1
|
| 116 |
+
size: [64, 64]
|
| 117 |
+
minecraft:
|
| 118 |
+
break_speed: 100.0
|
| 119 |
+
length: 36000
|
| 120 |
+
logs: false
|
| 121 |
+
size: [64, 64]
|
| 122 |
+
procgen:
|
| 123 |
+
resize: pillow
|
| 124 |
+
size: [96, 96]
|
| 125 |
+
errfile: false
|
| 126 |
+
ipv6: false
|
| 127 |
+
jax:
|
| 128 |
+
compute_dtype: bfloat16
|
| 129 |
+
coordinator_address: ''
|
| 130 |
+
debug: false
|
| 131 |
+
enable_policy: true
|
| 132 |
+
expect_devices: 0
|
| 133 |
+
jit: true
|
| 134 |
+
mock_devices: 0
|
| 135 |
+
platform: cuda
|
| 136 |
+
policy_devices: [0]
|
| 137 |
+
prealloc: true
|
| 138 |
+
train_devices: [0]
|
| 139 |
+
logdir: ./logdir/dreamer/reacher_easy/20260830T125224
|
| 140 |
+
logger:
|
| 141 |
+
filter: score|length|fps|ratio|train/loss/|train/rand/
|
| 142 |
+
fps: 15
|
| 143 |
+
outputs: [jsonl, scope]
|
| 144 |
+
timer: true
|
| 145 |
+
user: ''
|
| 146 |
+
method: name
|
| 147 |
+
random_agent: false
|
| 148 |
+
replay:
|
| 149 |
+
chunksize: 1024
|
| 150 |
+
fracs: {priority: 0.0, recency: 0.0, uniform: 1.0}
|
| 151 |
+
online: true
|
| 152 |
+
prio: {exponent: 0.8, initial: inf, maxfrac: 0.5, zero_on_sample: true}
|
| 153 |
+
priosignal: model
|
| 154 |
+
recexp: 1.0
|
| 155 |
+
size: 5000000.0
|
| 156 |
+
replay_context: 1
|
| 157 |
+
replica: 0
|
| 158 |
+
replicas: 1
|
| 159 |
+
report_length: 32
|
| 160 |
+
run:
|
| 161 |
+
actor_addr: localhost:{auto}
|
| 162 |
+
actor_batch: -1
|
| 163 |
+
actor_threads: 1
|
| 164 |
+
agent_process: false
|
| 165 |
+
debug: true
|
| 166 |
+
duration: 0
|
| 167 |
+
envs: 16
|
| 168 |
+
episode_timeout: 180
|
| 169 |
+
eval_envs: 4
|
| 170 |
+
eval_eps: 1
|
| 171 |
+
from_checkpoint: ''
|
| 172 |
+
log_every: 120
|
| 173 |
+
logger_addr: localhost:{auto}
|
| 174 |
+
remote_envs: false
|
| 175 |
+
remote_replay: false
|
| 176 |
+
replay_addr: localhost:{auto}
|
| 177 |
+
report_batches: 1
|
| 178 |
+
report_every: 300
|
| 179 |
+
save_every: 900
|
| 180 |
+
steps: 1100000.0
|
| 181 |
+
train_ratio: 256.0
|
| 182 |
+
usage: {gc: false, gputil: false, malloc: false, nvsmi: true, psutil: true}
|
| 183 |
+
script: train
|
| 184 |
+
seed: 0
|
| 185 |
+
task: dmc_reacher_easy
|
dreamerv3/runtime.json
ADDED
|
@@ -0,0 +1,126 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"python": "3.12.14",
|
| 3 |
+
"platform": "Linux x86_64",
|
| 4 |
+
"packages": {
|
| 5 |
+
"nvidia-cuda-nvcc-cu12": "12.9.86",
|
| 6 |
+
"google-resumable-media": "2.10.2",
|
| 7 |
+
"google-crc32c": "1.8.0",
|
| 8 |
+
"scope": "0.7.1",
|
| 9 |
+
"uvicorn": "0.52.4",
|
| 10 |
+
"websockets": "17.1",
|
| 11 |
+
"watchfiles": "1.2.0",
|
| 12 |
+
"anyio": "4.14.2",
|
| 13 |
+
"typing-extensions": "4.16.0",
|
| 14 |
+
"idna": "3.19",
|
| 15 |
+
"uvloop": "0.22.1",
|
| 16 |
+
"pyyaml": "6.0.3",
|
| 17 |
+
"python-dotenv": "1.2.3",
|
| 18 |
+
"httptools": "0.8.0",
|
| 19 |
+
"h11": "0.16.0",
|
| 20 |
+
"click": "8.5.0",
|
| 21 |
+
"pillow": "12.3.0",
|
| 22 |
+
"orjson": "3.12.0",
|
| 23 |
+
"numpy": "1.26.4",
|
| 24 |
+
"mediapy": "1.2.7",
|
| 25 |
+
"matplotlib": "3.11.1",
|
| 26 |
+
"python-dateutil": "2.9.0.post0",
|
| 27 |
+
"six": "1.17.0",
|
| 28 |
+
"pyparsing": "3.3.2",
|
| 29 |
+
"packaging": "26.3",
|
| 30 |
+
"kiwisolver": "1.5.1",
|
| 31 |
+
"fonttools": "4.63.0",
|
| 32 |
+
"cycler": "0.12.1",
|
| 33 |
+
"contourpy": "1.3.3",
|
| 34 |
+
"ipython": "9.17.0",
|
| 35 |
+
"traitlets": "5.16.1",
|
| 36 |
+
"stack-data": "0.6.3",
|
| 37 |
+
"pure-eval": "0.2.3",
|
| 38 |
+
"asttokens": "3.0.2",
|
| 39 |
+
"executing": "2.2.1",
|
| 40 |
+
"pygments": "2.21.0",
|
| 41 |
+
"psutil": "7.2.2",
|
| 42 |
+
"prompt-toolkit": "3.0.53",
|
| 43 |
+
"wcwidth": "0.8.3",
|
| 44 |
+
"pexpect": "4.9.0",
|
| 45 |
+
"ptyprocess": "0.7.0",
|
| 46 |
+
"matplotlib-inline": "0.2.2",
|
| 47 |
+
"jedi": "0.20.0",
|
| 48 |
+
"parso": "0.8.7",
|
| 49 |
+
"ipython-pygments-lexers": "1.1.1",
|
| 50 |
+
"filelock": "3.32.4",
|
| 51 |
+
"fastapi": "0.141.1",
|
| 52 |
+
"annotated-doc": "0.0.5",
|
| 53 |
+
"typing-inspection": "0.4.4",
|
| 54 |
+
"pydantic": "2.13.5",
|
| 55 |
+
"pydantic-core": "2.46.5",
|
| 56 |
+
"annotated-types": "0.8.0",
|
| 57 |
+
"starlette": "1.6.0",
|
| 58 |
+
"elements": "3.22.0",
|
| 59 |
+
"ruamel-yaml": "0.19.1",
|
| 60 |
+
"google-cloud-storage": "3.13.1",
|
| 61 |
+
"requests": "2.34.2",
|
| 62 |
+
"certifi": "2026.7.22",
|
| 63 |
+
"urllib3": "2.7.0",
|
| 64 |
+
"charset-normalizer": "3.5.1",
|
| 65 |
+
"google-cloud-core": "2.7.0",
|
| 66 |
+
"google-auth": "2.57.0",
|
| 67 |
+
"cryptography": "50.0.1",
|
| 68 |
+
"cffi": "2.1.1",
|
| 69 |
+
"pycparser": "3.0",
|
| 70 |
+
"pyasn1-modules": "0.4.2",
|
| 71 |
+
"pyasn1": "0.6.4",
|
| 72 |
+
"google-api-core": "2.34.0",
|
| 73 |
+
"proto-plus": "1.28.4",
|
| 74 |
+
"protobuf": "7.36.0",
|
| 75 |
+
"googleapis-common-protos": "1.75.2",
|
| 76 |
+
"av": "18.1.0",
|
| 77 |
+
"jaxtyping": "0.3.11",
|
| 78 |
+
"wadler-lindig": "0.1.7",
|
| 79 |
+
"colored-traceback": "0.4.2",
|
| 80 |
+
"cloudpickle": "3.1.2",
|
| 81 |
+
"mujoco": "3.12.0",
|
| 82 |
+
"pyopengl": "3.1.10",
|
| 83 |
+
"glfw": "2.10.2",
|
| 84 |
+
"etils": "1.14.0",
|
| 85 |
+
"zipp": "4.1.0",
|
| 86 |
+
"fsspec": "2026.7.0",
|
| 87 |
+
"absl-py": "2.5.0",
|
| 88 |
+
"dm-control": "1.0.45",
|
| 89 |
+
"tqdm": "4.70.0",
|
| 90 |
+
"scipy": "1.17.1",
|
| 91 |
+
"setuptools": "83.0.0",
|
| 92 |
+
"lxml": "6.1.2",
|
| 93 |
+
"labmaze": "1.0.6",
|
| 94 |
+
"dm-tree": "0.1.10",
|
| 95 |
+
"wrapt": "2.3.0",
|
| 96 |
+
"attrs": "26.1.0",
|
| 97 |
+
"dm-env": "1.6",
|
| 98 |
+
"einops": "0.8.2",
|
| 99 |
+
"chex": "0.1.90",
|
| 100 |
+
"toolz": "1.1.0",
|
| 101 |
+
"jaxlib": "0.7.1",
|
| 102 |
+
"ml-dtypes": "0.5.4",
|
| 103 |
+
"jax": "0.7.1",
|
| 104 |
+
"opt-einsum": "3.4.0",
|
| 105 |
+
"optax": "0.2.5",
|
| 106 |
+
"granular": "0.23.3",
|
| 107 |
+
"msgpack": "1.2.2",
|
| 108 |
+
"portal": "3.8.1",
|
| 109 |
+
"ninjax": "3.6.3",
|
| 110 |
+
"jax-cuda12-plugin": "0.7.1",
|
| 111 |
+
"nvidia-nvshmem-cu12": "3.7.2",
|
| 112 |
+
"nvidia-cuda-cccl-cu12": "12.9.27",
|
| 113 |
+
"nvidia-cuda-nvrtc-cu12": "12.9.86",
|
| 114 |
+
"nvidia-nvjitlink-cu12": "12.9.86",
|
| 115 |
+
"nvidia-nccl-cu12": "2.31.2",
|
| 116 |
+
"nvidia-cusparse-cu12": "12.5.10.65",
|
| 117 |
+
"nvidia-cusolver-cu12": "11.7.5.82",
|
| 118 |
+
"nvidia-cublas-cu12": "12.9.2.10",
|
| 119 |
+
"nvidia-cufft-cu12": "11.4.1.4",
|
| 120 |
+
"nvidia-cudnn-cu12": "9.25.1.1",
|
| 121 |
+
"nvidia-cuda-runtime-cu12": "12.9.79",
|
| 122 |
+
"nvidia-cuda-cupti-cu12": "12.9.79",
|
| 123 |
+
"jax-cuda12-pjrt": "0.7.1"
|
| 124 |
+
},
|
| 125 |
+
"metadata_conflicts": []
|
| 126 |
+
}
|
le-wm/cube/config.json
ADDED
|
@@ -0,0 +1,49 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_target_": "stable_worldmodel.wm.lewm.LeWM",
|
| 3 |
+
"encoder": {
|
| 4 |
+
"_target_": "stable_pretraining.backbone.utils.vit_hf",
|
| 5 |
+
"size": "tiny",
|
| 6 |
+
"patch_size": 14,
|
| 7 |
+
"image_size": 224,
|
| 8 |
+
"pretrained": false,
|
| 9 |
+
"use_mask_token": false
|
| 10 |
+
},
|
| 11 |
+
"predictor": {
|
| 12 |
+
"_target_": "stable_worldmodel.wm.lewm.module.Predictor",
|
| 13 |
+
"num_frames": 3,
|
| 14 |
+
"input_dim": 192,
|
| 15 |
+
"hidden_dim": 192,
|
| 16 |
+
"output_dim": 192,
|
| 17 |
+
"depth": 6,
|
| 18 |
+
"heads": 16,
|
| 19 |
+
"mlp_dim": 2048,
|
| 20 |
+
"dim_head": 64,
|
| 21 |
+
"dropout": 0.1,
|
| 22 |
+
"emb_dropout": 0.0
|
| 23 |
+
},
|
| 24 |
+
"action_encoder": {
|
| 25 |
+
"_target_": "stable_worldmodel.wm.lewm.module.Embedder",
|
| 26 |
+
"input_dim": 25,
|
| 27 |
+
"emb_dim": 192
|
| 28 |
+
},
|
| 29 |
+
"projector": {
|
| 30 |
+
"_target_": "stable_worldmodel.wm.lewm.module.MLP",
|
| 31 |
+
"input_dim": 192,
|
| 32 |
+
"output_dim": 192,
|
| 33 |
+
"hidden_dim": 2048,
|
| 34 |
+
"norm_fn": {
|
| 35 |
+
"_target_": "torch.nn.BatchNorm1d",
|
| 36 |
+
"_partial_": true
|
| 37 |
+
}
|
| 38 |
+
},
|
| 39 |
+
"pred_proj": {
|
| 40 |
+
"_target_": "stable_worldmodel.wm.lewm.module.MLP",
|
| 41 |
+
"input_dim": 192,
|
| 42 |
+
"output_dim": 192,
|
| 43 |
+
"hidden_dim": 2048,
|
| 44 |
+
"norm_fn": {
|
| 45 |
+
"_target_": "torch.nn.BatchNorm1d",
|
| 46 |
+
"_partial_": true
|
| 47 |
+
}
|
| 48 |
+
}
|
| 49 |
+
}
|
le-wm/cube/weights.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2839a907362f403f9136383016e91774373a295d958ae75121791f22a9fddf89
|
| 3 |
+
size 72291425
|
le-wm/defenses/cube/paper_hgd/hgd_paper.best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:04ed4a9b4d9edea42f4e4ec7b489ae79378890a3c2d505077a97d69d19fd1ec9
|
| 3 |
+
size 132576538
|
le-wm/defenses/cube/paper_hgd/hgd_paper.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9ebae97be8b61cefde85958dadea28d5ae34c959b3cd183f4932f6bfe1debdf2
|
| 3 |
+
size 132574859
|
le-wm/defenses/cube/residual_hgd/hgd_eps16.best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:68f58d28e6b39954331143c2f66e736f402882f15fce35ace8afc8d464c17f07
|
| 3 |
+
size 5917691
|
le-wm/defenses/cube/residual_hgd/hgd_eps16.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bc3af72c1ae2ed328c8907f87e5709b719e00c8dd61e7c32ccb9dc722951772a
|
| 3 |
+
size 5916747
|
le-wm/pusht/config.json
ADDED
|
@@ -0,0 +1,49 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_target_": "stable_worldmodel.wm.lewm.LeWM",
|
| 3 |
+
"encoder": {
|
| 4 |
+
"_target_": "stable_pretraining.backbone.utils.vit_hf",
|
| 5 |
+
"size": "tiny",
|
| 6 |
+
"patch_size": 14,
|
| 7 |
+
"image_size": 224,
|
| 8 |
+
"pretrained": false,
|
| 9 |
+
"use_mask_token": false
|
| 10 |
+
},
|
| 11 |
+
"predictor": {
|
| 12 |
+
"_target_": "stable_worldmodel.wm.lewm.module.Predictor",
|
| 13 |
+
"num_frames": 3,
|
| 14 |
+
"input_dim": 192,
|
| 15 |
+
"hidden_dim": 192,
|
| 16 |
+
"output_dim": 192,
|
| 17 |
+
"depth": 6,
|
| 18 |
+
"heads": 16,
|
| 19 |
+
"mlp_dim": 2048,
|
| 20 |
+
"dim_head": 64,
|
| 21 |
+
"dropout": 0.1,
|
| 22 |
+
"emb_dropout": 0.0
|
| 23 |
+
},
|
| 24 |
+
"action_encoder": {
|
| 25 |
+
"_target_": "stable_worldmodel.wm.lewm.module.Embedder",
|
| 26 |
+
"input_dim": 10,
|
| 27 |
+
"emb_dim": 192
|
| 28 |
+
},
|
| 29 |
+
"projector": {
|
| 30 |
+
"_target_": "stable_worldmodel.wm.lewm.module.MLP",
|
| 31 |
+
"input_dim": 192,
|
| 32 |
+
"output_dim": 192,
|
| 33 |
+
"hidden_dim": 2048,
|
| 34 |
+
"norm_fn": {
|
| 35 |
+
"_target_": "torch.nn.BatchNorm1d",
|
| 36 |
+
"_partial_": true
|
| 37 |
+
}
|
| 38 |
+
},
|
| 39 |
+
"pred_proj": {
|
| 40 |
+
"_target_": "stable_worldmodel.wm.lewm.module.MLP",
|
| 41 |
+
"input_dim": 192,
|
| 42 |
+
"output_dim": 192,
|
| 43 |
+
"hidden_dim": 2048,
|
| 44 |
+
"norm_fn": {
|
| 45 |
+
"_target_": "torch.nn.BatchNorm1d",
|
| 46 |
+
"_partial_": true
|
| 47 |
+
}
|
| 48 |
+
}
|
| 49 |
+
}
|
le-wm/pusht/weights.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:48938400ae3464c9680731287f583a9cb516f55a8ec64ea13a91be47fb15b607
|
| 3 |
+
size 72290721
|
le-wm/reacher/config.json
ADDED
|
@@ -0,0 +1,49 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_target_": "stable_worldmodel.wm.lewm.LeWM",
|
| 3 |
+
"encoder": {
|
| 4 |
+
"_target_": "stable_pretraining.backbone.utils.vit_hf",
|
| 5 |
+
"size": "tiny",
|
| 6 |
+
"patch_size": 14,
|
| 7 |
+
"image_size": 224,
|
| 8 |
+
"pretrained": false,
|
| 9 |
+
"use_mask_token": false
|
| 10 |
+
},
|
| 11 |
+
"predictor": {
|
| 12 |
+
"_target_": "stable_worldmodel.wm.lewm.module.Predictor",
|
| 13 |
+
"num_frames": 3,
|
| 14 |
+
"input_dim": 192,
|
| 15 |
+
"hidden_dim": 192,
|
| 16 |
+
"output_dim": 192,
|
| 17 |
+
"depth": 6,
|
| 18 |
+
"heads": 16,
|
| 19 |
+
"mlp_dim": 2048,
|
| 20 |
+
"dim_head": 64,
|
| 21 |
+
"dropout": 0.1,
|
| 22 |
+
"emb_dropout": 0.0
|
| 23 |
+
},
|
| 24 |
+
"action_encoder": {
|
| 25 |
+
"_target_": "stable_worldmodel.wm.lewm.module.Embedder",
|
| 26 |
+
"input_dim": 10,
|
| 27 |
+
"emb_dim": 192
|
| 28 |
+
},
|
| 29 |
+
"projector": {
|
| 30 |
+
"_target_": "stable_worldmodel.wm.lewm.module.MLP",
|
| 31 |
+
"input_dim": 192,
|
| 32 |
+
"output_dim": 192,
|
| 33 |
+
"hidden_dim": 2048,
|
| 34 |
+
"norm_fn": {
|
| 35 |
+
"_target_": "torch.nn.BatchNorm1d",
|
| 36 |
+
"_partial_": true
|
| 37 |
+
}
|
| 38 |
+
},
|
| 39 |
+
"pred_proj": {
|
| 40 |
+
"_target_": "stable_worldmodel.wm.lewm.module.MLP",
|
| 41 |
+
"input_dim": 192,
|
| 42 |
+
"output_dim": 192,
|
| 43 |
+
"hidden_dim": 2048,
|
| 44 |
+
"norm_fn": {
|
| 45 |
+
"_target_": "torch.nn.BatchNorm1d",
|
| 46 |
+
"_partial_": true
|
| 47 |
+
}
|
| 48 |
+
}
|
| 49 |
+
}
|
le-wm/reacher/weights.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:eb70b1fd5409f8f81875d62f5ee5a20dd220a3128a477de66b5760f475f0f469
|
| 3 |
+
size 72290849
|
le-wm/runtime.json
ADDED
|
@@ -0,0 +1,190 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"python": "3.10.20",
|
| 3 |
+
"platform": "Linux x86_64",
|
| 4 |
+
"packages": {
|
| 5 |
+
"einops": "0.8.2",
|
| 6 |
+
"huggingface-hub": "0.36.2",
|
| 7 |
+
"typing-extensions": "4.16.0",
|
| 8 |
+
"tqdm": "4.70.0",
|
| 9 |
+
"requests": "2.34.2",
|
| 10 |
+
"certifi": "2026.7.22",
|
| 11 |
+
"urllib3": "2.7.0",
|
| 12 |
+
"idna": "3.18",
|
| 13 |
+
"charset-normalizer": "3.4.9",
|
| 14 |
+
"pyyaml": "6.0.3",
|
| 15 |
+
"packaging": "26.2",
|
| 16 |
+
"hf-xet": "1.5.2",
|
| 17 |
+
"fsspec": "2025.10.0",
|
| 18 |
+
"filelock": "3.32.0",
|
| 19 |
+
"hydra-core": "1.3.4",
|
| 20 |
+
"antlr4-python3-runtime": "4.9.3",
|
| 21 |
+
"omegaconf": "2.3.1",
|
| 22 |
+
"h5py": "3.16.0",
|
| 23 |
+
"numpy": "2.2.6",
|
| 24 |
+
"matplotlib": "3.10.9",
|
| 25 |
+
"python-dateutil": "2.9.0.post0",
|
| 26 |
+
"six": "1.17.0",
|
| 27 |
+
"pyparsing": "3.3.2",
|
| 28 |
+
"pillow": "11.3.0",
|
| 29 |
+
"kiwisolver": "1.5.0",
|
| 30 |
+
"fonttools": "4.63.0",
|
| 31 |
+
"cycler": "0.12.1",
|
| 32 |
+
"contourpy": "1.3.2",
|
| 33 |
+
"imageio-ffmpeg": "0.6.0",
|
| 34 |
+
"imageio": "2.37.4",
|
| 35 |
+
"scipy": "1.15.3",
|
| 36 |
+
"torchvision": "0.28.0",
|
| 37 |
+
"torch": "2.13.0",
|
| 38 |
+
"triton": "3.7.1",
|
| 39 |
+
"nvidia-nvshmem-cu13": "3.4.5",
|
| 40 |
+
"nvidia-nccl-cu13": "2.29.7",
|
| 41 |
+
"nvidia-cusparselt-cu13": "0.8.1",
|
| 42 |
+
"nvidia-cudnn-cu13": "9.20.0.48",
|
| 43 |
+
"nvidia-cublas": "13.1.1.3",
|
| 44 |
+
"nvidia-cuda-nvrtc": "13.0.88",
|
| 45 |
+
"cuda-bindings": "13.3.1",
|
| 46 |
+
"cuda-pathfinder": "1.6.0",
|
| 47 |
+
"cuda-toolkit": "13.0.3.0",
|
| 48 |
+
"nvidia-nvtx": "13.0.85",
|
| 49 |
+
"nvidia-nvjitlink": "13.3.33",
|
| 50 |
+
"nvidia-cusparse": "12.6.3.3",
|
| 51 |
+
"nvidia-cusolver": "12.0.4.66",
|
| 52 |
+
"nvidia-curand": "10.4.0.35",
|
| 53 |
+
"nvidia-cuda-cupti": "13.0.85",
|
| 54 |
+
"nvidia-cufile": "1.15.1.6",
|
| 55 |
+
"nvidia-cufft": "12.0.0.61",
|
| 56 |
+
"nvidia-cuda-runtime": "13.0.96",
|
| 57 |
+
"jinja2": "3.1.6",
|
| 58 |
+
"markupsafe": "3.0.3",
|
| 59 |
+
"networkx": "3.4.2",
|
| 60 |
+
"sympy": "1.14.0",
|
| 61 |
+
"mpmath": "1.3.0",
|
| 62 |
+
"setuptools": "83.0.0",
|
| 63 |
+
"stable-worldmodel": "0.1.1",
|
| 64 |
+
"wandb": "0.28.1",
|
| 65 |
+
"sentry-sdk": "2.66.1",
|
| 66 |
+
"pydantic": "2.13.4",
|
| 67 |
+
"typing-inspection": "0.4.2",
|
| 68 |
+
"pydantic-core": "2.46.4",
|
| 69 |
+
"annotated-types": "0.8.0",
|
| 70 |
+
"protobuf": "7.35.1",
|
| 71 |
+
"platformdirs": "4.11.0",
|
| 72 |
+
"click": "8.4.2",
|
| 73 |
+
"hydra-submitit-launcher": "1.2.0",
|
| 74 |
+
"submitit": "1.5.4",
|
| 75 |
+
"cloudpickle": "3.1.2",
|
| 76 |
+
"stable-pretraining": "0.1.7",
|
| 77 |
+
"opencv-python-headless": "5.0.0.93",
|
| 78 |
+
"pylance": "9.0.0",
|
| 79 |
+
"lance-namespace": "0.8.6",
|
| 80 |
+
"lance-namespace-urllib3-client": "0.8.6",
|
| 81 |
+
"pyarrow": "20.0.0",
|
| 82 |
+
"opt-einsum": "3.4.0",
|
| 83 |
+
"richuru": "0.1.1",
|
| 84 |
+
"rich": "15.0.0",
|
| 85 |
+
"pygments": "2.20.0",
|
| 86 |
+
"markdown-it-py": "4.2.0",
|
| 87 |
+
"mdurl": "0.1.2",
|
| 88 |
+
"loguru": "0.7.3",
|
| 89 |
+
"scikit-learn": "1.7.2",
|
| 90 |
+
"threadpoolctl": "3.6.0",
|
| 91 |
+
"joblib": "1.5.3",
|
| 92 |
+
"minari": "0.5.3",
|
| 93 |
+
"gymnasium": "1.3.0",
|
| 94 |
+
"farama-notifications": "0.0.6",
|
| 95 |
+
"typer": "0.27.0",
|
| 96 |
+
"annotated-doc": "0.0.5",
|
| 97 |
+
"shellingham": "1.5.4",
|
| 98 |
+
"datasets": "4.4.0",
|
| 99 |
+
"aiohttp": "3.14.3",
|
| 100 |
+
"yarl": "1.24.5",
|
| 101 |
+
"propcache": "0.5.2",
|
| 102 |
+
"multidict": "6.7.1",
|
| 103 |
+
"frozenlist": "1.8.0",
|
| 104 |
+
"attrs": "26.1.0",
|
| 105 |
+
"async-timeout": "5.0.1",
|
| 106 |
+
"aiosignal": "1.4.0",
|
| 107 |
+
"aiohappyeyeballs": "2.7.1",
|
| 108 |
+
"multiprocess": "0.70.18",
|
| 109 |
+
"dill": "0.4.0",
|
| 110 |
+
"xxhash": "3.8.1",
|
| 111 |
+
"httpx": "0.28.1",
|
| 112 |
+
"httpcore": "1.0.9",
|
| 113 |
+
"h11": "0.16.0",
|
| 114 |
+
"anyio": "4.14.2",
|
| 115 |
+
"exceptiongroup": "1.3.1",
|
| 116 |
+
"pandas": "2.3.3",
|
| 117 |
+
"tzdata": "2026.3",
|
| 118 |
+
"pytz": "2026.3.post1",
|
| 119 |
+
"transformers": "4.57.3",
|
| 120 |
+
"safetensors": "0.8.0",
|
| 121 |
+
"tokenizers": "0.22.2",
|
| 122 |
+
"regex": "2026.7.19",
|
| 123 |
+
"zstandard": "0.25.0",
|
| 124 |
+
"timm": "1.0.28",
|
| 125 |
+
"prettytable": "3.18.0",
|
| 126 |
+
"wcwidth": "0.8.2",
|
| 127 |
+
"requests-cache": "1.3.3",
|
| 128 |
+
"url-normalize": "3.0.0",
|
| 129 |
+
"cattrs": "26.1.0",
|
| 130 |
+
"tabulate": "0.10.0",
|
| 131 |
+
"lightning": "2.6.5",
|
| 132 |
+
"pytorch-lightning": "2.6.5",
|
| 133 |
+
"lightning-utilities": "0.15.3",
|
| 134 |
+
"torchmetrics": "1.9.0",
|
| 135 |
+
"ale-py": "0.12.0",
|
| 136 |
+
"craftax": "1.6.1",
|
| 137 |
+
"pygame": "2.6.1",
|
| 138 |
+
"flax": "0.10.7",
|
| 139 |
+
"treescope": "0.1.10",
|
| 140 |
+
"tensorstore": "0.1.78",
|
| 141 |
+
"ml-dtypes": "0.5.4",
|
| 142 |
+
"orbax-checkpoint": "0.11.39",
|
| 143 |
+
"uvloop": "0.22.1",
|
| 144 |
+
"psutil": "7.2.2",
|
| 145 |
+
"simplejson": "4.1.1",
|
| 146 |
+
"humanize": "4.16.0",
|
| 147 |
+
"aiofiles": "25.1.0",
|
| 148 |
+
"jax": "0.6.2",
|
| 149 |
+
"jaxlib": "0.6.2",
|
| 150 |
+
"msgpack": "1.2.1",
|
| 151 |
+
"etils": "1.13.0",
|
| 152 |
+
"zipp": "4.1.0",
|
| 153 |
+
"importlib-resources": "7.1.0",
|
| 154 |
+
"absl-py": "2.5.0",
|
| 155 |
+
"optax": "0.2.8",
|
| 156 |
+
"stable-baselines3": "2.9.0",
|
| 157 |
+
"gymnasium-robotics": "1.4.2",
|
| 158 |
+
"pettingzoo": "1.26.1",
|
| 159 |
+
"mujoco": "3.11.0",
|
| 160 |
+
"pyopengl": "3.1.10",
|
| 161 |
+
"glfw": "2.10.2",
|
| 162 |
+
"moviepy": "2.2.1",
|
| 163 |
+
"python-dotenv": "1.2.2",
|
| 164 |
+
"proglog": "0.1.12",
|
| 165 |
+
"decorator": "5.3.1",
|
| 166 |
+
"opencv-python": "5.0.0.93",
|
| 167 |
+
"array-api-compat": "1.15.0",
|
| 168 |
+
"pygame-ce": "2.5.7",
|
| 169 |
+
"swig": "4.4.1",
|
| 170 |
+
"box2d-py": "2.3.5",
|
| 171 |
+
"minigrid": "3.1.0",
|
| 172 |
+
"ogbench": "1.2.1",
|
| 173 |
+
"dm-control": "1.0.44",
|
| 174 |
+
"lxml": "6.1.1",
|
| 175 |
+
"labmaze": "1.0.6",
|
| 176 |
+
"dm-tree": "0.1.10",
|
| 177 |
+
"wrapt": "2.3.0",
|
| 178 |
+
"dm-env": "1.6",
|
| 179 |
+
"shapely": "2.1.2",
|
| 180 |
+
"pymunk": "7.3.0",
|
| 181 |
+
"cffi": "2.1.0",
|
| 182 |
+
"pycparser": "3.0",
|
| 183 |
+
"lancedb": "0.34.0",
|
| 184 |
+
"overrides": "7.7.0",
|
| 185 |
+
"deprecation": "2.1.0"
|
| 186 |
+
},
|
| 187 |
+
"metadata_conflicts": [
|
| 188 |
+
"datasets==4.4.0 requires pyarrow>=21.0.0; observed pyarrow==20.0.0"
|
| 189 |
+
]
|
| 190 |
+
}
|
le-wm/tworoom/config.json
ADDED
|
@@ -0,0 +1,49 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_target_": "stable_worldmodel.wm.lewm.LeWM",
|
| 3 |
+
"encoder": {
|
| 4 |
+
"_target_": "stable_pretraining.backbone.utils.vit_hf",
|
| 5 |
+
"size": "tiny",
|
| 6 |
+
"patch_size": 14,
|
| 7 |
+
"image_size": 224,
|
| 8 |
+
"pretrained": false,
|
| 9 |
+
"use_mask_token": false
|
| 10 |
+
},
|
| 11 |
+
"predictor": {
|
| 12 |
+
"_target_": "stable_worldmodel.wm.lewm.module.Predictor",
|
| 13 |
+
"num_frames": 3,
|
| 14 |
+
"input_dim": 192,
|
| 15 |
+
"hidden_dim": 192,
|
| 16 |
+
"output_dim": 192,
|
| 17 |
+
"depth": 6,
|
| 18 |
+
"heads": 16,
|
| 19 |
+
"mlp_dim": 2048,
|
| 20 |
+
"dim_head": 64,
|
| 21 |
+
"dropout": 0.1,
|
| 22 |
+
"emb_dropout": 0.0
|
| 23 |
+
},
|
| 24 |
+
"action_encoder": {
|
| 25 |
+
"_target_": "stable_worldmodel.wm.lewm.module.Embedder",
|
| 26 |
+
"input_dim": 10,
|
| 27 |
+
"emb_dim": 192
|
| 28 |
+
},
|
| 29 |
+
"projector": {
|
| 30 |
+
"_target_": "stable_worldmodel.wm.lewm.module.MLP",
|
| 31 |
+
"input_dim": 192,
|
| 32 |
+
"output_dim": 192,
|
| 33 |
+
"hidden_dim": 2048,
|
| 34 |
+
"norm_fn": {
|
| 35 |
+
"_target_": "torch.nn.BatchNorm1d",
|
| 36 |
+
"_partial_": true
|
| 37 |
+
}
|
| 38 |
+
},
|
| 39 |
+
"pred_proj": {
|
| 40 |
+
"_target_": "stable_worldmodel.wm.lewm.module.MLP",
|
| 41 |
+
"input_dim": 192,
|
| 42 |
+
"output_dim": 192,
|
| 43 |
+
"hidden_dim": 2048,
|
| 44 |
+
"norm_fn": {
|
| 45 |
+
"_target_": "torch.nn.BatchNorm1d",
|
| 46 |
+
"_partial_": true
|
| 47 |
+
}
|
| 48 |
+
}
|
| 49 |
+
}
|
le-wm/tworoom/weights.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:566f223624ea4bfb39dbfe6ae731198dd6ea73b7b8919fed6b1ecafca810f7dd
|
| 3 |
+
size 72290849
|
licenses/dino_wm.txt
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
MIT License
|
| 2 |
+
|
| 3 |
+
Copyright (c) 2025 gaoyuezhou
|
| 4 |
+
|
| 5 |
+
Permission is hereby granted, free of charge, to any person obtaining a copy
|
| 6 |
+
of this software and associated documentation files (the "Software"), to deal
|
| 7 |
+
in the Software without restriction, including without limitation the rights
|
| 8 |
+
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
| 9 |
+
copies of the Software, and to permit persons to whom the Software is
|
| 10 |
+
furnished to do so, subject to the following conditions:
|
| 11 |
+
|
| 12 |
+
The above copyright notice and this permission notice shall be included in all
|
| 13 |
+
copies or substantial portions of the Software.
|
| 14 |
+
|
| 15 |
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
| 16 |
+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
| 17 |
+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
| 18 |
+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
| 19 |
+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
| 20 |
+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
| 21 |
+
SOFTWARE.
|
licenses/dreamerv3.txt
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
Copyright (c) 2024 Danijar Hafner
|
| 2 |
+
|
| 3 |
+
Permission is hereby granted, free of charge, to any person obtaining a copy
|
| 4 |
+
of this software and associated documentation files (the "Software"), to deal
|
| 5 |
+
in the Software without restriction, including without limitation the rights
|
| 6 |
+
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
| 7 |
+
copies of the Software, and to permit persons to whom the Software is
|
| 8 |
+
furnished to do so, subject to the following conditions:
|
| 9 |
+
|
| 10 |
+
The above copyright notice and this permission notice shall be included in all
|
| 11 |
+
copies or substantial portions of the Software.
|
| 12 |
+
|
| 13 |
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
| 14 |
+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
| 15 |
+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
| 16 |
+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
| 17 |
+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
| 18 |
+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
| 19 |
+
SOFTWARE.
|
licenses/le-wm.txt
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
MIT License
|
| 2 |
+
|
| 3 |
+
Copyright (c) 2026 Lucas Maes
|
| 4 |
+
|
| 5 |
+
Permission is hereby granted, free of charge, to any person obtaining a copy
|
| 6 |
+
of this software and associated documentation files (the "Software"), to deal
|
| 7 |
+
in the Software without restriction, including without limitation the rights
|
| 8 |
+
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
| 9 |
+
copies of the Software, and to permit persons to whom the Software is
|
| 10 |
+
furnished to do so, subject to the following conditions:
|
| 11 |
+
|
| 12 |
+
The above copyright notice and this permission notice shall be included in all
|
| 13 |
+
copies or substantial portions of the Software.
|
| 14 |
+
|
| 15 |
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
| 16 |
+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
| 17 |
+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
| 18 |
+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
| 19 |
+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
| 20 |
+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
| 21 |
+
SOFTWARE.
|
licenses/planet.txt
ADDED
|
@@ -0,0 +1,202 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
Apache License
|
| 3 |
+
Version 2.0, January 2004
|
| 4 |
+
http://www.apache.org/licenses/
|
| 5 |
+
|
| 6 |
+
TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
|
| 7 |
+
|
| 8 |
+
1. Definitions.
|
| 9 |
+
|
| 10 |
+
"License" shall mean the terms and conditions for use, reproduction,
|
| 11 |
+
and distribution as defined by Sections 1 through 9 of this document.
|
| 12 |
+
|
| 13 |
+
"Licensor" shall mean the copyright owner or entity authorized by
|
| 14 |
+
the copyright owner that is granting the License.
|
| 15 |
+
|
| 16 |
+
"Legal Entity" shall mean the union of the acting entity and all
|
| 17 |
+
other entities that control, are controlled by, or are under common
|
| 18 |
+
control with that entity. For the purposes of this definition,
|
| 19 |
+
"control" means (i) the power, direct or indirect, to cause the
|
| 20 |
+
direction or management of such entity, whether by contract or
|
| 21 |
+
otherwise, or (ii) ownership of fifty percent (50%) or more of the
|
| 22 |
+
outstanding shares, or (iii) beneficial ownership of such entity.
|
| 23 |
+
|
| 24 |
+
"You" (or "Your") shall mean an individual or Legal Entity
|
| 25 |
+
exercising permissions granted by this License.
|
| 26 |
+
|
| 27 |
+
"Source" form shall mean the preferred form for making modifications,
|
| 28 |
+
including but not limited to software source code, documentation
|
| 29 |
+
source, and configuration files.
|
| 30 |
+
|
| 31 |
+
"Object" form shall mean any form resulting from mechanical
|
| 32 |
+
transformation or translation of a Source form, including but
|
| 33 |
+
not limited to compiled object code, generated documentation,
|
| 34 |
+
and conversions to other media types.
|
| 35 |
+
|
| 36 |
+
"Work" shall mean the work of authorship, whether in Source or
|
| 37 |
+
Object form, made available under the License, as indicated by a
|
| 38 |
+
copyright notice that is included in or attached to the work
|
| 39 |
+
(an example is provided in the Appendix below).
|
| 40 |
+
|
| 41 |
+
"Derivative Works" shall mean any work, whether in Source or Object
|
| 42 |
+
form, that is based on (or derived from) the Work and for which the
|
| 43 |
+
editorial revisions, annotations, elaborations, or other modifications
|
| 44 |
+
represent, as a whole, an original work of authorship. For the purposes
|
| 45 |
+
of this License, Derivative Works shall not include works that remain
|
| 46 |
+
separable from, or merely link (or bind by name) to the interfaces of,
|
| 47 |
+
the Work and Derivative Works thereof.
|
| 48 |
+
|
| 49 |
+
"Contribution" shall mean any work of authorship, including
|
| 50 |
+
the original version of the Work and any modifications or additions
|
| 51 |
+
to that Work or Derivative Works thereof, that is intentionally
|
| 52 |
+
submitted to Licensor for inclusion in the Work by the copyright owner
|
| 53 |
+
or by an individual or Legal Entity authorized to submit on behalf of
|
| 54 |
+
the copyright owner. For the purposes of this definition, "submitted"
|
| 55 |
+
means any form of electronic, verbal, or written communication sent
|
| 56 |
+
to the Licensor or its representatives, including but not limited to
|
| 57 |
+
communication on electronic mailing lists, source code control systems,
|
| 58 |
+
and issue tracking systems that are managed by, or on behalf of, the
|
| 59 |
+
Licensor for the purpose of discussing and improving the Work, but
|
| 60 |
+
excluding communication that is conspicuously marked or otherwise
|
| 61 |
+
designated in writing by the copyright owner as "Not a Contribution."
|
| 62 |
+
|
| 63 |
+
"Contributor" shall mean Licensor and any individual or Legal Entity
|
| 64 |
+
on behalf of whom a Contribution has been received by Licensor and
|
| 65 |
+
subsequently incorporated within the Work.
|
| 66 |
+
|
| 67 |
+
2. Grant of Copyright License. Subject to the terms and conditions of
|
| 68 |
+
this License, each Contributor hereby grants to You a perpetual,
|
| 69 |
+
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
| 70 |
+
copyright license to reproduce, prepare Derivative Works of,
|
| 71 |
+
publicly display, publicly perform, sublicense, and distribute the
|
| 72 |
+
Work and such Derivative Works in Source or Object form.
|
| 73 |
+
|
| 74 |
+
3. Grant of Patent License. Subject to the terms and conditions of
|
| 75 |
+
this License, each Contributor hereby grants to You a perpetual,
|
| 76 |
+
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
| 77 |
+
(except as stated in this section) patent license to make, have made,
|
| 78 |
+
use, offer to sell, sell, import, and otherwise transfer the Work,
|
| 79 |
+
where such license applies only to those patent claims licensable
|
| 80 |
+
by such Contributor that are necessarily infringed by their
|
| 81 |
+
Contribution(s) alone or by combination of their Contribution(s)
|
| 82 |
+
with the Work to which such Contribution(s) was submitted. If You
|
| 83 |
+
institute patent litigation against any entity (including a
|
| 84 |
+
cross-claim or counterclaim in a lawsuit) alleging that the Work
|
| 85 |
+
or a Contribution incorporated within the Work constitutes direct
|
| 86 |
+
or contributory patent infringement, then any patent licenses
|
| 87 |
+
granted to You under this License for that Work shall terminate
|
| 88 |
+
as of the date such litigation is filed.
|
| 89 |
+
|
| 90 |
+
4. Redistribution. You may reproduce and distribute copies of the
|
| 91 |
+
Work or Derivative Works thereof in any medium, with or without
|
| 92 |
+
modifications, and in Source or Object form, provided that You
|
| 93 |
+
meet the following conditions:
|
| 94 |
+
|
| 95 |
+
(a) You must give any other recipients of the Work or
|
| 96 |
+
Derivative Works a copy of this License; and
|
| 97 |
+
|
| 98 |
+
(b) You must cause any modified files to carry prominent notices
|
| 99 |
+
stating that You changed the files; and
|
| 100 |
+
|
| 101 |
+
(c) You must retain, in the Source form of any Derivative Works
|
| 102 |
+
that You distribute, all copyright, patent, trademark, and
|
| 103 |
+
attribution notices from the Source form of the Work,
|
| 104 |
+
excluding those notices that do not pertain to any part of
|
| 105 |
+
the Derivative Works; and
|
| 106 |
+
|
| 107 |
+
(d) If the Work includes a "NOTICE" text file as part of its
|
| 108 |
+
distribution, then any Derivative Works that You distribute must
|
| 109 |
+
include a readable copy of the attribution notices contained
|
| 110 |
+
within such NOTICE file, excluding those notices that do not
|
| 111 |
+
pertain to any part of the Derivative Works, in at least one
|
| 112 |
+
of the following places: within a NOTICE text file distributed
|
| 113 |
+
as part of the Derivative Works; within the Source form or
|
| 114 |
+
documentation, if provided along with the Derivative Works; or,
|
| 115 |
+
within a display generated by the Derivative Works, if and
|
| 116 |
+
wherever such third-party notices normally appear. The contents
|
| 117 |
+
of the NOTICE file are for informational purposes only and
|
| 118 |
+
do not modify the License. You may add Your own attribution
|
| 119 |
+
notices within Derivative Works that You distribute, alongside
|
| 120 |
+
or as an addendum to the NOTICE text from the Work, provided
|
| 121 |
+
that such additional attribution notices cannot be construed
|
| 122 |
+
as modifying the License.
|
| 123 |
+
|
| 124 |
+
You may add Your own copyright statement to Your modifications and
|
| 125 |
+
may provide additional or different license terms and conditions
|
| 126 |
+
for use, reproduction, or distribution of Your modifications, or
|
| 127 |
+
for any such Derivative Works as a whole, provided Your use,
|
| 128 |
+
reproduction, and distribution of the Work otherwise complies with
|
| 129 |
+
the conditions stated in this License.
|
| 130 |
+
|
| 131 |
+
5. Submission of Contributions. Unless You explicitly state otherwise,
|
| 132 |
+
any Contribution intentionally submitted for inclusion in the Work
|
| 133 |
+
by You to the Licensor shall be under the terms and conditions of
|
| 134 |
+
this License, without any additional terms or conditions.
|
| 135 |
+
Notwithstanding the above, nothing herein shall supersede or modify
|
| 136 |
+
the terms of any separate license agreement you may have executed
|
| 137 |
+
with Licensor regarding such Contributions.
|
| 138 |
+
|
| 139 |
+
6. Trademarks. This License does not grant permission to use the trade
|
| 140 |
+
names, trademarks, service marks, or product names of the Licensor,
|
| 141 |
+
except as required for reasonable and customary use in describing the
|
| 142 |
+
origin of the Work and reproducing the content of the NOTICE file.
|
| 143 |
+
|
| 144 |
+
7. Disclaimer of Warranty. Unless required by applicable law or
|
| 145 |
+
agreed to in writing, Licensor provides the Work (and each
|
| 146 |
+
Contributor provides its Contributions) on an "AS IS" BASIS,
|
| 147 |
+
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
| 148 |
+
implied, including, without limitation, any warranties or conditions
|
| 149 |
+
of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
|
| 150 |
+
PARTICULAR PURPOSE. You are solely responsible for determining the
|
| 151 |
+
appropriateness of using or redistributing the Work and assume any
|
| 152 |
+
risks associated with Your exercise of permissions under this License.
|
| 153 |
+
|
| 154 |
+
8. Limitation of Liability. In no event and under no legal theory,
|
| 155 |
+
whether in tort (including negligence), contract, or otherwise,
|
| 156 |
+
unless required by applicable law (such as deliberate and grossly
|
| 157 |
+
negligent acts) or agreed to in writing, shall any Contributor be
|
| 158 |
+
liable to You for damages, including any direct, indirect, special,
|
| 159 |
+
incidental, or consequential damages of any character arising as a
|
| 160 |
+
result of this License or out of the use or inability to use the
|
| 161 |
+
Work (including but not limited to damages for loss of goodwill,
|
| 162 |
+
work stoppage, computer failure or malfunction, or any and all
|
| 163 |
+
other commercial damages or losses), even if such Contributor
|
| 164 |
+
has been advised of the possibility of such damages.
|
| 165 |
+
|
| 166 |
+
9. Accepting Warranty or Additional Liability. While redistributing
|
| 167 |
+
the Work or Derivative Works thereof, You may choose to offer,
|
| 168 |
+
and charge a fee for, acceptance of support, warranty, indemnity,
|
| 169 |
+
or other liability obligations and/or rights consistent with this
|
| 170 |
+
License. However, in accepting such obligations, You may act only
|
| 171 |
+
on Your own behalf and on Your sole responsibility, not on behalf
|
| 172 |
+
of any other Contributor, and only if You agree to indemnify,
|
| 173 |
+
defend, and hold each Contributor harmless for any liability
|
| 174 |
+
incurred by, or claims asserted against, such Contributor by reason
|
| 175 |
+
of your accepting any such warranty or additional liability.
|
| 176 |
+
|
| 177 |
+
END OF TERMS AND CONDITIONS
|
| 178 |
+
|
| 179 |
+
APPENDIX: How to apply the Apache License to your work.
|
| 180 |
+
|
| 181 |
+
To apply the Apache License to your work, attach the following
|
| 182 |
+
boilerplate notice, with the fields enclosed by brackets "[]"
|
| 183 |
+
replaced with your own identifying information. (Don't include
|
| 184 |
+
the brackets!) The text should be enclosed in the appropriate
|
| 185 |
+
comment syntax for the file format. We also recommend that a
|
| 186 |
+
file or class name and description of purpose be included on the
|
| 187 |
+
same "printed page" as the copyright notice for easier
|
| 188 |
+
identification within third-party archives.
|
| 189 |
+
|
| 190 |
+
Copyright [yyyy] [name of copyright owner]
|
| 191 |
+
|
| 192 |
+
Licensed under the Apache License, Version 2.0 (the "License");
|
| 193 |
+
you may not use this file except in compliance with the License.
|
| 194 |
+
You may obtain a copy of the License at
|
| 195 |
+
|
| 196 |
+
http://www.apache.org/licenses/LICENSE-2.0
|
| 197 |
+
|
| 198 |
+
Unless required by applicable law or agreed to in writing, software
|
| 199 |
+
distributed under the License is distributed on an "AS IS" BASIS,
|
| 200 |
+
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
| 201 |
+
See the License for the specific language governing permissions and
|
| 202 |
+
limitations under the License.
|
licenses/tdmpc2.txt
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
MIT License
|
| 2 |
+
|
| 3 |
+
Copyright (c) Nicklas Hansen (2023).
|
| 4 |
+
|
| 5 |
+
Permission is hereby granted, free of charge, to any person obtaining a copy
|
| 6 |
+
of this software and associated documentation files (the "Software"), to deal
|
| 7 |
+
in the Software without restriction, including without limitation the rights
|
| 8 |
+
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
| 9 |
+
copies of the Software, and to permit persons to whom the Software is
|
| 10 |
+
furnished to do so, subject to the following conditions:
|
| 11 |
+
|
| 12 |
+
The above copyright notice and this permission notice shall be included in all
|
| 13 |
+
copies or substantial portions of the Software.
|
| 14 |
+
|
| 15 |
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
| 16 |
+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
| 17 |
+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
| 18 |
+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
| 19 |
+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
| 20 |
+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
| 21 |
+
SOFTWARE.
|
manifest.json
ADDED
|
@@ -0,0 +1,481 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"format_version": 1,
|
| 3 |
+
"repo_id": "mohendy/world-model-attack-checkpoints",
|
| 4 |
+
"files": [
|
| 5 |
+
{
|
| 6 |
+
"path": "tdmpc2/finger-spin/step-1000000.pt",
|
| 7 |
+
"model": "tdmpc2",
|
| 8 |
+
"bytes": 42677713,
|
| 9 |
+
"sha256": "bbc1c9b519784a661839bfe97707a5c5395ca03655fbc17f24fe13cf23857be8",
|
| 10 |
+
"source_sha256": "bbc1c9b519784a661839bfe97707a5c5395ca03655fbc17f24fe13cf23857be8",
|
| 11 |
+
"source_relative_path": "tdmpc2/tdmpc2/logs/finger-spin/1/rgb_seed1_2m/models/step-1000000.pt",
|
| 12 |
+
"sanitized": false
|
| 13 |
+
},
|
| 14 |
+
{
|
| 15 |
+
"path": "tdmpc2/cup-catch/step-1500000.pt",
|
| 16 |
+
"model": "tdmpc2",
|
| 17 |
+
"bytes": 42677713,
|
| 18 |
+
"sha256": "df67f9c68655c3443c8c5e36799091ee9debfebc4ca11b1a59abf370b4083e47",
|
| 19 |
+
"source_sha256": "df67f9c68655c3443c8c5e36799091ee9debfebc4ca11b1a59abf370b4083e47",
|
| 20 |
+
"source_relative_path": "tdmpc2/tdmpc2/logs/cup-catch/1/rgb_seed1_2m/models/step-1500000.pt",
|
| 21 |
+
"sanitized": false
|
| 22 |
+
},
|
| 23 |
+
{
|
| 24 |
+
"path": "tdmpc2/cheetah-run/step-1500000.pt",
|
| 25 |
+
"model": "tdmpc2",
|
| 26 |
+
"bytes": 42833361,
|
| 27 |
+
"sha256": "5654ea3c9c69a2640848b9d0efc043dc7d55896eeb9de62c7cb5c958f8652cd8",
|
| 28 |
+
"source_sha256": "5654ea3c9c69a2640848b9d0efc043dc7d55896eeb9de62c7cb5c958f8652cd8",
|
| 29 |
+
"source_relative_path": "tdmpc2/tdmpc2/logs/cheetah-run/1/rgb_seed1_2m/models/step-1500000.pt",
|
| 30 |
+
"sanitized": false
|
| 31 |
+
},
|
| 32 |
+
{
|
| 33 |
+
"path": "tdmpc2/cartpole-balance/step-250000.pt",
|
| 34 |
+
"model": "tdmpc2",
|
| 35 |
+
"bytes": 42638723,
|
| 36 |
+
"sha256": "74cf0a67dc4f2487c270381f32f1027a07a3299fa8fbe282cb7b7a632ce84b6e",
|
| 37 |
+
"source_sha256": "74cf0a67dc4f2487c270381f32f1027a07a3299fa8fbe282cb7b7a632ce84b6e",
|
| 38 |
+
"source_relative_path": "tdmpc2/tdmpc2/logs/cartpole-balance/1/rgb_seed1_2m/models/step-250000.pt",
|
| 39 |
+
"sanitized": false
|
| 40 |
+
},
|
| 41 |
+
{
|
| 42 |
+
"path": "planet/reacher_easy/config.yaml",
|
| 43 |
+
"model": "planet",
|
| 44 |
+
"bytes": 12461,
|
| 45 |
+
"sha256": "da3a9fa154e89037fc27ac9b2f88db4d7ae821660a6428ce1cb9240d2dcae324",
|
| 46 |
+
"source_sha256": "1a91c863aea1d70dd07c625ef7e7736c774b18f4cb6b0a1ddf8347d643dce2d3",
|
| 47 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/reacher_easy/config.yaml",
|
| 48 |
+
"sanitized": true
|
| 49 |
+
},
|
| 50 |
+
{
|
| 51 |
+
"path": "planet/reacher_easy/attack_export.json",
|
| 52 |
+
"model": "planet",
|
| 53 |
+
"bytes": 464,
|
| 54 |
+
"sha256": "512ce35d92afb78d89106cfa0279c4cd22110d0cf6b1bd65549c88501052a487",
|
| 55 |
+
"source_sha256": "512ce35d92afb78d89106cfa0279c4cd22110d0cf6b1bd65549c88501052a487",
|
| 56 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/reacher_easy/attack_export.json",
|
| 57 |
+
"sanitized": false
|
| 58 |
+
},
|
| 59 |
+
{
|
| 60 |
+
"path": "planet/reacher_easy/attack_weights.npz",
|
| 61 |
+
"model": "planet",
|
| 62 |
+
"bytes": 5342758,
|
| 63 |
+
"sha256": "a1e9829fbd16c2d1b18336534c561006938a4b59a5f55b0de526f7cebfac5e38",
|
| 64 |
+
"source_sha256": "a1e9829fbd16c2d1b18336534c561006938a4b59a5f55b0de526f7cebfac5e38",
|
| 65 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/reacher_easy/attack_weights.npz",
|
| 66 |
+
"sanitized": false
|
| 67 |
+
},
|
| 68 |
+
{
|
| 69 |
+
"path": "planet/reacher_easy/tf_reference.npz",
|
| 70 |
+
"model": "planet",
|
| 71 |
+
"bytes": 112084,
|
| 72 |
+
"sha256": "22d50e7b59c1c661bc1ffb889c4196bffe9fe6d28f4cf2b62d7700ff99e8d165",
|
| 73 |
+
"source_sha256": "22d50e7b59c1c661bc1ffb889c4196bffe9fe6d28f4cf2b62d7700ff99e8d165",
|
| 74 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/reacher_easy/tf_reference.npz",
|
| 75 |
+
"sanitized": false
|
| 76 |
+
},
|
| 77 |
+
{
|
| 78 |
+
"path": "planet/cartpole_swingup/config.yaml",
|
| 79 |
+
"model": "planet",
|
| 80 |
+
"bytes": 12497,
|
| 81 |
+
"sha256": "6e645402217101ca5151be97f37448913d6b89787161422a019b80b9f547ab36",
|
| 82 |
+
"source_sha256": "4e9de7dfe7bf9d706caf6720a164e640ed68221acfbe099386dfed75c4a1ca30",
|
| 83 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/cartpole_swingup/config.yaml",
|
| 84 |
+
"sanitized": true
|
| 85 |
+
},
|
| 86 |
+
{
|
| 87 |
+
"path": "planet/cartpole_swingup/attack_export.json",
|
| 88 |
+
"model": "planet",
|
| 89 |
+
"bytes": 464,
|
| 90 |
+
"sha256": "dc7e1134643a0bce63521befb2eaf708aa83d24fdb8e361d0b8280b1cbecd80e",
|
| 91 |
+
"source_sha256": "dc7e1134643a0bce63521befb2eaf708aa83d24fdb8e361d0b8280b1cbecd80e",
|
| 92 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/cartpole_swingup/attack_export.json",
|
| 93 |
+
"sanitized": false
|
| 94 |
+
},
|
| 95 |
+
{
|
| 96 |
+
"path": "planet/cartpole_swingup/attack_weights.npz",
|
| 97 |
+
"model": "planet",
|
| 98 |
+
"bytes": 5341958,
|
| 99 |
+
"sha256": "f75ced2942b489b578c750ed99ce5daa9e14a8727ce911b469a7785ce6231580",
|
| 100 |
+
"source_sha256": "f75ced2942b489b578c750ed99ce5daa9e14a8727ce911b469a7785ce6231580",
|
| 101 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/cartpole_swingup/attack_weights.npz",
|
| 102 |
+
"sanitized": false
|
| 103 |
+
},
|
| 104 |
+
{
|
| 105 |
+
"path": "planet/cartpole_swingup/tf_reference.npz",
|
| 106 |
+
"model": "planet",
|
| 107 |
+
"bytes": 111936,
|
| 108 |
+
"sha256": "62b9163ef512d1a397af51cd1320b5721f59ea8cb7283b8be1f320d6dd7035a0",
|
| 109 |
+
"source_sha256": "62b9163ef512d1a397af51cd1320b5721f59ea8cb7283b8be1f320d6dd7035a0",
|
| 110 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/cartpole_swingup/tf_reference.npz",
|
| 111 |
+
"sanitized": false
|
| 112 |
+
},
|
| 113 |
+
{
|
| 114 |
+
"path": "planet/cheetah_run/config.yaml",
|
| 115 |
+
"model": "planet",
|
| 116 |
+
"bytes": 12422,
|
| 117 |
+
"sha256": "7da0ca0f3486fe410848afd46fd1c000c8ef8241dbe47ec5e5d98ef169f51fa0",
|
| 118 |
+
"source_sha256": "0e234a9f150d9e0e301e3e8d770cb787c56ec018406ac8b9432b4c9a31551e06",
|
| 119 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/cheetah_run/config.yaml",
|
| 120 |
+
"sanitized": true
|
| 121 |
+
},
|
| 122 |
+
{
|
| 123 |
+
"path": "planet/cheetah_run/attack_export.json",
|
| 124 |
+
"model": "planet",
|
| 125 |
+
"bytes": 464,
|
| 126 |
+
"sha256": "0d7d9b3940ead701528d14eca0fa05ed6fca471cb009ded68448b8251cea6740",
|
| 127 |
+
"source_sha256": "0d7d9b3940ead701528d14eca0fa05ed6fca471cb009ded68448b8251cea6740",
|
| 128 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/cheetah_run/attack_export.json",
|
| 129 |
+
"sanitized": false
|
| 130 |
+
},
|
| 131 |
+
{
|
| 132 |
+
"path": "planet/cheetah_run/attack_weights.npz",
|
| 133 |
+
"model": "planet",
|
| 134 |
+
"bytes": 5345958,
|
| 135 |
+
"sha256": "fdf26210c8304305b8b5903d730dbdfe8b56c6a358524dba3c35d15048f4c968",
|
| 136 |
+
"source_sha256": "fdf26210c8304305b8b5903d730dbdfe8b56c6a358524dba3c35d15048f4c968",
|
| 137 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/cheetah_run/attack_weights.npz",
|
| 138 |
+
"sanitized": false
|
| 139 |
+
},
|
| 140 |
+
{
|
| 141 |
+
"path": "planet/cheetah_run/tf_reference.npz",
|
| 142 |
+
"model": "planet",
|
| 143 |
+
"bytes": 112676,
|
| 144 |
+
"sha256": "7bb4dfe7beea345abd585c5db704f40c668e1356c0b7644c4a5135d44ec7d271",
|
| 145 |
+
"source_sha256": "7bb4dfe7beea345abd585c5db704f40c668e1356c0b7644c4a5135d44ec7d271",
|
| 146 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/cheetah_run/tf_reference.npz",
|
| 147 |
+
"sanitized": false
|
| 148 |
+
},
|
| 149 |
+
{
|
| 150 |
+
"path": "dino_wm/pusht/hydra.yaml",
|
| 151 |
+
"model": "dino_wm",
|
| 152 |
+
"bytes": 1927,
|
| 153 |
+
"sha256": "33f40be9f62d4dae9553ea99011692b5b9c20aae9110595d1807d876b13f585a",
|
| 154 |
+
"source_sha256": "e4d7f21b85c778ab8b83c0bb83bc7625fdfb193a6a0619fef25f4a2afd7f8049",
|
| 155 |
+
"source_relative_path": "dino_wm/checkpoints/outputs/pusht/hydra.yaml",
|
| 156 |
+
"sanitized": true
|
| 157 |
+
},
|
| 158 |
+
{
|
| 159 |
+
"path": "dino_wm/wall_single/hydra.yaml",
|
| 160 |
+
"model": "dino_wm",
|
| 161 |
+
"bytes": 1920,
|
| 162 |
+
"sha256": "f0b13250e49408c4890190eda837641e030ff6cda073cbc1238e4d11068c4370",
|
| 163 |
+
"source_sha256": "4ed3109219ed6820ce51fd19e4c97199552b3443ed58c502fddd6441784967e1",
|
| 164 |
+
"source_relative_path": "dino_wm/checkpoints/outputs/wall_single/hydra.yaml",
|
| 165 |
+
"sanitized": true
|
| 166 |
+
},
|
| 167 |
+
{
|
| 168 |
+
"path": "dino_wm/wall_single/model_latest.pth",
|
| 169 |
+
"model": "dino_wm",
|
| 170 |
+
"bytes": 368656057,
|
| 171 |
+
"sha256": "8441971becdae934fe08de5b163398390a32f6fe1fb0a8df113290e2468e142b",
|
| 172 |
+
"source_sha256": "8441971becdae934fe08de5b163398390a32f6fe1fb0a8df113290e2468e142b",
|
| 173 |
+
"source_relative_path": "dino_wm/checkpoints/outputs/wall_single/checkpoints/model_latest.pth",
|
| 174 |
+
"sanitized": false
|
| 175 |
+
},
|
| 176 |
+
{
|
| 177 |
+
"path": "dino_wm/pusht/model_latest.pth",
|
| 178 |
+
"model": "dino_wm",
|
| 179 |
+
"bytes": 377929744,
|
| 180 |
+
"sha256": "e909f0cec958fc0b49a79f2e85730ae6b5f83dee61a224705f91883eb3bb74c5",
|
| 181 |
+
"source_sha256": "e909f0cec958fc0b49a79f2e85730ae6b5f83dee61a224705f91883eb3bb74c5",
|
| 182 |
+
"source_relative_path": "dino_wm/checkpoints/outputs/pusht/checkpoints/model_latest.pth",
|
| 183 |
+
"sanitized": false
|
| 184 |
+
},
|
| 185 |
+
{
|
| 186 |
+
"path": "le-wm/cube/config.json",
|
| 187 |
+
"model": "le-wm",
|
| 188 |
+
"bytes": 1313,
|
| 189 |
+
"sha256": "4d446944fe28922cc2c5763f43d4ef9132a457bd89e9a0ce5dbceac183994999",
|
| 190 |
+
"source_sha256": "4d446944fe28922cc2c5763f43d4ef9132a457bd89e9a0ce5dbceac183994999",
|
| 191 |
+
"source_relative_path": "le-wm/.stable-wm/checkpoints/cube/lewm/config.json",
|
| 192 |
+
"sanitized": false
|
| 193 |
+
},
|
| 194 |
+
{
|
| 195 |
+
"path": "le-wm/cube/weights.pt",
|
| 196 |
+
"model": "le-wm",
|
| 197 |
+
"bytes": 72291425,
|
| 198 |
+
"sha256": "2839a907362f403f9136383016e91774373a295d958ae75121791f22a9fddf89",
|
| 199 |
+
"source_sha256": "2839a907362f403f9136383016e91774373a295d958ae75121791f22a9fddf89",
|
| 200 |
+
"source_relative_path": "le-wm/.stable-wm/checkpoints/cube/lewm/weights.pt",
|
| 201 |
+
"sanitized": false
|
| 202 |
+
},
|
| 203 |
+
{
|
| 204 |
+
"path": "le-wm/pusht/config.json",
|
| 205 |
+
"model": "le-wm",
|
| 206 |
+
"bytes": 1313,
|
| 207 |
+
"sha256": "2564086e961e7b5c7c04dffc451091115b389a590645ff19653c64fd0bc16e09",
|
| 208 |
+
"source_sha256": "2564086e961e7b5c7c04dffc451091115b389a590645ff19653c64fd0bc16e09",
|
| 209 |
+
"source_relative_path": "le-wm/.stable-wm/checkpoints/pusht/lewm/config.json",
|
| 210 |
+
"sanitized": false
|
| 211 |
+
},
|
| 212 |
+
{
|
| 213 |
+
"path": "le-wm/pusht/weights.pt",
|
| 214 |
+
"model": "le-wm",
|
| 215 |
+
"bytes": 72290721,
|
| 216 |
+
"sha256": "48938400ae3464c9680731287f583a9cb516f55a8ec64ea13a91be47fb15b607",
|
| 217 |
+
"source_sha256": "48938400ae3464c9680731287f583a9cb516f55a8ec64ea13a91be47fb15b607",
|
| 218 |
+
"source_relative_path": "le-wm/.stable-wm/checkpoints/pusht/lewm/weights.pt",
|
| 219 |
+
"sanitized": false
|
| 220 |
+
},
|
| 221 |
+
{
|
| 222 |
+
"path": "le-wm/reacher/config.json",
|
| 223 |
+
"model": "le-wm",
|
| 224 |
+
"bytes": 1313,
|
| 225 |
+
"sha256": "2564086e961e7b5c7c04dffc451091115b389a590645ff19653c64fd0bc16e09",
|
| 226 |
+
"source_sha256": "2564086e961e7b5c7c04dffc451091115b389a590645ff19653c64fd0bc16e09",
|
| 227 |
+
"source_relative_path": "le-wm/.stable-wm/checkpoints/reacher/lewm/config.json",
|
| 228 |
+
"sanitized": false
|
| 229 |
+
},
|
| 230 |
+
{
|
| 231 |
+
"path": "le-wm/reacher/weights.pt",
|
| 232 |
+
"model": "le-wm",
|
| 233 |
+
"bytes": 72290849,
|
| 234 |
+
"sha256": "eb70b1fd5409f8f81875d62f5ee5a20dd220a3128a477de66b5760f475f0f469",
|
| 235 |
+
"source_sha256": "eb70b1fd5409f8f81875d62f5ee5a20dd220a3128a477de66b5760f475f0f469",
|
| 236 |
+
"source_relative_path": "le-wm/.stable-wm/checkpoints/reacher/lewm/weights.pt",
|
| 237 |
+
"sanitized": false
|
| 238 |
+
},
|
| 239 |
+
{
|
| 240 |
+
"path": "le-wm/tworoom/config.json",
|
| 241 |
+
"model": "le-wm",
|
| 242 |
+
"bytes": 1313,
|
| 243 |
+
"sha256": "2564086e961e7b5c7c04dffc451091115b389a590645ff19653c64fd0bc16e09",
|
| 244 |
+
"source_sha256": "2564086e961e7b5c7c04dffc451091115b389a590645ff19653c64fd0bc16e09",
|
| 245 |
+
"source_relative_path": "le-wm/.stable-wm/checkpoints/tworoom/lewm/config.json",
|
| 246 |
+
"sanitized": false
|
| 247 |
+
},
|
| 248 |
+
{
|
| 249 |
+
"path": "le-wm/tworoom/weights.pt",
|
| 250 |
+
"model": "le-wm",
|
| 251 |
+
"bytes": 72290849,
|
| 252 |
+
"sha256": "566f223624ea4bfb39dbfe6ae731198dd6ea73b7b8919fed6b1ecafca810f7dd",
|
| 253 |
+
"source_sha256": "566f223624ea4bfb39dbfe6ae731198dd6ea73b7b8919fed6b1ecafca810f7dd",
|
| 254 |
+
"source_relative_path": "le-wm/.stable-wm/checkpoints/tworoom/lewm/weights.pt",
|
| 255 |
+
"sanitized": false
|
| 256 |
+
},
|
| 257 |
+
{
|
| 258 |
+
"path": "dreamerv3/cheetah_run/config.yaml",
|
| 259 |
+
"model": "dreamerv3",
|
| 260 |
+
"bytes": 4403,
|
| 261 |
+
"sha256": "b28b6a00ac52e29df5d31accca25595942f1ea8e6986e02ad1c27d94daba0ad5",
|
| 262 |
+
"source_sha256": "f73845c27ef1eff92eff2acf5fd196c9a5bd25aaf43aed4a2a8a16ef78457f74",
|
| 263 |
+
"source_relative_path": "logdir/dreamer/cheetah_run/slurm-23188/config.yaml",
|
| 264 |
+
"sanitized": true
|
| 265 |
+
},
|
| 266 |
+
{
|
| 267 |
+
"path": "dreamerv3/cheetah_run/ckpt/20260901T093748F019112/agent.pkl",
|
| 268 |
+
"model": "dreamerv3",
|
| 269 |
+
"bytes": 2042806149,
|
| 270 |
+
"sha256": "1d94c5824d8c77ba8191ccd5207409986a08bbf7e04ae5d55a9d468c7d158af3",
|
| 271 |
+
"source_sha256": "1d94c5824d8c77ba8191ccd5207409986a08bbf7e04ae5d55a9d468c7d158af3",
|
| 272 |
+
"source_relative_path": "logdir/dreamer/cheetah_run/slurm-23188/ckpt/20260901T093748F019112/agent.pkl",
|
| 273 |
+
"sanitized": false
|
| 274 |
+
},
|
| 275 |
+
{
|
| 276 |
+
"path": "dreamerv3/cheetah_run/ckpt/20260901T093748F019112/step.pkl",
|
| 277 |
+
"model": "dreamerv3",
|
| 278 |
+
"bytes": 17,
|
| 279 |
+
"sha256": "69901686be1d13680e0c0ddbfd0a0d2f2a70bb0ae46a2e636bf78bf5bc45cce2",
|
| 280 |
+
"source_sha256": "69901686be1d13680e0c0ddbfd0a0d2f2a70bb0ae46a2e636bf78bf5bc45cce2",
|
| 281 |
+
"source_relative_path": "logdir/dreamer/cheetah_run/slurm-23188/ckpt/20260901T093748F019112/step.pkl",
|
| 282 |
+
"sanitized": false
|
| 283 |
+
},
|
| 284 |
+
{
|
| 285 |
+
"path": "dreamerv3/cheetah_run/ckpt/20260901T093748F019112/done",
|
| 286 |
+
"model": "dreamerv3",
|
| 287 |
+
"bytes": 0,
|
| 288 |
+
"sha256": "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",
|
| 289 |
+
"source_sha256": "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",
|
| 290 |
+
"source_relative_path": "logdir/dreamer/cheetah_run/slurm-23188/ckpt/20260901T093748F019112/done",
|
| 291 |
+
"sanitized": false
|
| 292 |
+
},
|
| 293 |
+
{
|
| 294 |
+
"path": "dreamerv3/reacher_easy/config.yaml",
|
| 295 |
+
"model": "dreamerv3",
|
| 296 |
+
"bytes": 4409,
|
| 297 |
+
"sha256": "5a264f9cfc744b358798b053447f8df1d936a91592b04e96850fe3dda5ca32e4",
|
| 298 |
+
"source_sha256": "431174ded7bf75a544ba3717917cff2487c3b997c6a4c5c22ddbb9260e88e897",
|
| 299 |
+
"source_relative_path": "logdir/dreamer/reacher_easy/20260830T125224/config.yaml",
|
| 300 |
+
"sanitized": true
|
| 301 |
+
},
|
| 302 |
+
{
|
| 303 |
+
"path": "dreamerv3/reacher_easy/ckpt/20260830T195435F325934/agent.pkl",
|
| 304 |
+
"model": "dreamerv3",
|
| 305 |
+
"bytes": 2042658588,
|
| 306 |
+
"sha256": "a1fb60436f02fcfa6ba1fbd0b5b5078142cb6f6c34b13c42e4a5967ef047b394",
|
| 307 |
+
"source_sha256": "a1fb60436f02fcfa6ba1fbd0b5b5078142cb6f6c34b13c42e4a5967ef047b394",
|
| 308 |
+
"source_relative_path": "logdir/dreamer/reacher_easy/20260830T125224/ckpt/20260830T195435F325934/agent.pkl",
|
| 309 |
+
"sanitized": false
|
| 310 |
+
},
|
| 311 |
+
{
|
| 312 |
+
"path": "dreamerv3/reacher_easy/ckpt/20260830T195435F325934/step.pkl",
|
| 313 |
+
"model": "dreamerv3",
|
| 314 |
+
"bytes": 17,
|
| 315 |
+
"sha256": "7e5c5685c2d39badee13ea957052bbe8a3f71e577e2369e55151ad875842cb2f",
|
| 316 |
+
"source_sha256": "7e5c5685c2d39badee13ea957052bbe8a3f71e577e2369e55151ad875842cb2f",
|
| 317 |
+
"source_relative_path": "logdir/dreamer/reacher_easy/20260830T125224/ckpt/20260830T195435F325934/step.pkl",
|
| 318 |
+
"sanitized": false
|
| 319 |
+
},
|
| 320 |
+
{
|
| 321 |
+
"path": "dreamerv3/reacher_easy/ckpt/20260830T195435F325934/done",
|
| 322 |
+
"model": "dreamerv3",
|
| 323 |
+
"bytes": 0,
|
| 324 |
+
"sha256": "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",
|
| 325 |
+
"source_sha256": "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",
|
| 326 |
+
"source_relative_path": "logdir/dreamer/reacher_easy/20260830T125224/ckpt/20260830T195435F325934/done",
|
| 327 |
+
"sanitized": false
|
| 328 |
+
},
|
| 329 |
+
{
|
| 330 |
+
"path": "planet/cartpole_swingup/tensorflow/model.ckpt-4504500.data-00000-of-00001",
|
| 331 |
+
"model": "planet",
|
| 332 |
+
"bytes": 61543552,
|
| 333 |
+
"sha256": "22b7d2959cab4e3b38fd55a579bb718d59516053f6cb92530c7cabf785cd5ce5",
|
| 334 |
+
"source_sha256": "22b7d2959cab4e3b38fd55a579bb718d59516053f6cb92530c7cabf785cd5ce5",
|
| 335 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/cartpole_swingup/model.ckpt-4504500.data-00000-of-00001",
|
| 336 |
+
"sanitized": false
|
| 337 |
+
},
|
| 338 |
+
{
|
| 339 |
+
"path": "planet/cartpole_swingup/tensorflow/model.ckpt-4504500.index",
|
| 340 |
+
"model": "planet",
|
| 341 |
+
"bytes": 5797,
|
| 342 |
+
"sha256": "b34dd58607ccf858446024be305a73757eb9b65a83ccecde57768974ad13a509",
|
| 343 |
+
"source_sha256": "b34dd58607ccf858446024be305a73757eb9b65a83ccecde57768974ad13a509",
|
| 344 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/cartpole_swingup/model.ckpt-4504500.index",
|
| 345 |
+
"sanitized": false
|
| 346 |
+
},
|
| 347 |
+
{
|
| 348 |
+
"path": "planet/cartpole_swingup/tensorflow/model.ckpt-4504500.meta",
|
| 349 |
+
"model": "planet",
|
| 350 |
+
"bytes": 7264371,
|
| 351 |
+
"sha256": "cdd30a7fd3c3bff6ca5e0bcce9c87bfd5056a02462730ac3d05d91e9cce4b1df",
|
| 352 |
+
"source_sha256": "cdd30a7fd3c3bff6ca5e0bcce9c87bfd5056a02462730ac3d05d91e9cce4b1df",
|
| 353 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/cartpole_swingup/model.ckpt-4504500.meta",
|
| 354 |
+
"sanitized": false
|
| 355 |
+
},
|
| 356 |
+
{
|
| 357 |
+
"path": "planet/cartpole_swingup/tensorflow/checkpoint",
|
| 358 |
+
"model": "planet",
|
| 359 |
+
"bytes": 93,
|
| 360 |
+
"sha256": "c4b1b9ef54b5185a72880247eb4ab7c71e54cc1b3e15a6ef82cbaf9510ac35d6",
|
| 361 |
+
"source_sha256": "c4b1b9ef54b5185a72880247eb4ab7c71e54cc1b3e15a6ef82cbaf9510ac35d6",
|
| 362 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/cartpole_swingup/checkpoint",
|
| 363 |
+
"sanitized": false
|
| 364 |
+
},
|
| 365 |
+
{
|
| 366 |
+
"path": "planet/reacher_easy/tensorflow/model.ckpt-4504500.data-00000-of-00001",
|
| 367 |
+
"model": "planet",
|
| 368 |
+
"bytes": 61545952,
|
| 369 |
+
"sha256": "4d1cbb7fe86a05756a2433fef62f8b024b8b846c642c7e2354b40a37e0a9f4c0",
|
| 370 |
+
"source_sha256": "4d1cbb7fe86a05756a2433fef62f8b024b8b846c642c7e2354b40a37e0a9f4c0",
|
| 371 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/reacher_easy/model.ckpt-4504500.data-00000-of-00001",
|
| 372 |
+
"sanitized": false
|
| 373 |
+
},
|
| 374 |
+
{
|
| 375 |
+
"path": "planet/reacher_easy/tensorflow/model.ckpt-4504500.index",
|
| 376 |
+
"model": "planet",
|
| 377 |
+
"bytes": 5789,
|
| 378 |
+
"sha256": "5844b59c53652eadad60eca39dbd8e1cd0273510ca78a56b221a7fe904a41df8",
|
| 379 |
+
"source_sha256": "5844b59c53652eadad60eca39dbd8e1cd0273510ca78a56b221a7fe904a41df8",
|
| 380 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/reacher_easy/model.ckpt-4504500.index",
|
| 381 |
+
"sanitized": false
|
| 382 |
+
},
|
| 383 |
+
{
|
| 384 |
+
"path": "planet/reacher_easy/tensorflow/model.ckpt-4504500.meta",
|
| 385 |
+
"model": "planet",
|
| 386 |
+
"bytes": 7112337,
|
| 387 |
+
"sha256": "ca8e8aef416be0b50d17dd164c53f352f1a001e065e2cd3d477c4666a796418e",
|
| 388 |
+
"source_sha256": "ca8e8aef416be0b50d17dd164c53f352f1a001e065e2cd3d477c4666a796418e",
|
| 389 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/reacher_easy/model.ckpt-4504500.meta",
|
| 390 |
+
"sanitized": false
|
| 391 |
+
},
|
| 392 |
+
{
|
| 393 |
+
"path": "planet/reacher_easy/tensorflow/checkpoint",
|
| 394 |
+
"model": "planet",
|
| 395 |
+
"bytes": 93,
|
| 396 |
+
"sha256": "c4b1b9ef54b5185a72880247eb4ab7c71e54cc1b3e15a6ef82cbaf9510ac35d6",
|
| 397 |
+
"source_sha256": "c4b1b9ef54b5185a72880247eb4ab7c71e54cc1b3e15a6ef82cbaf9510ac35d6",
|
| 398 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/reacher_easy/checkpoint",
|
| 399 |
+
"sanitized": false
|
| 400 |
+
},
|
| 401 |
+
{
|
| 402 |
+
"path": "planet/cheetah_run/tensorflow/model.ckpt-4504500.data-00000-of-00001",
|
| 403 |
+
"model": "planet",
|
| 404 |
+
"bytes": 61555552,
|
| 405 |
+
"sha256": "fe62149d086a11b89dcb259d8a6af18b41583747c5b34baecc96ae46d47a0431",
|
| 406 |
+
"source_sha256": "fe62149d086a11b89dcb259d8a6af18b41583747c5b34baecc96ae46d47a0431",
|
| 407 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/cheetah_run/model.ckpt-4504500.data-00000-of-00001",
|
| 408 |
+
"sanitized": false
|
| 409 |
+
},
|
| 410 |
+
{
|
| 411 |
+
"path": "planet/cheetah_run/tensorflow/model.ckpt-4504500.index",
|
| 412 |
+
"model": "planet",
|
| 413 |
+
"bytes": 5787,
|
| 414 |
+
"sha256": "f3c20abc873aa178695036d8d913acc41f1c61d1753f9e87a6fa8fccb8df80f2",
|
| 415 |
+
"source_sha256": "f3c20abc873aa178695036d8d913acc41f1c61d1753f9e87a6fa8fccb8df80f2",
|
| 416 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/cheetah_run/model.ckpt-4504500.index",
|
| 417 |
+
"sanitized": false
|
| 418 |
+
},
|
| 419 |
+
{
|
| 420 |
+
"path": "planet/cheetah_run/tensorflow/model.ckpt-4504500.meta",
|
| 421 |
+
"model": "planet",
|
| 422 |
+
"bytes": 7068294,
|
| 423 |
+
"sha256": "bd92bc2d5226976ffb7cca03252f2ee29753cb40025187cdf29961beb6cdc7c2",
|
| 424 |
+
"source_sha256": "bd92bc2d5226976ffb7cca03252f2ee29753cb40025187cdf29961beb6cdc7c2",
|
| 425 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/cheetah_run/model.ckpt-4504500.meta",
|
| 426 |
+
"sanitized": false
|
| 427 |
+
},
|
| 428 |
+
{
|
| 429 |
+
"path": "planet/cheetah_run/tensorflow/checkpoint",
|
| 430 |
+
"model": "planet",
|
| 431 |
+
"bytes": 93,
|
| 432 |
+
"sha256": "c4b1b9ef54b5185a72880247eb4ab7c71e54cc1b3e15a6ef82cbaf9510ac35d6",
|
| 433 |
+
"source_sha256": "c4b1b9ef54b5185a72880247eb4ab7c71e54cc1b3e15a6ef82cbaf9510ac35d6",
|
| 434 |
+
"source_relative_path": "planet/checkpoints/planet-dmc-rgb-seed0/cheetah_run/checkpoint",
|
| 435 |
+
"sanitized": false
|
| 436 |
+
},
|
| 437 |
+
{
|
| 438 |
+
"path": "le-wm/defenses/cube/residual_hgd/hgd_eps16.best.pt",
|
| 439 |
+
"model": "le-wm",
|
| 440 |
+
"bytes": 5917691,
|
| 441 |
+
"sha256": "68f58d28e6b39954331143c2f66e736f402882f15fce35ace8afc8d464c17f07",
|
| 442 |
+
"source_sha256": "6532554c98c2bfa8982942d06562cf10ed81f96579667855cdf0bac328232480",
|
| 443 |
+
"source_relative_path": "le-wm/.stable-wm/defenses/cube/eps16_encrel_lam0p1/checkpoints/hgd_eps16.best.pt",
|
| 444 |
+
"sanitized": true
|
| 445 |
+
},
|
| 446 |
+
{
|
| 447 |
+
"path": "le-wm/defenses/cube/residual_hgd/hgd_eps16.pt",
|
| 448 |
+
"model": "le-wm",
|
| 449 |
+
"bytes": 5916747,
|
| 450 |
+
"sha256": "bc3af72c1ae2ed328c8907f87e5709b719e00c8dd61e7c32ccb9dc722951772a",
|
| 451 |
+
"source_sha256": "373c0ec72c661465ea7cccadbad0b472443c7dd48ebfd2af4f78052e6457afce",
|
| 452 |
+
"source_relative_path": "le-wm/.stable-wm/defenses/cube/eps16_encrel_lam0p1/checkpoints/hgd_eps16.pt",
|
| 453 |
+
"sanitized": true
|
| 454 |
+
},
|
| 455 |
+
{
|
| 456 |
+
"path": "le-wm/defenses/cube/paper_hgd/hgd_paper.best.pt",
|
| 457 |
+
"model": "le-wm",
|
| 458 |
+
"bytes": 132576538,
|
| 459 |
+
"sha256": "04ed4a9b4d9edea42f4e4ec7b489ae79378890a3c2d505077a97d69d19fd1ec9",
|
| 460 |
+
"source_sha256": "2354e376b01155e7253677e0e16273e31b32416c5d5db3f182423e3bc99ab2cf",
|
| 461 |
+
"source_relative_path": "le-wm/.stable-wm/defenses/cube/hgd_paper_train1000_val100/checkpoints/hgd_paper.best.pt",
|
| 462 |
+
"sanitized": true
|
| 463 |
+
},
|
| 464 |
+
{
|
| 465 |
+
"path": "le-wm/defenses/cube/paper_hgd/hgd_paper.pt",
|
| 466 |
+
"model": "le-wm",
|
| 467 |
+
"bytes": 132574859,
|
| 468 |
+
"sha256": "9ebae97be8b61cefde85958dadea28d5ae34c959b3cd183f4932f6bfe1debdf2",
|
| 469 |
+
"source_sha256": "ca01417994b99b2ef0f9938e4a28b4d6e2a819a52865e2d8246aa5cddca6e087",
|
| 470 |
+
"source_relative_path": "le-wm/.stable-wm/defenses/cube/hgd_paper_train1000_val100/checkpoints/hgd_paper.pt",
|
| 471 |
+
"sanitized": true
|
| 472 |
+
}
|
| 473 |
+
],
|
| 474 |
+
"notes": [
|
| 475 |
+
"Binary research model checkpoints are byte-identical to local sources, except HGD metadata sanitization with exact state_dict tensor equality verified.",
|
| 476 |
+
"Configuration output paths are relative; set them for the destination machine.",
|
| 477 |
+
"Native TensorFlow PlaNet checkpoints, PyTorch inference exports, and TensorFlow parity fixtures are included.",
|
| 478 |
+
"Datasets, videos, episode perturbations, logs, and credentials are excluded.",
|
| 479 |
+
"The source_relative_path field maps this repository back to the research code artifact layout."
|
| 480 |
+
]
|
| 481 |
+
}
|
planet/cartpole_swingup/attack_export.json
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"source_revision": "b26a7fe027c018f62b3178c8e126371fbdee33e5",
|
| 3 |
+
"tensorflow": "2.15.0",
|
| 4 |
+
"inference_tensor_count": 32,
|
| 5 |
+
"source_sha256": {
|
| 6 |
+
"model.ckpt-4504500.data-00000-of-00001": "22b7d2959cab4e3b38fd55a579bb718d59516053f6cb92530c7cabf785cd5ce5",
|
| 7 |
+
"model.ckpt-4504500.index": "b34dd58607ccf858446024be305a73757eb9b65a83ccecde57768974ad13a509",
|
| 8 |
+
"model.ckpt-4504500.meta": "cdd30a7fd3c3bff6ca5e0bcce9c87bfd5056a02462730ac3d05d91e9cce4b1df"
|
| 9 |
+
}
|
| 10 |
+
}
|
planet/cartpole_swingup/attack_weights.npz
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f75ced2942b489b578c750ed99ce5daa9e14a8727ce911b469a7785ce6231580
|
| 3 |
+
size 5341958
|
planet/cartpole_swingup/config.yaml
ADDED
|
@@ -0,0 +1,388 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
!!python/object/new:planet.tools.attr_dict.AttrDict
|
| 2 |
+
dictitems:
|
| 3 |
+
activation: &id001 !!python/name:tensorflow.python.ops.gen_nn_ops.relu ''
|
| 4 |
+
batch_shape: &id004
|
| 5 |
+
- 50
|
| 6 |
+
- 50
|
| 7 |
+
bound_action: !!python/object:planet.tools.bind.bind
|
| 8 |
+
_args: !!python/tuple []
|
| 9 |
+
_fn: !!python/name:planet.tools.bound_action.bound_action ''
|
| 10 |
+
_kwargs:
|
| 11 |
+
strategy: clip
|
| 12 |
+
cell: !!python/object:planet.tools.bind.bind
|
| 13 |
+
_args: !!python/tuple
|
| 14 |
+
- 30
|
| 15 |
+
- 200
|
| 16 |
+
- 200
|
| 17 |
+
- true
|
| 18 |
+
- false
|
| 19 |
+
- 0.1
|
| 20 |
+
- *id001
|
| 21 |
+
- 1
|
| 22 |
+
_fn: !!python/name:planet.models.rssm.RSSM ''
|
| 23 |
+
_kwargs: {}
|
| 24 |
+
checkpoint_to_load: null
|
| 25 |
+
data_loader: !!python/object:planet.tools.bind.bind
|
| 26 |
+
_args: !!python/tuple []
|
| 27 |
+
_fn: !!python/name:planet.tools.numpy_episodes.recent_loader ''
|
| 28 |
+
_kwargs:
|
| 29 |
+
every: 1000
|
| 30 |
+
data_reader: !!python/name:planet.tools.numpy_episodes.episode_reader ''
|
| 31 |
+
debug: false
|
| 32 |
+
decoder: &id002 !!python/name:planet.networks.conv_ha.decoder ''
|
| 33 |
+
encoder: !!python/name:planet.networks.conv_ha.encoder ''
|
| 34 |
+
free_nats: 3.0
|
| 35 |
+
gradient_heads:
|
| 36 |
+
- image
|
| 37 |
+
- reward
|
| 38 |
+
head_network: &id003 !!python/object:planet.tools.bind.bind
|
| 39 |
+
_args: !!python/tuple []
|
| 40 |
+
_fn: !!python/name:planet.networks.basic.feed_forward ''
|
| 41 |
+
_kwargs:
|
| 42 |
+
activation: *id001
|
| 43 |
+
num_layers: 2
|
| 44 |
+
units: 200
|
| 45 |
+
heads: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 46 |
+
dictitems:
|
| 47 |
+
image: *id002
|
| 48 |
+
reward: !!python/object:planet.tools.bind.bind
|
| 49 |
+
_args: !!python/tuple []
|
| 50 |
+
_fn: *id003
|
| 51 |
+
_kwargs:
|
| 52 |
+
stop_gradient: false
|
| 53 |
+
state:
|
| 54 |
+
_defaults: {}
|
| 55 |
+
_touched: !!set
|
| 56 |
+
__getstate__: null
|
| 57 |
+
_unlocked: false
|
| 58 |
+
isolate_envs: thread
|
| 59 |
+
logdir: ./planet/results/planet-single-seed/cartpole_swingup/seed_00/00001
|
| 60 |
+
loss_scales: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 61 |
+
dictitems:
|
| 62 |
+
divergence: 1.0
|
| 63 |
+
global_divergence: 0.0
|
| 64 |
+
image: 1.0
|
| 65 |
+
overshooting: 0.0
|
| 66 |
+
reward: 10.0
|
| 67 |
+
state:
|
| 68 |
+
_defaults: {}
|
| 69 |
+
_touched: !!set
|
| 70 |
+
__getstate__: null
|
| 71 |
+
_unlocked: false
|
| 72 |
+
max_steps: 5000000
|
| 73 |
+
num_chunks: 1
|
| 74 |
+
num_layers: 2
|
| 75 |
+
num_units: 200
|
| 76 |
+
open_loop_context: 5
|
| 77 |
+
optimizers: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 78 |
+
dictitems:
|
| 79 |
+
main: !!python/object:planet.tools.bind.bind
|
| 80 |
+
_args: !!python/tuple []
|
| 81 |
+
_fn: !!python/name:planet.tools.custom_optimizer.CustomOptimizer ''
|
| 82 |
+
_kwargs:
|
| 83 |
+
clipping: 1000.0
|
| 84 |
+
learning_rate: 0.001
|
| 85 |
+
optimizer_cls: !!python/object:planet.tools.bind.bind
|
| 86 |
+
_args: !!python/tuple []
|
| 87 |
+
_fn: !!python/name:tensorflow.python.training.adam.AdamOptimizer ''
|
| 88 |
+
_kwargs:
|
| 89 |
+
epsilon: 0.0001
|
| 90 |
+
state:
|
| 91 |
+
_defaults: {}
|
| 92 |
+
_touched: !!set
|
| 93 |
+
__getstate__: null
|
| 94 |
+
_unlocked: false
|
| 95 |
+
os_stop_posterior_grad: true
|
| 96 |
+
overshooting_distance: 0
|
| 97 |
+
postprocess_fn: !!python/object:planet.tools.bind.bind
|
| 98 |
+
_args: !!python/tuple []
|
| 99 |
+
_fn: !!python/name:planet.tools.preprocess.postprocess ''
|
| 100 |
+
_kwargs:
|
| 101 |
+
bits: 5
|
| 102 |
+
preprocess_fn: !!python/object:planet.tools.bind.bind
|
| 103 |
+
_args: !!python/tuple []
|
| 104 |
+
_fn: !!python/name:planet.tools.preprocess.preprocess ''
|
| 105 |
+
_kwargs:
|
| 106 |
+
bits: 5
|
| 107 |
+
print_metrics_every: 5000
|
| 108 |
+
random_collects: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 109 |
+
dictitems:
|
| 110 |
+
test-cartpole_swingup: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 111 |
+
dictitems:
|
| 112 |
+
num_episodes: 5
|
| 113 |
+
save_episode_dir:
|
| 114 |
+
./planet/results/planet-single-seed/cartpole_swingup/seed_00/00001/test_episodes
|
| 115 |
+
task: &id005 !!python/object/new:planet.scripts.tasks.Task
|
| 116 |
+
- cartpole_swingup
|
| 117 |
+
- !!python/object:planet.tools.bind.bind
|
| 118 |
+
_args: !!python/tuple
|
| 119 |
+
- 8
|
| 120 |
+
- 125
|
| 121 |
+
- cartpole
|
| 122 |
+
- swingup
|
| 123 |
+
- &id006 !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 124 |
+
dictitems:
|
| 125 |
+
activation: relu
|
| 126 |
+
batch_shape: *id004
|
| 127 |
+
collect_every: 5000
|
| 128 |
+
divergence_scale: 1.0
|
| 129 |
+
free_nats: 3.0
|
| 130 |
+
global_div_scale: 0.0
|
| 131 |
+
image_bits: 5
|
| 132 |
+
isolate_envs: thread
|
| 133 |
+
logdir: ./planet/results/planet-single-seed/cartpole_swingup/seed_00/00001
|
| 134 |
+
main_gradient_clipping: 1000.0
|
| 135 |
+
main_learning_rate: 0.001
|
| 136 |
+
max_steps: 5000000
|
| 137 |
+
model: rssm
|
| 138 |
+
model_size: 200
|
| 139 |
+
num_layers: 2
|
| 140 |
+
num_seed_episodes: 5
|
| 141 |
+
num_units: 200
|
| 142 |
+
overshooting_scale: 0.0
|
| 143 |
+
planner_amount: 1000
|
| 144 |
+
planner_horizon: 12
|
| 145 |
+
planner_iterations: 10
|
| 146 |
+
planner_topk: 100
|
| 147 |
+
seed: 0
|
| 148 |
+
state_size: 30
|
| 149 |
+
tasks:
|
| 150 |
+
- cartpole_swingup
|
| 151 |
+
test_collects:
|
| 152 |
+
- batch_size: 10
|
| 153 |
+
test_steps: 50
|
| 154 |
+
train_action_noise: 0.3
|
| 155 |
+
train_steps: 50000
|
| 156 |
+
state:
|
| 157 |
+
_defaults: {}
|
| 158 |
+
_touched: !!set
|
| 159 |
+
__getstate__: null
|
| 160 |
+
action_noise_min: null
|
| 161 |
+
action_noise_ramp: null
|
| 162 |
+
action_repeat: null
|
| 163 |
+
activation: null
|
| 164 |
+
batch_shape: null
|
| 165 |
+
bound_action: null
|
| 166 |
+
camera_id: null
|
| 167 |
+
checkpoint_every: null
|
| 168 |
+
collect_every: null
|
| 169 |
+
collect_objective: null
|
| 170 |
+
divergence_scale: null
|
| 171 |
+
free_nats: null
|
| 172 |
+
future_rnn: null
|
| 173 |
+
global_div_scale: null
|
| 174 |
+
gradient_heads: null
|
| 175 |
+
image_bits: null
|
| 176 |
+
image_loss_scale: null
|
| 177 |
+
isolate_envs: null
|
| 178 |
+
loader: null
|
| 179 |
+
loader_every: null
|
| 180 |
+
logdir: null
|
| 181 |
+
main_gradient_clipping: null
|
| 182 |
+
main_learning_rate: null
|
| 183 |
+
max_steps: null
|
| 184 |
+
mean_only: null
|
| 185 |
+
min_stddev: null
|
| 186 |
+
model: null
|
| 187 |
+
model_layers: null
|
| 188 |
+
model_size: null
|
| 189 |
+
network: null
|
| 190 |
+
num_chunks: null
|
| 191 |
+
num_layers: null
|
| 192 |
+
num_seed_episodes: null
|
| 193 |
+
num_units: null
|
| 194 |
+
os_stop_posterior_grad: null
|
| 195 |
+
overshooting_distance: null
|
| 196 |
+
overshooting_scale: null
|
| 197 |
+
planner: null
|
| 198 |
+
planner_amount: null
|
| 199 |
+
planner_horizon: null
|
| 200 |
+
planner_iterations: null
|
| 201 |
+
planner_topk: null
|
| 202 |
+
reward_loss_scale: null
|
| 203 |
+
seed: null
|
| 204 |
+
state_diagnostics: null
|
| 205 |
+
state_size: null
|
| 206 |
+
tasks: null
|
| 207 |
+
test_collects: null
|
| 208 |
+
test_steps: null
|
| 209 |
+
train_action_noise: null
|
| 210 |
+
train_collects: null
|
| 211 |
+
train_steps: null
|
| 212 |
+
_unlocked: false
|
| 213 |
+
_fn: !!python/name:planet.scripts.tasks._dm_control_env ''
|
| 214 |
+
_kwargs: {}
|
| 215 |
+
- 125
|
| 216 |
+
- - reward
|
| 217 |
+
- position
|
| 218 |
+
- velocity
|
| 219 |
+
state:
|
| 220 |
+
_defaults: {}
|
| 221 |
+
_touched: !!set
|
| 222 |
+
__getstate__: null
|
| 223 |
+
num_episodes: null
|
| 224 |
+
save_episode_dir: null
|
| 225 |
+
task: null
|
| 226 |
+
_unlocked: false
|
| 227 |
+
train-cartpole_swingup: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 228 |
+
dictitems:
|
| 229 |
+
num_episodes: 5
|
| 230 |
+
save_episode_dir:
|
| 231 |
+
./planet/results/planet-single-seed/cartpole_swingup/seed_00/00001/train_episodes
|
| 232 |
+
task: *id005
|
| 233 |
+
state:
|
| 234 |
+
_defaults: {}
|
| 235 |
+
_touched: !!set
|
| 236 |
+
__getstate__: null
|
| 237 |
+
num_episodes: null
|
| 238 |
+
save_episode_dir: null
|
| 239 |
+
task: null
|
| 240 |
+
_unlocked: false
|
| 241 |
+
state:
|
| 242 |
+
_defaults: {}
|
| 243 |
+
_touched: !!set
|
| 244 |
+
__getstate__: null
|
| 245 |
+
_unlocked: false
|
| 246 |
+
savers:
|
| 247 |
+
- !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 248 |
+
dictitems:
|
| 249 |
+
exclude: !!python/tuple
|
| 250 |
+
- .*_temporary.*
|
| 251 |
+
state:
|
| 252 |
+
_defaults: {}
|
| 253 |
+
_touched: !!set
|
| 254 |
+
__getstate__: null
|
| 255 |
+
_unlocked: false
|
| 256 |
+
tasks:
|
| 257 |
+
- *id005
|
| 258 |
+
test_checkpoint_every: 500
|
| 259 |
+
test_collects: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 260 |
+
dictitems:
|
| 261 |
+
test_main_cartpole_swingup: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 262 |
+
dictitems:
|
| 263 |
+
exploration: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 264 |
+
dictitems:
|
| 265 |
+
scale: 0.0
|
| 266 |
+
schedule: !!python/object:planet.tools.bind.bind
|
| 267 |
+
_args: !!python/tuple []
|
| 268 |
+
_fn: &id007 !!python/name:planet.tools.schedule.linear ''
|
| 269 |
+
_kwargs:
|
| 270 |
+
min: 0.0
|
| 271 |
+
ramp: 0
|
| 272 |
+
state:
|
| 273 |
+
_defaults: {}
|
| 274 |
+
_touched: !!set
|
| 275 |
+
__getstate__: null
|
| 276 |
+
_unlocked: false
|
| 277 |
+
num_agents: 10
|
| 278 |
+
objective: !!python/object:planet.tools.bind.bind
|
| 279 |
+
_args: !!python/tuple []
|
| 280 |
+
_fn: &id008 !!python/name:planet.scripts.objectives.reward ''
|
| 281 |
+
_kwargs:
|
| 282 |
+
params: *id006
|
| 283 |
+
planner: !!python/object:planet.tools.bind.bind
|
| 284 |
+
_args: !!python/tuple []
|
| 285 |
+
_fn: &id009 !!python/name:planet.control.planning.cross_entropy_method ''
|
| 286 |
+
_kwargs:
|
| 287 |
+
amount: 1000
|
| 288 |
+
horizon: 12
|
| 289 |
+
iterations: 10
|
| 290 |
+
topk: 100
|
| 291 |
+
save_episode_dir:
|
| 292 |
+
./planet/results/planet-single-seed/cartpole_swingup/seed_00/00001/test_episodes
|
| 293 |
+
steps_after: 5000
|
| 294 |
+
steps_every: 5000
|
| 295 |
+
steps_until: -1
|
| 296 |
+
task: *id005
|
| 297 |
+
state:
|
| 298 |
+
_defaults: {}
|
| 299 |
+
_touched: !!set
|
| 300 |
+
__getstate__: null
|
| 301 |
+
_unlocked: false
|
| 302 |
+
state:
|
| 303 |
+
_defaults: {}
|
| 304 |
+
_touched: !!set
|
| 305 |
+
__getstate__: null
|
| 306 |
+
_unlocked: false
|
| 307 |
+
test_dir:
|
| 308 |
+
./planet/results/planet-single-seed/cartpole_swingup/seed_00/00001/test_episodes
|
| 309 |
+
test_steps: 50
|
| 310 |
+
train_checkpoint_every: null
|
| 311 |
+
train_collects: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 312 |
+
dictitems:
|
| 313 |
+
train_main_cartpole_swingup: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 314 |
+
dictitems:
|
| 315 |
+
exploration: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 316 |
+
dictitems:
|
| 317 |
+
scale: 0.3
|
| 318 |
+
schedule: !!python/object:planet.tools.bind.bind
|
| 319 |
+
_args: !!python/tuple []
|
| 320 |
+
_fn: *id007
|
| 321 |
+
_kwargs:
|
| 322 |
+
min: 0.0
|
| 323 |
+
ramp: 0
|
| 324 |
+
state:
|
| 325 |
+
_defaults: {}
|
| 326 |
+
_touched: !!set
|
| 327 |
+
__getstate__: null
|
| 328 |
+
_unlocked: false
|
| 329 |
+
num_agents: 1
|
| 330 |
+
objective: !!python/object:planet.tools.bind.bind
|
| 331 |
+
_args: !!python/tuple []
|
| 332 |
+
_fn: *id008
|
| 333 |
+
_kwargs:
|
| 334 |
+
params: *id006
|
| 335 |
+
planner: !!python/object:planet.tools.bind.bind
|
| 336 |
+
_args: !!python/tuple []
|
| 337 |
+
_fn: *id009
|
| 338 |
+
_kwargs:
|
| 339 |
+
amount: 1000
|
| 340 |
+
horizon: 12
|
| 341 |
+
iterations: 10
|
| 342 |
+
topk: 100
|
| 343 |
+
save_episode_dir:
|
| 344 |
+
./planet/results/planet-single-seed/cartpole_swingup/seed_00/00001/train_episodes
|
| 345 |
+
steps_after: 5000
|
| 346 |
+
steps_every: 5000
|
| 347 |
+
steps_until: -1
|
| 348 |
+
task: *id005
|
| 349 |
+
state:
|
| 350 |
+
_defaults: {}
|
| 351 |
+
_touched: !!set
|
| 352 |
+
__getstate__: null
|
| 353 |
+
_unlocked: false
|
| 354 |
+
state:
|
| 355 |
+
_defaults: {}
|
| 356 |
+
_touched: !!set
|
| 357 |
+
__getstate__: null
|
| 358 |
+
_unlocked: false
|
| 359 |
+
train_dir:
|
| 360 |
+
./planet/results/planet-single-seed/cartpole_swingup/seed_00/00001/train_episodes
|
| 361 |
+
train_log_every: 50000
|
| 362 |
+
train_steps: 50000
|
| 363 |
+
state:
|
| 364 |
+
_defaults: {}
|
| 365 |
+
_touched: !!set
|
| 366 |
+
__getstate__: null
|
| 367 |
+
activation: null
|
| 368 |
+
batch_shape: null
|
| 369 |
+
data_loader: null
|
| 370 |
+
data_reader: null
|
| 371 |
+
decoder: null
|
| 372 |
+
gradient_heads: null
|
| 373 |
+
head_network: null
|
| 374 |
+
heads: null
|
| 375 |
+
logdir: null
|
| 376 |
+
loss_scales: null
|
| 377 |
+
num_chunks: null
|
| 378 |
+
num_layers: null
|
| 379 |
+
num_units: null
|
| 380 |
+
optimizers: null
|
| 381 |
+
preprocess_fn: null
|
| 382 |
+
random_collects: null
|
| 383 |
+
tasks: null
|
| 384 |
+
test_dir: null
|
| 385 |
+
test_steps: null
|
| 386 |
+
train_dir: null
|
| 387 |
+
train_steps: null
|
| 388 |
+
_unlocked: false
|
planet/cartpole_swingup/tensorflow/checkpoint
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
model_checkpoint_path: "model.ckpt-4504500"
|
| 2 |
+
all_model_checkpoint_paths: "model.ckpt-4504500"
|
planet/cartpole_swingup/tensorflow/model.ckpt-4504500.data-00000-of-00001
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:22b7d2959cab4e3b38fd55a579bb718d59516053f6cb92530c7cabf785cd5ce5
|
| 3 |
+
size 61543552
|
planet/cartpole_swingup/tensorflow/model.ckpt-4504500.index
ADDED
|
Binary file (5.8 kB). View file
|
|
|
planet/cartpole_swingup/tensorflow/model.ckpt-4504500.meta
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cdd30a7fd3c3bff6ca5e0bcce9c87bfd5056a02462730ac3d05d91e9cce4b1df
|
| 3 |
+
size 7264371
|
planet/cartpole_swingup/tf_reference.npz
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:62b9163ef512d1a397af51cd1320b5721f59ea8cb7283b8be1f320d6dd7035a0
|
| 3 |
+
size 111936
|
planet/cheetah_run/attack_export.json
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"source_revision": "b26a7fe027c018f62b3178c8e126371fbdee33e5",
|
| 3 |
+
"tensorflow": "2.15.0",
|
| 4 |
+
"inference_tensor_count": 32,
|
| 5 |
+
"source_sha256": {
|
| 6 |
+
"model.ckpt-4504500.data-00000-of-00001": "fe62149d086a11b89dcb259d8a6af18b41583747c5b34baecc96ae46d47a0431",
|
| 7 |
+
"model.ckpt-4504500.index": "f3c20abc873aa178695036d8d913acc41f1c61d1753f9e87a6fa8fccb8df80f2",
|
| 8 |
+
"model.ckpt-4504500.meta": "bd92bc2d5226976ffb7cca03252f2ee29753cb40025187cdf29961beb6cdc7c2"
|
| 9 |
+
}
|
| 10 |
+
}
|
planet/cheetah_run/attack_weights.npz
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:fdf26210c8304305b8b5903d730dbdfe8b56c6a358524dba3c35d15048f4c968
|
| 3 |
+
size 5345958
|
planet/cheetah_run/config.yaml
ADDED
|
@@ -0,0 +1,388 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
!!python/object/new:planet.tools.attr_dict.AttrDict
|
| 2 |
+
dictitems:
|
| 3 |
+
activation: &id001 !!python/name:tensorflow.python.ops.gen_nn_ops.relu ''
|
| 4 |
+
batch_shape: &id004
|
| 5 |
+
- 50
|
| 6 |
+
- 50
|
| 7 |
+
bound_action: !!python/object:planet.tools.bind.bind
|
| 8 |
+
_args: !!python/tuple []
|
| 9 |
+
_fn: !!python/name:planet.tools.bound_action.bound_action ''
|
| 10 |
+
_kwargs:
|
| 11 |
+
strategy: clip
|
| 12 |
+
cell: !!python/object:planet.tools.bind.bind
|
| 13 |
+
_args: !!python/tuple
|
| 14 |
+
- 30
|
| 15 |
+
- 200
|
| 16 |
+
- 200
|
| 17 |
+
- true
|
| 18 |
+
- false
|
| 19 |
+
- 0.1
|
| 20 |
+
- *id001
|
| 21 |
+
- 1
|
| 22 |
+
_fn: !!python/name:planet.models.rssm.RSSM ''
|
| 23 |
+
_kwargs: {}
|
| 24 |
+
checkpoint_to_load: null
|
| 25 |
+
data_loader: !!python/object:planet.tools.bind.bind
|
| 26 |
+
_args: !!python/tuple []
|
| 27 |
+
_fn: !!python/name:planet.tools.numpy_episodes.recent_loader ''
|
| 28 |
+
_kwargs:
|
| 29 |
+
every: 1000
|
| 30 |
+
data_reader: !!python/name:planet.tools.numpy_episodes.episode_reader ''
|
| 31 |
+
debug: false
|
| 32 |
+
decoder: &id002 !!python/name:planet.networks.conv_ha.decoder ''
|
| 33 |
+
encoder: !!python/name:planet.networks.conv_ha.encoder ''
|
| 34 |
+
free_nats: 3.0
|
| 35 |
+
gradient_heads:
|
| 36 |
+
- image
|
| 37 |
+
- reward
|
| 38 |
+
head_network: &id003 !!python/object:planet.tools.bind.bind
|
| 39 |
+
_args: !!python/tuple []
|
| 40 |
+
_fn: !!python/name:planet.networks.basic.feed_forward ''
|
| 41 |
+
_kwargs:
|
| 42 |
+
activation: *id001
|
| 43 |
+
num_layers: 2
|
| 44 |
+
units: 200
|
| 45 |
+
heads: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 46 |
+
dictitems:
|
| 47 |
+
image: *id002
|
| 48 |
+
reward: !!python/object:planet.tools.bind.bind
|
| 49 |
+
_args: !!python/tuple []
|
| 50 |
+
_fn: *id003
|
| 51 |
+
_kwargs:
|
| 52 |
+
stop_gradient: false
|
| 53 |
+
state:
|
| 54 |
+
_defaults: {}
|
| 55 |
+
_touched: !!set
|
| 56 |
+
__getstate__: null
|
| 57 |
+
_unlocked: false
|
| 58 |
+
isolate_envs: thread
|
| 59 |
+
logdir: ./planet/results/planet-single-seed/cheetah_run/seed_00/00001
|
| 60 |
+
loss_scales: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 61 |
+
dictitems:
|
| 62 |
+
divergence: 1.0
|
| 63 |
+
global_divergence: 0.0
|
| 64 |
+
image: 1.0
|
| 65 |
+
overshooting: 0.0
|
| 66 |
+
reward: 10.0
|
| 67 |
+
state:
|
| 68 |
+
_defaults: {}
|
| 69 |
+
_touched: !!set
|
| 70 |
+
__getstate__: null
|
| 71 |
+
_unlocked: false
|
| 72 |
+
max_steps: 5000000
|
| 73 |
+
num_chunks: 1
|
| 74 |
+
num_layers: 2
|
| 75 |
+
num_units: 200
|
| 76 |
+
open_loop_context: 5
|
| 77 |
+
optimizers: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 78 |
+
dictitems:
|
| 79 |
+
main: !!python/object:planet.tools.bind.bind
|
| 80 |
+
_args: !!python/tuple []
|
| 81 |
+
_fn: !!python/name:planet.tools.custom_optimizer.CustomOptimizer ''
|
| 82 |
+
_kwargs:
|
| 83 |
+
clipping: 1000.0
|
| 84 |
+
learning_rate: 0.001
|
| 85 |
+
optimizer_cls: !!python/object:planet.tools.bind.bind
|
| 86 |
+
_args: !!python/tuple []
|
| 87 |
+
_fn: !!python/name:tensorflow.python.training.adam.AdamOptimizer ''
|
| 88 |
+
_kwargs:
|
| 89 |
+
epsilon: 0.0001
|
| 90 |
+
state:
|
| 91 |
+
_defaults: {}
|
| 92 |
+
_touched: !!set
|
| 93 |
+
__getstate__: null
|
| 94 |
+
_unlocked: false
|
| 95 |
+
os_stop_posterior_grad: true
|
| 96 |
+
overshooting_distance: 0
|
| 97 |
+
postprocess_fn: !!python/object:planet.tools.bind.bind
|
| 98 |
+
_args: !!python/tuple []
|
| 99 |
+
_fn: !!python/name:planet.tools.preprocess.postprocess ''
|
| 100 |
+
_kwargs:
|
| 101 |
+
bits: 5
|
| 102 |
+
preprocess_fn: !!python/object:planet.tools.bind.bind
|
| 103 |
+
_args: !!python/tuple []
|
| 104 |
+
_fn: !!python/name:planet.tools.preprocess.preprocess ''
|
| 105 |
+
_kwargs:
|
| 106 |
+
bits: 5
|
| 107 |
+
print_metrics_every: 5000
|
| 108 |
+
random_collects: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 109 |
+
dictitems:
|
| 110 |
+
test-cheetah_run: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 111 |
+
dictitems:
|
| 112 |
+
num_episodes: 5
|
| 113 |
+
save_episode_dir:
|
| 114 |
+
./planet/results/planet-single-seed/cheetah_run/seed_00/00001/test_episodes
|
| 115 |
+
task: &id005 !!python/object/new:planet.scripts.tasks.Task
|
| 116 |
+
- cheetah_run
|
| 117 |
+
- !!python/object:planet.tools.bind.bind
|
| 118 |
+
_args: !!python/tuple
|
| 119 |
+
- 4
|
| 120 |
+
- 250
|
| 121 |
+
- cheetah
|
| 122 |
+
- run
|
| 123 |
+
- &id006 !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 124 |
+
dictitems:
|
| 125 |
+
activation: relu
|
| 126 |
+
batch_shape: *id004
|
| 127 |
+
collect_every: 5000
|
| 128 |
+
divergence_scale: 1.0
|
| 129 |
+
free_nats: 3.0
|
| 130 |
+
global_div_scale: 0.0
|
| 131 |
+
image_bits: 5
|
| 132 |
+
isolate_envs: thread
|
| 133 |
+
logdir: ./planet/results/planet-single-seed/cheetah_run/seed_00/00001
|
| 134 |
+
main_gradient_clipping: 1000.0
|
| 135 |
+
main_learning_rate: 0.001
|
| 136 |
+
max_steps: 5000000
|
| 137 |
+
model: rssm
|
| 138 |
+
model_size: 200
|
| 139 |
+
num_layers: 2
|
| 140 |
+
num_seed_episodes: 5
|
| 141 |
+
num_units: 200
|
| 142 |
+
overshooting_scale: 0.0
|
| 143 |
+
planner_amount: 1000
|
| 144 |
+
planner_horizon: 12
|
| 145 |
+
planner_iterations: 10
|
| 146 |
+
planner_topk: 100
|
| 147 |
+
seed: 0
|
| 148 |
+
state_size: 30
|
| 149 |
+
tasks:
|
| 150 |
+
- cheetah_run
|
| 151 |
+
test_collects:
|
| 152 |
+
- batch_size: 10
|
| 153 |
+
test_steps: 50
|
| 154 |
+
train_action_noise: 0.3
|
| 155 |
+
train_steps: 50000
|
| 156 |
+
state:
|
| 157 |
+
_defaults: {}
|
| 158 |
+
_touched: !!set
|
| 159 |
+
__getstate__: null
|
| 160 |
+
action_noise_min: null
|
| 161 |
+
action_noise_ramp: null
|
| 162 |
+
action_repeat: null
|
| 163 |
+
activation: null
|
| 164 |
+
batch_shape: null
|
| 165 |
+
bound_action: null
|
| 166 |
+
camera_id: null
|
| 167 |
+
checkpoint_every: null
|
| 168 |
+
collect_every: null
|
| 169 |
+
collect_objective: null
|
| 170 |
+
divergence_scale: null
|
| 171 |
+
free_nats: null
|
| 172 |
+
future_rnn: null
|
| 173 |
+
global_div_scale: null
|
| 174 |
+
gradient_heads: null
|
| 175 |
+
image_bits: null
|
| 176 |
+
image_loss_scale: null
|
| 177 |
+
isolate_envs: null
|
| 178 |
+
loader: null
|
| 179 |
+
loader_every: null
|
| 180 |
+
logdir: null
|
| 181 |
+
main_gradient_clipping: null
|
| 182 |
+
main_learning_rate: null
|
| 183 |
+
max_steps: null
|
| 184 |
+
mean_only: null
|
| 185 |
+
min_stddev: null
|
| 186 |
+
model: null
|
| 187 |
+
model_layers: null
|
| 188 |
+
model_size: null
|
| 189 |
+
network: null
|
| 190 |
+
num_chunks: null
|
| 191 |
+
num_layers: null
|
| 192 |
+
num_seed_episodes: null
|
| 193 |
+
num_units: null
|
| 194 |
+
os_stop_posterior_grad: null
|
| 195 |
+
overshooting_distance: null
|
| 196 |
+
overshooting_scale: null
|
| 197 |
+
planner: null
|
| 198 |
+
planner_amount: null
|
| 199 |
+
planner_horizon: null
|
| 200 |
+
planner_iterations: null
|
| 201 |
+
planner_topk: null
|
| 202 |
+
reward_loss_scale: null
|
| 203 |
+
seed: null
|
| 204 |
+
state_diagnostics: null
|
| 205 |
+
state_size: null
|
| 206 |
+
tasks: null
|
| 207 |
+
test_collects: null
|
| 208 |
+
test_steps: null
|
| 209 |
+
train_action_noise: null
|
| 210 |
+
train_collects: null
|
| 211 |
+
train_steps: null
|
| 212 |
+
_unlocked: false
|
| 213 |
+
_fn: !!python/name:planet.scripts.tasks._dm_control_env ''
|
| 214 |
+
_kwargs: {}
|
| 215 |
+
- 250
|
| 216 |
+
- - reward
|
| 217 |
+
- position
|
| 218 |
+
- velocity
|
| 219 |
+
state:
|
| 220 |
+
_defaults: {}
|
| 221 |
+
_touched: !!set
|
| 222 |
+
__getstate__: null
|
| 223 |
+
num_episodes: null
|
| 224 |
+
save_episode_dir: null
|
| 225 |
+
task: null
|
| 226 |
+
_unlocked: false
|
| 227 |
+
train-cheetah_run: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 228 |
+
dictitems:
|
| 229 |
+
num_episodes: 5
|
| 230 |
+
save_episode_dir:
|
| 231 |
+
./planet/results/planet-single-seed/cheetah_run/seed_00/00001/train_episodes
|
| 232 |
+
task: *id005
|
| 233 |
+
state:
|
| 234 |
+
_defaults: {}
|
| 235 |
+
_touched: !!set
|
| 236 |
+
__getstate__: null
|
| 237 |
+
num_episodes: null
|
| 238 |
+
save_episode_dir: null
|
| 239 |
+
task: null
|
| 240 |
+
_unlocked: false
|
| 241 |
+
state:
|
| 242 |
+
_defaults: {}
|
| 243 |
+
_touched: !!set
|
| 244 |
+
__getstate__: null
|
| 245 |
+
_unlocked: false
|
| 246 |
+
savers:
|
| 247 |
+
- !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 248 |
+
dictitems:
|
| 249 |
+
exclude: !!python/tuple
|
| 250 |
+
- .*_temporary.*
|
| 251 |
+
state:
|
| 252 |
+
_defaults: {}
|
| 253 |
+
_touched: !!set
|
| 254 |
+
__getstate__: null
|
| 255 |
+
_unlocked: false
|
| 256 |
+
tasks:
|
| 257 |
+
- *id005
|
| 258 |
+
test_checkpoint_every: 500
|
| 259 |
+
test_collects: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 260 |
+
dictitems:
|
| 261 |
+
test_main_cheetah_run: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 262 |
+
dictitems:
|
| 263 |
+
exploration: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 264 |
+
dictitems:
|
| 265 |
+
scale: 0.0
|
| 266 |
+
schedule: !!python/object:planet.tools.bind.bind
|
| 267 |
+
_args: !!python/tuple []
|
| 268 |
+
_fn: &id007 !!python/name:planet.tools.schedule.linear ''
|
| 269 |
+
_kwargs:
|
| 270 |
+
min: 0.0
|
| 271 |
+
ramp: 0
|
| 272 |
+
state:
|
| 273 |
+
_defaults: {}
|
| 274 |
+
_touched: !!set
|
| 275 |
+
__getstate__: null
|
| 276 |
+
_unlocked: false
|
| 277 |
+
num_agents: 10
|
| 278 |
+
objective: !!python/object:planet.tools.bind.bind
|
| 279 |
+
_args: !!python/tuple []
|
| 280 |
+
_fn: &id008 !!python/name:planet.scripts.objectives.reward ''
|
| 281 |
+
_kwargs:
|
| 282 |
+
params: *id006
|
| 283 |
+
planner: !!python/object:planet.tools.bind.bind
|
| 284 |
+
_args: !!python/tuple []
|
| 285 |
+
_fn: &id009 !!python/name:planet.control.planning.cross_entropy_method ''
|
| 286 |
+
_kwargs:
|
| 287 |
+
amount: 1000
|
| 288 |
+
horizon: 12
|
| 289 |
+
iterations: 10
|
| 290 |
+
topk: 100
|
| 291 |
+
save_episode_dir:
|
| 292 |
+
./planet/results/planet-single-seed/cheetah_run/seed_00/00001/test_episodes
|
| 293 |
+
steps_after: 5000
|
| 294 |
+
steps_every: 5000
|
| 295 |
+
steps_until: -1
|
| 296 |
+
task: *id005
|
| 297 |
+
state:
|
| 298 |
+
_defaults: {}
|
| 299 |
+
_touched: !!set
|
| 300 |
+
__getstate__: null
|
| 301 |
+
_unlocked: false
|
| 302 |
+
state:
|
| 303 |
+
_defaults: {}
|
| 304 |
+
_touched: !!set
|
| 305 |
+
__getstate__: null
|
| 306 |
+
_unlocked: false
|
| 307 |
+
test_dir:
|
| 308 |
+
./planet/results/planet-single-seed/cheetah_run/seed_00/00001/test_episodes
|
| 309 |
+
test_steps: 50
|
| 310 |
+
train_checkpoint_every: null
|
| 311 |
+
train_collects: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 312 |
+
dictitems:
|
| 313 |
+
train_main_cheetah_run: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 314 |
+
dictitems:
|
| 315 |
+
exploration: !!python/object/new:planet.tools.attr_dict.AttrDict
|
| 316 |
+
dictitems:
|
| 317 |
+
scale: 0.3
|
| 318 |
+
schedule: !!python/object:planet.tools.bind.bind
|
| 319 |
+
_args: !!python/tuple []
|
| 320 |
+
_fn: *id007
|
| 321 |
+
_kwargs:
|
| 322 |
+
min: 0.0
|
| 323 |
+
ramp: 0
|
| 324 |
+
state:
|
| 325 |
+
_defaults: {}
|
| 326 |
+
_touched: !!set
|
| 327 |
+
__getstate__: null
|
| 328 |
+
_unlocked: false
|
| 329 |
+
num_agents: 1
|
| 330 |
+
objective: !!python/object:planet.tools.bind.bind
|
| 331 |
+
_args: !!python/tuple []
|
| 332 |
+
_fn: *id008
|
| 333 |
+
_kwargs:
|
| 334 |
+
params: *id006
|
| 335 |
+
planner: !!python/object:planet.tools.bind.bind
|
| 336 |
+
_args: !!python/tuple []
|
| 337 |
+
_fn: *id009
|
| 338 |
+
_kwargs:
|
| 339 |
+
amount: 1000
|
| 340 |
+
horizon: 12
|
| 341 |
+
iterations: 10
|
| 342 |
+
topk: 100
|
| 343 |
+
save_episode_dir:
|
| 344 |
+
./planet/results/planet-single-seed/cheetah_run/seed_00/00001/train_episodes
|
| 345 |
+
steps_after: 5000
|
| 346 |
+
steps_every: 5000
|
| 347 |
+
steps_until: -1
|
| 348 |
+
task: *id005
|
| 349 |
+
state:
|
| 350 |
+
_defaults: {}
|
| 351 |
+
_touched: !!set
|
| 352 |
+
__getstate__: null
|
| 353 |
+
_unlocked: false
|
| 354 |
+
state:
|
| 355 |
+
_defaults: {}
|
| 356 |
+
_touched: !!set
|
| 357 |
+
__getstate__: null
|
| 358 |
+
_unlocked: false
|
| 359 |
+
train_dir:
|
| 360 |
+
./planet/results/planet-single-seed/cheetah_run/seed_00/00001/train_episodes
|
| 361 |
+
train_log_every: 50000
|
| 362 |
+
train_steps: 50000
|
| 363 |
+
state:
|
| 364 |
+
_defaults: {}
|
| 365 |
+
_touched: !!set
|
| 366 |
+
__getstate__: null
|
| 367 |
+
activation: null
|
| 368 |
+
batch_shape: null
|
| 369 |
+
data_loader: null
|
| 370 |
+
data_reader: null
|
| 371 |
+
decoder: null
|
| 372 |
+
gradient_heads: null
|
| 373 |
+
head_network: null
|
| 374 |
+
heads: null
|
| 375 |
+
logdir: null
|
| 376 |
+
loss_scales: null
|
| 377 |
+
num_chunks: null
|
| 378 |
+
num_layers: null
|
| 379 |
+
num_units: null
|
| 380 |
+
optimizers: null
|
| 381 |
+
preprocess_fn: null
|
| 382 |
+
random_collects: null
|
| 383 |
+
tasks: null
|
| 384 |
+
test_dir: null
|
| 385 |
+
test_steps: null
|
| 386 |
+
train_dir: null
|
| 387 |
+
train_steps: null
|
| 388 |
+
_unlocked: false
|
planet/cheetah_run/tensorflow/checkpoint
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
model_checkpoint_path: "model.ckpt-4504500"
|
| 2 |
+
all_model_checkpoint_paths: "model.ckpt-4504500"
|
planet/cheetah_run/tensorflow/model.ckpt-4504500.data-00000-of-00001
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:fe62149d086a11b89dcb259d8a6af18b41583747c5b34baecc96ae46d47a0431
|
| 3 |
+
size 61555552
|
planet/cheetah_run/tensorflow/model.ckpt-4504500.index
ADDED
|
Binary file (5.79 kB). View file
|
|
|