mohendy commited on
Commit
27dce55
·
verified ·
1 Parent(s): 2de3da0

Add sanitized world-model and HGD research checkpoints

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. .gitattributes +6 -0
  2. LICENSES.md +17 -0
  3. README.md +87 -0
  4. dino_wm/pusht/hydra.yaml +96 -0
  5. dino_wm/pusht/model_latest.pth +3 -0
  6. dino_wm/runtime.json +103 -0
  7. dino_wm/wall_single/hydra.yaml +96 -0
  8. dino_wm/wall_single/model_latest.pth +3 -0
  9. dreamerv3/cheetah_run/ckpt/20260901T093748F019112/agent.pkl +3 -0
  10. dreamerv3/cheetah_run/ckpt/20260901T093748F019112/done +0 -0
  11. dreamerv3/cheetah_run/ckpt/20260901T093748F019112/step.pkl +3 -0
  12. dreamerv3/cheetah_run/config.yaml +185 -0
  13. dreamerv3/reacher_easy/ckpt/20260830T195435F325934/agent.pkl +3 -0
  14. dreamerv3/reacher_easy/ckpt/20260830T195435F325934/done +0 -0
  15. dreamerv3/reacher_easy/ckpt/20260830T195435F325934/step.pkl +3 -0
  16. dreamerv3/reacher_easy/config.yaml +185 -0
  17. dreamerv3/runtime.json +126 -0
  18. le-wm/cube/config.json +49 -0
  19. le-wm/cube/weights.pt +3 -0
  20. le-wm/defenses/cube/paper_hgd/hgd_paper.best.pt +3 -0
  21. le-wm/defenses/cube/paper_hgd/hgd_paper.pt +3 -0
  22. le-wm/defenses/cube/residual_hgd/hgd_eps16.best.pt +3 -0
  23. le-wm/defenses/cube/residual_hgd/hgd_eps16.pt +3 -0
  24. le-wm/pusht/config.json +49 -0
  25. le-wm/pusht/weights.pt +3 -0
  26. le-wm/reacher/config.json +49 -0
  27. le-wm/reacher/weights.pt +3 -0
  28. le-wm/runtime.json +190 -0
  29. le-wm/tworoom/config.json +49 -0
  30. le-wm/tworoom/weights.pt +3 -0
  31. licenses/dino_wm.txt +21 -0
  32. licenses/dreamerv3.txt +19 -0
  33. licenses/le-wm.txt +21 -0
  34. licenses/planet.txt +202 -0
  35. licenses/tdmpc2.txt +21 -0
  36. manifest.json +481 -0
  37. planet/cartpole_swingup/attack_export.json +10 -0
  38. planet/cartpole_swingup/attack_weights.npz +3 -0
  39. planet/cartpole_swingup/config.yaml +388 -0
  40. planet/cartpole_swingup/tensorflow/checkpoint +2 -0
  41. planet/cartpole_swingup/tensorflow/model.ckpt-4504500.data-00000-of-00001 +3 -0
  42. planet/cartpole_swingup/tensorflow/model.ckpt-4504500.index +0 -0
  43. planet/cartpole_swingup/tensorflow/model.ckpt-4504500.meta +3 -0
  44. planet/cartpole_swingup/tf_reference.npz +3 -0
  45. planet/cheetah_run/attack_export.json +10 -0
  46. planet/cheetah_run/attack_weights.npz +3 -0
  47. planet/cheetah_run/config.yaml +388 -0
  48. planet/cheetah_run/tensorflow/checkpoint +2 -0
  49. planet/cheetah_run/tensorflow/model.ckpt-4504500.data-00000-of-00001 +3 -0
  50. 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