Hugging Face's logo

Potestates
/
TERRA

Robotics
Safetensors
openvla
vision-language-action
libero
Model card Files
xet
 Community
Potestates commited on
Commit
00e900a
·
verified ·
1 Parent(s): a8b98cf

piper_pick_can_10k: 20k phase-2 base + LoRA/flow/q4, 10k steps on the Piper pick-can data

Browse files
Files changed (22) hide show
  1. .gitattributes +1 -0
  2. piper_pick_can_10k/action_decoder.pt +3 -0
  3. piper_pick_can_10k/added_tokens.json +7 -0
  4. piper_pick_can_10k/config.json +3169 -0
  5. piper_pick_can_10k/configuration_prismatic.py +140 -0
  6. piper_pick_can_10k/ctrl_meta.json +24 -0
  7. piper_pick_can_10k/dataset_statistics.json +127 -0
  8. piper_pick_can_10k/generation_config.json +7 -0
  9. piper_pick_can_10k/model-00001-of-00004.safetensors +3 -0
  10. piper_pick_can_10k/model-00002-of-00004.safetensors +3 -0
  11. piper_pick_can_10k/model-00003-of-00004.safetensors +3 -0
  12. piper_pick_can_10k/model-00004-of-00004.safetensors +3 -0
  13. piper_pick_can_10k/model.safetensors.index.json +989 -0
  14. piper_pick_can_10k/modeling_prismatic.py +562 -0
  15. piper_pick_can_10k/preprocessor_config.json +114 -0
  16. piper_pick_can_10k/processing_prismatic.py +257 -0
  17. piper_pick_can_10k/processor_config.json +6 -0
  18. piper_pick_can_10k/special_tokens_map.json +60 -0
  19. piper_pick_can_10k/tokenizer.json +3 -0
  20. piper_pick_can_10k/tokenizer.model +3 -0
  21. piper_pick_can_10k/tokenizer_config.json +91 -0
  22. piper_pick_can_10k/train_log.jsonl +400 -0
.gitattributes CHANGED
@@ -2,3 +2,4 @@
2
  *.pt filter=lfs diff=lfs merge=lfs -text
3
  *.bin filter=lfs diff=lfs merge=lfs -text
4
  tokenizer.json filter=lfs diff=lfs merge=lfs -text
 
 
2
  *.pt filter=lfs diff=lfs merge=lfs -text
3
  *.bin filter=lfs diff=lfs merge=lfs -text
4
  tokenizer.json filter=lfs diff=lfs merge=lfs -text
5
+ piper_pick_can_10k/tokenizer.model filter=lfs diff=lfs merge=lfs -text
piper_pick_can_10k/action_decoder.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f6d1a87fe191a23c525b3627f0a6c0e01a363f70d475fd7e27c56f8811ce14ca
3
+ size 105271865
piper_pick_can_10k/added_tokens.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "<ACT_0>": 32001,
3
+ "<ACT_1>": 32002,
4
+ "<ACT_2>": 32003,
5
+ "<ACT_3>": 32004,
6
+ "<PAD>": 32000
7
+ }
piper_pick_can_10k/config.json ADDED
@@ -0,0 +1,3169 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "/home/ti874297/Projects/latent_free/cache/vla/openvla-7b",
3
+ "arch_specifier": "no-align+fused-gelu-mlp",
4
+ "architectures": [
5
+ "OpenVLAForActionPrediction"
6
+ ],
7
+ "auto_map": {
8
+ "AutoConfig": "configuration_prismatic.OpenVLAConfig",
9
+ "AutoModelForVision2Seq": "modeling_prismatic.OpenVLAForActionPrediction"
10
+ },
11
+ "hf_llm_id": "meta-llama/Llama-2-7b-hf",
12
+ "image_resize_strategy": "resize-naive",
13
+ "image_sizes": [
14
+ 224,
15
+ 224
16
+ ],
17
+ "llm_backbone_id": "llama2-7b-pure",
18
+ "llm_max_length": 2048,
19
+ "model_type": "openvla",
20
+ "n_action_bins": 256,
21
+ "norm_stats": {
22
+ "austin_buds_dataset_converted_externally_to_rlds": {
23
+ "action": {
24
+ "mask": [
25
+ true,
26
+ true,
27
+ true,
28
+ true,
29
+ true,
30
+ true,
31
+ false
32
+ ],
33
+ "max": [
34
+ 1.0,
35
+ 1.0,
36
+ 1.0,
37
+ 0.0,
38
+ 0.0,
39
+ 0.0,
40
+ 1.0
41
+ ],
42
+ "mean": [
43
+ -0.07678354531526566,
44
+ 0.0036849044263362885,
45
+ 0.05644911900162697,
46
+ 0.0,
47
+ 0.0,
48
+ 0.0,
49
+ 0.3510494828224182
50
+ ],
51
+ "min": [
52
+ -1.0,
53
+ -1.0,
54
+ -1.0,
55
+ 0.0,
56
+ 0.0,
57
+ 0.0,
58
+ 0.0
59
+ ],
60
+ "q01": [
61
+ -1.0,
62
+ -0.9599999785423279,
63
+ -0.8714285492897034,
64
+ 0.0,
65
+ 0.0,
66
+ 0.0,
67
+ 0.0
68
+ ],
69
+ "q99": [
70
+ 1.0,
71
+ 0.8600000143051147,
72
+ 1.0,
73
+ 0.0,
74
+ 0.0,
75
+ 0.0,
76
+ 1.0
77
+ ],
78
+ "std": [
79
+ 0.6367740631103516,
80
+ 0.37889179587364197,
81
+ 0.47796326875686646,
82
+ 0.0,
83
+ 0.0,
84
+ 0.0,
85
+ 0.47721168398857117
86
+ ]
87
+ },
88
+ "num_trajectories": 50,
89
+ "num_transitions": 34112,
90
+ "proprio": {
91
+ "max": [
92
+ 0.0,
93
+ 0.0,
94
+ 0.0,
95
+ 0.0,
96
+ 0.0,
97
+ 0.0,
98
+ 0.0
99
+ ],
100
+ "mean": [
101
+ 0.0,
102
+ 0.0,
103
+ 0.0,
104
+ 0.0,
105
+ 0.0,
106
+ 0.0,
107
+ 0.0
108
+ ],
109
+ "min": [
110
+ 0.0,
111
+ 0.0,
112
+ 0.0,
113
+ 0.0,
114
+ 0.0,
115
+ 0.0,
116
+ 0.0
117
+ ],
118
+ "q01": [
119
+ 0.0,
120
+ 0.0,
121
+ 0.0,
122
+ 0.0,
123
+ 0.0,
124
+ 0.0,
125
+ 0.0
126
+ ],
127
+ "q99": [
128
+ 0.0,
129
+ 0.0,
130
+ 0.0,
131
+ 0.0,
132
+ 0.0,
133
+ 0.0,
134
+ 0.0
135
+ ],
136
+ "std": [
137
+ 0.0,
138
+ 0.0,
139
+ 0.0,
140
+ 0.0,
141
+ 0.0,
142
+ 0.0,
143
+ 0.0
144
+ ]
145
+ }
146
+ },
147
+ "austin_sailor_dataset_converted_externally_to_rlds": {
148
+ "action": {
149
+ "mask": [
150
+ true,
151
+ true,
152
+ true,
153
+ true,
154
+ true,
155
+ true,
156
+ false
157
+ ],
158
+ "max": [
159
+ 1.0,
160
+ 1.0,
161
+ 1.0,
162
+ 0.0,
163
+ 0.0,
164
+ 0.375,
165
+ 1.0
166
+ ],
167
+ "mean": [
168
+ 0.011825348250567913,
169
+ 0.006461074110120535,
170
+ 0.06023626774549484,
171
+ 0.0,
172
+ 0.0,
173
+ 0.0016465914668515325,
174
+ 0.5260950326919556
175
+ ],
176
+ "min": [
177
+ -1.0,
178
+ -1.0,
179
+ -1.0,
180
+ 0.0,
181
+ 0.0,
182
+ -0.375,
183
+ 0.0
184
+ ],
185
+ "q01": [
186
+ -1.0,
187
+ -0.9828571677207947,
188
+ -0.6000000238418579,
189
+ 0.0,
190
+ 0.0,
191
+ -0.17249999940395355,
192
+ 0.0
193
+ ],
194
+ "q99": [
195
+ 1.0,
196
+ 0.9457142949104309,
197
+ 1.0,
198
+ 0.0,
199
+ 0.0,
200
+ 0.17892856895923615,
201
+ 1.0
202
+ ],
203
+ "std": [
204
+ 0.46348899602890015,
205
+ 0.41240179538726807,
206
+ 0.411862850189209,
207
+ 0.0,
208
+ 0.0,
209
+ 0.0578610822558403,
210
+ 0.49894046783447266
211
+ ]
212
+ },
213
+ "num_trajectories": 240,
214
+ "num_transitions": 353094,
215
+ "proprio": {
216
+ "max": [
217
+ 0.0,
218
+ 0.0,
219
+ 0.0,
220
+ 0.0,
221
+ 0.0,
222
+ 0.0,
223
+ 0.0
224
+ ],
225
+ "mean": [
226
+ 0.0,
227
+ 0.0,
228
+ 0.0,
229
+ 0.0,
230
+ 0.0,
231
+ 0.0,
232
+ 0.0
233
+ ],
234
+ "min": [
235
+ 0.0,
236
+ 0.0,
237
+ 0.0,
238
+ 0.0,
239
+ 0.0,
240
+ 0.0,
241
+ 0.0
242
+ ],
243
+ "q01": [
244
+ 0.0,
245
+ 0.0,
246
+ 0.0,
247
+ 0.0,
248
+ 0.0,
249
+ 0.0,
250
+ 0.0
251
+ ],
252
+ "q99": [
253
+ 0.0,
254
+ 0.0,
255
+ 0.0,
256
+ 0.0,
257
+ 0.0,
258
+ 0.0,
259
+ 0.0
260
+ ],
261
+ "std": [
262
+ 0.0,
263
+ 0.0,
264
+ 0.0,
265
+ 0.0,
266
+ 0.0,
267
+ 0.0,
268
+ 0.0
269
+ ]
270
+ }
271
+ },
272
+ "austin_sirius_dataset_converted_externally_to_rlds": {
273
+ "action": {
274
+ "mask": [
275
+ true,
276
+ true,
277
+ true,
278
+ true,
279
+ true,
280
+ true,
281
+ false
282
+ ],
283
+ "max": [
284
+ 1.0002285242080688,
285
+ 0.960608720779419,
286
+ 1.105179786682129,
287
+ 0.0,
288
+ 0.0,
289
+ 0.341785728931427,
290
+ 1.0
291
+ ],
292
+ "mean": [
293
+ 0.07747682929039001,
294
+ 0.03195561468601227,
295
+ 0.04244732856750488,
296
+ 0.0,
297
+ 0.0,
298
+ -0.01603456400334835,
299
+ 0.43260177969932556
300
+ ],
301
+ "min": [
302
+ -1.0183025598526,
303
+ -0.9800000190734863,
304
+ -0.9774575233459473,
305
+ 0.0,
306
+ 0.0,
307
+ -0.34607142210006714,
308
+ 0.0
309
+ ],
310
+ "q01": [
311
+ -0.780905865430832,
312
+ -0.5667179036140442,
313
+ -0.5254343223571777,
314
+ 0.0,
315
+ 0.0,
316
+ -0.28495091378688814,
317
+ 0.0
318
+ ],
319
+ "q99": [
320
+ 0.9569637751579284,
321
+ 0.6971374487876891,
322
+ 0.8124888157844541,
323
+ 0.0,
324
+ 0.0,
325
+ 0.1971428543329239,
326
+ 1.0
327
+ ],
328
+ "std": [
329
+ 0.3906329572200775,
330
+ 0.2998155355453491,
331
+ 0.2782271206378937,
332
+ 0.0,
333
+ 0.0,
334
+ 0.08120622485876083,
335
+ 0.49528297781944275
336
+ ]
337
+ },
338
+ "num_trajectories": 559,
339
+ "num_transitions": 279939,
340
+ "proprio": {
341
+ "max": [
342
+ 0.0,
343
+ 0.0,
344
+ 0.0,
345
+ 0.0,
346
+ 0.0,
347
+ 0.0,
348
+ 0.0
349
+ ],
350
+ "mean": [
351
+ 0.0,
352
+ 0.0,
353
+ 0.0,
354
+ 0.0,
355
+ 0.0,
356
+ 0.0,
357
+ 0.0
358
+ ],
359
+ "min": [
360
+ 0.0,
361
+ 0.0,
362
+ 0.0,
363
+ 0.0,
364
+ 0.0,
365
+ 0.0,
366
+ 0.0
367
+ ],
368
+ "q01": [
369
+ 0.0,
370
+ 0.0,
371
+ 0.0,
372
+ 0.0,
373
+ 0.0,
374
+ 0.0,
375
+ 0.0
376
+ ],
377
+ "q99": [
378
+ 0.0,
379
+ 0.0,
380
+ 0.0,
381
+ 0.0,
382
+ 0.0,
383
+ 0.0,
384
+ 0.0
385
+ ],
386
+ "std": [
387
+ 0.0,
388
+ 0.0,
389
+ 0.0,
390
+ 0.0,
391
+ 0.0,
392
+ 0.0,
393
+ 0.0
394
+ ]
395
+ }
396
+ },
397
+ "bc_z": {
398
+ "action": {
399
+ "mask": [
400
+ true,
401
+ true,
402
+ true,
403
+ true,
404
+ true,
405
+ true,
406
+ false
407
+ ],
408
+ "max": [
409
+ 0.2165454924106598,
410
+ 0.1251407265663147,
411
+ 0.10772687941789627,
412
+ 0.33544227480888367,
413
+ 0.28117990493774414,
414
+ 0.40614867210388184,
415
+ 1.0
416
+ ],
417
+ "mean": [
418
+ -0.009958467446267605,
419
+ 0.0008958321413956583,
420
+ 0.004995597992092371,
421
+ 0.00029755113064311445,
422
+ -0.008735382929444313,
423
+ -0.030693737789988518,
424
+ 0.8344562649726868
425
+ ],
426
+ "min": [
427
+ -0.1677047461271286,
428
+ -0.14630407094955444,
429
+ -0.10066790133714676,
430
+ -0.29421567916870117,
431
+ -0.32101404666900635,
432
+ -0.4635624885559082,
433
+ 0.0
434
+ ],
435
+ "q01": [
436
+ -0.09220654994249344,
437
+ -0.06456145539879798,
438
+ -0.049121275544166565,
439
+ -0.11594625547528267,
440
+ -0.14152548640966414,
441
+ -0.2251061636209488,
442
+ 0.0
443
+ ],
444
+ "q99": [
445
+ 0.07628866866230968,
446
+ 0.058019736707210584,
447
+ 0.052540797740221024,
448
+ 0.11740604028105736,
449
+ 0.11703975558280955,
450
+ 0.16729306846857078,
451
+ 1.0
452
+ ],
453
+ "std": [
454
+ 0.03053455986082554,
455
+ 0.0231423731893301,
456
+ 0.020641816779971123,
457
+ 0.04155943542718887,
458
+ 0.046427831053733826,
459
+ 0.0769818127155304,
460
+ 0.3610210120677948
461
+ ]
462
+ },
463
+ "num_trajectories": 43264,
464
+ "num_transitions": 6015535,
465
+ "proprio": {
466
+ "max": [
467
+ 0.0,
468
+ 0.0,
469
+ 0.0,
470
+ 0.0,
471
+ 0.0,
472
+ 0.0,
473
+ 0.0
474
+ ],
475
+ "mean": [
476
+ 0.0,
477
+ 0.0,
478
+ 0.0,
479
+ 0.0,
480
+ 0.0,
481
+ 0.0,
482
+ 0.0
483
+ ],
484
+ "min": [
485
+ 0.0,
486
+ 0.0,
487
+ 0.0,
488
+ 0.0,
489
+ 0.0,
490
+ 0.0,
491
+ 0.0
492
+ ],
493
+ "q01": [
494
+ 0.0,
495
+ 0.0,
496
+ 0.0,
497
+ 0.0,
498
+ 0.0,
499
+ 0.0,
500
+ 0.0
501
+ ],
502
+ "q99": [
503
+ 0.0,
504
+ 0.0,
505
+ 0.0,
506
+ 0.0,
507
+ 0.0,
508
+ 0.0,
509
+ 0.0
510
+ ],
511
+ "std": [
512
+ 0.0,
513
+ 0.0,
514
+ 0.0,
515
+ 0.0,
516
+ 0.0,
517
+ 0.0,
518
+ 0.0
519
+ ]
520
+ }
521
+ },
522
+ "berkeley_autolab_ur5": {
523
+ "action": {
524
+ "mask": [
525
+ true,
526
+ true,
527
+ true,
528
+ true,
529
+ true,
530
+ true,
531
+ false
532
+ ],
533
+ "max": [
534
+ 0.019999999552965164,
535
+ 0.019999999552965164,
536
+ 0.019999999552965164,
537
+ 0.06666667014360428,
538
+ 0.06666667014360428,
539
+ 0.06666667014360428,
540
+ 1.0
541
+ ],
542
+ "mean": [
543
+ 0.0005683620693162084,
544
+ 0.001217700308188796,
545
+ -0.0005296372692100704,
546
+ 0.00021029810886830091,
547
+ 6.0695128922816366e-05,
548
+ 0.001204986940138042,
549
+ 0.6298308372497559
550
+ ],
551
+ "min": [
552
+ -0.019999999552965164,
553
+ -0.019999999552965164,
554
+ -0.019999999552965164,
555
+ -0.06666667014360428,
556
+ -0.06666667014360428,
557
+ -0.06666667014360428,
558
+ 0.0
559
+ ],
560
+ "q01": [
561
+ -0.019999999552965164,
562
+ -0.019999999552965164,
563
+ -0.019999999552965164,
564
+ -0.02628571353852749,
565
+ -0.06666667014360428,
566
+ -0.03847619146108627,
567
+ 0.0
568
+ ],
569
+ "q99": [
570
+ 0.019999999552965164,
571
+ 0.019999999552965164,
572
+ 0.019999999552965164,
573
+ 0.031809523701667786,
574
+ 0.06666667014360428,
575
+ 0.036571428179740906,
576
+ 1.0
577
+ ],
578
+ "std": [
579
+ 0.0115329809486866,
580
+ 0.007990492507815361,
581
+ 0.009577835910022259,
582
+ 0.009432995691895485,
583
+ 0.016427582129836082,
584
+ 0.011053967289626598,
585
+ 0.48267969489097595
586
+ ]
587
+ },
588
+ "num_trajectories": 1000,
589
+ "num_transitions": 97939,
590
+ "proprio": {
591
+ "max": [
592
+ 0.0,
593
+ 0.0,
594
+ 0.0,
595
+ 0.0,
596
+ 0.0,
597
+ 0.0,
598
+ 0.0
599
+ ],
600
+ "mean": [
601
+ 0.0,
602
+ 0.0,
603
+ 0.0,
604
+ 0.0,
605
+ 0.0,
606
+ 0.0,
607
+ 0.0
608
+ ],
609
+ "min": [
610
+ 0.0,
611
+ 0.0,
612
+ 0.0,
613
+ 0.0,
614
+ 0.0,
615
+ 0.0,
616
+ 0.0
617
+ ],
618
+ "q01": [
619
+ 0.0,
620
+ 0.0,
621
+ 0.0,
622
+ 0.0,
623
+ 0.0,
624
+ 0.0,
625
+ 0.0
626
+ ],
627
+ "q99": [
628
+ 0.0,
629
+ 0.0,
630
+ 0.0,
631
+ 0.0,
632
+ 0.0,
633
+ 0.0,
634
+ 0.0
635
+ ],
636
+ "std": [
637
+ 0.0,
638
+ 0.0,
639
+ 0.0,
640
+ 0.0,
641
+ 0.0,
642
+ 0.0,
643
+ 0.0
644
+ ]
645
+ }
646
+ },
647
+ "berkeley_cable_routing": {
648
+ "action": {
649
+ "mask": [
650
+ true,
651
+ true,
652
+ true,
653
+ true,
654
+ true,
655
+ true,
656
+ false
657
+ ],
658
+ "max": [
659
+ 0.9633283019065857,
660
+ 1.0,
661
+ 1.0,
662
+ 0.0,
663
+ 0.0,
664
+ 1.0,
665
+ 0.0
666
+ ],
667
+ "mean": [
668
+ -0.07139874249696732,
669
+ 0.023609008640050888,
670
+ 0.10241943597793579,
671
+ 0.0,
672
+ 0.0,
673
+ 0.049671024084091187,
674
+ 0.0
675
+ ],
676
+ "min": [
677
+ -0.9809081554412842,
678
+ -0.9554349184036255,
679
+ -0.9994775056838989,
680
+ 0.0,
681
+ 0.0,
682
+ -1.0,
683
+ 0.0
684
+ ],
685
+ "q01": [
686
+ -0.5534318816661835,
687
+ -0.4797285574674606,
688
+ -0.5314934802055359,
689
+ 0.0,
690
+ 0.0,
691
+ -0.8855219376087189,
692
+ 0.0
693
+ ],
694
+ "q99": [
695
+ 0.42652835428714786,
696
+ 0.5000944086909298,
697
+ 0.639823433756829,
698
+ 0.0,
699
+ 0.0,
700
+ 0.984243879914284,
701
+ 0.0
702
+ ],
703
+ "std": [
704
+ 0.1815500408411026,
705
+ 0.1810990273952484,
706
+ 0.21220779418945312,
707
+ 0.0,
708
+ 0.0,
709
+ 0.3475511968135834,
710
+ 0.0
711
+ ]
712
+ },
713
+ "num_trajectories": 1647,
714
+ "num_transitions": 42328,
715
+ "proprio": {
716
+ "max": [
717
+ 0.0,
718
+ 0.0,
719
+ 0.0,
720
+ 0.0,
721
+ 0.0,
722
+ 0.0,
723
+ 0.0
724
+ ],
725
+ "mean": [
726
+ 0.0,
727
+ 0.0,
728
+ 0.0,
729
+ 0.0,
730
+ 0.0,
731
+ 0.0,
732
+ 0.0
733
+ ],
734
+ "min": [
735
+ 0.0,
736
+ 0.0,
737
+ 0.0,
738
+ 0.0,
739
+ 0.0,
740
+ 0.0,
741
+ 0.0
742
+ ],
743
+ "q01": [
744
+ 0.0,
745
+ 0.0,
746
+ 0.0,
747
+ 0.0,
748
+ 0.0,
749
+ 0.0,
750
+ 0.0
751
+ ],
752
+ "q99": [
753
+ 0.0,
754
+ 0.0,
755
+ 0.0,
756
+ 0.0,
757
+ 0.0,
758
+ 0.0,
759
+ 0.0
760
+ ],
761
+ "std": [
762
+ 0.0,
763
+ 0.0,
764
+ 0.0,
765
+ 0.0,
766
+ 0.0,
767
+ 0.0,
768
+ 0.0
769
+ ]
770
+ }
771
+ },
772
+ "berkeley_fanuc_manipulation": {
773
+ "action": {
774
+ "mask": [
775
+ true,
776
+ true,
777
+ true,
778
+ true,
779
+ true,
780
+ true,
781
+ false
782
+ ],
783
+ "max": [
784
+ 0.009999999776482582,
785
+ 0.009999999776482582,
786
+ 0.009999999776482582,
787
+ 0.03490658476948738,
788
+ 0.03490658476948738,
789
+ 0.03490658476948738,
790
+ 1.0
791
+ ],
792
+ "mean": [
793
+ 0.0007744057802483439,
794
+ -0.00031240080716088414,
795
+ -0.0015001941937953234,
796
+ -0.0007515158504247665,
797
+ -0.00015832878125365824,
798
+ 0.00014327642566058785,
799
+ 0.699295699596405
800
+ ],
801
+ "min": [
802
+ -0.009999999776482582,
803
+ -0.009999999776482582,
804
+ -0.009999999776482582,
805
+ -0.03490658476948738,
806
+ -0.03490658476948738,
807
+ -0.03490658476948738,
808
+ 0.0
809
+ ],
810
+ "q01": [
811
+ -0.009999999776482582,
812
+ -0.009999999776482582,
813
+ -0.009999999776482582,
814
+ -0.03490658476948738,
815
+ 0.0,
816
+ -0.03490658476948738,
817
+ 0.0
818
+ ],
819
+ "q99": [
820
+ 0.009999999776482582,
821
+ 0.009999999776482582,
822
+ 0.009999999776482582,
823
+ 0.03490658476948738,
824
+ 0.0,
825
+ 0.03490658476948738,
826
+ 1.0
827
+ ],
828
+ "std": [
829
+ 0.0034070091787725687,
830
+ 0.0049921851605176926,
831
+ 0.005344334989786148,
832
+ 0.00759894959628582,
833
+ 0.004081866703927517,
834
+ 0.008568956516683102,
835
+ 0.4586937427520752
836
+ ]
837
+ },
838
+ "num_trajectories": 415,
839
+ "num_transitions": 62613,
840
+ "proprio": {
841
+ "max": [
842
+ 0.0,
843
+ 0.0,
844
+ 0.0,
845
+ 0.0,
846
+ 0.0,
847
+ 0.0,
848
+ 0.0
849
+ ],
850
+ "mean": [
851
+ 0.0,
852
+ 0.0,
853
+ 0.0,
854
+ 0.0,
855
+ 0.0,
856
+ 0.0,
857
+ 0.0
858
+ ],
859
+ "min": [
860
+ 0.0,
861
+ 0.0,
862
+ 0.0,
863
+ 0.0,
864
+ 0.0,
865
+ 0.0,
866
+ 0.0
867
+ ],
868
+ "q01": [
869
+ 0.0,
870
+ 0.0,
871
+ 0.0,
872
+ 0.0,
873
+ 0.0,
874
+ 0.0,
875
+ 0.0
876
+ ],
877
+ "q99": [
878
+ 0.0,
879
+ 0.0,
880
+ 0.0,
881
+ 0.0,
882
+ 0.0,
883
+ 0.0,
884
+ 0.0
885
+ ],
886
+ "std": [
887
+ 0.0,
888
+ 0.0,
889
+ 0.0,
890
+ 0.0,
891
+ 0.0,
892
+ 0.0,
893
+ 0.0
894
+ ]
895
+ }
896
+ },
897
+ "bridge_orig": {
898
+ "action": {
899
+ "mask": [
900
+ true,
901
+ true,
902
+ true,
903
+ true,
904
+ true,
905
+ true,
906
+ false
907
+ ],
908
+ "max": [
909
+ 0.41691166162490845,
910
+ 0.25864794850349426,
911
+ 0.21218234300613403,
912
+ 3.122201919555664,
913
+ 1.8618112802505493,
914
+ 6.280478477478027,
915
+ 1.0
916
+ ],
917
+ "mean": [
918
+ 0.0002334194869035855,
919
+ 0.00013004911306779832,
920
+ -0.00012762474943883717,
921
+ -0.0001556558854645118,
922
+ -0.0004039328487124294,
923
+ 0.00023557482927571982,
924
+ 0.5764579176902771
925
+ ],
926
+ "min": [
927
+ -0.4007510244846344,
928
+ -0.13874775171279907,
929
+ -0.22553899884223938,
930
+ -3.2010786533355713,
931
+ -1.8618112802505493,
932
+ -6.279075622558594,
933
+ 0.0
934
+ ],
935
+ "q01": [
936
+ -0.02872725307941437,
937
+ -0.04170349963009357,
938
+ -0.026093858778476715,
939
+ -0.08092105075716972,
940
+ -0.09288699507713317,
941
+ -0.20718276381492615,
942
+ 0.0
943
+ ],
944
+ "q99": [
945
+ 0.028309678435325586,
946
+ 0.040855254605412394,
947
+ 0.040161586627364146,
948
+ 0.08192047759890528,
949
+ 0.07792850524187081,
950
+ 0.20382574498653397,
951
+ 1.0
952
+ ],
953
+ "std": [
954
+ 0.009765930473804474,
955
+ 0.013689135201275349,
956
+ 0.012667362578213215,
957
+ 0.028534092009067535,
958
+ 0.030637972056865692,
959
+ 0.07691419124603271,
960
+ 0.4973701536655426
961
+ ]
962
+ },
963
+ "num_trajectories": 60064,
964
+ "num_transitions": 2135463,
965
+ "proprio": {
966
+ "max": [
967
+ 0.0,
968
+ 0.0,
969
+ 0.0,
970
+ 0.0,
971
+ 0.0,
972
+ 0.0,
973
+ 0.0
974
+ ],
975
+ "mean": [
976
+ 0.0,
977
+ 0.0,
978
+ 0.0,
979
+ 0.0,
980
+ 0.0,
981
+ 0.0,
982
+ 0.0
983
+ ],
984
+ "min": [
985
+ 0.0,
986
+ 0.0,
987
+ 0.0,
988
+ 0.0,
989
+ 0.0,
990
+ 0.0,
991
+ 0.0
992
+ ],
993
+ "q01": [
994
+ 0.0,
995
+ 0.0,
996
+ 0.0,
997
+ 0.0,
998
+ 0.0,
999
+ 0.0,
1000
+ 0.0
1001
+ ],
1002
+ "q99": [
1003
+ 0.0,
1004
+ 0.0,
1005
+ 0.0,
1006
+ 0.0,
1007
+ 0.0,
1008
+ 0.0,
1009
+ 0.0
1010
+ ],
1011
+ "std": [
1012
+ 0.0,
1013
+ 0.0,
1014
+ 0.0,
1015
+ 0.0,
1016
+ 0.0,
1017
+ 0.0,
1018
+ 0.0
1019
+ ]
1020
+ }
1021
+ },
1022
+ "cmu_stretch": {
1023
+ "action": {
1024
+ "mask": [
1025
+ true,
1026
+ true,
1027
+ true,
1028
+ true,
1029
+ true,
1030
+ true,
1031
+ false
1032
+ ],
1033
+ "max": [
1034
+ 0.02338407188653946,
1035
+ 0.0,
1036
+ 0.023404927924275398,
1037
+ 0.0,
1038
+ 0.0,
1039
+ 0.0,
1040
+ 1.0
1041
+ ],
1042
+ "mean": [
1043
+ 0.00036304505192674696,
1044
+ 0.0,
1045
+ 0.0016466958913952112,
1046
+ 0.0,
1047
+ 0.0,
1048
+ 0.0,
1049
+ 0.3987048268318176
1050
+ ],
1051
+ "min": [
1052
+ -0.019353797659277916,
1053
+ 0.0,
1054
+ -0.02019215188920498,
1055
+ 0.0,
1056
+ 0.0,
1057
+ 0.0,
1058
+ 0.0
1059
+ ],
1060
+ "q01": [
1061
+ -0.011175686959177256,
1062
+ 0.0,
1063
+ -0.0032206363626755773,
1064
+ 0.0,
1065
+ 0.0,
1066
+ 0.0,
1067
+ 0.0
1068
+ ],
1069
+ "q99": [
1070
+ 0.014501785952597848,
1071
+ 0.0,
1072
+ 0.015056106168776728,
1073
+ 0.0,
1074
+ 0.0,
1075
+ 0.0,
1076
+ 1.0
1077
+ ],
1078
+ "std": [
1079
+ 0.004081828519701958,
1080
+ 0.0,
1081
+ 0.0037743328139185905,
1082
+ 0.0,
1083
+ 0.0,
1084
+ 0.0,
1085
+ 0.48963725566864014
1086
+ ]
1087
+ },
1088
+ "num_trajectories": 135,
1089
+ "num_transitions": 25016,
1090
+ "proprio": {
1091
+ "max": [
1092
+ 0.0,
1093
+ 0.0,
1094
+ 0.0,
1095
+ 0.0,
1096
+ 0.0,
1097
+ 0.0,
1098
+ 0.0
1099
+ ],
1100
+ "mean": [
1101
+ 0.0,
1102
+ 0.0,
1103
+ 0.0,
1104
+ 0.0,
1105
+ 0.0,
1106
+ 0.0,
1107
+ 0.0
1108
+ ],
1109
+ "min": [
1110
+ 0.0,
1111
+ 0.0,
1112
+ 0.0,
1113
+ 0.0,
1114
+ 0.0,
1115
+ 0.0,
1116
+ 0.0
1117
+ ],
1118
+ "q01": [
1119
+ 0.0,
1120
+ 0.0,
1121
+ 0.0,
1122
+ 0.0,
1123
+ 0.0,
1124
+ 0.0,
1125
+ 0.0
1126
+ ],
1127
+ "q99": [
1128
+ 0.0,
1129
+ 0.0,
1130
+ 0.0,
1131
+ 0.0,
1132
+ 0.0,
1133
+ 0.0,
1134
+ 0.0
1135
+ ],
1136
+ "std": [
1137
+ 0.0,
1138
+ 0.0,
1139
+ 0.0,
1140
+ 0.0,
1141
+ 0.0,
1142
+ 0.0,
1143
+ 0.0
1144
+ ]
1145
+ }
1146
+ },
1147
+ "dlr_edan_shared_control_converted_externally_to_rlds": {
1148
+ "action": {
1149
+ "mask": [
1150
+ true,
1151
+ true,
1152
+ true,
1153
+ true,
1154
+ true,
1155
+ true,
1156
+ false
1157
+ ],
1158
+ "max": [
1159
+ 0.18991442024707794,
1160
+ 0.0739002525806427,
1161
+ 0.18064819276332855,
1162
+ 0.0866486132144928,
1163
+ 0.13464981317520142,
1164
+ 0.16910280287265778,
1165
+ 1.0
1166
+ ],
1167
+ "mean": [
1168
+ 0.006647810339927673,
1169
+ -0.0007657372043468058,
1170
+ 0.006522852927446365,
1171
+ 0.0011679717572405934,
1172
+ -0.006395625416189432,
1173
+ -0.011902998201549053,
1174
+ 0.6985887289047241
1175
+ ],
1176
+ "min": [
1177
+ -0.10054297000169754,
1178
+ -0.08427435159683228,
1179
+ -0.13533438742160797,
1180
+ -0.17556548118591309,
1181
+ -0.18485672771930695,
1182
+ -0.2680685818195343,
1183
+ 0.0
1184
+ ],
1185
+ "q01": [
1186
+ -0.02987122368067503,
1187
+ -0.06013262912631035,
1188
+ -0.08286409199237824,
1189
+ -0.05924444157630205,
1190
+ -0.15986866518855095,
1191
+ -0.15636983573436739,
1192
+ 0.0
1193
+ ],
1194
+ "q99": [
1195
+ 0.08832092039287087,
1196
+ 0.042126184627413736,
1197
+ 0.11311905644834042,
1198
+ 0.0643695573508739,
1199
+ 0.03941855944693088,
1200
+ 0.156646853685379,
1201
+ 1.0
1202
+ ],
1203
+ "std": [
1204
+ 0.021393608301877975,
1205
+ 0.01814231649041176,
1206
+ 0.03374375030398369,
1207
+ 0.01743541844189167,
1208
+ 0.03394376486539841,
1209
+ 0.04641875624656677,
1210
+ 0.4588589072227478
1211
+ ]
1212
+ },
1213
+ "num_trajectories": 104,
1214
+ "num_transitions": 8928,
1215
+ "proprio": {
1216
+ "max": [
1217
+ 0.0,
1218
+ 0.0,
1219
+ 0.0,
1220
+ 0.0,
1221
+ 0.0,
1222
+ 0.0,
1223
+ 0.0
1224
+ ],
1225
+ "mean": [
1226
+ 0.0,
1227
+ 0.0,
1228
+ 0.0,
1229
+ 0.0,
1230
+ 0.0,
1231
+ 0.0,
1232
+ 0.0
1233
+ ],
1234
+ "min": [
1235
+ 0.0,
1236
+ 0.0,
1237
+ 0.0,
1238
+ 0.0,
1239
+ 0.0,
1240
+ 0.0,
1241
+ 0.0
1242
+ ],
1243
+ "q01": [
1244
+ 0.0,
1245
+ 0.0,
1246
+ 0.0,
1247
+ 0.0,
1248
+ 0.0,
1249
+ 0.0,
1250
+ 0.0
1251
+ ],
1252
+ "q99": [
1253
+ 0.0,
1254
+ 0.0,
1255
+ 0.0,
1256
+ 0.0,
1257
+ 0.0,
1258
+ 0.0,
1259
+ 0.0
1260
+ ],
1261
+ "std": [
1262
+ 0.0,
1263
+ 0.0,
1264
+ 0.0,
1265
+ 0.0,
1266
+ 0.0,
1267
+ 0.0,
1268
+ 0.0
1269
+ ]
1270
+ }
1271
+ },
1272
+ "dobbe": {
1273
+ "action": {
1274
+ "mask": [
1275
+ true,
1276
+ true,
1277
+ true,
1278
+ true,
1279
+ true,
1280
+ true,
1281
+ false
1282
+ ],
1283
+ "max": [
1284
+ 38.590423583984375,
1285
+ 17.932697296142578,
1286
+ 4.843764305114746,
1287
+ 1.4372116327285767,
1288
+ 0.4340403974056244,
1289
+ 1.2057193517684937,
1290
+ 0.9998947381973267
1291
+ ],
1292
+ "mean": [
1293
+ -0.0001120665911003016,
1294
+ 0.0011229600058868527,
1295
+ -0.00010194431524723768,
1296
+ -7.371398532995954e-05,
1297
+ -0.00067531579406932,
1298
+ -5.6643435527803376e-05,
1299
+ 0.6318281888961792
1300
+ ],
1301
+ "min": [
1302
+ -5.700923442840576,
1303
+ -21.605947494506836,
1304
+ -123.72489929199219,
1305
+ -1.7229845523834229,
1306
+ -0.4998578727245331,
1307
+ -0.8867913484573364,
1308
+ 1.4196479014572105e-06
1309
+ ],
1310
+ "q01": [
1311
+ -0.01119564864784479,
1312
+ -0.014266146533191203,
1313
+ -0.0071747214533388615,
1314
+ -0.009444301575422287,
1315
+ -0.03990109823644161,
1316
+ -0.017422311007976532,
1317
+ 4.003279136668425e-05
1318
+ ],
1319
+ "q99": [
1320
+ 0.01015154086053368,
1321
+ 0.017181577533483497,
1322
+ 0.007216989761218411,
1323
+ 0.010380979906767595,
1324
+ 0.03556173853576176,
1325
+ 0.018032474815845446,
1326
+ 0.9982578039169312
1327
+ ],
1328
+ "std": [
1329
+ 0.04264938458800316,
1330
+ 0.04428559169173241,
1331
+ 0.12224084138870239,
1332
+ 0.005388413090258837,
1333
+ 0.011246449314057827,
1334
+ 0.006287882570177317,
1335
+ 0.39732322096824646
1336
+ ]
1337
+ },
1338
+ "num_trajectories": 5208,
1339
+ "num_transitions": 1139911,
1340
+ "proprio": {
1341
+ "max": [
1342
+ 0.0,
1343
+ 0.0,
1344
+ 0.0,
1345
+ 0.0,
1346
+ 0.0,
1347
+ 0.0,
1348
+ 0.0
1349
+ ],
1350
+ "mean": [
1351
+ 0.0,
1352
+ 0.0,
1353
+ 0.0,
1354
+ 0.0,
1355
+ 0.0,
1356
+ 0.0,
1357
+ 0.0
1358
+ ],
1359
+ "min": [
1360
+ 0.0,
1361
+ 0.0,
1362
+ 0.0,
1363
+ 0.0,
1364
+ 0.0,
1365
+ 0.0,
1366
+ 0.0
1367
+ ],
1368
+ "q01": [
1369
+ 0.0,
1370
+ 0.0,
1371
+ 0.0,
1372
+ 0.0,
1373
+ 0.0,
1374
+ 0.0,
1375
+ 0.0
1376
+ ],
1377
+ "q99": [
1378
+ 0.0,
1379
+ 0.0,
1380
+ 0.0,
1381
+ 0.0,
1382
+ 0.0,
1383
+ 0.0,
1384
+ 0.0
1385
+ ],
1386
+ "std": [
1387
+ 0.0,
1388
+ 0.0,
1389
+ 0.0,
1390
+ 0.0,
1391
+ 0.0,
1392
+ 0.0,
1393
+ 0.0
1394
+ ]
1395
+ }
1396
+ },
1397
+ "fmb_dataset": {
1398
+ "action": {
1399
+ "mask": [
1400
+ true,
1401
+ true,
1402
+ true,
1403
+ true,
1404
+ true,
1405
+ true,
1406
+ false
1407
+ ],
1408
+ "max": [
1409
+ 1.399999976158142,
1410
+ 1.0,
1411
+ 1.399999976158142,
1412
+ 1.0,
1413
+ 1.0,
1414
+ 1.0,
1415
+ 1.0
1416
+ ],
1417
+ "mean": [
1418
+ 0.059029702097177505,
1419
+ -0.06476633995771408,
1420
+ -0.09787475317716599,
1421
+ 0.004325388930737972,
1422
+ 0.00028963794466108084,
1423
+ -0.04457257315516472,
1424
+ 0.7336440086364746
1425
+ ],
1426
+ "min": [
1427
+ -1.399999976158142,
1428
+ -1.399999976158142,
1429
+ -1.0,
1430
+ -1.0,
1431
+ -1.0,
1432
+ -1.0,
1433
+ 0.0
1434
+ ],
1435
+ "q01": [
1436
+ -0.8257142901420593,
1437
+ -1.399999976158142,
1438
+ -1.0,
1439
+ -1.0,
1440
+ -0.3028571307659149,
1441
+ -1.0,
1442
+ 0.0
1443
+ ],
1444
+ "q99": [
1445
+ 1.0,
1446
+ 0.5257142782211304,
1447
+ 1.0,
1448
+ 1.0,
1449
+ 0.3400000035762787,
1450
+ 1.0,
1451
+ 1.0
1452
+ ],
1453
+ "std": [
1454
+ 0.28809213638305664,
1455
+ 0.2820415794849396,
1456
+ 0.4626740515232086,
1457
+ 0.3266514539718628,
1458
+ 0.10842999070882797,
1459
+ 0.3440099358558655,
1460
+ 0.4435282051563263
1461
+ ]
1462
+ },
1463
+ "num_trajectories": 8612,
1464
+ "num_transitions": 1137459,
1465
+ "proprio": {
1466
+ "max": [
1467
+ 0.0,
1468
+ 0.0,
1469
+ 0.0,
1470
+ 0.0,
1471
+ 0.0,
1472
+ 0.0,
1473
+ 0.0
1474
+ ],
1475
+ "mean": [
1476
+ 0.0,
1477
+ 0.0,
1478
+ 0.0,
1479
+ 0.0,
1480
+ 0.0,
1481
+ 0.0,
1482
+ 0.0
1483
+ ],
1484
+ "min": [
1485
+ 0.0,
1486
+ 0.0,
1487
+ 0.0,
1488
+ 0.0,
1489
+ 0.0,
1490
+ 0.0,
1491
+ 0.0
1492
+ ],
1493
+ "q01": [
1494
+ 0.0,
1495
+ 0.0,
1496
+ 0.0,
1497
+ 0.0,
1498
+ 0.0,
1499
+ 0.0,
1500
+ 0.0
1501
+ ],
1502
+ "q99": [
1503
+ 0.0,
1504
+ 0.0,
1505
+ 0.0,
1506
+ 0.0,
1507
+ 0.0,
1508
+ 0.0,
1509
+ 0.0
1510
+ ],
1511
+ "std": [
1512
+ 0.0,
1513
+ 0.0,
1514
+ 0.0,
1515
+ 0.0,
1516
+ 0.0,
1517
+ 0.0,
1518
+ 0.0
1519
+ ]
1520
+ }
1521
+ },
1522
+ "fractal20220817_data": {
1523
+ "action": {
1524
+ "mask": [
1525
+ true,
1526
+ true,
1527
+ true,
1528
+ true,
1529
+ true,
1530
+ true,
1531
+ false
1532
+ ],
1533
+ "max": [
1534
+ 2.9984593391418457,
1535
+ 22.09052848815918,
1536
+ 2.7507524490356445,
1537
+ 1.570636510848999,
1538
+ 1.5321086645126343,
1539
+ 1.5691522359848022,
1540
+ 1.0
1541
+ ],
1542
+ "mean": [
1543
+ 0.006987582892179489,
1544
+ 0.006265917327255011,
1545
+ -0.01262515690177679,
1546
+ 0.04333311319351196,
1547
+ -0.005756212864071131,
1548
+ 0.0009130256366916001,
1549
+ 0.5354204773902893
1550
+ ],
1551
+ "min": [
1552
+ -2.0204520225524902,
1553
+ -5.497899532318115,
1554
+ -2.031663417816162,
1555
+ -1.569917917251587,
1556
+ -1.569892168045044,
1557
+ -1.570419430732727,
1558
+ 0.0
1559
+ ],
1560
+ "q01": [
1561
+ -0.22453527510166169,
1562
+ -0.14820013284683228,
1563
+ -0.231589707583189,
1564
+ -0.3517994859814644,
1565
+ -0.4193011274933815,
1566
+ -0.43643461108207704,
1567
+ 0.0
1568
+ ],
1569
+ "q99": [
1570
+ 0.17824687153100965,
1571
+ 0.14938379630446405,
1572
+ 0.21842354819178575,
1573
+ 0.5892666035890578,
1574
+ 0.35272657424211445,
1575
+ 0.44796681255102094,
1576
+ 1.0
1577
+ ],
1578
+ "std": [
1579
+ 0.0692116990685463,
1580
+ 0.05970962345600128,
1581
+ 0.07353084534406662,
1582
+ 0.15610496699810028,
1583
+ 0.13164450228214264,
1584
+ 0.14593800902366638,
1585
+ 0.497110515832901
1586
+ ]
1587
+ },
1588
+ "num_trajectories": 87212,
1589
+ "num_transitions": 3786400,
1590
+ "proprio": {
1591
+ "max": [
1592
+ 0.0,
1593
+ 0.0,
1594
+ 0.0,
1595
+ 0.0,
1596
+ 0.0,
1597
+ 0.0,
1598
+ 0.0
1599
+ ],
1600
+ "mean": [
1601
+ 0.0,
1602
+ 0.0,
1603
+ 0.0,
1604
+ 0.0,
1605
+ 0.0,
1606
+ 0.0,
1607
+ 0.0
1608
+ ],
1609
+ "min": [
1610
+ 0.0,
1611
+ 0.0,
1612
+ 0.0,
1613
+ 0.0,
1614
+ 0.0,
1615
+ 0.0,
1616
+ 0.0
1617
+ ],
1618
+ "q01": [
1619
+ 0.0,
1620
+ 0.0,
1621
+ 0.0,
1622
+ 0.0,
1623
+ 0.0,
1624
+ 0.0,
1625
+ 0.0
1626
+ ],
1627
+ "q99": [
1628
+ 0.0,
1629
+ 0.0,
1630
+ 0.0,
1631
+ 0.0,
1632
+ 0.0,
1633
+ 0.0,
1634
+ 0.0
1635
+ ],
1636
+ "std": [
1637
+ 0.0,
1638
+ 0.0,
1639
+ 0.0,
1640
+ 0.0,
1641
+ 0.0,
1642
+ 0.0,
1643
+ 0.0
1644
+ ]
1645
+ }
1646
+ },
1647
+ "furniture_bench_dataset_converted_externally_to_rlds": {
1648
+ "action": {
1649
+ "mask": [
1650
+ true,
1651
+ true,
1652
+ true,
1653
+ true,
1654
+ true,
1655
+ true,
1656
+ false
1657
+ ],
1658
+ "max": [
1659
+ 0.10000000149011612,
1660
+ 0.10000000149011612,
1661
+ 0.10000000149011612,
1662
+ 0.8651833534240723,
1663
+ 1.0909736156463623,
1664
+ 2.863185405731201,
1665
+ 1.0
1666
+ ],
1667
+ "mean": [
1668
+ 0.00014610752987209707,
1669
+ 0.0010830952087417245,
1670
+ 0.0006224989192560315,
1671
+ -0.003303206292912364,
1672
+ -0.0026880695950239897,
1673
+ 0.018242603167891502,
1674
+ 0.48854944109916687
1675
+ ],
1676
+ "min": [
1677
+ -0.10495579987764359,
1678
+ -0.10939455777406693,
1679
+ -0.10000000149011612,
1680
+ -0.971906840801239,
1681
+ -1.0475432872772217,
1682
+ -3.06000018119812,
1683
+ 0.0
1684
+ ],
1685
+ "q01": [
1686
+ -0.053988199681043625,
1687
+ -0.05049169331789017,
1688
+ -0.032499241530895236,
1689
+ -0.1953887003660202,
1690
+ -0.41674559473991396,
1691
+ -0.8886768388748169,
1692
+ 0.0
1693
+ ],
1694
+ "q99": [
1695
+ 0.05414841488003723,
1696
+ 0.04965164884924884,
1697
+ 0.060055799782276154,
1698
+ 0.18231668293476103,
1699
+ 0.39867786407470646,
1700
+ 0.8772023963928218,
1701
+ 1.0
1702
+ ],
1703
+ "std": [
1704
+ 0.01610708422958851,
1705
+ 0.014891477301716805,
1706
+ 0.014014219865202904,
1707
+ 0.058274295181035995,
1708
+ 0.11417088657617569,
1709
+ 0.33479776978492737,
1710
+ 0.49991825222969055
1711
+ ]
1712
+ },
1713
+ "num_trajectories": 5100,
1714
+ "num_transitions": 3948057,
1715
+ "proprio": {
1716
+ "max": [
1717
+ 0.0,
1718
+ 0.0,
1719
+ 0.0,
1720
+ 0.0,
1721
+ 0.0,
1722
+ 0.0,
1723
+ 0.0
1724
+ ],
1725
+ "mean": [
1726
+ 0.0,
1727
+ 0.0,
1728
+ 0.0,
1729
+ 0.0,
1730
+ 0.0,
1731
+ 0.0,
1732
+ 0.0
1733
+ ],
1734
+ "min": [
1735
+ 0.0,
1736
+ 0.0,
1737
+ 0.0,
1738
+ 0.0,
1739
+ 0.0,
1740
+ 0.0,
1741
+ 0.0
1742
+ ],
1743
+ "q01": [
1744
+ 0.0,
1745
+ 0.0,
1746
+ 0.0,
1747
+ 0.0,
1748
+ 0.0,
1749
+ 0.0,
1750
+ 0.0
1751
+ ],
1752
+ "q99": [
1753
+ 0.0,
1754
+ 0.0,
1755
+ 0.0,
1756
+ 0.0,
1757
+ 0.0,
1758
+ 0.0,
1759
+ 0.0
1760
+ ],
1761
+ "std": [
1762
+ 0.0,
1763
+ 0.0,
1764
+ 0.0,
1765
+ 0.0,
1766
+ 0.0,
1767
+ 0.0,
1768
+ 0.0
1769
+ ]
1770
+ }
1771
+ },
1772
+ "iamlab_cmu_pickup_insert_converted_externally_to_rlds": {
1773
+ "action": {
1774
+ "mask": [
1775
+ true,
1776
+ true,
1777
+ true,
1778
+ true,
1779
+ true,
1780
+ true,
1781
+ false
1782
+ ],
1783
+ "max": [
1784
+ 0.6634981632232666,
1785
+ 0.23428471386432648,
1786
+ 0.4308285415172577,
1787
+ 3.1415927410125732,
1788
+ 0.13647015392780304,
1789
+ 3.141592502593994,
1790
+ 1.0
1791
+ ],
1792
+ "mean": [
1793
+ 0.5274372696876526,
1794
+ 0.02858201041817665,
1795
+ 0.18712575733661652,
1796
+ 1.2339589595794678,
1797
+ 0.03226623684167862,
1798
+ -1.4199490547180176,
1799
+ 0.5550631880760193
1800
+ ],
1801
+ "min": [
1802
+ 0.3071657121181488,
1803
+ -0.29754969477653503,
1804
+ 0.06578229367733002,
1805
+ -3.1415927410125732,
1806
+ -0.04584203287959099,
1807
+ -3.141592502593994,
1808
+ 0.0
1809
+ ],
1810
+ "q01": [
1811
+ 0.3148897051811218,
1812
+ -0.20317550599575043,
1813
+ 0.06785467118024827,
1814
+ -3.140952730178833,
1815
+ -0.029743434861302376,
1816
+ -3.141091251373291,
1817
+ 0.0
1818
+ ],
1819
+ "q99": [
1820
+ 0.6472805738449097,
1821
+ 0.20846802592277527,
1822
+ 0.36855655312538155,
1823
+ 3.1409926891326903,
1824
+ 0.11424950212240226,
1825
+ 3.1410969257354737,
1826
+ 1.0
1827
+ ],
1828
+ "std": [
1829
+ 0.08108345419168472,
1830
+ 0.1116757020354271,
1831
+ 0.07747554779052734,
1832
+ 2.8737246990203857,
1833
+ 0.02774704433977604,
1834
+ 2.7678682804107666,
1835
+ 0.49695101380348206
1836
+ ]
1837
+ },
1838
+ "num_trajectories": 631,
1839
+ "num_transitions": 146241,
1840
+ "proprio": {
1841
+ "max": [
1842
+ 0.0,
1843
+ 0.0,
1844
+ 0.0,
1845
+ 0.0,
1846
+ 0.0,
1847
+ 0.0,
1848
+ 0.0
1849
+ ],
1850
+ "mean": [
1851
+ 0.0,
1852
+ 0.0,
1853
+ 0.0,
1854
+ 0.0,
1855
+ 0.0,
1856
+ 0.0,
1857
+ 0.0
1858
+ ],
1859
+ "min": [
1860
+ 0.0,
1861
+ 0.0,
1862
+ 0.0,
1863
+ 0.0,
1864
+ 0.0,
1865
+ 0.0,
1866
+ 0.0
1867
+ ],
1868
+ "q01": [
1869
+ 0.0,
1870
+ 0.0,
1871
+ 0.0,
1872
+ 0.0,
1873
+ 0.0,
1874
+ 0.0,
1875
+ 0.0
1876
+ ],
1877
+ "q99": [
1878
+ 0.0,
1879
+ 0.0,
1880
+ 0.0,
1881
+ 0.0,
1882
+ 0.0,
1883
+ 0.0,
1884
+ 0.0
1885
+ ],
1886
+ "std": [
1887
+ 0.0,
1888
+ 0.0,
1889
+ 0.0,
1890
+ 0.0,
1891
+ 0.0,
1892
+ 0.0,
1893
+ 0.0
1894
+ ]
1895
+ }
1896
+ },
1897
+ "jaco_play": {
1898
+ "action": {
1899
+ "mask": [
1900
+ true,
1901
+ true,
1902
+ true,
1903
+ true,
1904
+ true,
1905
+ true,
1906
+ false
1907
+ ],
1908
+ "max": [
1909
+ 0.20000000298023224,
1910
+ 0.20000000298023224,
1911
+ 0.20000000298023224,
1912
+ 0.0,
1913
+ 0.0,
1914
+ 0.0,
1915
+ 1.0
1916
+ ],
1917
+ "mean": [
1918
+ 0.0009658430935814977,
1919
+ -0.00580078037455678,
1920
+ -0.00395062193274498,
1921
+ 0.0,
1922
+ 0.0,
1923
+ 0.0,
1924
+ 0.34934908151626587
1925
+ ],
1926
+ "min": [
1927
+ -0.20000000298023224,
1928
+ -0.20000000298023224,
1929
+ -0.20000000298023224,
1930
+ 0.0,
1931
+ 0.0,
1932
+ 0.0,
1933
+ 0.0
1934
+ ],
1935
+ "q01": [
1936
+ -0.20000000298023224,
1937
+ -0.20000000298023224,
1938
+ -0.20000000298023224,
1939
+ 0.0,
1940
+ 0.0,
1941
+ 0.0,
1942
+ 0.0
1943
+ ],
1944
+ "q99": [
1945
+ 0.20000000298023224,
1946
+ 0.20000000298023224,
1947
+ 0.20000000298023224,
1948
+ 0.0,
1949
+ 0.0,
1950
+ 0.0,
1951
+ 1.0
1952
+ ],
1953
+ "std": [
1954
+ 0.12235074490308762,
1955
+ 0.09678777307271957,
1956
+ 0.11155334860086441,
1957
+ 0.0,
1958
+ 0.0,
1959
+ 0.0,
1960
+ 0.4768252968788147
1961
+ ]
1962
+ },
1963
+ "num_trajectories": 1085,
1964
+ "num_transitions": 77965,
1965
+ "proprio": {
1966
+ "max": [
1967
+ 0.0,
1968
+ 0.0,
1969
+ 0.0,
1970
+ 0.0,
1971
+ 0.0,
1972
+ 0.0,
1973
+ 0.0
1974
+ ],
1975
+ "mean": [
1976
+ 0.0,
1977
+ 0.0,
1978
+ 0.0,
1979
+ 0.0,
1980
+ 0.0,
1981
+ 0.0,
1982
+ 0.0
1983
+ ],
1984
+ "min": [
1985
+ 0.0,
1986
+ 0.0,
1987
+ 0.0,
1988
+ 0.0,
1989
+ 0.0,
1990
+ 0.0,
1991
+ 0.0
1992
+ ],
1993
+ "q01": [
1994
+ 0.0,
1995
+ 0.0,
1996
+ 0.0,
1997
+ 0.0,
1998
+ 0.0,
1999
+ 0.0,
2000
+ 0.0
2001
+ ],
2002
+ "q99": [
2003
+ 0.0,
2004
+ 0.0,
2005
+ 0.0,
2006
+ 0.0,
2007
+ 0.0,
2008
+ 0.0,
2009
+ 0.0
2010
+ ],
2011
+ "std": [
2012
+ 0.0,
2013
+ 0.0,
2014
+ 0.0,
2015
+ 0.0,
2016
+ 0.0,
2017
+ 0.0,
2018
+ 0.0
2019
+ ]
2020
+ }
2021
+ },
2022
+ "kuka": {
2023
+ "action": {
2024
+ "mask": [
2025
+ true,
2026
+ true,
2027
+ true,
2028
+ true,
2029
+ true,
2030
+ true,
2031
+ false
2032
+ ],
2033
+ "max": [
2034
+ 0.1697135865688324,
2035
+ 0.2777623236179352,
2036
+ 0.43710532784461975,
2037
+ 0.0,
2038
+ 0.0,
2039
+ 1.9684287309646606,
2040
+ 1.0
2041
+ ],
2042
+ "mean": [
2043
+ -0.0004668905457947403,
2044
+ 0.00040138536132872105,
2045
+ -0.001280792523175478,
2046
+ 0.0,
2047
+ 0.0,
2048
+ -0.03722453489899635,
2049
+ 0.4131543040275574
2050
+ ],
2051
+ "min": [
2052
+ -0.159867063164711,
2053
+ -0.2892282009124756,
2054
+ -0.2795473635196686,
2055
+ 0.0,
2056
+ 0.0,
2057
+ -1.9875637292861938,
2058
+ 0.0
2059
+ ],
2060
+ "q01": [
2061
+ -0.06619441494345665,
2062
+ -0.08713878810405731,
2063
+ -0.15083016991615295,
2064
+ 0.0,
2065
+ 0.0,
2066
+ -0.5415697038173676,
2067
+ 0.0
2068
+ ],
2069
+ "q99": [
2070
+ 0.06601839080452929,
2071
+ 0.08732476785779003,
2072
+ 0.18168179214000715,
2073
+ 0.0,
2074
+ 0.0,
2075
+ 0.2923380345106127,
2076
+ 1.0
2077
+ ],
2078
+ "std": [
2079
+ 0.02083250693976879,
2080
+ 0.02915887162089348,
2081
+ 0.06422865390777588,
2082
+ 0.0,
2083
+ 0.0,
2084
+ 0.14224295318126678,
2085
+ 0.49086448550224304
2086
+ ]
2087
+ },
2088
+ "num_trajectories": 209880,
2089
+ "num_transitions": 2455879,
2090
+ "proprio": {
2091
+ "max": [
2092
+ 0.0,
2093
+ 0.0,
2094
+ 0.0,
2095
+ 0.0,
2096
+ 0.0,
2097
+ 0.0,
2098
+ 0.0
2099
+ ],
2100
+ "mean": [
2101
+ 0.0,
2102
+ 0.0,
2103
+ 0.0,
2104
+ 0.0,
2105
+ 0.0,
2106
+ 0.0,
2107
+ 0.0
2108
+ ],
2109
+ "min": [
2110
+ 0.0,
2111
+ 0.0,
2112
+ 0.0,
2113
+ 0.0,
2114
+ 0.0,
2115
+ 0.0,
2116
+ 0.0
2117
+ ],
2118
+ "q01": [
2119
+ 0.0,
2120
+ 0.0,
2121
+ 0.0,
2122
+ 0.0,
2123
+ 0.0,
2124
+ 0.0,
2125
+ 0.0
2126
+ ],
2127
+ "q99": [
2128
+ 0.0,
2129
+ 0.0,
2130
+ 0.0,
2131
+ 0.0,
2132
+ 0.0,
2133
+ 0.0,
2134
+ 0.0
2135
+ ],
2136
+ "std": [
2137
+ 0.0,
2138
+ 0.0,
2139
+ 0.0,
2140
+ 0.0,
2141
+ 0.0,
2142
+ 0.0,
2143
+ 0.0
2144
+ ]
2145
+ }
2146
+ },
2147
+ "nyu_franka_play_dataset_converted_externally_to_rlds": {
2148
+ "action": {
2149
+ "mask": [
2150
+ true,
2151
+ true,
2152
+ true,
2153
+ true,
2154
+ true,
2155
+ true,
2156
+ false
2157
+ ],
2158
+ "max": [
2159
+ 0.06424188613891602,
2160
+ 0.07027634978294373,
2161
+ 0.06129661202430725,
2162
+ 6.281067848205566,
2163
+ 0.1967729926109314,
2164
+ 0.26377415657043457,
2165
+ 1.0
2166
+ ],
2167
+ "mean": [
2168
+ 0.001021989737637341,
2169
+ -0.00012002651783404872,
2170
+ 0.00032894269679673016,
2171
+ 0.0015034361276775599,
2172
+ -0.002198522910475731,
2173
+ -0.001663230243138969,
2174
+ 0.7230083346366882
2175
+ ],
2176
+ "min": [
2177
+ -0.05952230095863342,
2178
+ -0.07232445478439331,
2179
+ -0.06730806827545166,
2180
+ -6.278434753417969,
2181
+ -0.21479034423828125,
2182
+ -0.3627619743347168,
2183
+ 0.0
2184
+ ],
2185
+ "q01": [
2186
+ -0.03199600875377655,
2187
+ -0.032861671447753905,
2188
+ -0.03368805110454559,
2189
+ -0.12080862045288086,
2190
+ -0.12175218224525451,
2191
+ -0.11370223641395569,
2192
+ 0.0
2193
+ ],
2194
+ "q99": [
2195
+ 0.03101520001888276,
2196
+ 0.0373908892273903,
2197
+ 0.03646374464035038,
2198
+ 0.11764093399047852,
2199
+ 0.1258920183777809,
2200
+ 0.09366151213645942,
2201
+ 1.0
2202
+ ],
2203
+ "std": [
2204
+ 0.01327415369451046,
2205
+ 0.013215910643339157,
2206
+ 0.012822109274566174,
2207
+ 0.2732451558113098,
2208
+ 0.057022541761398315,
2209
+ 0.039172880351543427,
2210
+ 0.44752755761146545
2211
+ ]
2212
+ },
2213
+ "num_trajectories": 456,
2214
+ "num_transitions": 44875,
2215
+ "proprio": {
2216
+ "max": [
2217
+ 0.0,
2218
+ 0.0,
2219
+ 0.0,
2220
+ 0.0,
2221
+ 0.0,
2222
+ 0.0,
2223
+ 0.0
2224
+ ],
2225
+ "mean": [
2226
+ 0.0,
2227
+ 0.0,
2228
+ 0.0,
2229
+ 0.0,
2230
+ 0.0,
2231
+ 0.0,
2232
+ 0.0
2233
+ ],
2234
+ "min": [
2235
+ 0.0,
2236
+ 0.0,
2237
+ 0.0,
2238
+ 0.0,
2239
+ 0.0,
2240
+ 0.0,
2241
+ 0.0
2242
+ ],
2243
+ "q01": [
2244
+ 0.0,
2245
+ 0.0,
2246
+ 0.0,
2247
+ 0.0,
2248
+ 0.0,
2249
+ 0.0,
2250
+ 0.0
2251
+ ],
2252
+ "q99": [
2253
+ 0.0,
2254
+ 0.0,
2255
+ 0.0,
2256
+ 0.0,
2257
+ 0.0,
2258
+ 0.0,
2259
+ 0.0
2260
+ ],
2261
+ "std": [
2262
+ 0.0,
2263
+ 0.0,
2264
+ 0.0,
2265
+ 0.0,
2266
+ 0.0,
2267
+ 0.0,
2268
+ 0.0
2269
+ ]
2270
+ }
2271
+ },
2272
+ "roboturk": {
2273
+ "action": {
2274
+ "mask": [
2275
+ true,
2276
+ true,
2277
+ true,
2278
+ true,
2279
+ true,
2280
+ true,
2281
+ false
2282
+ ],
2283
+ "max": [
2284
+ 0.39124172925949097,
2285
+ 0.4601028263568878,
2286
+ 0.4870833456516266,
2287
+ 1.816888689994812,
2288
+ 1.8240282535552979,
2289
+ 1.4824820756912231,
2290
+ 1.0
2291
+ ],
2292
+ "mean": [
2293
+ 0.0014448732836171985,
2294
+ -0.0015945249469950795,
2295
+ -0.0011753785656765103,
2296
+ 0.0023012510500848293,
2297
+ -0.0009382463176734746,
2298
+ -0.00011485807772260159,
2299
+ 0.5746025443077087
2300
+ ],
2301
+ "min": [
2302
+ -0.6546999216079712,
2303
+ -0.6365841031074524,
2304
+ -0.4217723608016968,
2305
+ -1.6695482730865479,
2306
+ -1.8023357391357422,
2307
+ -1.4630827903747559,
2308
+ 0.0
2309
+ ],
2310
+ "q01": [
2311
+ -0.1342635464668274,
2312
+ -0.19996687173843383,
2313
+ -0.1482972100377083,
2314
+ -0.20720748245716095,
2315
+ -0.09676413893699647,
2316
+ -0.18075634717941286,
2317
+ 0.0
2318
+ ],
2319
+ "q99": [
2320
+ 0.14956976801157001,
2321
+ 0.1805950567126275,
2322
+ 0.18841815620660796,
2323
+ 0.21615413755178453,
2324
+ 0.09457383215427405,
2325
+ 0.18543301910162005,
2326
+ 1.0
2327
+ ],
2328
+ "std": [
2329
+ 0.04935386776924133,
2330
+ 0.0635455846786499,
2331
+ 0.061164740473032,
2332
+ 0.09553450345993042,
2333
+ 0.08420111238956451,
2334
+ 0.06517903506755829,
2335
+ 0.49452081322669983
2336
+ ]
2337
+ },
2338
+ "num_trajectories": 1995,
2339
+ "num_transitions": 187507,
2340
+ "proprio": {
2341
+ "max": [
2342
+ 0.0,
2343
+ 0.0,
2344
+ 0.0,
2345
+ 0.0,
2346
+ 0.0,
2347
+ 0.0,
2348
+ 0.0
2349
+ ],
2350
+ "mean": [
2351
+ 0.0,
2352
+ 0.0,
2353
+ 0.0,
2354
+ 0.0,
2355
+ 0.0,
2356
+ 0.0,
2357
+ 0.0
2358
+ ],
2359
+ "min": [
2360
+ 0.0,
2361
+ 0.0,
2362
+ 0.0,
2363
+ 0.0,
2364
+ 0.0,
2365
+ 0.0,
2366
+ 0.0
2367
+ ],
2368
+ "q01": [
2369
+ 0.0,
2370
+ 0.0,
2371
+ 0.0,
2372
+ 0.0,
2373
+ 0.0,
2374
+ 0.0,
2375
+ 0.0
2376
+ ],
2377
+ "q99": [
2378
+ 0.0,
2379
+ 0.0,
2380
+ 0.0,
2381
+ 0.0,
2382
+ 0.0,
2383
+ 0.0,
2384
+ 0.0
2385
+ ],
2386
+ "std": [
2387
+ 0.0,
2388
+ 0.0,
2389
+ 0.0,
2390
+ 0.0,
2391
+ 0.0,
2392
+ 0.0,
2393
+ 0.0
2394
+ ]
2395
+ }
2396
+ },
2397
+ "stanford_hydra_dataset_converted_externally_to_rlds": {
2398
+ "action": {
2399
+ "mask": [
2400
+ true,
2401
+ true,
2402
+ true,
2403
+ true,
2404
+ true,
2405
+ true,
2406
+ false
2407
+ ],
2408
+ "max": [
2409
+ 0.02499854564666748,
2410
+ 0.02499903365969658,
2411
+ 0.024999922141432762,
2412
+ 0.24974457919597626,
2413
+ 0.24997030198574066,
2414
+ 0.24999946355819702,
2415
+ 1.0
2416
+ ],
2417
+ "mean": [
2418
+ 0.0007790001109242439,
2419
+ 0.00013707754260394722,
2420
+ -0.0002548607881180942,
2421
+ 0.0012903271708637476,
2422
+ -0.004751681815832853,
2423
+ 0.002692886395379901,
2424
+ 0.48855218291282654
2425
+ ],
2426
+ "min": [
2427
+ -0.024999044835567474,
2428
+ -0.024999700486660004,
2429
+ -0.02499929815530777,
2430
+ -0.24993225932121277,
2431
+ -0.2499666064977646,
2432
+ -0.2499932497739792,
2433
+ 0.0
2434
+ ],
2435
+ "q01": [
2436
+ -0.019992006458342076,
2437
+ -0.02415412735193968,
2438
+ -0.022941758055239916,
2439
+ -0.11085530579090118,
2440
+ -0.12024572037160397,
2441
+ -0.13314770206809043,
2442
+ 0.0
2443
+ ],
2444
+ "q99": [
2445
+ 0.022886231057345868,
2446
+ 0.022358838934451335,
2447
+ 0.02410089675337076,
2448
+ 0.12370114490389822,
2449
+ 0.11323311634361738,
2450
+ 0.18474749639630164,
2451
+ 1.0
2452
+ ],
2453
+ "std": [
2454
+ 0.008022161200642586,
2455
+ 0.009131459519267082,
2456
+ 0.009574338793754578,
2457
+ 0.04122216999530792,
2458
+ 0.0384303517639637,
2459
+ 0.04606688767671585,
2460
+ 0.49976691603660583
2461
+ ]
2462
+ },
2463
+ "num_trajectories": 570,
2464
+ "num_transitions": 358234,
2465
+ "proprio": {
2466
+ "max": [
2467
+ 0.0,
2468
+ 0.0,
2469
+ 0.0,
2470
+ 0.0,
2471
+ 0.0,
2472
+ 0.0,
2473
+ 0.0
2474
+ ],
2475
+ "mean": [
2476
+ 0.0,
2477
+ 0.0,
2478
+ 0.0,
2479
+ 0.0,
2480
+ 0.0,
2481
+ 0.0,
2482
+ 0.0
2483
+ ],
2484
+ "min": [
2485
+ 0.0,
2486
+ 0.0,
2487
+ 0.0,
2488
+ 0.0,
2489
+ 0.0,
2490
+ 0.0,
2491
+ 0.0
2492
+ ],
2493
+ "q01": [
2494
+ 0.0,
2495
+ 0.0,
2496
+ 0.0,
2497
+ 0.0,
2498
+ 0.0,
2499
+ 0.0,
2500
+ 0.0
2501
+ ],
2502
+ "q99": [
2503
+ 0.0,
2504
+ 0.0,
2505
+ 0.0,
2506
+ 0.0,
2507
+ 0.0,
2508
+ 0.0,
2509
+ 0.0
2510
+ ],
2511
+ "std": [
2512
+ 0.0,
2513
+ 0.0,
2514
+ 0.0,
2515
+ 0.0,
2516
+ 0.0,
2517
+ 0.0,
2518
+ 0.0
2519
+ ]
2520
+ }
2521
+ },
2522
+ "taco_play": {
2523
+ "action": {
2524
+ "mask": [
2525
+ true,
2526
+ true,
2527
+ true,
2528
+ true,
2529
+ true,
2530
+ true,
2531
+ false
2532
+ ],
2533
+ "max": [
2534
+ 1.4915844202041626,
2535
+ 2.1842432022094727,
2536
+ 2.6836395263671875,
2537
+ 5.035226821899414,
2538
+ 2.665864944458008,
2539
+ 4.250768661499023,
2540
+ 1.0
2541
+ ],
2542
+ "mean": [
2543
+ -0.003845922416076064,
2544
+ 0.009671456180512905,
2545
+ 0.012780580669641495,
2546
+ -0.005403771996498108,
2547
+ -0.009606587700545788,
2548
+ -0.002480733208358288,
2549
+ 0.4263913035392761
2550
+ ],
2551
+ "min": [
2552
+ -4.242457866668701,
2553
+ -3.192805051803589,
2554
+ -1.3371467590332031,
2555
+ -4.202683448791504,
2556
+ -2.6722638607025146,
2557
+ -3.3467135429382324,
2558
+ 0.0
2559
+ ],
2560
+ "q01": [
2561
+ -0.7106140398979186,
2562
+ -1.056944659948349,
2563
+ -0.5878450274467468,
2564
+ -0.7682853937149048,
2565
+ -0.7180147767066956,
2566
+ -1.5527938604354858,
2567
+ 0.0
2568
+ ],
2569
+ "q99": [
2570
+ 0.6482916426658629,
2571
+ 1.0051310062408447,
2572
+ 0.9480248689651489,
2573
+ 0.6926478147506714,
2574
+ 0.6351067513227462,
2575
+ 1.628010264635086,
2576
+ 1.0
2577
+ ],
2578
+ "std": [
2579
+ 0.23254038393497467,
2580
+ 0.36298269033432007,
2581
+ 0.28692901134490967,
2582
+ 0.2617705166339874,
2583
+ 0.2438892275094986,
2584
+ 0.5216503143310547,
2585
+ 0.4946896731853485
2586
+ ]
2587
+ },
2588
+ "num_trajectories": 3603,
2589
+ "num_transitions": 237798,
2590
+ "proprio": {
2591
+ "max": [
2592
+ 0.0,
2593
+ 0.0,
2594
+ 0.0,
2595
+ 0.0,
2596
+ 0.0,
2597
+ 0.0,
2598
+ 0.0
2599
+ ],
2600
+ "mean": [
2601
+ 0.0,
2602
+ 0.0,
2603
+ 0.0,
2604
+ 0.0,
2605
+ 0.0,
2606
+ 0.0,
2607
+ 0.0
2608
+ ],
2609
+ "min": [
2610
+ 0.0,
2611
+ 0.0,
2612
+ 0.0,
2613
+ 0.0,
2614
+ 0.0,
2615
+ 0.0,
2616
+ 0.0
2617
+ ],
2618
+ "q01": [
2619
+ 0.0,
2620
+ 0.0,
2621
+ 0.0,
2622
+ 0.0,
2623
+ 0.0,
2624
+ 0.0,
2625
+ 0.0
2626
+ ],
2627
+ "q99": [
2628
+ 0.0,
2629
+ 0.0,
2630
+ 0.0,
2631
+ 0.0,
2632
+ 0.0,
2633
+ 0.0,
2634
+ 0.0
2635
+ ],
2636
+ "std": [
2637
+ 0.0,
2638
+ 0.0,
2639
+ 0.0,
2640
+ 0.0,
2641
+ 0.0,
2642
+ 0.0,
2643
+ 0.0
2644
+ ]
2645
+ }
2646
+ },
2647
+ "toto": {
2648
+ "action": {
2649
+ "mask": [
2650
+ true,
2651
+ true,
2652
+ true,
2653
+ true,
2654
+ true,
2655
+ true,
2656
+ false
2657
+ ],
2658
+ "max": [
2659
+ 0.6839867234230042,
2660
+ 0.4454185664653778,
2661
+ 0.7984078526496887,
2662
+ 2.120781660079956,
2663
+ 1.371164321899414,
2664
+ 1.4118704795837402,
2665
+ 0.0
2666
+ ],
2667
+ "mean": [
2668
+ 0.38542115688323975,
2669
+ 0.007769413758069277,
2670
+ 0.3632740378379822,
2671
+ -0.6652036905288696,
2672
+ 0.1890396922826767,
2673
+ 0.03298724442720413,
2674
+ 0.0
2675
+ ],
2676
+ "min": [
2677
+ 0.09922284632921219,
2678
+ -0.5180193781852722,
2679
+ 0.13791072368621826,
2680
+ -2.635117530822754,
2681
+ -1.0734480619430542,
2682
+ -1.9282547235488892,
2683
+ 0.0
2684
+ ],
2685
+ "q01": [
2686
+ 0.1756722891330719,
2687
+ -0.3077590811252594,
2688
+ 0.235383919775486,
2689
+ -2.0908505964279174,
2690
+ -0.6191593289375306,
2691
+ -0.7488683319091797,
2692
+ 0.0
2693
+ ],
2694
+ "q99": [
2695
+ 0.6136963081359863,
2696
+ 0.33704194784164443,
2697
+ 0.6681221985816956,
2698
+ 0.7422861719131538,
2699
+ 0.7955395007133507,
2700
+ 0.740464625358582,
2701
+ 0.0
2702
+ ],
2703
+ "std": [
2704
+ 0.12211652100086212,
2705
+ 0.19378550350666046,
2706
+ 0.10178236663341522,
2707
+ 0.5725259184837341,
2708
+ 0.29884573817253113,
2709
+ 0.3259911835193634,
2710
+ 0.0
2711
+ ]
2712
+ },
2713
+ "num_trajectories": 1003,
2714
+ "num_transitions": 325699,
2715
+ "proprio": {
2716
+ "max": [
2717
+ 0.0,
2718
+ 0.0,
2719
+ 0.0,
2720
+ 0.0,
2721
+ 0.0,
2722
+ 0.0,
2723
+ 0.0
2724
+ ],
2725
+ "mean": [
2726
+ 0.0,
2727
+ 0.0,
2728
+ 0.0,
2729
+ 0.0,
2730
+ 0.0,
2731
+ 0.0,
2732
+ 0.0
2733
+ ],
2734
+ "min": [
2735
+ 0.0,
2736
+ 0.0,
2737
+ 0.0,
2738
+ 0.0,
2739
+ 0.0,
2740
+ 0.0,
2741
+ 0.0
2742
+ ],
2743
+ "q01": [
2744
+ 0.0,
2745
+ 0.0,
2746
+ 0.0,
2747
+ 0.0,
2748
+ 0.0,
2749
+ 0.0,
2750
+ 0.0
2751
+ ],
2752
+ "q99": [
2753
+ 0.0,
2754
+ 0.0,
2755
+ 0.0,
2756
+ 0.0,
2757
+ 0.0,
2758
+ 0.0,
2759
+ 0.0
2760
+ ],
2761
+ "std": [
2762
+ 0.0,
2763
+ 0.0,
2764
+ 0.0,
2765
+ 0.0,
2766
+ 0.0,
2767
+ 0.0,
2768
+ 0.0
2769
+ ]
2770
+ }
2771
+ },
2772
+ "ucsd_kitchen_dataset_converted_externally_to_rlds": {
2773
+ "action": {
2774
+ "mask": [
2775
+ true,
2776
+ true,
2777
+ true,
2778
+ true,
2779
+ true,
2780
+ true,
2781
+ false
2782
+ ],
2783
+ "max": [
2784
+ 678.0,
2785
+ 400.0,
2786
+ 507.0,
2787
+ 180.00001525878906,
2788
+ 6.000013828277588,
2789
+ 116.99998474121094,
2790
+ 1.0
2791
+ ],
2792
+ "mean": [
2793
+ 410.37567138671875,
2794
+ 116.9518814086914,
2795
+ 192.35032653808594,
2796
+ -121.22441864013672,
2797
+ -33.84893035888672,
2798
+ 50.016136169433594,
2799
+ 0.741813600063324
2800
+ ],
2801
+ "min": [
2802
+ 172.0,
2803
+ -166.0,
2804
+ -99.99999237060547,
2805
+ -180.00001525878906,
2806
+ -89.0,
2807
+ -96.00010681152344,
2808
+ 0.0
2809
+ ],
2810
+ "q01": [
2811
+ 200.00001052856445,
2812
+ -102.31004211425781,
2813
+ -94.99993370056153,
2814
+ -180.00001525878906,
2815
+ -88.00001525878906,
2816
+ -38.999977111816406,
2817
+ 0.0
2818
+ ],
2819
+ "q99": [
2820
+ 637.0,
2821
+ 368.30999999999995,
2822
+ 493.0,
2823
+ 180.00001525878906,
2824
+ 0.999983012676239,
2825
+ 105.00001525878906,
2826
+ 1.0
2827
+ ],
2828
+ "std": [
2829
+ 122.81494903564453,
2830
+ 108.8009033203125,
2831
+ 130.303466796875,
2832
+ 116.28205108642578,
2833
+ 27.621843338012695,
2834
+ 41.02094650268555,
2835
+ 0.43763357400894165
2836
+ ]
2837
+ },
2838
+ "num_trajectories": 150,
2839
+ "num_transitions": 3970,
2840
+ "proprio": {
2841
+ "max": [
2842
+ 0.0,
2843
+ 0.0,
2844
+ 0.0,
2845
+ 0.0,
2846
+ 0.0,
2847
+ 0.0,
2848
+ 0.0
2849
+ ],
2850
+ "mean": [
2851
+ 0.0,
2852
+ 0.0,
2853
+ 0.0,
2854
+ 0.0,
2855
+ 0.0,
2856
+ 0.0,
2857
+ 0.0
2858
+ ],
2859
+ "min": [
2860
+ 0.0,
2861
+ 0.0,
2862
+ 0.0,
2863
+ 0.0,
2864
+ 0.0,
2865
+ 0.0,
2866
+ 0.0
2867
+ ],
2868
+ "q01": [
2869
+ 0.0,
2870
+ 0.0,
2871
+ 0.0,
2872
+ 0.0,
2873
+ 0.0,
2874
+ 0.0,
2875
+ 0.0
2876
+ ],
2877
+ "q99": [
2878
+ 0.0,
2879
+ 0.0,
2880
+ 0.0,
2881
+ 0.0,
2882
+ 0.0,
2883
+ 0.0,
2884
+ 0.0
2885
+ ],
2886
+ "std": [
2887
+ 0.0,
2888
+ 0.0,
2889
+ 0.0,
2890
+ 0.0,
2891
+ 0.0,
2892
+ 0.0,
2893
+ 0.0
2894
+ ]
2895
+ }
2896
+ },
2897
+ "utaustin_mutex": {
2898
+ "action": {
2899
+ "mask": [
2900
+ true,
2901
+ true,
2902
+ true,
2903
+ true,
2904
+ true,
2905
+ true,
2906
+ false
2907
+ ],
2908
+ "max": [
2909
+ 1.0,
2910
+ 1.0,
2911
+ 1.0,
2912
+ 0.375,
2913
+ 0.375,
2914
+ 0.375,
2915
+ 1.0
2916
+ ],
2917
+ "mean": [
2918
+ 0.06176406890153885,
2919
+ -0.005005486309528351,
2920
+ 0.10216785222291946,
2921
+ -0.03314131125807762,
2922
+ 0.013895004987716675,
2923
+ -0.011317633092403412,
2924
+ 0.5038976669311523
2925
+ ],
2926
+ "min": [
2927
+ -1.0,
2928
+ -1.0,
2929
+ -1.0,
2930
+ -0.375,
2931
+ -0.375,
2932
+ -0.375,
2933
+ 0.0
2934
+ ],
2935
+ "q01": [
2936
+ -0.4285714328289032,
2937
+ -0.9800000190734863,
2938
+ -0.5571428537368774,
2939
+ -0.375,
2940
+ -0.15642857551574707,
2941
+ -0.335357129573822,
2942
+ 0.0
2943
+ ],
2944
+ "q99": [
2945
+ 0.5914285778999329,
2946
+ 0.9714285731315613,
2947
+ 1.0,
2948
+ 0.3278571367263794,
2949
+ 0.207857146859169,
2950
+ 0.25607141852378845,
2951
+ 1.0
2952
+ ],
2953
+ "std": [
2954
+ 0.1875014752149582,
2955
+ 0.4468473494052887,
2956
+ 0.3792876601219177,
2957
+ 0.14097853004932404,
2958
+ 0.06453701853752136,
2959
+ 0.11765272170305252,
2960
+ 0.501045286655426
2961
+ ]
2962
+ },
2963
+ "num_trajectories": 1500,
2964
+ "num_transitions": 361883,
2965
+ "proprio": {
2966
+ "max": [
2967
+ 0.0,
2968
+ 0.0,
2969
+ 0.0,
2970
+ 0.0,
2971
+ 0.0,
2972
+ 0.0,
2973
+ 0.0
2974
+ ],
2975
+ "mean": [
2976
+ 0.0,
2977
+ 0.0,
2978
+ 0.0,
2979
+ 0.0,
2980
+ 0.0,
2981
+ 0.0,
2982
+ 0.0
2983
+ ],
2984
+ "min": [
2985
+ 0.0,
2986
+ 0.0,
2987
+ 0.0,
2988
+ 0.0,
2989
+ 0.0,
2990
+ 0.0,
2991
+ 0.0
2992
+ ],
2993
+ "q01": [
2994
+ 0.0,
2995
+ 0.0,
2996
+ 0.0,
2997
+ 0.0,
2998
+ 0.0,
2999
+ 0.0,
3000
+ 0.0
3001
+ ],
3002
+ "q99": [
3003
+ 0.0,
3004
+ 0.0,
3005
+ 0.0,
3006
+ 0.0,
3007
+ 0.0,
3008
+ 0.0,
3009
+ 0.0
3010
+ ],
3011
+ "std": [
3012
+ 0.0,
3013
+ 0.0,
3014
+ 0.0,
3015
+ 0.0,
3016
+ 0.0,
3017
+ 0.0,
3018
+ 0.0
3019
+ ]
3020
+ }
3021
+ },
3022
+ "viola": {
3023
+ "action": {
3024
+ "mask": [
3025
+ true,
3026
+ true,
3027
+ true,
3028
+ true,
3029
+ true,
3030
+ true,
3031
+ false
3032
+ ],
3033
+ "max": [
3034
+ 1.0,
3035
+ 1.0,
3036
+ 1.0,
3037
+ 0.375,
3038
+ 0.36321428418159485,
3039
+ 0.375,
3040
+ 1.0
3041
+ ],
3042
+ "mean": [
3043
+ 0.04761844128370285,
3044
+ -0.029204415157437325,
3045
+ 0.05586736649274826,
3046
+ -0.002618510741740465,
3047
+ 0.006867344491183758,
3048
+ -0.01682133786380291,
3049
+ 0.7323777675628662
3050
+ ],
3051
+ "min": [
3052
+ -1.0,
3053
+ -1.0,
3054
+ -1.0,
3055
+ -0.375,
3056
+ -0.375,
3057
+ -0.375,
3058
+ 0.0
3059
+ ],
3060
+ "q01": [
3061
+ -0.9628571271896362,
3062
+ -1.0,
3063
+ -1.0,
3064
+ -0.26249998807907104,
3065
+ -0.21321429312229156,
3066
+ -0.3385714292526245,
3067
+ 0.0
3068
+ ],
3069
+ "q99": [
3070
+ 0.9114285707473755,
3071
+ 0.868571400642395,
3072
+ 1.0,
3073
+ 0.2817857265472412,
3074
+ 0.2239285707473755,
3075
+ 0.3557142913341522,
3076
+ 1.0
3077
+ ],
3078
+ "std": [
3079
+ 0.39157867431640625,
3080
+ 0.4076525568962097,
3081
+ 0.40077948570251465,
3082
+ 0.10023996233940125,
3083
+ 0.0844319611787796,
3084
+ 0.10375042259693146,
3085
+ 0.44260647892951965
3086
+ ]
3087
+ },
3088
+ "num_trajectories": 150,
3089
+ "num_transitions": 76324,
3090
+ "proprio": {
3091
+ "max": [
3092
+ 0.0,
3093
+ 0.0,
3094
+ 0.0,
3095
+ 0.0,
3096
+ 0.0,
3097
+ 0.0,
3098
+ 0.0
3099
+ ],
3100
+ "mean": [
3101
+ 0.0,
3102
+ 0.0,
3103
+ 0.0,
3104
+ 0.0,
3105
+ 0.0,
3106
+ 0.0,
3107
+ 0.0
3108
+ ],
3109
+ "min": [
3110
+ 0.0,
3111
+ 0.0,
3112
+ 0.0,
3113
+ 0.0,
3114
+ 0.0,
3115
+ 0.0,
3116
+ 0.0
3117
+ ],
3118
+ "q01": [
3119
+ 0.0,
3120
+ 0.0,
3121
+ 0.0,
3122
+ 0.0,
3123
+ 0.0,
3124
+ 0.0,
3125
+ 0.0
3126
+ ],
3127
+ "q99": [
3128
+ 0.0,
3129
+ 0.0,
3130
+ 0.0,
3131
+ 0.0,
3132
+ 0.0,
3133
+ 0.0,
3134
+ 0.0
3135
+ ],
3136
+ "std": [
3137
+ 0.0,
3138
+ 0.0,
3139
+ 0.0,
3140
+ 0.0,
3141
+ 0.0,
3142
+ 0.0,
3143
+ 0.0
3144
+ ]
3145
+ }
3146
+ }
3147
+ },
3148
+ "output_projector_states": false,
3149
+ "pad_to_multiple_of": 64,
3150
+ "pad_token_id": 32000,
3151
+ "text_config": {
3152
+ "model_type": "llama",
3153
+ "pad_token_id": 32000,
3154
+ "torch_dtype": "bfloat16",
3155
+ "vocab_size": 32005
3156
+ },
3157
+ "timm_model_ids": [
3158
+ "vit_large_patch14_reg4_dinov2.lvd142m",
3159
+ "vit_so400m_patch14_siglip_224"
3160
+ ],
3161
+ "timm_override_act_layers": [
3162
+ null,
3163
+ null
3164
+ ],
3165
+ "torch_dtype": "bfloat16",
3166
+ "transformers_version": "4.40.1",
3167
+ "use_fused_vision_backbone": true,
3168
+ "vision_backbone_id": "dinosiglip-vit-so-224px"
3169
+ }
piper_pick_can_10k/configuration_prismatic.py ADDED
@@ -0,0 +1,140 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """
2
+ configuration_prismatic.py
3
+
4
+ HuggingFace-style configuration definition for Prismatic VLMs, inheriting from `transformers.PretrainedConfig`.
5
+ Default configuration specifies `siglip-224px+7b`.
6
+ """
7
+
8
+ from typing import Any, Dict, List, Optional
9
+
10
+ from transformers import PretrainedConfig
11
+ from transformers.models.auto import CONFIG_MAPPING
12
+
13
+ # === Utilities for Mapping Prismatic names to HF names ===
14
+ # fmt: off
15
+ VISION_BACKBONE_TO_RESOLUTION: Dict[str, List[int]] = {
16
+ "clip-vit-l": [224], "siglip-vit-so400m": [224], "dinov2-vit-l": [224], "in1k-vit-l": [224],
17
+
18
+ "clip-vit-l-336px": [336],
19
+ "siglip-vit-so400m-384px": [384],
20
+
21
+ "dinoclip-vit-l-336px": [336, 336],
22
+ "dinosiglip-vit-so-224px": [224, 224],
23
+ "dinosiglip-vit-so-384px": [384, 384],
24
+ }
25
+ VISION_BACKBONE_TO_TIMM_ID: Dict[str, List[str]] = {
26
+ "clip-vit-l": ["vit_large_patch14_clip_224.openai"],
27
+ "clip-vit-l-336px": ["vit_large_patch14_clip_336.openai"],
28
+
29
+ "dinov2-vit-l": ["vit_large_patch14_reg4_dinov2.lvd142m"],
30
+ "in1k-vit-l": ["vit_large_patch16_224.augreg_in21k_ft_in1k"],
31
+
32
+ "siglip-vit-so400m": ["vit_so400m_patch14_siglip_224"],
33
+ "siglip-vit-so400m-384px": ["vit_so400m_patch14_siglip_384"],
34
+
35
+ "dinoclip-vit-l-336px": ["vit_large_patch14_reg4_dinov2.lvd142m", "vit_large_patch14_clip_336.openai"],
36
+ "dinosiglip-vit-so-224px": ["vit_large_patch14_reg4_dinov2.lvd142m", "vit_so400m_patch14_siglip_224"],
37
+ "dinosiglip-vit-so-384px": ["vit_large_patch14_reg4_dinov2.lvd142m", "vit_so400m_patch14_siglip_384"],
38
+ }
39
+ TIMM_OVERRIDE_ACT_LAYER: Dict[str, List[Optional[str]]] = {
40
+ "clip-vit-l": ["quick_gelu"], "clip-vit-l-336px": ["quick_gelu"],
41
+ "dinov2-vit-l": [None], "in1k-vit-l": [None],
42
+ "siglip-vit-so400m": [None], "siglip-vit-so400m-384px": [None],
43
+ "dinoclip-vit-l-336px": [None, "quick_gelu"],
44
+ "dinosiglip-vit-so-224px": [None, None], "dinosiglip-vit-so-384px": [None, None]
45
+ }
46
+
47
+ LLM_BACKBONE_TO_HF_PATH = {
48
+ "llama2-7b-pure": "meta-llama/Llama-2-7b-hf", "llama2-13b-pure": "meta-llama/Llama-2-13b-hf",
49
+ "llama2-7b-chat": "meta-llama/Llama-2-7b-chat-hf", "llama2-13b-chat": "meta-llama/Llama-2-13b-chat-hf",
50
+
51
+ "vicuna-v15-7b": "lmsys/vicuna-7b-v1.5", "vicuna-v15-13b": "lmsys/vicuna-13b-v1.5",
52
+
53
+ "mistral-v0.1-7b-pure": "mistralai/Mistral-7B-v0.1",
54
+ "mistral-v0.1-7b-instruct": "mistralai/Mistral-7B-Instruct-v0.1",
55
+
56
+ "phi-2-3b": "microsoft/phi-2",
57
+ }
58
+ LLM_BACKBONE_TO_HF_METACLASS = {
59
+ "llama2-7b-pure": "llama", "llama2-13b-pure": "llama", "llama2-7b-chat": "llama", "llama2-13b-chat": "llama",
60
+ "vicuna-v15-7b": "llama", "vicuna-v15-13b": "llama",
61
+
62
+ "mistral-v0.1-7b-pure": "mistral", "mistral-v0.1-7b-instruct": "mistral",
63
+
64
+ "phi-2-3b": "phi",
65
+ }
66
+
67
+ VALID_VISION_BACKBONES = set(VISION_BACKBONE_TO_RESOLUTION.keys())
68
+ VALID_LLM_BACKBONES = set(LLM_BACKBONE_TO_HF_PATH)
69
+ # fmt: on
70
+
71
+
72
+ class PrismaticConfig(PretrainedConfig):
73
+ model_type: str = "prismatic"
74
+ is_composition: bool = False
75
+
76
+ def __init__(
77
+ self,
78
+ vision_backbone_id: str = "siglip-vit-so400m",
79
+ llm_backbone_id: str = "vicuna-v15-7b",
80
+ arch_specifier: str = "no-align+gelu-mlp",
81
+ use_fused_vision_backbone: Optional[bool] = None,
82
+ image_resize_strategy: str = "letterbox",
83
+ text_config: Optional[Dict[str, Any]] = None,
84
+ llm_max_length: int = 2048,
85
+ pad_token_id: int = 32000,
86
+ pad_to_multiple_of: int = 64,
87
+ output_projector_states: bool = False,
88
+ **kwargs: str,
89
+ ) -> None:
90
+ if vision_backbone_id not in VALID_VISION_BACKBONES:
91
+ raise ValueError(f"Vision backbone `{vision_backbone_id}` not in {VALID_VISION_BACKBONES = }")
92
+
93
+ if llm_backbone_id not in VALID_LLM_BACKBONES:
94
+ raise ValueError(f"LLM backbone `{llm_backbone_id}` not in {VALID_LLM_BACKBONES = }")
95
+
96
+ # Set Prismatic Configuration Fields
97
+ self.vision_backbone_id = vision_backbone_id
98
+ self.llm_backbone_id = llm_backbone_id
99
+ self.arch_specifier = arch_specifier
100
+ self.output_projector_states = output_projector_states
101
+
102
+ # [Contract] All vision backbone parameters are lists =>> supports fused backbones with different preprocessing
103
+ self.use_fused_vision_backbone = (
104
+ use_fused_vision_backbone
105
+ if use_fused_vision_backbone is not None
106
+ else any(self.vision_backbone_id.startswith(v) for v in ["dinoclip", "dinosiglip"])
107
+ )
108
+
109
+ self.timm_model_ids = VISION_BACKBONE_TO_TIMM_ID[self.vision_backbone_id]
110
+ self.timm_override_act_layers = TIMM_OVERRIDE_ACT_LAYER[self.vision_backbone_id]
111
+ self.image_sizes = VISION_BACKBONE_TO_RESOLUTION[self.vision_backbone_id]
112
+ self.image_resize_strategy = image_resize_strategy
113
+
114
+ self.hf_llm_id = LLM_BACKBONE_TO_HF_PATH[self.llm_backbone_id]
115
+ self.llm_max_length = llm_max_length
116
+ self.pad_token_id, self.pad_to_multiple_of = pad_token_id, pad_to_multiple_of
117
+
118
+ # [IMPORTANT] HF Utilities actually look for a `text_config` field... we need to use that specific naming!
119
+ self.text_config = (
120
+ CONFIG_MAPPING[LLM_BACKBONE_TO_HF_METACLASS[self.llm_backbone_id]](**text_config)
121
+ if text_config is not None
122
+ else CONFIG_MAPPING[LLM_BACKBONE_TO_HF_METACLASS[self.llm_backbone_id]]()
123
+ )
124
+
125
+ # Dispatch **kwargs to super() =>> note that `pad_token_id` collides, so we pass it in here as well...
126
+ super().__init__(pad_token_id=pad_token_id, **kwargs)
127
+
128
+
129
+ class OpenVLAConfig(PrismaticConfig):
130
+ model_type: str = "openvla"
131
+
132
+ def __init__(
133
+ self,
134
+ norm_stats: Optional[Dict[str, Dict[str, Dict[str, Dict[str, List[float]]]]]] = None,
135
+ n_action_bins: int = 256,
136
+ **kwargs: str,
137
+ ) -> None:
138
+ self.norm_stats, self.n_action_bins = norm_stats, n_action_bins
139
+
140
+ super().__init__(**kwargs)
piper_pick_can_10k/ctrl_meta.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "latent": "tc_cont",
3
+ "n_tokens": 4,
4
+ "codebook_size": 0,
5
+ "dz": 256,
6
+ "window_size": 12,
7
+ "step": 10000,
8
+ "vla_path": "/home/ti874297/Projects/latent_free/cache/vla/openvla-7b",
9
+ "lam_path": "/home/ti874297/Projects/latent_free/claude_survey/ckpt/s2mix_1ep.pt",
10
+ "acdc_tag": "s1_acdc_1ep",
11
+ "acdc_codebook": "vq_s1_acdc_1ep_K64.npz",
12
+ "tc_tag": "latp2_inv",
13
+ "dataset_name": "piper_pick_can",
14
+ "train_frac": null,
15
+ "override": {},
16
+ "head": "flow",
17
+ "query_tokens": 4,
18
+ "flow": {
19
+ "width": 512,
20
+ "layers": 4,
21
+ "z_tok": 32,
22
+ "steps": 10
23
+ }
24
+ }
piper_pick_can_10k/dataset_statistics.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "piper_pick_can": {
3
+ "action": {
4
+ "mean": [
5
+ -2.9980573654174805,
6
+ 111.5005874633789,
7
+ -68.02422332763672,
8
+ -3.6513760089874268,
9
+ -20.165210723876953,
10
+ -0.7962256669998169,
11
+ 0.6070931553840637
12
+ ],
13
+ "std": [
14
+ 20.342702865600586,
15
+ 38.4741325378418,
16
+ 37.62217330932617,
17
+ 5.898085594177246,
18
+ 13.74382495880127,
19
+ 2.4010581970214844,
20
+ 0.40146440267562866
21
+ ],
22
+ "max": [
23
+ 38.301998138427734,
24
+ 173.63600158691406,
25
+ 0.4320000112056732,
26
+ 43.90800094604492,
27
+ 49.02299880981445,
28
+ 9.246000289916992,
29
+ 0.9930487871170044
30
+ ],
31
+ "min": [
32
+ -45.207000732421875,
33
+ 0.04500000178813934,
34
+ -157.5919952392578,
35
+ -24.059999465942383,
36
+ -58.637001037597656,
37
+ -21.71299934387207,
38
+ 0.0
39
+ ],
40
+ "q01": [
41
+ -38.527000427246094,
42
+ 2.4509999752044678,
43
+ -147.5540008544922,
44
+ -18.913559226989747,
45
+ -47.396848602294924,
46
+ -9.83550971031189,
47
+ 0.0012195121962577105
48
+ ],
49
+ "q99": [
50
+ 31.479000091552734,
51
+ 166.66000366210938,
52
+ 0.0,
53
+ 10.813340072631838,
54
+ 18.056560783386264,
55
+ 4.296999931335449,
56
+ 0.9921951293945312
57
+ ],
58
+ "mask": [
59
+ true,
60
+ true,
61
+ true,
62
+ true,
63
+ true,
64
+ true,
65
+ false
66
+ ]
67
+ },
68
+ "proprio": {
69
+ "mean": [
70
+ 0.0,
71
+ 0.0,
72
+ 0.0,
73
+ 0.0,
74
+ 0.0,
75
+ 0.0,
76
+ 0.0
77
+ ],
78
+ "std": [
79
+ 0.0,
80
+ 0.0,
81
+ 0.0,
82
+ 0.0,
83
+ 0.0,
84
+ 0.0,
85
+ 0.0
86
+ ],
87
+ "max": [
88
+ 0.0,
89
+ 0.0,
90
+ 0.0,
91
+ 0.0,
92
+ 0.0,
93
+ 0.0,
94
+ 0.0
95
+ ],
96
+ "min": [
97
+ 0.0,
98
+ 0.0,
99
+ 0.0,
100
+ 0.0,
101
+ 0.0,
102
+ 0.0,
103
+ 0.0
104
+ ],
105
+ "q01": [
106
+ 0.0,
107
+ 0.0,
108
+ 0.0,
109
+ 0.0,
110
+ 0.0,
111
+ 0.0,
112
+ 0.0
113
+ ],
114
+ "q99": [
115
+ 0.0,
116
+ 0.0,
117
+ 0.0,
118
+ 0.0,
119
+ 0.0,
120
+ 0.0,
121
+ 0.0
122
+ ]
123
+ },
124
+ "num_transitions": 6462,
125
+ "num_trajectories": 150
126
+ }
127
+ }
piper_pick_can_10k/generation_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 1,
4
+ "eos_token_id": 2,
5
+ "pad_token_id": 32000,
6
+ "transformers_version": "4.40.1"
7
+ }
piper_pick_can_10k/model-00001-of-00004.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:812abf2dd26634ff0093332a9e9dc77af86bc6d47e1474bd112b5df2f75467fc
3
+ size 4924639120
piper_pick_can_10k/model-00002-of-00004.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:42389a98358cadf6afbc4d2c4be0c8cead4ca5bbd3e9c51faaa76f09e9326f3f
3
+ size 4947392496
piper_pick_can_10k/model-00003-of-00004.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:18844fff4fe7c2d846b18414677bd74f27584c49badde9b16c6c24d3389aefcf
3
+ size 4947417456
piper_pick_can_10k/model-00004-of-00004.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c02efd18fd0cbed4fed8d1fe3dfad42f6da6c452b58f880f0faecd1e92e911f7
3
+ size 262185104
piper_pick_can_10k/model.safetensors.index.json ADDED
@@ -0,0 +1,989 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metadata": {
3
+ "total_size": 15081507712
4
+ },
5
+ "weight_map": {
6
+ "language_model.lm_head.weight": "model-00004-of-00004.safetensors",
7
+ "language_model.model.embed_tokens.weight": "model-00001-of-00004.safetensors",
8
+ "language_model.model.layers.0.input_layernorm.weight": "model-00001-of-00004.safetensors",
9
+ "language_model.model.layers.0.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
10
+ "language_model.model.layers.0.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
11
+ "language_model.model.layers.0.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
12
+ "language_model.model.layers.0.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
13
+ "language_model.model.layers.0.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
14
+ "language_model.model.layers.0.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
15
+ "language_model.model.layers.0.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
16
+ "language_model.model.layers.0.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
17
+ "language_model.model.layers.1.input_layernorm.weight": "model-00001-of-00004.safetensors",
18
+ "language_model.model.layers.1.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
19
+ "language_model.model.layers.1.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
20
+ "language_model.model.layers.1.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
21
+ "language_model.model.layers.1.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
22
+ "language_model.model.layers.1.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
23
+ "language_model.model.layers.1.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
24
+ "language_model.model.layers.1.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
25
+ "language_model.model.layers.1.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
26
+ "language_model.model.layers.10.input_layernorm.weight": "model-00002-of-00004.safetensors",
27
+ "language_model.model.layers.10.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
28
+ "language_model.model.layers.10.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
29
+ "language_model.model.layers.10.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
30
+ "language_model.model.layers.10.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
31
+ "language_model.model.layers.10.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
32
+ "language_model.model.layers.10.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
33
+ "language_model.model.layers.10.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
34
+ "language_model.model.layers.10.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
35
+ "language_model.model.layers.11.input_layernorm.weight": "model-00002-of-00004.safetensors",
36
+ "language_model.model.layers.11.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
37
+ "language_model.model.layers.11.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
38
+ "language_model.model.layers.11.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
39
+ "language_model.model.layers.11.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
40
+ "language_model.model.layers.11.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
41
+ "language_model.model.layers.11.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
42
+ "language_model.model.layers.11.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
43
+ "language_model.model.layers.11.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
44
+ "language_model.model.layers.12.input_layernorm.weight": "model-00002-of-00004.safetensors",
45
+ "language_model.model.layers.12.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
46
+ "language_model.model.layers.12.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
47
+ "language_model.model.layers.12.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
48
+ "language_model.model.layers.12.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
49
+ "language_model.model.layers.12.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
50
+ "language_model.model.layers.12.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
51
+ "language_model.model.layers.12.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
52
+ "language_model.model.layers.12.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
53
+ "language_model.model.layers.13.input_layernorm.weight": "model-00002-of-00004.safetensors",
54
+ "language_model.model.layers.13.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
55
+ "language_model.model.layers.13.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
56
+ "language_model.model.layers.13.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
57
+ "language_model.model.layers.13.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
58
+ "language_model.model.layers.13.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
59
+ "language_model.model.layers.13.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
60
+ "language_model.model.layers.13.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
61
+ "language_model.model.layers.13.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
62
+ "language_model.model.layers.14.input_layernorm.weight": "model-00002-of-00004.safetensors",
63
+ "language_model.model.layers.14.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
64
+ "language_model.model.layers.14.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
65
+ "language_model.model.layers.14.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
66
+ "language_model.model.layers.14.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
67
+ "language_model.model.layers.14.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
68
+ "language_model.model.layers.14.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
69
+ "language_model.model.layers.14.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
70
+ "language_model.model.layers.14.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
71
+ "language_model.model.layers.15.input_layernorm.weight": "model-00002-of-00004.safetensors",
72
+ "language_model.model.layers.15.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
73
+ "language_model.model.layers.15.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
74
+ "language_model.model.layers.15.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
75
+ "language_model.model.layers.15.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
76
+ "language_model.model.layers.15.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
77
+ "language_model.model.layers.15.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
78
+ "language_model.model.layers.15.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
79
+ "language_model.model.layers.15.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
80
+ "language_model.model.layers.16.input_layernorm.weight": "model-00002-of-00004.safetensors",
81
+ "language_model.model.layers.16.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
82
+ "language_model.model.layers.16.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
83
+ "language_model.model.layers.16.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
84
+ "language_model.model.layers.16.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
85
+ "language_model.model.layers.16.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
86
+ "language_model.model.layers.16.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
87
+ "language_model.model.layers.16.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
88
+ "language_model.model.layers.16.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
89
+ "language_model.model.layers.17.input_layernorm.weight": "model-00002-of-00004.safetensors",
90
+ "language_model.model.layers.17.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
91
+ "language_model.model.layers.17.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
92
+ "language_model.model.layers.17.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
93
+ "language_model.model.layers.17.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
94
+ "language_model.model.layers.17.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
95
+ "language_model.model.layers.17.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
96
+ "language_model.model.layers.17.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
97
+ "language_model.model.layers.17.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
98
+ "language_model.model.layers.18.input_layernorm.weight": "model-00002-of-00004.safetensors",
99
+ "language_model.model.layers.18.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
100
+ "language_model.model.layers.18.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
101
+ "language_model.model.layers.18.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
102
+ "language_model.model.layers.18.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
103
+ "language_model.model.layers.18.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
104
+ "language_model.model.layers.18.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
105
+ "language_model.model.layers.18.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
106
+ "language_model.model.layers.18.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
107
+ "language_model.model.layers.19.input_layernorm.weight": "model-00003-of-00004.safetensors",
108
+ "language_model.model.layers.19.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
109
+ "language_model.model.layers.19.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
110
+ "language_model.model.layers.19.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
111
+ "language_model.model.layers.19.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
112
+ "language_model.model.layers.19.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
113
+ "language_model.model.layers.19.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
114
+ "language_model.model.layers.19.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
115
+ "language_model.model.layers.19.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
116
+ "language_model.model.layers.2.input_layernorm.weight": "model-00001-of-00004.safetensors",
117
+ "language_model.model.layers.2.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
118
+ "language_model.model.layers.2.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
119
+ "language_model.model.layers.2.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
120
+ "language_model.model.layers.2.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
121
+ "language_model.model.layers.2.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
122
+ "language_model.model.layers.2.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
123
+ "language_model.model.layers.2.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
124
+ "language_model.model.layers.2.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
125
+ "language_model.model.layers.20.input_layernorm.weight": "model-00003-of-00004.safetensors",
126
+ "language_model.model.layers.20.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
127
+ "language_model.model.layers.20.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
128
+ "language_model.model.layers.20.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
129
+ "language_model.model.layers.20.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
130
+ "language_model.model.layers.20.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
131
+ "language_model.model.layers.20.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
132
+ "language_model.model.layers.20.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
133
+ "language_model.model.layers.20.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
134
+ "language_model.model.layers.21.input_layernorm.weight": "model-00003-of-00004.safetensors",
135
+ "language_model.model.layers.21.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
136
+ "language_model.model.layers.21.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
137
+ "language_model.model.layers.21.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
138
+ "language_model.model.layers.21.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
139
+ "language_model.model.layers.21.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
140
+ "language_model.model.layers.21.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
141
+ "language_model.model.layers.21.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
142
+ "language_model.model.layers.21.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
143
+ "language_model.model.layers.22.input_layernorm.weight": "model-00003-of-00004.safetensors",
144
+ "language_model.model.layers.22.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
145
+ "language_model.model.layers.22.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
146
+ "language_model.model.layers.22.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
147
+ "language_model.model.layers.22.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
148
+ "language_model.model.layers.22.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
149
+ "language_model.model.layers.22.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
150
+ "language_model.model.layers.22.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
151
+ "language_model.model.layers.22.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
152
+ "language_model.model.layers.23.input_layernorm.weight": "model-00003-of-00004.safetensors",
153
+ "language_model.model.layers.23.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
154
+ "language_model.model.layers.23.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
155
+ "language_model.model.layers.23.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
156
+ "language_model.model.layers.23.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
157
+ "language_model.model.layers.23.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
158
+ "language_model.model.layers.23.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
159
+ "language_model.model.layers.23.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
160
+ "language_model.model.layers.23.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
161
+ "language_model.model.layers.24.input_layernorm.weight": "model-00003-of-00004.safetensors",
162
+ "language_model.model.layers.24.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
163
+ "language_model.model.layers.24.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
164
+ "language_model.model.layers.24.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
165
+ "language_model.model.layers.24.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
166
+ "language_model.model.layers.24.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
167
+ "language_model.model.layers.24.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
168
+ "language_model.model.layers.24.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
169
+ "language_model.model.layers.24.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
170
+ "language_model.model.layers.25.input_layernorm.weight": "model-00003-of-00004.safetensors",
171
+ "language_model.model.layers.25.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
172
+ "language_model.model.layers.25.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
173
+ "language_model.model.layers.25.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
174
+ "language_model.model.layers.25.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
175
+ "language_model.model.layers.25.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
176
+ "language_model.model.layers.25.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
177
+ "language_model.model.layers.25.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
178
+ "language_model.model.layers.25.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
179
+ "language_model.model.layers.26.input_layernorm.weight": "model-00003-of-00004.safetensors",
180
+ "language_model.model.layers.26.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
181
+ "language_model.model.layers.26.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
182
+ "language_model.model.layers.26.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
183
+ "language_model.model.layers.26.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
184
+ "language_model.model.layers.26.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
185
+ "language_model.model.layers.26.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
186
+ "language_model.model.layers.26.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
187
+ "language_model.model.layers.26.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
188
+ "language_model.model.layers.27.input_layernorm.weight": "model-00003-of-00004.safetensors",
189
+ "language_model.model.layers.27.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
190
+ "language_model.model.layers.27.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
191
+ "language_model.model.layers.27.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
192
+ "language_model.model.layers.27.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
193
+ "language_model.model.layers.27.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
194
+ "language_model.model.layers.27.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
195
+ "language_model.model.layers.27.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
196
+ "language_model.model.layers.27.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
197
+ "language_model.model.layers.28.input_layernorm.weight": "model-00003-of-00004.safetensors",
198
+ "language_model.model.layers.28.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
199
+ "language_model.model.layers.28.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
200
+ "language_model.model.layers.28.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
201
+ "language_model.model.layers.28.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
202
+ "language_model.model.layers.28.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
203
+ "language_model.model.layers.28.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
204
+ "language_model.model.layers.28.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
205
+ "language_model.model.layers.28.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
206
+ "language_model.model.layers.29.input_layernorm.weight": "model-00003-of-00004.safetensors",
207
+ "language_model.model.layers.29.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
208
+ "language_model.model.layers.29.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
209
+ "language_model.model.layers.29.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
210
+ "language_model.model.layers.29.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
211
+ "language_model.model.layers.29.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
212
+ "language_model.model.layers.29.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
213
+ "language_model.model.layers.29.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
214
+ "language_model.model.layers.29.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
215
+ "language_model.model.layers.3.input_layernorm.weight": "model-00001-of-00004.safetensors",
216
+ "language_model.model.layers.3.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
217
+ "language_model.model.layers.3.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
218
+ "language_model.model.layers.3.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
219
+ "language_model.model.layers.3.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
220
+ "language_model.model.layers.3.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
221
+ "language_model.model.layers.3.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
222
+ "language_model.model.layers.3.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
223
+ "language_model.model.layers.3.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
224
+ "language_model.model.layers.30.input_layernorm.weight": "model-00003-of-00004.safetensors",
225
+ "language_model.model.layers.30.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
226
+ "language_model.model.layers.30.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
227
+ "language_model.model.layers.30.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
228
+ "language_model.model.layers.30.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
229
+ "language_model.model.layers.30.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
230
+ "language_model.model.layers.30.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
231
+ "language_model.model.layers.30.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
232
+ "language_model.model.layers.30.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
233
+ "language_model.model.layers.31.input_layernorm.weight": "model-00003-of-00004.safetensors",
234
+ "language_model.model.layers.31.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
235
+ "language_model.model.layers.31.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
236
+ "language_model.model.layers.31.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
237
+ "language_model.model.layers.31.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
238
+ "language_model.model.layers.31.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
239
+ "language_model.model.layers.31.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
240
+ "language_model.model.layers.31.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
241
+ "language_model.model.layers.31.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
242
+ "language_model.model.layers.4.input_layernorm.weight": "model-00001-of-00004.safetensors",
243
+ "language_model.model.layers.4.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
244
+ "language_model.model.layers.4.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
245
+ "language_model.model.layers.4.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
246
+ "language_model.model.layers.4.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
247
+ "language_model.model.layers.4.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
248
+ "language_model.model.layers.4.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
249
+ "language_model.model.layers.4.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
250
+ "language_model.model.layers.4.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
251
+ "language_model.model.layers.5.input_layernorm.weight": "model-00001-of-00004.safetensors",
252
+ "language_model.model.layers.5.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
253
+ "language_model.model.layers.5.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
254
+ "language_model.model.layers.5.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
255
+ "language_model.model.layers.5.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
256
+ "language_model.model.layers.5.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
257
+ "language_model.model.layers.5.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
258
+ "language_model.model.layers.5.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
259
+ "language_model.model.layers.5.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
260
+ "language_model.model.layers.6.input_layernorm.weight": "model-00001-of-00004.safetensors",
261
+ "language_model.model.layers.6.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
262
+ "language_model.model.layers.6.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
263
+ "language_model.model.layers.6.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
264
+ "language_model.model.layers.6.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
265
+ "language_model.model.layers.6.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
266
+ "language_model.model.layers.6.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
267
+ "language_model.model.layers.6.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
268
+ "language_model.model.layers.6.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
269
+ "language_model.model.layers.7.input_layernorm.weight": "model-00002-of-00004.safetensors",
270
+ "language_model.model.layers.7.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
271
+ "language_model.model.layers.7.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
272
+ "language_model.model.layers.7.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
273
+ "language_model.model.layers.7.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
274
+ "language_model.model.layers.7.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
275
+ "language_model.model.layers.7.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
276
+ "language_model.model.layers.7.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
277
+ "language_model.model.layers.7.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
278
+ "language_model.model.layers.8.input_layernorm.weight": "model-00002-of-00004.safetensors",
279
+ "language_model.model.layers.8.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
280
+ "language_model.model.layers.8.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
281
+ "language_model.model.layers.8.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
282
+ "language_model.model.layers.8.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
283
+ "language_model.model.layers.8.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
284
+ "language_model.model.layers.8.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
285
+ "language_model.model.layers.8.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
286
+ "language_model.model.layers.8.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
287
+ "language_model.model.layers.9.input_layernorm.weight": "model-00002-of-00004.safetensors",
288
+ "language_model.model.layers.9.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
289
+ "language_model.model.layers.9.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
290
+ "language_model.model.layers.9.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
291
+ "language_model.model.layers.9.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
292
+ "language_model.model.layers.9.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
293
+ "language_model.model.layers.9.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
294
+ "language_model.model.layers.9.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
295
+ "language_model.model.layers.9.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
296
+ "language_model.model.norm.weight": "model-00003-of-00004.safetensors",
297
+ "projector.fc1.bias": "model-00001-of-00004.safetensors",
298
+ "projector.fc1.weight": "model-00001-of-00004.safetensors",
299
+ "projector.fc2.bias": "model-00001-of-00004.safetensors",
300
+ "projector.fc2.weight": "model-00001-of-00004.safetensors",
301
+ "projector.fc3.bias": "model-00001-of-00004.safetensors",
302
+ "projector.fc3.weight": "model-00001-of-00004.safetensors",
303
+ "vision_backbone.featurizer.blocks.0.attn.proj.bias": "model-00001-of-00004.safetensors",
304
+ "vision_backbone.featurizer.blocks.0.attn.proj.weight": "model-00001-of-00004.safetensors",
305
+ "vision_backbone.featurizer.blocks.0.attn.qkv.bias": "model-00001-of-00004.safetensors",
306
+ "vision_backbone.featurizer.blocks.0.attn.qkv.weight": "model-00001-of-00004.safetensors",
307
+ "vision_backbone.featurizer.blocks.0.ls1.scale_factor": "model-00001-of-00004.safetensors",
308
+ "vision_backbone.featurizer.blocks.0.ls2.scale_factor": "model-00001-of-00004.safetensors",
309
+ "vision_backbone.featurizer.blocks.0.mlp.fc1.bias": "model-00001-of-00004.safetensors",
310
+ "vision_backbone.featurizer.blocks.0.mlp.fc1.weight": "model-00001-of-00004.safetensors",
311
+ "vision_backbone.featurizer.blocks.0.mlp.fc2.bias": "model-00001-of-00004.safetensors",
312
+ "vision_backbone.featurizer.blocks.0.mlp.fc2.weight": "model-00001-of-00004.safetensors",
313
+ "vision_backbone.featurizer.blocks.0.norm1.bias": "model-00001-of-00004.safetensors",
314
+ "vision_backbone.featurizer.blocks.0.norm1.weight": "model-00001-of-00004.safetensors",
315
+ "vision_backbone.featurizer.blocks.0.norm2.bias": "model-00001-of-00004.safetensors",
316
+ "vision_backbone.featurizer.blocks.0.norm2.weight": "model-00001-of-00004.safetensors",
317
+ "vision_backbone.featurizer.blocks.1.attn.proj.bias": "model-00001-of-00004.safetensors",
318
+ "vision_backbone.featurizer.blocks.1.attn.proj.weight": "model-00001-of-00004.safetensors",
319
+ "vision_backbone.featurizer.blocks.1.attn.qkv.bias": "model-00001-of-00004.safetensors",
320
+ "vision_backbone.featurizer.blocks.1.attn.qkv.weight": "model-00001-of-00004.safetensors",
321
+ "vision_backbone.featurizer.blocks.1.ls1.scale_factor": "model-00001-of-00004.safetensors",
322
+ "vision_backbone.featurizer.blocks.1.ls2.scale_factor": "model-00001-of-00004.safetensors",
323
+ "vision_backbone.featurizer.blocks.1.mlp.fc1.bias": "model-00001-of-00004.safetensors",
324
+ "vision_backbone.featurizer.blocks.1.mlp.fc1.weight": "model-00001-of-00004.safetensors",
325
+ "vision_backbone.featurizer.blocks.1.mlp.fc2.bias": "model-00001-of-00004.safetensors",
326
+ "vision_backbone.featurizer.blocks.1.mlp.fc2.weight": "model-00001-of-00004.safetensors",
327
+ "vision_backbone.featurizer.blocks.1.norm1.bias": "model-00001-of-00004.safetensors",
328
+ "vision_backbone.featurizer.blocks.1.norm1.weight": "model-00001-of-00004.safetensors",
329
+ "vision_backbone.featurizer.blocks.1.norm2.bias": "model-00001-of-00004.safetensors",
330
+ "vision_backbone.featurizer.blocks.1.norm2.weight": "model-00001-of-00004.safetensors",
331
+ "vision_backbone.featurizer.blocks.10.attn.proj.bias": "model-00001-of-00004.safetensors",
332
+ "vision_backbone.featurizer.blocks.10.attn.proj.weight": "model-00001-of-00004.safetensors",
333
+ "vision_backbone.featurizer.blocks.10.attn.qkv.bias": "model-00001-of-00004.safetensors",
334
+ "vision_backbone.featurizer.blocks.10.attn.qkv.weight": "model-00001-of-00004.safetensors",
335
+ "vision_backbone.featurizer.blocks.10.ls1.scale_factor": "model-00001-of-00004.safetensors",
336
+ "vision_backbone.featurizer.blocks.10.ls2.scale_factor": "model-00001-of-00004.safetensors",
337
+ "vision_backbone.featurizer.blocks.10.mlp.fc1.bias": "model-00001-of-00004.safetensors",
338
+ "vision_backbone.featurizer.blocks.10.mlp.fc1.weight": "model-00001-of-00004.safetensors",
339
+ "vision_backbone.featurizer.blocks.10.mlp.fc2.bias": "model-00001-of-00004.safetensors",
340
+ "vision_backbone.featurizer.blocks.10.mlp.fc2.weight": "model-00001-of-00004.safetensors",
341
+ "vision_backbone.featurizer.blocks.10.norm1.bias": "model-00001-of-00004.safetensors",
342
+ "vision_backbone.featurizer.blocks.10.norm1.weight": "model-00001-of-00004.safetensors",
343
+ "vision_backbone.featurizer.blocks.10.norm2.bias": "model-00001-of-00004.safetensors",
344
+ "vision_backbone.featurizer.blocks.10.norm2.weight": "model-00001-of-00004.safetensors",
345
+ "vision_backbone.featurizer.blocks.11.attn.proj.bias": "model-00001-of-00004.safetensors",
346
+ "vision_backbone.featurizer.blocks.11.attn.proj.weight": "model-00001-of-00004.safetensors",
347
+ "vision_backbone.featurizer.blocks.11.attn.qkv.bias": "model-00001-of-00004.safetensors",
348
+ "vision_backbone.featurizer.blocks.11.attn.qkv.weight": "model-00001-of-00004.safetensors",
349
+ "vision_backbone.featurizer.blocks.11.ls1.scale_factor": "model-00001-of-00004.safetensors",
350
+ "vision_backbone.featurizer.blocks.11.ls2.scale_factor": "model-00001-of-00004.safetensors",
351
+ "vision_backbone.featurizer.blocks.11.mlp.fc1.bias": "model-00001-of-00004.safetensors",
352
+ "vision_backbone.featurizer.blocks.11.mlp.fc1.weight": "model-00001-of-00004.safetensors",
353
+ "vision_backbone.featurizer.blocks.11.mlp.fc2.bias": "model-00001-of-00004.safetensors",
354
+ "vision_backbone.featurizer.blocks.11.mlp.fc2.weight": "model-00001-of-00004.safetensors",
355
+ "vision_backbone.featurizer.blocks.11.norm1.bias": "model-00001-of-00004.safetensors",
356
+ "vision_backbone.featurizer.blocks.11.norm1.weight": "model-00001-of-00004.safetensors",
357
+ "vision_backbone.featurizer.blocks.11.norm2.bias": "model-00001-of-00004.safetensors",
358
+ "vision_backbone.featurizer.blocks.11.norm2.weight": "model-00001-of-00004.safetensors",
359
+ "vision_backbone.featurizer.blocks.12.attn.proj.bias": "model-00001-of-00004.safetensors",
360
+ "vision_backbone.featurizer.blocks.12.attn.proj.weight": "model-00001-of-00004.safetensors",
361
+ "vision_backbone.featurizer.blocks.12.attn.qkv.bias": "model-00001-of-00004.safetensors",
362
+ "vision_backbone.featurizer.blocks.12.attn.qkv.weight": "model-00001-of-00004.safetensors",
363
+ "vision_backbone.featurizer.blocks.12.ls1.scale_factor": "model-00001-of-00004.safetensors",
364
+ "vision_backbone.featurizer.blocks.12.ls2.scale_factor": "model-00001-of-00004.safetensors",
365
+ "vision_backbone.featurizer.blocks.12.mlp.fc1.bias": "model-00001-of-00004.safetensors",
366
+ "vision_backbone.featurizer.blocks.12.mlp.fc1.weight": "model-00001-of-00004.safetensors",
367
+ "vision_backbone.featurizer.blocks.12.mlp.fc2.bias": "model-00001-of-00004.safetensors",
368
+ "vision_backbone.featurizer.blocks.12.mlp.fc2.weight": "model-00001-of-00004.safetensors",
369
+ "vision_backbone.featurizer.blocks.12.norm1.bias": "model-00001-of-00004.safetensors",
370
+ "vision_backbone.featurizer.blocks.12.norm1.weight": "model-00001-of-00004.safetensors",
371
+ "vision_backbone.featurizer.blocks.12.norm2.bias": "model-00001-of-00004.safetensors",
372
+ "vision_backbone.featurizer.blocks.12.norm2.weight": "model-00001-of-00004.safetensors",
373
+ "vision_backbone.featurizer.blocks.13.attn.proj.bias": "model-00001-of-00004.safetensors",
374
+ "vision_backbone.featurizer.blocks.13.attn.proj.weight": "model-00001-of-00004.safetensors",
375
+ "vision_backbone.featurizer.blocks.13.attn.qkv.bias": "model-00001-of-00004.safetensors",
376
+ "vision_backbone.featurizer.blocks.13.attn.qkv.weight": "model-00001-of-00004.safetensors",
377
+ "vision_backbone.featurizer.blocks.13.ls1.scale_factor": "model-00001-of-00004.safetensors",
378
+ "vision_backbone.featurizer.blocks.13.ls2.scale_factor": "model-00001-of-00004.safetensors",
379
+ "vision_backbone.featurizer.blocks.13.mlp.fc1.bias": "model-00001-of-00004.safetensors",
380
+ "vision_backbone.featurizer.blocks.13.mlp.fc1.weight": "model-00001-of-00004.safetensors",
381
+ "vision_backbone.featurizer.blocks.13.mlp.fc2.bias": "model-00001-of-00004.safetensors",
382
+ "vision_backbone.featurizer.blocks.13.mlp.fc2.weight": "model-00001-of-00004.safetensors",
383
+ "vision_backbone.featurizer.blocks.13.norm1.bias": "model-00001-of-00004.safetensors",
384
+ "vision_backbone.featurizer.blocks.13.norm1.weight": "model-00001-of-00004.safetensors",
385
+ "vision_backbone.featurizer.blocks.13.norm2.bias": "model-00001-of-00004.safetensors",
386
+ "vision_backbone.featurizer.blocks.13.norm2.weight": "model-00001-of-00004.safetensors",
387
+ "vision_backbone.featurizer.blocks.14.attn.proj.bias": "model-00001-of-00004.safetensors",
388
+ "vision_backbone.featurizer.blocks.14.attn.proj.weight": "model-00001-of-00004.safetensors",
389
+ "vision_backbone.featurizer.blocks.14.attn.qkv.bias": "model-00001-of-00004.safetensors",
390
+ "vision_backbone.featurizer.blocks.14.attn.qkv.weight": "model-00001-of-00004.safetensors",
391
+ "vision_backbone.featurizer.blocks.14.ls1.scale_factor": "model-00001-of-00004.safetensors",
392
+ "vision_backbone.featurizer.blocks.14.ls2.scale_factor": "model-00001-of-00004.safetensors",
393
+ "vision_backbone.featurizer.blocks.14.mlp.fc1.bias": "model-00001-of-00004.safetensors",
394
+ "vision_backbone.featurizer.blocks.14.mlp.fc1.weight": "model-00001-of-00004.safetensors",
395
+ "vision_backbone.featurizer.blocks.14.mlp.fc2.bias": "model-00001-of-00004.safetensors",
396
+ "vision_backbone.featurizer.blocks.14.mlp.fc2.weight": "model-00001-of-00004.safetensors",
397
+ "vision_backbone.featurizer.blocks.14.norm1.bias": "model-00001-of-00004.safetensors",
398
+ "vision_backbone.featurizer.blocks.14.norm1.weight": "model-00001-of-00004.safetensors",
399
+ "vision_backbone.featurizer.blocks.14.norm2.bias": "model-00001-of-00004.safetensors",
400
+ "vision_backbone.featurizer.blocks.14.norm2.weight": "model-00001-of-00004.safetensors",
401
+ "vision_backbone.featurizer.blocks.15.attn.proj.bias": "model-00001-of-00004.safetensors",
402
+ "vision_backbone.featurizer.blocks.15.attn.proj.weight": "model-00001-of-00004.safetensors",
403
+ "vision_backbone.featurizer.blocks.15.attn.qkv.bias": "model-00001-of-00004.safetensors",
404
+ "vision_backbone.featurizer.blocks.15.attn.qkv.weight": "model-00001-of-00004.safetensors",
405
+ "vision_backbone.featurizer.blocks.15.ls1.scale_factor": "model-00001-of-00004.safetensors",
406
+ "vision_backbone.featurizer.blocks.15.ls2.scale_factor": "model-00001-of-00004.safetensors",
407
+ "vision_backbone.featurizer.blocks.15.mlp.fc1.bias": "model-00001-of-00004.safetensors",
408
+ "vision_backbone.featurizer.blocks.15.mlp.fc1.weight": "model-00001-of-00004.safetensors",
409
+ "vision_backbone.featurizer.blocks.15.mlp.fc2.bias": "model-00001-of-00004.safetensors",
410
+ "vision_backbone.featurizer.blocks.15.mlp.fc2.weight": "model-00001-of-00004.safetensors",
411
+ "vision_backbone.featurizer.blocks.15.norm1.bias": "model-00001-of-00004.safetensors",
412
+ "vision_backbone.featurizer.blocks.15.norm1.weight": "model-00001-of-00004.safetensors",
413
+ "vision_backbone.featurizer.blocks.15.norm2.bias": "model-00001-of-00004.safetensors",
414
+ "vision_backbone.featurizer.blocks.15.norm2.weight": "model-00001-of-00004.safetensors",
415
+ "vision_backbone.featurizer.blocks.16.attn.proj.bias": "model-00001-of-00004.safetensors",
416
+ "vision_backbone.featurizer.blocks.16.attn.proj.weight": "model-00001-of-00004.safetensors",
417
+ "vision_backbone.featurizer.blocks.16.attn.qkv.bias": "model-00001-of-00004.safetensors",
418
+ "vision_backbone.featurizer.blocks.16.attn.qkv.weight": "model-00001-of-00004.safetensors",
419
+ "vision_backbone.featurizer.blocks.16.ls1.scale_factor": "model-00001-of-00004.safetensors",
420
+ "vision_backbone.featurizer.blocks.16.ls2.scale_factor": "model-00001-of-00004.safetensors",
421
+ "vision_backbone.featurizer.blocks.16.mlp.fc1.bias": "model-00001-of-00004.safetensors",
422
+ "vision_backbone.featurizer.blocks.16.mlp.fc1.weight": "model-00001-of-00004.safetensors",
423
+ "vision_backbone.featurizer.blocks.16.mlp.fc2.bias": "model-00001-of-00004.safetensors",
424
+ "vision_backbone.featurizer.blocks.16.mlp.fc2.weight": "model-00001-of-00004.safetensors",
425
+ "vision_backbone.featurizer.blocks.16.norm1.bias": "model-00001-of-00004.safetensors",
426
+ "vision_backbone.featurizer.blocks.16.norm1.weight": "model-00001-of-00004.safetensors",
427
+ "vision_backbone.featurizer.blocks.16.norm2.bias": "model-00001-of-00004.safetensors",
428
+ "vision_backbone.featurizer.blocks.16.norm2.weight": "model-00001-of-00004.safetensors",
429
+ "vision_backbone.featurizer.blocks.17.attn.proj.bias": "model-00001-of-00004.safetensors",
430
+ "vision_backbone.featurizer.blocks.17.attn.proj.weight": "model-00001-of-00004.safetensors",
431
+ "vision_backbone.featurizer.blocks.17.attn.qkv.bias": "model-00001-of-00004.safetensors",
432
+ "vision_backbone.featurizer.blocks.17.attn.qkv.weight": "model-00001-of-00004.safetensors",
433
+ "vision_backbone.featurizer.blocks.17.ls1.scale_factor": "model-00001-of-00004.safetensors",
434
+ "vision_backbone.featurizer.blocks.17.ls2.scale_factor": "model-00001-of-00004.safetensors",
435
+ "vision_backbone.featurizer.blocks.17.mlp.fc1.bias": "model-00001-of-00004.safetensors",
436
+ "vision_backbone.featurizer.blocks.17.mlp.fc1.weight": "model-00001-of-00004.safetensors",
437
+ "vision_backbone.featurizer.blocks.17.mlp.fc2.bias": "model-00001-of-00004.safetensors",
438
+ "vision_backbone.featurizer.blocks.17.mlp.fc2.weight": "model-00001-of-00004.safetensors",
439
+ "vision_backbone.featurizer.blocks.17.norm1.bias": "model-00001-of-00004.safetensors",
440
+ "vision_backbone.featurizer.blocks.17.norm1.weight": "model-00001-of-00004.safetensors",
441
+ "vision_backbone.featurizer.blocks.17.norm2.bias": "model-00001-of-00004.safetensors",
442
+ "vision_backbone.featurizer.blocks.17.norm2.weight": "model-00001-of-00004.safetensors",
443
+ "vision_backbone.featurizer.blocks.18.attn.proj.bias": "model-00001-of-00004.safetensors",
444
+ "vision_backbone.featurizer.blocks.18.attn.proj.weight": "model-00001-of-00004.safetensors",
445
+ "vision_backbone.featurizer.blocks.18.attn.qkv.bias": "model-00001-of-00004.safetensors",
446
+ "vision_backbone.featurizer.blocks.18.attn.qkv.weight": "model-00001-of-00004.safetensors",
447
+ "vision_backbone.featurizer.blocks.18.ls1.scale_factor": "model-00001-of-00004.safetensors",
448
+ "vision_backbone.featurizer.blocks.18.ls2.scale_factor": "model-00001-of-00004.safetensors",
449
+ "vision_backbone.featurizer.blocks.18.mlp.fc1.bias": "model-00001-of-00004.safetensors",
450
+ "vision_backbone.featurizer.blocks.18.mlp.fc1.weight": "model-00001-of-00004.safetensors",
451
+ "vision_backbone.featurizer.blocks.18.mlp.fc2.bias": "model-00001-of-00004.safetensors",
452
+ "vision_backbone.featurizer.blocks.18.mlp.fc2.weight": "model-00001-of-00004.safetensors",
453
+ "vision_backbone.featurizer.blocks.18.norm1.bias": "model-00001-of-00004.safetensors",
454
+ "vision_backbone.featurizer.blocks.18.norm1.weight": "model-00001-of-00004.safetensors",
455
+ "vision_backbone.featurizer.blocks.18.norm2.bias": "model-00001-of-00004.safetensors",
456
+ "vision_backbone.featurizer.blocks.18.norm2.weight": "model-00001-of-00004.safetensors",
457
+ "vision_backbone.featurizer.blocks.19.attn.proj.bias": "model-00001-of-00004.safetensors",
458
+ "vision_backbone.featurizer.blocks.19.attn.proj.weight": "model-00001-of-00004.safetensors",
459
+ "vision_backbone.featurizer.blocks.19.attn.qkv.bias": "model-00001-of-00004.safetensors",
460
+ "vision_backbone.featurizer.blocks.19.attn.qkv.weight": "model-00001-of-00004.safetensors",
461
+ "vision_backbone.featurizer.blocks.19.ls1.scale_factor": "model-00001-of-00004.safetensors",
462
+ "vision_backbone.featurizer.blocks.19.ls2.scale_factor": "model-00001-of-00004.safetensors",
463
+ "vision_backbone.featurizer.blocks.19.mlp.fc1.bias": "model-00001-of-00004.safetensors",
464
+ "vision_backbone.featurizer.blocks.19.mlp.fc1.weight": "model-00001-of-00004.safetensors",
465
+ "vision_backbone.featurizer.blocks.19.mlp.fc2.bias": "model-00001-of-00004.safetensors",
466
+ "vision_backbone.featurizer.blocks.19.mlp.fc2.weight": "model-00001-of-00004.safetensors",
467
+ "vision_backbone.featurizer.blocks.19.norm1.bias": "model-00001-of-00004.safetensors",
468
+ "vision_backbone.featurizer.blocks.19.norm1.weight": "model-00001-of-00004.safetensors",
469
+ "vision_backbone.featurizer.blocks.19.norm2.bias": "model-00001-of-00004.safetensors",
470
+ "vision_backbone.featurizer.blocks.19.norm2.weight": "model-00001-of-00004.safetensors",
471
+ "vision_backbone.featurizer.blocks.2.attn.proj.bias": "model-00001-of-00004.safetensors",
472
+ "vision_backbone.featurizer.blocks.2.attn.proj.weight": "model-00001-of-00004.safetensors",
473
+ "vision_backbone.featurizer.blocks.2.attn.qkv.bias": "model-00001-of-00004.safetensors",
474
+ "vision_backbone.featurizer.blocks.2.attn.qkv.weight": "model-00001-of-00004.safetensors",
475
+ "vision_backbone.featurizer.blocks.2.ls1.scale_factor": "model-00001-of-00004.safetensors",
476
+ "vision_backbone.featurizer.blocks.2.ls2.scale_factor": "model-00001-of-00004.safetensors",
477
+ "vision_backbone.featurizer.blocks.2.mlp.fc1.bias": "model-00001-of-00004.safetensors",
478
+ "vision_backbone.featurizer.blocks.2.mlp.fc1.weight": "model-00001-of-00004.safetensors",
479
+ "vision_backbone.featurizer.blocks.2.mlp.fc2.bias": "model-00001-of-00004.safetensors",
480
+ "vision_backbone.featurizer.blocks.2.mlp.fc2.weight": "model-00001-of-00004.safetensors",
481
+ "vision_backbone.featurizer.blocks.2.norm1.bias": "model-00001-of-00004.safetensors",
482
+ "vision_backbone.featurizer.blocks.2.norm1.weight": "model-00001-of-00004.safetensors",
483
+ "vision_backbone.featurizer.blocks.2.norm2.bias": "model-00001-of-00004.safetensors",
484
+ "vision_backbone.featurizer.blocks.2.norm2.weight": "model-00001-of-00004.safetensors",
485
+ "vision_backbone.featurizer.blocks.20.attn.proj.bias": "model-00001-of-00004.safetensors",
486
+ "vision_backbone.featurizer.blocks.20.attn.proj.weight": "model-00001-of-00004.safetensors",
487
+ "vision_backbone.featurizer.blocks.20.attn.qkv.bias": "model-00001-of-00004.safetensors",
488
+ "vision_backbone.featurizer.blocks.20.attn.qkv.weight": "model-00001-of-00004.safetensors",
489
+ "vision_backbone.featurizer.blocks.20.ls1.scale_factor": "model-00001-of-00004.safetensors",
490
+ "vision_backbone.featurizer.blocks.20.ls2.scale_factor": "model-00001-of-00004.safetensors",
491
+ "vision_backbone.featurizer.blocks.20.mlp.fc1.bias": "model-00001-of-00004.safetensors",
492
+ "vision_backbone.featurizer.blocks.20.mlp.fc1.weight": "model-00001-of-00004.safetensors",
493
+ "vision_backbone.featurizer.blocks.20.mlp.fc2.bias": "model-00001-of-00004.safetensors",
494
+ "vision_backbone.featurizer.blocks.20.mlp.fc2.weight": "model-00001-of-00004.safetensors",
495
+ "vision_backbone.featurizer.blocks.20.norm1.bias": "model-00001-of-00004.safetensors",
496
+ "vision_backbone.featurizer.blocks.20.norm1.weight": "model-00001-of-00004.safetensors",
497
+ "vision_backbone.featurizer.blocks.20.norm2.bias": "model-00001-of-00004.safetensors",
498
+ "vision_backbone.featurizer.blocks.20.norm2.weight": "model-00001-of-00004.safetensors",
499
+ "vision_backbone.featurizer.blocks.21.attn.proj.bias": "model-00001-of-00004.safetensors",
500
+ "vision_backbone.featurizer.blocks.21.attn.proj.weight": "model-00001-of-00004.safetensors",
501
+ "vision_backbone.featurizer.blocks.21.attn.qkv.bias": "model-00001-of-00004.safetensors",
502
+ "vision_backbone.featurizer.blocks.21.attn.qkv.weight": "model-00001-of-00004.safetensors",
503
+ "vision_backbone.featurizer.blocks.21.ls1.scale_factor": "model-00001-of-00004.safetensors",
504
+ "vision_backbone.featurizer.blocks.21.ls2.scale_factor": "model-00001-of-00004.safetensors",
505
+ "vision_backbone.featurizer.blocks.21.mlp.fc1.bias": "model-00001-of-00004.safetensors",
506
+ "vision_backbone.featurizer.blocks.21.mlp.fc1.weight": "model-00001-of-00004.safetensors",
507
+ "vision_backbone.featurizer.blocks.21.mlp.fc2.bias": "model-00001-of-00004.safetensors",
508
+ "vision_backbone.featurizer.blocks.21.mlp.fc2.weight": "model-00001-of-00004.safetensors",
509
+ "vision_backbone.featurizer.blocks.21.norm1.bias": "model-00001-of-00004.safetensors",
510
+ "vision_backbone.featurizer.blocks.21.norm1.weight": "model-00001-of-00004.safetensors",
511
+ "vision_backbone.featurizer.blocks.21.norm2.bias": "model-00001-of-00004.safetensors",
512
+ "vision_backbone.featurizer.blocks.21.norm2.weight": "model-00001-of-00004.safetensors",
513
+ "vision_backbone.featurizer.blocks.22.attn.proj.bias": "model-00001-of-00004.safetensors",
514
+ "vision_backbone.featurizer.blocks.22.attn.proj.weight": "model-00001-of-00004.safetensors",
515
+ "vision_backbone.featurizer.blocks.22.attn.qkv.bias": "model-00001-of-00004.safetensors",
516
+ "vision_backbone.featurizer.blocks.22.attn.qkv.weight": "model-00001-of-00004.safetensors",
517
+ "vision_backbone.featurizer.blocks.22.ls1.scale_factor": "model-00001-of-00004.safetensors",
518
+ "vision_backbone.featurizer.blocks.22.ls2.scale_factor": "model-00001-of-00004.safetensors",
519
+ "vision_backbone.featurizer.blocks.22.mlp.fc1.bias": "model-00001-of-00004.safetensors",
520
+ "vision_backbone.featurizer.blocks.22.mlp.fc1.weight": "model-00001-of-00004.safetensors",
521
+ "vision_backbone.featurizer.blocks.22.mlp.fc2.bias": "model-00001-of-00004.safetensors",
522
+ "vision_backbone.featurizer.blocks.22.mlp.fc2.weight": "model-00001-of-00004.safetensors",
523
+ "vision_backbone.featurizer.blocks.22.norm1.bias": "model-00001-of-00004.safetensors",
524
+ "vision_backbone.featurizer.blocks.22.norm1.weight": "model-00001-of-00004.safetensors",
525
+ "vision_backbone.featurizer.blocks.22.norm2.bias": "model-00001-of-00004.safetensors",
526
+ "vision_backbone.featurizer.blocks.22.norm2.weight": "model-00001-of-00004.safetensors",
527
+ "vision_backbone.featurizer.blocks.23.attn.proj.bias": "model-00001-of-00004.safetensors",
528
+ "vision_backbone.featurizer.blocks.23.attn.proj.weight": "model-00001-of-00004.safetensors",
529
+ "vision_backbone.featurizer.blocks.23.attn.qkv.bias": "model-00001-of-00004.safetensors",
530
+ "vision_backbone.featurizer.blocks.23.attn.qkv.weight": "model-00001-of-00004.safetensors",
531
+ "vision_backbone.featurizer.blocks.23.ls1.scale_factor": "model-00001-of-00004.safetensors",
532
+ "vision_backbone.featurizer.blocks.23.ls2.scale_factor": "model-00001-of-00004.safetensors",
533
+ "vision_backbone.featurizer.blocks.23.mlp.fc1.bias": "model-00001-of-00004.safetensors",
534
+ "vision_backbone.featurizer.blocks.23.mlp.fc1.weight": "model-00001-of-00004.safetensors",
535
+ "vision_backbone.featurizer.blocks.23.mlp.fc2.bias": "model-00001-of-00004.safetensors",
536
+ "vision_backbone.featurizer.blocks.23.mlp.fc2.weight": "model-00001-of-00004.safetensors",
537
+ "vision_backbone.featurizer.blocks.23.norm1.bias": "model-00001-of-00004.safetensors",
538
+ "vision_backbone.featurizer.blocks.23.norm1.weight": "model-00001-of-00004.safetensors",
539
+ "vision_backbone.featurizer.blocks.23.norm2.bias": "model-00001-of-00004.safetensors",
540
+ "vision_backbone.featurizer.blocks.23.norm2.weight": "model-00001-of-00004.safetensors",
541
+ "vision_backbone.featurizer.blocks.3.attn.proj.bias": "model-00001-of-00004.safetensors",
542
+ "vision_backbone.featurizer.blocks.3.attn.proj.weight": "model-00001-of-00004.safetensors",
543
+ "vision_backbone.featurizer.blocks.3.attn.qkv.bias": "model-00001-of-00004.safetensors",
544
+ "vision_backbone.featurizer.blocks.3.attn.qkv.weight": "model-00001-of-00004.safetensors",
545
+ "vision_backbone.featurizer.blocks.3.ls1.scale_factor": "model-00001-of-00004.safetensors",
546
+ "vision_backbone.featurizer.blocks.3.ls2.scale_factor": "model-00001-of-00004.safetensors",
547
+ "vision_backbone.featurizer.blocks.3.mlp.fc1.bias": "model-00001-of-00004.safetensors",
548
+ "vision_backbone.featurizer.blocks.3.mlp.fc1.weight": "model-00001-of-00004.safetensors",
549
+ "vision_backbone.featurizer.blocks.3.mlp.fc2.bias": "model-00001-of-00004.safetensors",
550
+ "vision_backbone.featurizer.blocks.3.mlp.fc2.weight": "model-00001-of-00004.safetensors",
551
+ "vision_backbone.featurizer.blocks.3.norm1.bias": "model-00001-of-00004.safetensors",
552
+ "vision_backbone.featurizer.blocks.3.norm1.weight": "model-00001-of-00004.safetensors",
553
+ "vision_backbone.featurizer.blocks.3.norm2.bias": "model-00001-of-00004.safetensors",
554
+ "vision_backbone.featurizer.blocks.3.norm2.weight": "model-00001-of-00004.safetensors",
555
+ "vision_backbone.featurizer.blocks.4.attn.proj.bias": "model-00001-of-00004.safetensors",
556
+ "vision_backbone.featurizer.blocks.4.attn.proj.weight": "model-00001-of-00004.safetensors",
557
+ "vision_backbone.featurizer.blocks.4.attn.qkv.bias": "model-00001-of-00004.safetensors",
558
+ "vision_backbone.featurizer.blocks.4.attn.qkv.weight": "model-00001-of-00004.safetensors",
559
+ "vision_backbone.featurizer.blocks.4.ls1.scale_factor": "model-00001-of-00004.safetensors",
560
+ "vision_backbone.featurizer.blocks.4.ls2.scale_factor": "model-00001-of-00004.safetensors",
561
+ "vision_backbone.featurizer.blocks.4.mlp.fc1.bias": "model-00001-of-00004.safetensors",
562
+ "vision_backbone.featurizer.blocks.4.mlp.fc1.weight": "model-00001-of-00004.safetensors",
563
+ "vision_backbone.featurizer.blocks.4.mlp.fc2.bias": "model-00001-of-00004.safetensors",
564
+ "vision_backbone.featurizer.blocks.4.mlp.fc2.weight": "model-00001-of-00004.safetensors",
565
+ "vision_backbone.featurizer.blocks.4.norm1.bias": "model-00001-of-00004.safetensors",
566
+ "vision_backbone.featurizer.blocks.4.norm1.weight": "model-00001-of-00004.safetensors",
567
+ "vision_backbone.featurizer.blocks.4.norm2.bias": "model-00001-of-00004.safetensors",
568
+ "vision_backbone.featurizer.blocks.4.norm2.weight": "model-00001-of-00004.safetensors",
569
+ "vision_backbone.featurizer.blocks.5.attn.proj.bias": "model-00001-of-00004.safetensors",
570
+ "vision_backbone.featurizer.blocks.5.attn.proj.weight": "model-00001-of-00004.safetensors",
571
+ "vision_backbone.featurizer.blocks.5.attn.qkv.bias": "model-00001-of-00004.safetensors",
572
+ "vision_backbone.featurizer.blocks.5.attn.qkv.weight": "model-00001-of-00004.safetensors",
573
+ "vision_backbone.featurizer.blocks.5.ls1.scale_factor": "model-00001-of-00004.safetensors",
574
+ "vision_backbone.featurizer.blocks.5.ls2.scale_factor": "model-00001-of-00004.safetensors",
575
+ "vision_backbone.featurizer.blocks.5.mlp.fc1.bias": "model-00001-of-00004.safetensors",
576
+ "vision_backbone.featurizer.blocks.5.mlp.fc1.weight": "model-00001-of-00004.safetensors",
577
+ "vision_backbone.featurizer.blocks.5.mlp.fc2.bias": "model-00001-of-00004.safetensors",
578
+ "vision_backbone.featurizer.blocks.5.mlp.fc2.weight": "model-00001-of-00004.safetensors",
579
+ "vision_backbone.featurizer.blocks.5.norm1.bias": "model-00001-of-00004.safetensors",
580
+ "vision_backbone.featurizer.blocks.5.norm1.weight": "model-00001-of-00004.safetensors",
581
+ "vision_backbone.featurizer.blocks.5.norm2.bias": "model-00001-of-00004.safetensors",
582
+ "vision_backbone.featurizer.blocks.5.norm2.weight": "model-00001-of-00004.safetensors",
583
+ "vision_backbone.featurizer.blocks.6.attn.proj.bias": "model-00001-of-00004.safetensors",
584
+ "vision_backbone.featurizer.blocks.6.attn.proj.weight": "model-00001-of-00004.safetensors",
585
+ "vision_backbone.featurizer.blocks.6.attn.qkv.bias": "model-00001-of-00004.safetensors",
586
+ "vision_backbone.featurizer.blocks.6.attn.qkv.weight": "model-00001-of-00004.safetensors",
587
+ "vision_backbone.featurizer.blocks.6.ls1.scale_factor": "model-00001-of-00004.safetensors",
588
+ "vision_backbone.featurizer.blocks.6.ls2.scale_factor": "model-00001-of-00004.safetensors",
589
+ "vision_backbone.featurizer.blocks.6.mlp.fc1.bias": "model-00001-of-00004.safetensors",
590
+ "vision_backbone.featurizer.blocks.6.mlp.fc1.weight": "model-00001-of-00004.safetensors",
591
+ "vision_backbone.featurizer.blocks.6.mlp.fc2.bias": "model-00001-of-00004.safetensors",
592
+ "vision_backbone.featurizer.blocks.6.mlp.fc2.weight": "model-00001-of-00004.safetensors",
593
+ "vision_backbone.featurizer.blocks.6.norm1.bias": "model-00001-of-00004.safetensors",
594
+ "vision_backbone.featurizer.blocks.6.norm1.weight": "model-00001-of-00004.safetensors",
595
+ "vision_backbone.featurizer.blocks.6.norm2.bias": "model-00001-of-00004.safetensors",
596
+ "vision_backbone.featurizer.blocks.6.norm2.weight": "model-00001-of-00004.safetensors",
597
+ "vision_backbone.featurizer.blocks.7.attn.proj.bias": "model-00001-of-00004.safetensors",
598
+ "vision_backbone.featurizer.blocks.7.attn.proj.weight": "model-00001-of-00004.safetensors",
599
+ "vision_backbone.featurizer.blocks.7.attn.qkv.bias": "model-00001-of-00004.safetensors",
600
+ "vision_backbone.featurizer.blocks.7.attn.qkv.weight": "model-00001-of-00004.safetensors",
601
+ "vision_backbone.featurizer.blocks.7.ls1.scale_factor": "model-00001-of-00004.safetensors",
602
+ "vision_backbone.featurizer.blocks.7.ls2.scale_factor": "model-00001-of-00004.safetensors",
603
+ "vision_backbone.featurizer.blocks.7.mlp.fc1.bias": "model-00001-of-00004.safetensors",
604
+ "vision_backbone.featurizer.blocks.7.mlp.fc1.weight": "model-00001-of-00004.safetensors",
605
+ "vision_backbone.featurizer.blocks.7.mlp.fc2.bias": "model-00001-of-00004.safetensors",
606
+ "vision_backbone.featurizer.blocks.7.mlp.fc2.weight": "model-00001-of-00004.safetensors",
607
+ "vision_backbone.featurizer.blocks.7.norm1.bias": "model-00001-of-00004.safetensors",
608
+ "vision_backbone.featurizer.blocks.7.norm1.weight": "model-00001-of-00004.safetensors",
609
+ "vision_backbone.featurizer.blocks.7.norm2.bias": "model-00001-of-00004.safetensors",
610
+ "vision_backbone.featurizer.blocks.7.norm2.weight": "model-00001-of-00004.safetensors",
611
+ "vision_backbone.featurizer.blocks.8.attn.proj.bias": "model-00001-of-00004.safetensors",
612
+ "vision_backbone.featurizer.blocks.8.attn.proj.weight": "model-00001-of-00004.safetensors",
613
+ "vision_backbone.featurizer.blocks.8.attn.qkv.bias": "model-00001-of-00004.safetensors",
614
+ "vision_backbone.featurizer.blocks.8.attn.qkv.weight": "model-00001-of-00004.safetensors",
615
+ "vision_backbone.featurizer.blocks.8.ls1.scale_factor": "model-00001-of-00004.safetensors",
616
+ "vision_backbone.featurizer.blocks.8.ls2.scale_factor": "model-00001-of-00004.safetensors",
617
+ "vision_backbone.featurizer.blocks.8.mlp.fc1.bias": "model-00001-of-00004.safetensors",
618
+ "vision_backbone.featurizer.blocks.8.mlp.fc1.weight": "model-00001-of-00004.safetensors",
619
+ "vision_backbone.featurizer.blocks.8.mlp.fc2.bias": "model-00001-of-00004.safetensors",
620
+ "vision_backbone.featurizer.blocks.8.mlp.fc2.weight": "model-00001-of-00004.safetensors",
621
+ "vision_backbone.featurizer.blocks.8.norm1.bias": "model-00001-of-00004.safetensors",
622
+ "vision_backbone.featurizer.blocks.8.norm1.weight": "model-00001-of-00004.safetensors",
623
+ "vision_backbone.featurizer.blocks.8.norm2.bias": "model-00001-of-00004.safetensors",
624
+ "vision_backbone.featurizer.blocks.8.norm2.weight": "model-00001-of-00004.safetensors",
625
+ "vision_backbone.featurizer.blocks.9.attn.proj.bias": "model-00001-of-00004.safetensors",
626
+ "vision_backbone.featurizer.blocks.9.attn.proj.weight": "model-00001-of-00004.safetensors",
627
+ "vision_backbone.featurizer.blocks.9.attn.qkv.bias": "model-00001-of-00004.safetensors",
628
+ "vision_backbone.featurizer.blocks.9.attn.qkv.weight": "model-00001-of-00004.safetensors",
629
+ "vision_backbone.featurizer.blocks.9.ls1.scale_factor": "model-00001-of-00004.safetensors",
630
+ "vision_backbone.featurizer.blocks.9.ls2.scale_factor": "model-00001-of-00004.safetensors",
631
+ "vision_backbone.featurizer.blocks.9.mlp.fc1.bias": "model-00001-of-00004.safetensors",
632
+ "vision_backbone.featurizer.blocks.9.mlp.fc1.weight": "model-00001-of-00004.safetensors",
633
+ "vision_backbone.featurizer.blocks.9.mlp.fc2.bias": "model-00001-of-00004.safetensors",
634
+ "vision_backbone.featurizer.blocks.9.mlp.fc2.weight": "model-00001-of-00004.safetensors",
635
+ "vision_backbone.featurizer.blocks.9.norm1.bias": "model-00001-of-00004.safetensors",
636
+ "vision_backbone.featurizer.blocks.9.norm1.weight": "model-00001-of-00004.safetensors",
637
+ "vision_backbone.featurizer.blocks.9.norm2.bias": "model-00001-of-00004.safetensors",
638
+ "vision_backbone.featurizer.blocks.9.norm2.weight": "model-00001-of-00004.safetensors",
639
+ "vision_backbone.featurizer.cls_token": "model-00001-of-00004.safetensors",
640
+ "vision_backbone.featurizer.norm.bias": "model-00001-of-00004.safetensors",
641
+ "vision_backbone.featurizer.norm.weight": "model-00001-of-00004.safetensors",
642
+ "vision_backbone.featurizer.patch_embed.proj.bias": "model-00001-of-00004.safetensors",
643
+ "vision_backbone.featurizer.patch_embed.proj.weight": "model-00001-of-00004.safetensors",
644
+ "vision_backbone.featurizer.pos_embed": "model-00001-of-00004.safetensors",
645
+ "vision_backbone.featurizer.reg_token": "model-00001-of-00004.safetensors",
646
+ "vision_backbone.fused_featurizer.attn_pool.kv.bias": "model-00001-of-00004.safetensors",
647
+ "vision_backbone.fused_featurizer.attn_pool.kv.weight": "model-00001-of-00004.safetensors",
648
+ "vision_backbone.fused_featurizer.attn_pool.latent": "model-00001-of-00004.safetensors",
649
+ "vision_backbone.fused_featurizer.attn_pool.mlp.fc1.bias": "model-00001-of-00004.safetensors",
650
+ "vision_backbone.fused_featurizer.attn_pool.mlp.fc1.weight": "model-00001-of-00004.safetensors",
651
+ "vision_backbone.fused_featurizer.attn_pool.mlp.fc2.bias": "model-00001-of-00004.safetensors",
652
+ "vision_backbone.fused_featurizer.attn_pool.mlp.fc2.weight": "model-00001-of-00004.safetensors",
653
+ "vision_backbone.fused_featurizer.attn_pool.norm.bias": "model-00001-of-00004.safetensors",
654
+ "vision_backbone.fused_featurizer.attn_pool.norm.weight": "model-00001-of-00004.safetensors",
655
+ "vision_backbone.fused_featurizer.attn_pool.proj.bias": "model-00001-of-00004.safetensors",
656
+ "vision_backbone.fused_featurizer.attn_pool.proj.weight": "model-00001-of-00004.safetensors",
657
+ "vision_backbone.fused_featurizer.attn_pool.q.bias": "model-00001-of-00004.safetensors",
658
+ "vision_backbone.fused_featurizer.attn_pool.q.weight": "model-00001-of-00004.safetensors",
659
+ "vision_backbone.fused_featurizer.blocks.0.attn.proj.bias": "model-00001-of-00004.safetensors",
660
+ "vision_backbone.fused_featurizer.blocks.0.attn.proj.weight": "model-00001-of-00004.safetensors",
661
+ "vision_backbone.fused_featurizer.blocks.0.attn.qkv.bias": "model-00001-of-00004.safetensors",
662
+ "vision_backbone.fused_featurizer.blocks.0.attn.qkv.weight": "model-00001-of-00004.safetensors",
663
+ "vision_backbone.fused_featurizer.blocks.0.mlp.fc1.bias": "model-00001-of-00004.safetensors",
664
+ "vision_backbone.fused_featurizer.blocks.0.mlp.fc1.weight": "model-00001-of-00004.safetensors",
665
+ "vision_backbone.fused_featurizer.blocks.0.mlp.fc2.bias": "model-00001-of-00004.safetensors",
666
+ "vision_backbone.fused_featurizer.blocks.0.mlp.fc2.weight": "model-00001-of-00004.safetensors",
667
+ "vision_backbone.fused_featurizer.blocks.0.norm1.bias": "model-00001-of-00004.safetensors",
668
+ "vision_backbone.fused_featurizer.blocks.0.norm1.weight": "model-00001-of-00004.safetensors",
669
+ "vision_backbone.fused_featurizer.blocks.0.norm2.bias": "model-00001-of-00004.safetensors",
670
+ "vision_backbone.fused_featurizer.blocks.0.norm2.weight": "model-00001-of-00004.safetensors",
671
+ "vision_backbone.fused_featurizer.blocks.1.attn.proj.bias": "model-00001-of-00004.safetensors",
672
+ "vision_backbone.fused_featurizer.blocks.1.attn.proj.weight": "model-00001-of-00004.safetensors",
673
+ "vision_backbone.fused_featurizer.blocks.1.attn.qkv.bias": "model-00001-of-00004.safetensors",
674
+ "vision_backbone.fused_featurizer.blocks.1.attn.qkv.weight": "model-00001-of-00004.safetensors",
675
+ "vision_backbone.fused_featurizer.blocks.1.mlp.fc1.bias": "model-00001-of-00004.safetensors",
676
+ "vision_backbone.fused_featurizer.blocks.1.mlp.fc1.weight": "model-00001-of-00004.safetensors",
677
+ "vision_backbone.fused_featurizer.blocks.1.mlp.fc2.bias": "model-00001-of-00004.safetensors",
678
+ "vision_backbone.fused_featurizer.blocks.1.mlp.fc2.weight": "model-00001-of-00004.safetensors",
679
+ "vision_backbone.fused_featurizer.blocks.1.norm1.bias": "model-00001-of-00004.safetensors",
680
+ "vision_backbone.fused_featurizer.blocks.1.norm1.weight": "model-00001-of-00004.safetensors",
681
+ "vision_backbone.fused_featurizer.blocks.1.norm2.bias": "model-00001-of-00004.safetensors",
682
+ "vision_backbone.fused_featurizer.blocks.1.norm2.weight": "model-00001-of-00004.safetensors",
683
+ "vision_backbone.fused_featurizer.blocks.10.attn.proj.bias": "model-00001-of-00004.safetensors",
684
+ "vision_backbone.fused_featurizer.blocks.10.attn.proj.weight": "model-00001-of-00004.safetensors",
685
+ "vision_backbone.fused_featurizer.blocks.10.attn.qkv.bias": "model-00001-of-00004.safetensors",
686
+ "vision_backbone.fused_featurizer.blocks.10.attn.qkv.weight": "model-00001-of-00004.safetensors",
687
+ "vision_backbone.fused_featurizer.blocks.10.mlp.fc1.bias": "model-00001-of-00004.safetensors",
688
+ "vision_backbone.fused_featurizer.blocks.10.mlp.fc1.weight": "model-00001-of-00004.safetensors",
689
+ "vision_backbone.fused_featurizer.blocks.10.mlp.fc2.bias": "model-00001-of-00004.safetensors",
690
+ "vision_backbone.fused_featurizer.blocks.10.mlp.fc2.weight": "model-00001-of-00004.safetensors",
691
+ "vision_backbone.fused_featurizer.blocks.10.norm1.bias": "model-00001-of-00004.safetensors",
692
+ "vision_backbone.fused_featurizer.blocks.10.norm1.weight": "model-00001-of-00004.safetensors",
693
+ "vision_backbone.fused_featurizer.blocks.10.norm2.bias": "model-00001-of-00004.safetensors",
694
+ "vision_backbone.fused_featurizer.blocks.10.norm2.weight": "model-00001-of-00004.safetensors",
695
+ "vision_backbone.fused_featurizer.blocks.11.attn.proj.bias": "model-00001-of-00004.safetensors",
696
+ "vision_backbone.fused_featurizer.blocks.11.attn.proj.weight": "model-00001-of-00004.safetensors",
697
+ "vision_backbone.fused_featurizer.blocks.11.attn.qkv.bias": "model-00001-of-00004.safetensors",
698
+ "vision_backbone.fused_featurizer.blocks.11.attn.qkv.weight": "model-00001-of-00004.safetensors",
699
+ "vision_backbone.fused_featurizer.blocks.11.mlp.fc1.bias": "model-00001-of-00004.safetensors",
700
+ "vision_backbone.fused_featurizer.blocks.11.mlp.fc1.weight": "model-00001-of-00004.safetensors",
701
+ "vision_backbone.fused_featurizer.blocks.11.mlp.fc2.bias": "model-00001-of-00004.safetensors",
702
+ "vision_backbone.fused_featurizer.blocks.11.mlp.fc2.weight": "model-00001-of-00004.safetensors",
703
+ "vision_backbone.fused_featurizer.blocks.11.norm1.bias": "model-00001-of-00004.safetensors",
704
+ "vision_backbone.fused_featurizer.blocks.11.norm1.weight": "model-00001-of-00004.safetensors",
705
+ "vision_backbone.fused_featurizer.blocks.11.norm2.bias": "model-00001-of-00004.safetensors",
706
+ "vision_backbone.fused_featurizer.blocks.11.norm2.weight": "model-00001-of-00004.safetensors",
707
+ "vision_backbone.fused_featurizer.blocks.12.attn.proj.bias": "model-00001-of-00004.safetensors",
708
+ "vision_backbone.fused_featurizer.blocks.12.attn.proj.weight": "model-00001-of-00004.safetensors",
709
+ "vision_backbone.fused_featurizer.blocks.12.attn.qkv.bias": "model-00001-of-00004.safetensors",
710
+ "vision_backbone.fused_featurizer.blocks.12.attn.qkv.weight": "model-00001-of-00004.safetensors",
711
+ "vision_backbone.fused_featurizer.blocks.12.mlp.fc1.bias": "model-00001-of-00004.safetensors",
712
+ "vision_backbone.fused_featurizer.blocks.12.mlp.fc1.weight": "model-00001-of-00004.safetensors",
713
+ "vision_backbone.fused_featurizer.blocks.12.mlp.fc2.bias": "model-00001-of-00004.safetensors",
714
+ "vision_backbone.fused_featurizer.blocks.12.mlp.fc2.weight": "model-00001-of-00004.safetensors",
715
+ "vision_backbone.fused_featurizer.blocks.12.norm1.bias": "model-00001-of-00004.safetensors",
716
+ "vision_backbone.fused_featurizer.blocks.12.norm1.weight": "model-00001-of-00004.safetensors",
717
+ "vision_backbone.fused_featurizer.blocks.12.norm2.bias": "model-00001-of-00004.safetensors",
718
+ "vision_backbone.fused_featurizer.blocks.12.norm2.weight": "model-00001-of-00004.safetensors",
719
+ "vision_backbone.fused_featurizer.blocks.13.attn.proj.bias": "model-00001-of-00004.safetensors",
720
+ "vision_backbone.fused_featurizer.blocks.13.attn.proj.weight": "model-00001-of-00004.safetensors",
721
+ "vision_backbone.fused_featurizer.blocks.13.attn.qkv.bias": "model-00001-of-00004.safetensors",
722
+ "vision_backbone.fused_featurizer.blocks.13.attn.qkv.weight": "model-00001-of-00004.safetensors",
723
+ "vision_backbone.fused_featurizer.blocks.13.mlp.fc1.bias": "model-00001-of-00004.safetensors",
724
+ "vision_backbone.fused_featurizer.blocks.13.mlp.fc1.weight": "model-00001-of-00004.safetensors",
725
+ "vision_backbone.fused_featurizer.blocks.13.mlp.fc2.bias": "model-00001-of-00004.safetensors",
726
+ "vision_backbone.fused_featurizer.blocks.13.mlp.fc2.weight": "model-00001-of-00004.safetensors",
727
+ "vision_backbone.fused_featurizer.blocks.13.norm1.bias": "model-00001-of-00004.safetensors",
728
+ "vision_backbone.fused_featurizer.blocks.13.norm1.weight": "model-00001-of-00004.safetensors",
729
+ "vision_backbone.fused_featurizer.blocks.13.norm2.bias": "model-00001-of-00004.safetensors",
730
+ "vision_backbone.fused_featurizer.blocks.13.norm2.weight": "model-00001-of-00004.safetensors",
731
+ "vision_backbone.fused_featurizer.blocks.14.attn.proj.bias": "model-00001-of-00004.safetensors",
732
+ "vision_backbone.fused_featurizer.blocks.14.attn.proj.weight": "model-00001-of-00004.safetensors",
733
+ "vision_backbone.fused_featurizer.blocks.14.attn.qkv.bias": "model-00001-of-00004.safetensors",
734
+ "vision_backbone.fused_featurizer.blocks.14.attn.qkv.weight": "model-00001-of-00004.safetensors",
735
+ "vision_backbone.fused_featurizer.blocks.14.mlp.fc1.bias": "model-00001-of-00004.safetensors",
736
+ "vision_backbone.fused_featurizer.blocks.14.mlp.fc1.weight": "model-00001-of-00004.safetensors",
737
+ "vision_backbone.fused_featurizer.blocks.14.mlp.fc2.bias": "model-00001-of-00004.safetensors",
738
+ "vision_backbone.fused_featurizer.blocks.14.mlp.fc2.weight": "model-00001-of-00004.safetensors",
739
+ "vision_backbone.fused_featurizer.blocks.14.norm1.bias": "model-00001-of-00004.safetensors",
740
+ "vision_backbone.fused_featurizer.blocks.14.norm1.weight": "model-00001-of-00004.safetensors",
741
+ "vision_backbone.fused_featurizer.blocks.14.norm2.bias": "model-00001-of-00004.safetensors",
742
+ "vision_backbone.fused_featurizer.blocks.14.norm2.weight": "model-00001-of-00004.safetensors",
743
+ "vision_backbone.fused_featurizer.blocks.15.attn.proj.bias": "model-00001-of-00004.safetensors",
744
+ "vision_backbone.fused_featurizer.blocks.15.attn.proj.weight": "model-00001-of-00004.safetensors",
745
+ "vision_backbone.fused_featurizer.blocks.15.attn.qkv.bias": "model-00001-of-00004.safetensors",
746
+ "vision_backbone.fused_featurizer.blocks.15.attn.qkv.weight": "model-00001-of-00004.safetensors",
747
+ "vision_backbone.fused_featurizer.blocks.15.mlp.fc1.bias": "model-00001-of-00004.safetensors",
748
+ "vision_backbone.fused_featurizer.blocks.15.mlp.fc1.weight": "model-00001-of-00004.safetensors",
749
+ "vision_backbone.fused_featurizer.blocks.15.mlp.fc2.bias": "model-00001-of-00004.safetensors",
750
+ "vision_backbone.fused_featurizer.blocks.15.mlp.fc2.weight": "model-00001-of-00004.safetensors",
751
+ "vision_backbone.fused_featurizer.blocks.15.norm1.bias": "model-00001-of-00004.safetensors",
752
+ "vision_backbone.fused_featurizer.blocks.15.norm1.weight": "model-00001-of-00004.safetensors",
753
+ "vision_backbone.fused_featurizer.blocks.15.norm2.bias": "model-00001-of-00004.safetensors",
754
+ "vision_backbone.fused_featurizer.blocks.15.norm2.weight": "model-00001-of-00004.safetensors",
755
+ "vision_backbone.fused_featurizer.blocks.16.attn.proj.bias": "model-00001-of-00004.safetensors",
756
+ "vision_backbone.fused_featurizer.blocks.16.attn.proj.weight": "model-00001-of-00004.safetensors",
757
+ "vision_backbone.fused_featurizer.blocks.16.attn.qkv.bias": "model-00001-of-00004.safetensors",
758
+ "vision_backbone.fused_featurizer.blocks.16.attn.qkv.weight": "model-00001-of-00004.safetensors",
759
+ "vision_backbone.fused_featurizer.blocks.16.mlp.fc1.bias": "model-00001-of-00004.safetensors",
760
+ "vision_backbone.fused_featurizer.blocks.16.mlp.fc1.weight": "model-00001-of-00004.safetensors",
761
+ "vision_backbone.fused_featurizer.blocks.16.mlp.fc2.bias": "model-00001-of-00004.safetensors",
762
+ "vision_backbone.fused_featurizer.blocks.16.mlp.fc2.weight": "model-00001-of-00004.safetensors",
763
+ "vision_backbone.fused_featurizer.blocks.16.norm1.bias": "model-00001-of-00004.safetensors",
764
+ "vision_backbone.fused_featurizer.blocks.16.norm1.weight": "model-00001-of-00004.safetensors",
765
+ "vision_backbone.fused_featurizer.blocks.16.norm2.bias": "model-00001-of-00004.safetensors",
766
+ "vision_backbone.fused_featurizer.blocks.16.norm2.weight": "model-00001-of-00004.safetensors",
767
+ "vision_backbone.fused_featurizer.blocks.17.attn.proj.bias": "model-00001-of-00004.safetensors",
768
+ "vision_backbone.fused_featurizer.blocks.17.attn.proj.weight": "model-00001-of-00004.safetensors",
769
+ "vision_backbone.fused_featurizer.blocks.17.attn.qkv.bias": "model-00001-of-00004.safetensors",
770
+ "vision_backbone.fused_featurizer.blocks.17.attn.qkv.weight": "model-00001-of-00004.safetensors",
771
+ "vision_backbone.fused_featurizer.blocks.17.mlp.fc1.bias": "model-00001-of-00004.safetensors",
772
+ "vision_backbone.fused_featurizer.blocks.17.mlp.fc1.weight": "model-00001-of-00004.safetensors",
773
+ "vision_backbone.fused_featurizer.blocks.17.mlp.fc2.bias": "model-00001-of-00004.safetensors",
774
+ "vision_backbone.fused_featurizer.blocks.17.mlp.fc2.weight": "model-00001-of-00004.safetensors",
775
+ "vision_backbone.fused_featurizer.blocks.17.norm1.bias": "model-00001-of-00004.safetensors",
776
+ "vision_backbone.fused_featurizer.blocks.17.norm1.weight": "model-00001-of-00004.safetensors",
777
+ "vision_backbone.fused_featurizer.blocks.17.norm2.bias": "model-00001-of-00004.safetensors",
778
+ "vision_backbone.fused_featurizer.blocks.17.norm2.weight": "model-00001-of-00004.safetensors",
779
+ "vision_backbone.fused_featurizer.blocks.18.attn.proj.bias": "model-00001-of-00004.safetensors",
780
+ "vision_backbone.fused_featurizer.blocks.18.attn.proj.weight": "model-00001-of-00004.safetensors",
781
+ "vision_backbone.fused_featurizer.blocks.18.attn.qkv.bias": "model-00001-of-00004.safetensors",
782
+ "vision_backbone.fused_featurizer.blocks.18.attn.qkv.weight": "model-00001-of-00004.safetensors",
783
+ "vision_backbone.fused_featurizer.blocks.18.mlp.fc1.bias": "model-00001-of-00004.safetensors",
784
+ "vision_backbone.fused_featurizer.blocks.18.mlp.fc1.weight": "model-00001-of-00004.safetensors",
785
+ "vision_backbone.fused_featurizer.blocks.18.mlp.fc2.bias": "model-00001-of-00004.safetensors",
786
+ "vision_backbone.fused_featurizer.blocks.18.mlp.fc2.weight": "model-00001-of-00004.safetensors",
787
+ "vision_backbone.fused_featurizer.blocks.18.norm1.bias": "model-00001-of-00004.safetensors",
788
+ "vision_backbone.fused_featurizer.blocks.18.norm1.weight": "model-00001-of-00004.safetensors",
789
+ "vision_backbone.fused_featurizer.blocks.18.norm2.bias": "model-00001-of-00004.safetensors",
790
+ "vision_backbone.fused_featurizer.blocks.18.norm2.weight": "model-00001-of-00004.safetensors",
791
+ "vision_backbone.fused_featurizer.blocks.19.attn.proj.bias": "model-00001-of-00004.safetensors",
792
+ "vision_backbone.fused_featurizer.blocks.19.attn.proj.weight": "model-00001-of-00004.safetensors",
793
+ "vision_backbone.fused_featurizer.blocks.19.attn.qkv.bias": "model-00001-of-00004.safetensors",
794
+ "vision_backbone.fused_featurizer.blocks.19.attn.qkv.weight": "model-00001-of-00004.safetensors",
795
+ "vision_backbone.fused_featurizer.blocks.19.mlp.fc1.bias": "model-00001-of-00004.safetensors",
796
+ "vision_backbone.fused_featurizer.blocks.19.mlp.fc1.weight": "model-00001-of-00004.safetensors",
797
+ "vision_backbone.fused_featurizer.blocks.19.mlp.fc2.bias": "model-00001-of-00004.safetensors",
798
+ "vision_backbone.fused_featurizer.blocks.19.mlp.fc2.weight": "model-00001-of-00004.safetensors",
799
+ "vision_backbone.fused_featurizer.blocks.19.norm1.bias": "model-00001-of-00004.safetensors",
800
+ "vision_backbone.fused_featurizer.blocks.19.norm1.weight": "model-00001-of-00004.safetensors",
801
+ "vision_backbone.fused_featurizer.blocks.19.norm2.bias": "model-00001-of-00004.safetensors",
802
+ "vision_backbone.fused_featurizer.blocks.19.norm2.weight": "model-00001-of-00004.safetensors",
803
+ "vision_backbone.fused_featurizer.blocks.2.attn.proj.bias": "model-00001-of-00004.safetensors",
804
+ "vision_backbone.fused_featurizer.blocks.2.attn.proj.weight": "model-00001-of-00004.safetensors",
805
+ "vision_backbone.fused_featurizer.blocks.2.attn.qkv.bias": "model-00001-of-00004.safetensors",
806
+ "vision_backbone.fused_featurizer.blocks.2.attn.qkv.weight": "model-00001-of-00004.safetensors",
807
+ "vision_backbone.fused_featurizer.blocks.2.mlp.fc1.bias": "model-00001-of-00004.safetensors",
808
+ "vision_backbone.fused_featurizer.blocks.2.mlp.fc1.weight": "model-00001-of-00004.safetensors",
809
+ "vision_backbone.fused_featurizer.blocks.2.mlp.fc2.bias": "model-00001-of-00004.safetensors",
810
+ "vision_backbone.fused_featurizer.blocks.2.mlp.fc2.weight": "model-00001-of-00004.safetensors",
811
+ "vision_backbone.fused_featurizer.blocks.2.norm1.bias": "model-00001-of-00004.safetensors",
812
+ "vision_backbone.fused_featurizer.blocks.2.norm1.weight": "model-00001-of-00004.safetensors",
813
+ "vision_backbone.fused_featurizer.blocks.2.norm2.bias": "model-00001-of-00004.safetensors",
814
+ "vision_backbone.fused_featurizer.blocks.2.norm2.weight": "model-00001-of-00004.safetensors",
815
+ "vision_backbone.fused_featurizer.blocks.20.attn.proj.bias": "model-00001-of-00004.safetensors",
816
+ "vision_backbone.fused_featurizer.blocks.20.attn.proj.weight": "model-00001-of-00004.safetensors",
817
+ "vision_backbone.fused_featurizer.blocks.20.attn.qkv.bias": "model-00001-of-00004.safetensors",
818
+ "vision_backbone.fused_featurizer.blocks.20.attn.qkv.weight": "model-00001-of-00004.safetensors",
819
+ "vision_backbone.fused_featurizer.blocks.20.mlp.fc1.bias": "model-00001-of-00004.safetensors",
820
+ "vision_backbone.fused_featurizer.blocks.20.mlp.fc1.weight": "model-00001-of-00004.safetensors",
821
+ "vision_backbone.fused_featurizer.blocks.20.mlp.fc2.bias": "model-00001-of-00004.safetensors",
822
+ "vision_backbone.fused_featurizer.blocks.20.mlp.fc2.weight": "model-00001-of-00004.safetensors",
823
+ "vision_backbone.fused_featurizer.blocks.20.norm1.bias": "model-00001-of-00004.safetensors",
824
+ "vision_backbone.fused_featurizer.blocks.20.norm1.weight": "model-00001-of-00004.safetensors",
825
+ "vision_backbone.fused_featurizer.blocks.20.norm2.bias": "model-00001-of-00004.safetensors",
826
+ "vision_backbone.fused_featurizer.blocks.20.norm2.weight": "model-00001-of-00004.safetensors",
827
+ "vision_backbone.fused_featurizer.blocks.21.attn.proj.bias": "model-00001-of-00004.safetensors",
828
+ "vision_backbone.fused_featurizer.blocks.21.attn.proj.weight": "model-00001-of-00004.safetensors",
829
+ "vision_backbone.fused_featurizer.blocks.21.attn.qkv.bias": "model-00001-of-00004.safetensors",
830
+ "vision_backbone.fused_featurizer.blocks.21.attn.qkv.weight": "model-00001-of-00004.safetensors",
831
+ "vision_backbone.fused_featurizer.blocks.21.mlp.fc1.bias": "model-00001-of-00004.safetensors",
832
+ "vision_backbone.fused_featurizer.blocks.21.mlp.fc1.weight": "model-00001-of-00004.safetensors",
833
+ "vision_backbone.fused_featurizer.blocks.21.mlp.fc2.bias": "model-00001-of-00004.safetensors",
834
+ "vision_backbone.fused_featurizer.blocks.21.mlp.fc2.weight": "model-00001-of-00004.safetensors",
835
+ "vision_backbone.fused_featurizer.blocks.21.norm1.bias": "model-00001-of-00004.safetensors",
836
+ "vision_backbone.fused_featurizer.blocks.21.norm1.weight": "model-00001-of-00004.safetensors",
837
+ "vision_backbone.fused_featurizer.blocks.21.norm2.bias": "model-00001-of-00004.safetensors",
838
+ "vision_backbone.fused_featurizer.blocks.21.norm2.weight": "model-00001-of-00004.safetensors",
839
+ "vision_backbone.fused_featurizer.blocks.22.attn.proj.bias": "model-00001-of-00004.safetensors",
840
+ "vision_backbone.fused_featurizer.blocks.22.attn.proj.weight": "model-00001-of-00004.safetensors",
841
+ "vision_backbone.fused_featurizer.blocks.22.attn.qkv.bias": "model-00001-of-00004.safetensors",
842
+ "vision_backbone.fused_featurizer.blocks.22.attn.qkv.weight": "model-00001-of-00004.safetensors",
843
+ "vision_backbone.fused_featurizer.blocks.22.mlp.fc1.bias": "model-00001-of-00004.safetensors",
844
+ "vision_backbone.fused_featurizer.blocks.22.mlp.fc1.weight": "model-00001-of-00004.safetensors",
845
+ "vision_backbone.fused_featurizer.blocks.22.mlp.fc2.bias": "model-00001-of-00004.safetensors",
846
+ "vision_backbone.fused_featurizer.blocks.22.mlp.fc2.weight": "model-00001-of-00004.safetensors",
847
+ "vision_backbone.fused_featurizer.blocks.22.norm1.bias": "model-00001-of-00004.safetensors",
848
+ "vision_backbone.fused_featurizer.blocks.22.norm1.weight": "model-00001-of-00004.safetensors",
849
+ "vision_backbone.fused_featurizer.blocks.22.norm2.bias": "model-00001-of-00004.safetensors",
850
+ "vision_backbone.fused_featurizer.blocks.22.norm2.weight": "model-00001-of-00004.safetensors",
851
+ "vision_backbone.fused_featurizer.blocks.23.attn.proj.bias": "model-00001-of-00004.safetensors",
852
+ "vision_backbone.fused_featurizer.blocks.23.attn.proj.weight": "model-00001-of-00004.safetensors",
853
+ "vision_backbone.fused_featurizer.blocks.23.attn.qkv.bias": "model-00001-of-00004.safetensors",
854
+ "vision_backbone.fused_featurizer.blocks.23.attn.qkv.weight": "model-00001-of-00004.safetensors",
855
+ "vision_backbone.fused_featurizer.blocks.23.mlp.fc1.bias": "model-00001-of-00004.safetensors",
856
+ "vision_backbone.fused_featurizer.blocks.23.mlp.fc1.weight": "model-00001-of-00004.safetensors",
857
+ "vision_backbone.fused_featurizer.blocks.23.mlp.fc2.bias": "model-00001-of-00004.safetensors",
858
+ "vision_backbone.fused_featurizer.blocks.23.mlp.fc2.weight": "model-00001-of-00004.safetensors",
859
+ "vision_backbone.fused_featurizer.blocks.23.norm1.bias": "model-00001-of-00004.safetensors",
860
+ "vision_backbone.fused_featurizer.blocks.23.norm1.weight": "model-00001-of-00004.safetensors",
861
+ "vision_backbone.fused_featurizer.blocks.23.norm2.bias": "model-00001-of-00004.safetensors",
862
+ "vision_backbone.fused_featurizer.blocks.23.norm2.weight": "model-00001-of-00004.safetensors",
863
+ "vision_backbone.fused_featurizer.blocks.24.attn.proj.bias": "model-00001-of-00004.safetensors",
864
+ "vision_backbone.fused_featurizer.blocks.24.attn.proj.weight": "model-00001-of-00004.safetensors",
865
+ "vision_backbone.fused_featurizer.blocks.24.attn.qkv.bias": "model-00001-of-00004.safetensors",
866
+ "vision_backbone.fused_featurizer.blocks.24.attn.qkv.weight": "model-00001-of-00004.safetensors",
867
+ "vision_backbone.fused_featurizer.blocks.24.mlp.fc1.bias": "model-00001-of-00004.safetensors",
868
+ "vision_backbone.fused_featurizer.blocks.24.mlp.fc1.weight": "model-00001-of-00004.safetensors",
869
+ "vision_backbone.fused_featurizer.blocks.24.mlp.fc2.bias": "model-00001-of-00004.safetensors",
870
+ "vision_backbone.fused_featurizer.blocks.24.mlp.fc2.weight": "model-00001-of-00004.safetensors",
871
+ "vision_backbone.fused_featurizer.blocks.24.norm1.bias": "model-00001-of-00004.safetensors",
872
+ "vision_backbone.fused_featurizer.blocks.24.norm1.weight": "model-00001-of-00004.safetensors",
873
+ "vision_backbone.fused_featurizer.blocks.24.norm2.bias": "model-00001-of-00004.safetensors",
874
+ "vision_backbone.fused_featurizer.blocks.24.norm2.weight": "model-00001-of-00004.safetensors",
875
+ "vision_backbone.fused_featurizer.blocks.25.attn.proj.bias": "model-00001-of-00004.safetensors",
876
+ "vision_backbone.fused_featurizer.blocks.25.attn.proj.weight": "model-00001-of-00004.safetensors",
877
+ "vision_backbone.fused_featurizer.blocks.25.attn.qkv.bias": "model-00001-of-00004.safetensors",
878
+ "vision_backbone.fused_featurizer.blocks.25.attn.qkv.weight": "model-00001-of-00004.safetensors",
879
+ "vision_backbone.fused_featurizer.blocks.25.mlp.fc1.bias": "model-00001-of-00004.safetensors",
880
+ "vision_backbone.fused_featurizer.blocks.25.mlp.fc1.weight": "model-00001-of-00004.safetensors",
881
+ "vision_backbone.fused_featurizer.blocks.25.mlp.fc2.bias": "model-00001-of-00004.safetensors",
882
+ "vision_backbone.fused_featurizer.blocks.25.mlp.fc2.weight": "model-00001-of-00004.safetensors",
883
+ "vision_backbone.fused_featurizer.blocks.25.norm1.bias": "model-00001-of-00004.safetensors",
884
+ "vision_backbone.fused_featurizer.blocks.25.norm1.weight": "model-00001-of-00004.safetensors",
885
+ "vision_backbone.fused_featurizer.blocks.25.norm2.bias": "model-00001-of-00004.safetensors",
886
+ "vision_backbone.fused_featurizer.blocks.25.norm2.weight": "model-00001-of-00004.safetensors",
887
+ "vision_backbone.fused_featurizer.blocks.26.attn.proj.bias": "model-00001-of-00004.safetensors",
888
+ "vision_backbone.fused_featurizer.blocks.26.attn.proj.weight": "model-00001-of-00004.safetensors",
889
+ "vision_backbone.fused_featurizer.blocks.26.attn.qkv.bias": "model-00001-of-00004.safetensors",
890
+ "vision_backbone.fused_featurizer.blocks.26.attn.qkv.weight": "model-00001-of-00004.safetensors",
891
+ "vision_backbone.fused_featurizer.blocks.26.mlp.fc1.bias": "model-00001-of-00004.safetensors",
892
+ "vision_backbone.fused_featurizer.blocks.26.mlp.fc1.weight": "model-00001-of-00004.safetensors",
893
+ "vision_backbone.fused_featurizer.blocks.26.mlp.fc2.bias": "model-00001-of-00004.safetensors",
894
+ "vision_backbone.fused_featurizer.blocks.26.mlp.fc2.weight": "model-00001-of-00004.safetensors",
895
+ "vision_backbone.fused_featurizer.blocks.26.norm1.bias": "model-00001-of-00004.safetensors",
896
+ "vision_backbone.fused_featurizer.blocks.26.norm1.weight": "model-00001-of-00004.safetensors",
897
+ "vision_backbone.fused_featurizer.blocks.26.norm2.bias": "model-00001-of-00004.safetensors",
898
+ "vision_backbone.fused_featurizer.blocks.26.norm2.weight": "model-00001-of-00004.safetensors",
899
+ "vision_backbone.fused_featurizer.blocks.3.attn.proj.bias": "model-00001-of-00004.safetensors",
900
+ "vision_backbone.fused_featurizer.blocks.3.attn.proj.weight": "model-00001-of-00004.safetensors",
901
+ "vision_backbone.fused_featurizer.blocks.3.attn.qkv.bias": "model-00001-of-00004.safetensors",
902
+ "vision_backbone.fused_featurizer.blocks.3.attn.qkv.weight": "model-00001-of-00004.safetensors",
903
+ "vision_backbone.fused_featurizer.blocks.3.mlp.fc1.bias": "model-00001-of-00004.safetensors",
904
+ "vision_backbone.fused_featurizer.blocks.3.mlp.fc1.weight": "model-00001-of-00004.safetensors",
905
+ "vision_backbone.fused_featurizer.blocks.3.mlp.fc2.bias": "model-00001-of-00004.safetensors",
906
+ "vision_backbone.fused_featurizer.blocks.3.mlp.fc2.weight": "model-00001-of-00004.safetensors",
907
+ "vision_backbone.fused_featurizer.blocks.3.norm1.bias": "model-00001-of-00004.safetensors",
908
+ "vision_backbone.fused_featurizer.blocks.3.norm1.weight": "model-00001-of-00004.safetensors",
909
+ "vision_backbone.fused_featurizer.blocks.3.norm2.bias": "model-00001-of-00004.safetensors",
910
+ "vision_backbone.fused_featurizer.blocks.3.norm2.weight": "model-00001-of-00004.safetensors",
911
+ "vision_backbone.fused_featurizer.blocks.4.attn.proj.bias": "model-00001-of-00004.safetensors",
912
+ "vision_backbone.fused_featurizer.blocks.4.attn.proj.weight": "model-00001-of-00004.safetensors",
913
+ "vision_backbone.fused_featurizer.blocks.4.attn.qkv.bias": "model-00001-of-00004.safetensors",
914
+ "vision_backbone.fused_featurizer.blocks.4.attn.qkv.weight": "model-00001-of-00004.safetensors",
915
+ "vision_backbone.fused_featurizer.blocks.4.mlp.fc1.bias": "model-00001-of-00004.safetensors",
916
+ "vision_backbone.fused_featurizer.blocks.4.mlp.fc1.weight": "model-00001-of-00004.safetensors",
917
+ "vision_backbone.fused_featurizer.blocks.4.mlp.fc2.bias": "model-00001-of-00004.safetensors",
918
+ "vision_backbone.fused_featurizer.blocks.4.mlp.fc2.weight": "model-00001-of-00004.safetensors",
919
+ "vision_backbone.fused_featurizer.blocks.4.norm1.bias": "model-00001-of-00004.safetensors",
920
+ "vision_backbone.fused_featurizer.blocks.4.norm1.weight": "model-00001-of-00004.safetensors",
921
+ "vision_backbone.fused_featurizer.blocks.4.norm2.bias": "model-00001-of-00004.safetensors",
922
+ "vision_backbone.fused_featurizer.blocks.4.norm2.weight": "model-00001-of-00004.safetensors",
923
+ "vision_backbone.fused_featurizer.blocks.5.attn.proj.bias": "model-00001-of-00004.safetensors",
924
+ "vision_backbone.fused_featurizer.blocks.5.attn.proj.weight": "model-00001-of-00004.safetensors",
925
+ "vision_backbone.fused_featurizer.blocks.5.attn.qkv.bias": "model-00001-of-00004.safetensors",
926
+ "vision_backbone.fused_featurizer.blocks.5.attn.qkv.weight": "model-00001-of-00004.safetensors",
927
+ "vision_backbone.fused_featurizer.blocks.5.mlp.fc1.bias": "model-00001-of-00004.safetensors",
928
+ "vision_backbone.fused_featurizer.blocks.5.mlp.fc1.weight": "model-00001-of-00004.safetensors",
929
+ "vision_backbone.fused_featurizer.blocks.5.mlp.fc2.bias": "model-00001-of-00004.safetensors",
930
+ "vision_backbone.fused_featurizer.blocks.5.mlp.fc2.weight": "model-00001-of-00004.safetensors",
931
+ "vision_backbone.fused_featurizer.blocks.5.norm1.bias": "model-00001-of-00004.safetensors",
932
+ "vision_backbone.fused_featurizer.blocks.5.norm1.weight": "model-00001-of-00004.safetensors",
933
+ "vision_backbone.fused_featurizer.blocks.5.norm2.bias": "model-00001-of-00004.safetensors",
934
+ "vision_backbone.fused_featurizer.blocks.5.norm2.weight": "model-00001-of-00004.safetensors",
935
+ "vision_backbone.fused_featurizer.blocks.6.attn.proj.bias": "model-00001-of-00004.safetensors",
936
+ "vision_backbone.fused_featurizer.blocks.6.attn.proj.weight": "model-00001-of-00004.safetensors",
937
+ "vision_backbone.fused_featurizer.blocks.6.attn.qkv.bias": "model-00001-of-00004.safetensors",
938
+ "vision_backbone.fused_featurizer.blocks.6.attn.qkv.weight": "model-00001-of-00004.safetensors",
939
+ "vision_backbone.fused_featurizer.blocks.6.mlp.fc1.bias": "model-00001-of-00004.safetensors",
940
+ "vision_backbone.fused_featurizer.blocks.6.mlp.fc1.weight": "model-00001-of-00004.safetensors",
941
+ "vision_backbone.fused_featurizer.blocks.6.mlp.fc2.bias": "model-00001-of-00004.safetensors",
942
+ "vision_backbone.fused_featurizer.blocks.6.mlp.fc2.weight": "model-00001-of-00004.safetensors",
943
+ "vision_backbone.fused_featurizer.blocks.6.norm1.bias": "model-00001-of-00004.safetensors",
944
+ "vision_backbone.fused_featurizer.blocks.6.norm1.weight": "model-00001-of-00004.safetensors",
945
+ "vision_backbone.fused_featurizer.blocks.6.norm2.bias": "model-00001-of-00004.safetensors",
946
+ "vision_backbone.fused_featurizer.blocks.6.norm2.weight": "model-00001-of-00004.safetensors",
947
+ "vision_backbone.fused_featurizer.blocks.7.attn.proj.bias": "model-00001-of-00004.safetensors",
948
+ "vision_backbone.fused_featurizer.blocks.7.attn.proj.weight": "model-00001-of-00004.safetensors",
949
+ "vision_backbone.fused_featurizer.blocks.7.attn.qkv.bias": "model-00001-of-00004.safetensors",
950
+ "vision_backbone.fused_featurizer.blocks.7.attn.qkv.weight": "model-00001-of-00004.safetensors",
951
+ "vision_backbone.fused_featurizer.blocks.7.mlp.fc1.bias": "model-00001-of-00004.safetensors",
952
+ "vision_backbone.fused_featurizer.blocks.7.mlp.fc1.weight": "model-00001-of-00004.safetensors",
953
+ "vision_backbone.fused_featurizer.blocks.7.mlp.fc2.bias": "model-00001-of-00004.safetensors",
954
+ "vision_backbone.fused_featurizer.blocks.7.mlp.fc2.weight": "model-00001-of-00004.safetensors",
955
+ "vision_backbone.fused_featurizer.blocks.7.norm1.bias": "model-00001-of-00004.safetensors",
956
+ "vision_backbone.fused_featurizer.blocks.7.norm1.weight": "model-00001-of-00004.safetensors",
957
+ "vision_backbone.fused_featurizer.blocks.7.norm2.bias": "model-00001-of-00004.safetensors",
958
+ "vision_backbone.fused_featurizer.blocks.7.norm2.weight": "model-00001-of-00004.safetensors",
959
+ "vision_backbone.fused_featurizer.blocks.8.attn.proj.bias": "model-00001-of-00004.safetensors",
960
+ "vision_backbone.fused_featurizer.blocks.8.attn.proj.weight": "model-00001-of-00004.safetensors",
961
+ "vision_backbone.fused_featurizer.blocks.8.attn.qkv.bias": "model-00001-of-00004.safetensors",
962
+ "vision_backbone.fused_featurizer.blocks.8.attn.qkv.weight": "model-00001-of-00004.safetensors",
963
+ "vision_backbone.fused_featurizer.blocks.8.mlp.fc1.bias": "model-00001-of-00004.safetensors",
964
+ "vision_backbone.fused_featurizer.blocks.8.mlp.fc1.weight": "model-00001-of-00004.safetensors",
965
+ "vision_backbone.fused_featurizer.blocks.8.mlp.fc2.bias": "model-00001-of-00004.safetensors",
966
+ "vision_backbone.fused_featurizer.blocks.8.mlp.fc2.weight": "model-00001-of-00004.safetensors",
967
+ "vision_backbone.fused_featurizer.blocks.8.norm1.bias": "model-00001-of-00004.safetensors",
968
+ "vision_backbone.fused_featurizer.blocks.8.norm1.weight": "model-00001-of-00004.safetensors",
969
+ "vision_backbone.fused_featurizer.blocks.8.norm2.bias": "model-00001-of-00004.safetensors",
970
+ "vision_backbone.fused_featurizer.blocks.8.norm2.weight": "model-00001-of-00004.safetensors",
971
+ "vision_backbone.fused_featurizer.blocks.9.attn.proj.bias": "model-00001-of-00004.safetensors",
972
+ "vision_backbone.fused_featurizer.blocks.9.attn.proj.weight": "model-00001-of-00004.safetensors",
973
+ "vision_backbone.fused_featurizer.blocks.9.attn.qkv.bias": "model-00001-of-00004.safetensors",
974
+ "vision_backbone.fused_featurizer.blocks.9.attn.qkv.weight": "model-00001-of-00004.safetensors",
975
+ "vision_backbone.fused_featurizer.blocks.9.mlp.fc1.bias": "model-00001-of-00004.safetensors",
976
+ "vision_backbone.fused_featurizer.blocks.9.mlp.fc1.weight": "model-00001-of-00004.safetensors",
977
+ "vision_backbone.fused_featurizer.blocks.9.mlp.fc2.bias": "model-00001-of-00004.safetensors",
978
+ "vision_backbone.fused_featurizer.blocks.9.mlp.fc2.weight": "model-00001-of-00004.safetensors",
979
+ "vision_backbone.fused_featurizer.blocks.9.norm1.bias": "model-00001-of-00004.safetensors",
980
+ "vision_backbone.fused_featurizer.blocks.9.norm1.weight": "model-00001-of-00004.safetensors",
981
+ "vision_backbone.fused_featurizer.blocks.9.norm2.bias": "model-00001-of-00004.safetensors",
982
+ "vision_backbone.fused_featurizer.blocks.9.norm2.weight": "model-00001-of-00004.safetensors",
983
+ "vision_backbone.fused_featurizer.norm.bias": "model-00001-of-00004.safetensors",
984
+ "vision_backbone.fused_featurizer.norm.weight": "model-00001-of-00004.safetensors",
985
+ "vision_backbone.fused_featurizer.patch_embed.proj.bias": "model-00001-of-00004.safetensors",
986
+ "vision_backbone.fused_featurizer.patch_embed.proj.weight": "model-00001-of-00004.safetensors",
987
+ "vision_backbone.fused_featurizer.pos_embed": "model-00001-of-00004.safetensors"
988
+ }
989
+ }
piper_pick_can_10k/modeling_prismatic.py ADDED
@@ -0,0 +1,562 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """
2
+ modeling_prismatic.py
3
+
4
+ Core HuggingFace-style PrismaticPreTrainedModel and PrismaticForConditionalGeneration class definitions, inheriting
5
+ from the default `transformers.PretrainedModel`. Meant to be standalone and self-contained, but exactly replicate the
6
+ logic in `prismatic.models.vlms.prismatic.py`.
7
+
8
+ Note =>> for the time being, not adding the custom HF "docstring" formatting.
9
+
10
+ References [LLaVa, IDEFICS-2]:
11
+ => https://github.com/huggingface/transformers/blob/main/src/transformers/models/llava/modeling_llava.py
12
+ => https://github.com/huggingface/transformers/blob/main/src/transformers/models/idefics2/modeling_idefics2.py
13
+ """
14
+
15
+ import logging
16
+ from dataclasses import dataclass
17
+ from functools import partial
18
+ from typing import Any, Callable, ClassVar, Dict, List, Optional, Tuple, Union
19
+
20
+ import numpy as np
21
+ import timm
22
+ import tokenizers
23
+ import torch
24
+ import torch.nn as nn
25
+ import transformers
26
+ from timm.models.vision_transformer import LayerScale
27
+ from transformers import AutoModelForCausalLM, PretrainedConfig, PreTrainedModel
28
+ from transformers.modeling_outputs import ModelOutput
29
+
30
+ from .configuration_prismatic import OpenVLAConfig, PrismaticConfig
31
+
32
+ # Get Logger
33
+ logger = logging.getLogger(__name__)
34
+
35
+
36
+ # === PyTorch/HuggingFace Default IGNORE_INDEX (for CrossEntropyLoss labels)
37
+ IGNORE_INDEX = -100
38
+
39
+
40
+ # === Utility Functions for Monkey-Patching ===
41
+ def unpack_tuple(fn: Callable[[Any], Tuple[Any]]) -> Callable[[Any], Any]:
42
+ def wrapper(*args: Any, **kwargs: Any) -> Any:
43
+ result = fn(*args, **kwargs)
44
+ return result[0] if isinstance(result, tuple) else result
45
+
46
+ return wrapper
47
+
48
+
49
+ # HF Transformers overwrites parameters with names containing `gamma`; we're going to patch VisionBackbone.LayerScale.
50
+ # =>> TIMM :: https://github.com/huggingface/pytorch-image-models/blob/main/timm/models/vision_transformer.py#L109
51
+ # =>> Transformers :: https://github.com/huggingface/transformers/blob/main/src/transformers/modeling_utils.py#L3960
52
+ def _ls_new_forward(self, x: torch.Tensor) -> torch.Tensor:
53
+ return x.mul_(self.scale_factor) if self.inplace else x * self.scale_factor
54
+
55
+
56
+ def ls_apply_patch(ls_module: LayerScale):
57
+ ls_module.scale_factor = nn.Parameter(ls_module.gamma.clone())
58
+ ls_module.forward = _ls_new_forward.__get__(ls_module, LayerScale)
59
+ del ls_module.gamma
60
+
61
+
62
+ # === Prismatic Vision Backbone (nn.Module) Definitions (w/ Fused Backbone Support) ===
63
+ class PrismaticVisionBackbone(nn.Module):
64
+ def __init__(
65
+ self,
66
+ use_fused_vision_backbone: bool,
67
+ image_sizes: List[int],
68
+ timm_model_ids: List[str],
69
+ timm_override_act_layers: List[Optional[str]],
70
+ ) -> None:
71
+ super().__init__()
72
+ self.use_fused_vision_backbone = use_fused_vision_backbone
73
+
74
+ # [Contract] Validate number of (fused) vision backbones, create "alpha" featurizer and Instantiate
75
+ # =>> Note :: Monkey-Patch the `forward()` function of the backbone to ensure FSDP-compatibility
76
+ # Hardcodes `get_intermediate_layers` to return the **SECOND-TO-LAST** layer patches!
77
+ assert len(timm_model_ids) <= 2, "Prismatic models only support up to 2 (fused) vision backbones!"
78
+ self.featurizer = timm.create_model(
79
+ timm_model_ids[0],
80
+ pretrained=False,
81
+ num_classes=0,
82
+ img_size=image_sizes[0],
83
+ act_layer=timm_override_act_layers[0],
84
+ )
85
+ self.featurizer.forward = unpack_tuple(
86
+ partial(self.featurizer.get_intermediate_layers, n={len(self.featurizer.blocks) - 2})
87
+ )
88
+ self.embed_dim = self.featurizer.embed_dim
89
+
90
+ # If `use_fused_vision_backbone` =>> create "beta" featurizer
91
+ if self.use_fused_vision_backbone:
92
+ self.fused_featurizer = timm.create_model(
93
+ timm_model_ids[1],
94
+ pretrained=False,
95
+ num_classes=0,
96
+ img_size=image_sizes[1],
97
+ act_layer=timm_override_act_layers[1],
98
+ )
99
+ self.fused_featurizer.forward = unpack_tuple(
100
+ partial(self.fused_featurizer.get_intermediate_layers, n={len(self.fused_featurizer.blocks) - 2})
101
+ )
102
+ self.embed_dim += self.fused_featurizer.embed_dim
103
+
104
+ # Patch `vision_backbone.featurizer` and `vision_backbone.fused_featurizer` with HF-Compatible LayerScale
105
+ for module in self.featurizer.modules():
106
+ if isinstance(module, LayerScale):
107
+ ls_apply_patch(module)
108
+
109
+ if self.use_fused_vision_backbone:
110
+ for module in self.fused_featurizer.modules():
111
+ if isinstance(module, LayerScale):
112
+ ls_apply_patch(module)
113
+
114
+ def forward(self, pixel_values: torch.Tensor) -> torch.Tensor:
115
+ """Run image (`pixel_values`) through featurizer; if channel-stacked, then dispatch and sequence stack."""
116
+ if not self.use_fused_vision_backbone:
117
+ return self.featurizer(pixel_values)
118
+
119
+ # Split `pixel_values :: [bsz, 2 * 3, resolution, resolution]` =>> featurize =>> channel stack
120
+ img, img_fused = torch.split(pixel_values, [3, 3], dim=1)
121
+ patches, patches_fused = self.featurizer(img), self.fused_featurizer(img_fused)
122
+
123
+ return torch.cat([patches, patches_fused], dim=2)
124
+
125
+
126
+ # === Prismatic Projector (nn.Module) Definitions ===
127
+ class PrismaticProjector(nn.Module):
128
+ def __init__(self, use_fused_vision_backbone: bool, vision_dim: int, llm_dim: int) -> None:
129
+ super().__init__()
130
+ self.use_fused_vision_backbone = use_fused_vision_backbone
131
+ self.vision_dim, self.llm_dim = vision_dim, llm_dim
132
+
133
+ # Switch on `use_fused_vision_backbone` =>> use slightly different MLPs and projection factors!
134
+ if not self.use_fused_vision_backbone:
135
+ self.fc1 = nn.Linear(self.vision_dim, self.llm_dim, bias=True)
136
+ self.fc2 = nn.Linear(self.llm_dim, self.llm_dim, bias=True)
137
+ self.act_fn1 = nn.GELU()
138
+ else:
139
+ initial_projection_dim = 4 * vision_dim
140
+ self.fc1 = nn.Linear(self.vision_dim, initial_projection_dim, bias=True)
141
+ self.fc2 = nn.Linear(initial_projection_dim, self.llm_dim, bias=True)
142
+ self.fc3 = nn.Linear(self.llm_dim, self.llm_dim, bias=True)
143
+ self.act_fn1 = nn.GELU()
144
+ self.act_fn2 = nn.GELU()
145
+
146
+ def forward(self, img_patches: torch.Tensor) -> torch.Tensor:
147
+ if not self.use_fused_vision_backbone:
148
+ projected_features = self.fc1(img_patches)
149
+ projected_features = self.act_fn1(projected_features)
150
+ projected_features = self.fc2(projected_features)
151
+ else:
152
+ projected_features = self.fc1(img_patches)
153
+ projected_features = self.act_fn1(projected_features)
154
+ projected_features = self.fc2(projected_features)
155
+ projected_features = self.act_fn2(projected_features)
156
+ projected_features = self.fc3(projected_features)
157
+
158
+ return projected_features
159
+
160
+
161
+ # === Main HF Class Definitions ===
162
+ @dataclass
163
+ class PrismaticCausalLMOutputWithPast(ModelOutput):
164
+ """Base class for Prismatic casual (visually-conditioned) language model outputs; also exposes visual features."""
165
+
166
+ loss: Optional[torch.FloatTensor] = None
167
+ logits: torch.FloatTensor = None
168
+ past_key_values: Optional[Tuple[Tuple[torch.FloatTensor]]] = None
169
+ hidden_states: Optional[Tuple[torch.FloatTensor, ...]] = None
170
+ attentions: Optional[Tuple[torch.FloatTensor]] = None
171
+
172
+ # Additions for VLMs
173
+ projector_features: Optional[torch.FloatTensor] = None
174
+
175
+
176
+ class PrismaticPreTrainedModel(PreTrainedModel):
177
+ config_class: PretrainedConfig = PrismaticConfig
178
+ base_model_prefix: str = "model"
179
+ supports_gradient_checkpointing: bool = True
180
+
181
+ _no_split_modules: ClassVar[List[str]] = ["PrismaticProjector"]
182
+ _skip_keys_device_placement: str = "past_key_values"
183
+ _supports_flash_attn_2: bool = True
184
+
185
+ def _init_weights(self, module: nn.Module) -> None:
186
+ # Important :: this HF ported version is *not* meant for training from scratch; only inference and fine-tuning!
187
+ # => As such, this init_weights code is not correct; if training VLMs from scratch, use the main codebase at
188
+ # https://github.com/TRI-ML/prismatic-vlms
189
+ std = (
190
+ self.config.initializer_range
191
+ if hasattr(self.config, "initializer_range")
192
+ else self.config.text_config.initializer_range
193
+ )
194
+
195
+ if hasattr(module, "class_embedding"):
196
+ module.class_embedding.data.normal_(mean=0.0, std=std)
197
+
198
+ if isinstance(module, (nn.Linear, nn.Conv2d)):
199
+ module.weight.data.normal_(mean=0.0, std=std)
200
+ if module.bias is not None:
201
+ module.bias.data.zero_()
202
+ elif isinstance(module, nn.Embedding):
203
+ module.weight.data.normal_(mean=0.0, std=std)
204
+ if module.padding_idx is not None:
205
+ module.weight.data[module.padding_idx].zero_()
206
+
207
+ @property
208
+ def _supports_sdpa(self) -> bool:
209
+ """Check LLM supports SDPA Attention"""
210
+ return self.language_model._supports_sdpa
211
+
212
+
213
+ class PrismaticForConditionalGeneration(PrismaticPreTrainedModel):
214
+ def __init__(self, config: PrismaticConfig) -> None:
215
+ super().__init__(config)
216
+
217
+ # [Validation] Lightweight Validate on `config` Fields + Dependency Versions
218
+ if config.use_fused_vision_backbone is None:
219
+ raise ValueError("Missing config field `use_fused_vision_backbone`")
220
+
221
+ if timm.__version__ not in {"0.9.10", "0.9.11", "0.9.12", "0.9.16"}:
222
+ raise NotImplementedError(
223
+ "TIMM Version must be >= 0.9.10 and < 1.0.0 (breaking); please raise a GitHub Issue "
224
+ "if you urgently need support for latest TIMM versions."
225
+ )
226
+
227
+ if (transformers.__version__ != "4.40.1") or (tokenizers.__version__ != "0.19.1"):
228
+ logger.warning(
229
+ f"Expected `transformers==4.40.1` and `tokenizers==0.19.1` but got "
230
+ f"`transformers=={transformers.__version__}` and `tokenizers=={tokenizers.__version__}`; "
231
+ f"there might be inference-time regressions due to dependency changes. If in doubt, please"
232
+ f"use the above versions."
233
+ )
234
+
235
+ # Instantiate PrismaticVisionBackbone (w/ Potential Fused Backbone)
236
+ self.vision_backbone = PrismaticVisionBackbone(
237
+ config.use_fused_vision_backbone, config.image_sizes, config.timm_model_ids, config.timm_override_act_layers
238
+ )
239
+
240
+ # Create Multimodal Projector
241
+ self.projector = PrismaticProjector(
242
+ config.use_fused_vision_backbone,
243
+ vision_dim=self.vision_backbone.embed_dim,
244
+ llm_dim=config.text_config.hidden_size,
245
+ )
246
+
247
+ # Instantiate LLM Backbone
248
+ self.language_model = AutoModelForCausalLM.from_config(
249
+ config.text_config, attn_implementation=config._attn_implementation
250
+ )
251
+ self.vocab_size = config.text_config.vocab_size
252
+ self.pad_token_id = config.pad_token_id
253
+
254
+ # HF Boilerplate =>> initializes weights via `_init_weights()` and sets gradient checkpointing
255
+ self.post_init()
256
+
257
+ # === `PreTrainedModel` Boilerplate ===
258
+ def get_input_embeddings(self) -> nn.Module:
259
+ return self.language_model.get_input_embeddings()
260
+
261
+ def set_input_embeddings(self, value: nn.Module) -> None:
262
+ self.language_model.set_input_embeddings(value)
263
+
264
+ def get_output_embeddings(self) -> nn.Module:
265
+ return self.language_model.get_output_embeddings()
266
+
267
+ def set_output_embeddings(self, new_embeddings: nn.Module) -> None:
268
+ self.language_model.set_output_embeddings(new_embeddings)
269
+
270
+ def get_decoder(self) -> nn.Module:
271
+ return self.language_model.get_decoder()
272
+
273
+ def set_decoder(self, decoder: nn.Module) -> None:
274
+ self.language_model.set_decoder(decoder)
275
+
276
+ def tie_weights(self) -> None:
277
+ self.language_model.tie_weights() # Note: `Llama-2` and `Mistral` don't tie weights (no-op)
278
+
279
+ def resize_token_embeddings(
280
+ self, new_num_tokens: Optional[int] = None, pad_to_multiple_of: Optional[int] = None
281
+ ) -> nn.Embedding:
282
+ updated_embeddings = self.language_model.resize_token_embeddings(new_num_tokens, pad_to_multiple_of)
283
+
284
+ # Update config/instance variables
285
+ self.config.text_config.vocab_size = updated_embeddings.num_embeddings
286
+ self.vocab_size = updated_embeddings.num_embeddings
287
+
288
+ return updated_embeddings
289
+
290
+ # === Core Prismatic VLM `forward()` Logic ===
291
+ def forward(
292
+ self,
293
+ input_ids: Optional[torch.LongTensor] = None,
294
+ attention_mask: Optional[torch.Tensor] = None,
295
+ pixel_values: Optional[torch.FloatTensor] = None,
296
+ labels: Optional[torch.LongTensor] = None,
297
+ inputs_embeds: Optional[torch.FloatTensor] = None,
298
+ past_key_values: Optional[List[torch.FloatTensor]] = None,
299
+ use_cache: Optional[bool] = None,
300
+ output_attentions: Optional[bool] = None,
301
+ output_hidden_states: Optional[bool] = None,
302
+ output_projector_features: Optional[bool] = None,
303
+ return_dict: Optional[bool] = None,
304
+ ) -> Union[Tuple, PrismaticCausalLMOutputWithPast]:
305
+ """Run a forward pass through the VLM, returning a PrismaticCausalLMOutputWithPast instance."""
306
+ output_attentions = output_attentions if output_attentions is not None else self.config.output_attentions
307
+ output_hidden_states = (
308
+ output_hidden_states if output_hidden_states is not None else self.config.output_hidden_states
309
+ )
310
+ output_projector_features = output_projector_features if output_projector_features is not None else False
311
+ return_dict = return_dict if return_dict is not None else self.config.use_return_dict
312
+
313
+ # Respect `use_cache` only if not training (even if `gradient_checkpointing` is off)
314
+ use_cache = use_cache and not self.training
315
+
316
+ # Instantiate Placeholder for Projector Features
317
+ projected_patch_embeddings = None
318
+
319
+ # Note :: We only support forward passes with the following cases:
320
+ # => Cached Generation :: (input_ids.shape[1] == 1) and (past_key_values is not None)
321
+ # => Unimodal Forward :: (pixel_values is None)
322
+ # => Multimodal Forward :: (pixel_values is not None) and (input_ids/embeds.shape[0] == pixel_values.shape[0])
323
+
324
+ # === Handle Generation with Cache (`input_ids.shape[1] == 1`) =>> requires `past_keys_values` ===
325
+ if input_ids.shape[1] == 1:
326
+ assert input_ids.shape[0] == 1, "Generation is only currently supported for batch size of 1!"
327
+ assert past_key_values is not None, "You must provide `past_key_values` during cached generation!"
328
+ assert labels is None, "Unexpected key `labels` provided during cached generation!"
329
+
330
+ language_model_output = self.language_model(
331
+ input_ids=input_ids,
332
+ attention_mask=None,
333
+ position_ids=None,
334
+ past_key_values=past_key_values,
335
+ inputs_embeds=None,
336
+ labels=None,
337
+ use_cache=use_cache,
338
+ output_attentions=output_attentions,
339
+ output_hidden_states=output_hidden_states,
340
+ return_dict=return_dict,
341
+ )
342
+
343
+ # === Handle Unimodal Forward ===
344
+ elif pixel_values is None:
345
+ assert (input_ids is not None) and (inputs_embeds is None), "Missing `input_ids` in language-only forward!"
346
+ assert past_key_values is None, "Unexpected key `past_key_values` provided during language-only forward!"
347
+
348
+ language_model_output = self.language_model(
349
+ input_ids=input_ids,
350
+ attention_mask=attention_mask,
351
+ position_ids=None,
352
+ past_key_values=None,
353
+ inputs_embeds=None,
354
+ labels=labels,
355
+ use_cache=use_cache,
356
+ output_attentions=output_attentions,
357
+ output_hidden_states=output_hidden_states,
358
+ return_dict=return_dict,
359
+ )
360
+
361
+ # === Handle Multimodal Forward ===
362
+ elif (input_ids.shape[0] == pixel_values.shape[0]) or (inputs_embeds.shape[0] == pixel_values.shape[0]):
363
+ assert past_key_values is None, "Unexpected key `past_key_values` provided during language-only forward!"
364
+
365
+ # Visual Feature Extraction
366
+ patch_features = self.vision_backbone(pixel_values)
367
+
368
+ # Projection Logic =>> Update Attention Mask
369
+ projected_patch_embeddings = self.projector(patch_features)
370
+ projected_patch_attention_mask = None
371
+ if attention_mask is not None:
372
+ projected_patch_attention_mask = torch.full(
373
+ (projected_patch_embeddings.shape[0], projected_patch_embeddings.shape[1]),
374
+ fill_value=True,
375
+ dtype=attention_mask.dtype,
376
+ device=attention_mask.device,
377
+ )
378
+
379
+ # Get Input Embeddings (from Language Model Embeddings)
380
+ input_embeddings = self.get_input_embeddings()(input_ids)
381
+
382
+ # Build Multimodal Embeddings & Attention Mask =>> Prismatic defaults to inserting after <BOS> token (1:)
383
+ multimodal_embeddings = torch.cat(
384
+ [input_embeddings[:, :1, :], projected_patch_embeddings, input_embeddings[:, 1:, :]], dim=1
385
+ )
386
+ multimodal_attention_mask = None
387
+ if attention_mask is not None:
388
+ multimodal_attention_mask = torch.cat(
389
+ [attention_mask[:, :1], projected_patch_attention_mask, attention_mask[:, 1:]], dim=1
390
+ )
391
+
392
+ # Build Labels (if specified) =>> Ignore Labels for Patch Embeddings
393
+ multimodal_labels = None
394
+ if labels is not None:
395
+ projected_patch_labels = torch.full(
396
+ (projected_patch_embeddings.shape[0], projected_patch_embeddings.shape[1]),
397
+ fill_value=IGNORE_INDEX,
398
+ dtype=labels.dtype,
399
+ device=labels.device,
400
+ )
401
+ multimodal_labels = torch.cat([labels[:, :1], projected_patch_labels, labels[:, 1:]], dim=1)
402
+
403
+ # Dispatch to Language Model
404
+ language_model_output = self.language_model(
405
+ input_ids=None,
406
+ attention_mask=multimodal_attention_mask,
407
+ position_ids=None,
408
+ past_key_values=None,
409
+ inputs_embeds=multimodal_embeddings,
410
+ labels=multimodal_labels,
411
+ use_cache=use_cache,
412
+ output_attentions=output_attentions,
413
+ output_hidden_states=output_hidden_states,
414
+ return_dict=return_dict,
415
+ )
416
+
417
+ # === Otherwise =>> Assume Invalid! ===
418
+ elif (input_ids.shape[0] != pixel_values.shape[0]) or (inputs_embeds.shape[0] != pixel_values.shape[0]):
419
+ raise ValueError("Non-homogenous batch of (text, image) input -- forward() does not support mixed batches!")
420
+
421
+ else:
422
+ raise ValueError(
423
+ "Invalid PrismaticForConditionalGeneration `forward()` call with provided arguments:\n"
424
+ f"=> `input_ids` = {input_ids is not None}\n"
425
+ f"=> `attention_mask` = {attention_mask is not None}\n"
426
+ f"=> `pixel_values` = {pixel_values is not None}\n"
427
+ f"=> `labels` = {labels is not None}\n"
428
+ f"=> `input_embeds` = {inputs_embeds is not None}\n"
429
+ f"=> `past_key_values` = {past_key_values is not None}\n"
430
+ f"=> `use_cache` = {use_cache}"
431
+ )
432
+
433
+ # Unpack `language_model_output` and return PrismaticCausalLMOutputWithPast (or tuple if not `return_dict`)
434
+ if not return_dict:
435
+ if output_projector_features and (projected_patch_embeddings is not None):
436
+ return *language_model_output, projected_patch_embeddings
437
+
438
+ return language_model_output
439
+
440
+ return PrismaticCausalLMOutputWithPast(
441
+ loss=language_model_output.loss,
442
+ logits=language_model_output.logits,
443
+ past_key_values=language_model_output.past_key_values,
444
+ hidden_states=language_model_output.hidden_states,
445
+ attentions=language_model_output.attentions,
446
+ projector_features=projected_patch_embeddings,
447
+ )
448
+
449
+ # === GenerationMixin Methods ===
450
+ def prepare_inputs_for_generation(
451
+ self,
452
+ input_ids: Optional[torch.Tensor] = None,
453
+ past_key_values: Optional[List[torch.FloatTensor]] = None,
454
+ inputs_embeds: Optional[torch.FloatTensor] = None,
455
+ pixel_values: Optional[torch.FloatTensor] = None,
456
+ attention_mask: Optional[torch.Tensor] = None,
457
+ **kwargs: str,
458
+ ) -> Dict[str, torch.Tensor]:
459
+ """Borrowed from `LlamaForCausalLM` and simplified for batch size = 1; mirrors original PrismaticVLM logic."""
460
+ if ((input_ids is not None) and (input_ids.shape[0] > 1)) or (
461
+ (inputs_embeds is not None) and (inputs_embeds.shape[0] > 1)
462
+ ):
463
+ raise ValueError("Generation with batch size > 1 is not currently supported!")
464
+
465
+ # Handle `past_key_values` (cache) =>> assume `input_ids` just has unprocessed tokens
466
+ if past_key_values is not None:
467
+ input_ids = input_ids[:, -1:]
468
+
469
+ # If `input_embeds` are passed, we only want to use them in the 1st generation step
470
+ if inputs_embeds is not None and past_key_values is None:
471
+ model_inputs = {"input_embeds": inputs_embeds}
472
+ else:
473
+ model_inputs = {"input_ids": input_ids}
474
+
475
+ # Make sure `pixel_values` are preserved in `model_inputs`
476
+ model_inputs.update(
477
+ {
478
+ "attention_mask": attention_mask,
479
+ "pixel_values": pixel_values,
480
+ "past_key_values": past_key_values,
481
+ "use_cache": kwargs.get("use_cache"),
482
+ }
483
+ )
484
+
485
+ return model_inputs
486
+
487
+ # Defer to Language Model (all handle this differently, with different return types)
488
+ def _reorder_cache(self, *args, **kwargs) -> Any:
489
+ return self.language_model._reorder_cache(*args, **kwargs)
490
+
491
+
492
+ class OpenVLAForActionPrediction(PrismaticForConditionalGeneration):
493
+ config_class: PretrainedConfig = OpenVLAConfig
494
+
495
+ def __init__(self, config: OpenVLAConfig) -> None:
496
+ super().__init__(config)
497
+ self.norm_stats = config.norm_stats
498
+
499
+ # Compute action bins
500
+ self.bins = np.linspace(-1, 1, config.n_action_bins)
501
+ self.bin_centers = (self.bins[:-1] + self.bins[1:]) / 2.0
502
+
503
+ # Compute vocab size for de-tokenization -- revert added "multiple of"
504
+ self.vocab_size = self.config.text_config.vocab_size - self.config.pad_to_multiple_of
505
+
506
+ def predict_action(
507
+ self, input_ids: Optional[torch.LongTensor] = None, unnorm_key: Optional[str] = None, **kwargs: str
508
+ ) -> np.ndarray:
509
+ """Thin wrapper around .generate() that decodes predicted actions and unnormalizes them."""
510
+ # If the special empty token ('') does not already appear after the colon (':') token in the prompt
511
+ # (after "OUT:" or "ASSISTANT:"), insert it to match the inputs seen at training time
512
+ if not torch.all(input_ids[:, -1] == 29871):
513
+ input_ids = torch.cat(
514
+ (input_ids, torch.unsqueeze(torch.Tensor([29871]).long(), dim=0).to(input_ids.device)), dim=1
515
+ )
516
+
517
+ # Run VLA inference
518
+ generated_ids = self.generate(input_ids, max_new_tokens=self.get_action_dim(unnorm_key), **kwargs)
519
+
520
+ # Extract predicted action tokens and translate into (normalized) continuous actions
521
+ predicted_action_token_ids = generated_ids[0, -self.get_action_dim(unnorm_key) :].cpu().numpy()
522
+ discretized_actions = self.vocab_size - predicted_action_token_ids
523
+ discretized_actions = np.clip(discretized_actions - 1, a_min=0, a_max=self.bin_centers.shape[0] - 1)
524
+ normalized_actions = self.bin_centers[discretized_actions]
525
+
526
+ # Unnormalize actions
527
+ action_norm_stats = self.get_action_stats(unnorm_key)
528
+ mask = action_norm_stats.get("mask", np.ones_like(action_norm_stats["q01"], dtype=bool))
529
+ action_high, action_low = np.array(action_norm_stats["q99"]), np.array(action_norm_stats["q01"])
530
+ actions = np.where(
531
+ mask,
532
+ 0.5 * (normalized_actions + 1) * (action_high - action_low) + action_low,
533
+ normalized_actions,
534
+ )
535
+
536
+ return actions
537
+
538
+ @staticmethod
539
+ def _check_unnorm_key(norm_stats: Dict[str, Dict[str, Any]], unnorm_key: Optional[str]) -> str:
540
+ if unnorm_key is None:
541
+ assert len(norm_stats) == 1, (
542
+ f"Your model was trained on more than one dataset, "
543
+ f"please pass a `unnorm_key` from the following options to choose the statistics "
544
+ f"used for un-normalizing actions: {norm_stats.keys()}"
545
+ )
546
+ unnorm_key = next(iter(norm_stats.keys()))
547
+
548
+ assert unnorm_key in norm_stats, (
549
+ f"The `unnorm_key` you chose is not in the set of available dataset statistics, "
550
+ f"please choose from: {norm_stats.keys()}"
551
+ )
552
+ return unnorm_key
553
+
554
+ def get_action_dim(self, unnorm_key: Optional[str] = None) -> int:
555
+ """Get the dimensionality of the policy's action space."""
556
+ unnorm_key = self._check_unnorm_key(self.norm_stats, unnorm_key)
557
+ return len(self.norm_stats[unnorm_key]["action"]["q01"])
558
+
559
+ def get_action_stats(self, unnorm_key: Optional[str] = None) -> Dict[str, Any]:
560
+ """Get all the logged statistics for the given dataset."""
561
+ unnorm_key = self._check_unnorm_key(self.norm_stats, unnorm_key)
562
+ return self.norm_stats[unnorm_key]["action"]
piper_pick_can_10k/preprocessor_config.json ADDED
@@ -0,0 +1,114 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "auto_map": {
3
+ "AutoImageProcessor": "processing_prismatic.PrismaticImageProcessor",
4
+ "AutoProcessor": "processing_prismatic.PrismaticProcessor"
5
+ },
6
+ "image_processor_type": "PrismaticImageProcessor",
7
+ "image_resize_strategy": "resize-naive",
8
+ "input_sizes": [
9
+ [
10
+ 3,
11
+ 224,
12
+ 224
13
+ ],
14
+ [
15
+ 3,
16
+ 224,
17
+ 224
18
+ ]
19
+ ],
20
+ "interpolations": [
21
+ "bicubic",
22
+ "bicubic"
23
+ ],
24
+ "means": [
25
+ [
26
+ 0.485,
27
+ 0.456,
28
+ 0.406
29
+ ],
30
+ [
31
+ 0.5,
32
+ 0.5,
33
+ 0.5
34
+ ]
35
+ ],
36
+ "processor_class": "PrismaticProcessor",
37
+ "stds": [
38
+ [
39
+ 0.229,
40
+ 0.224,
41
+ 0.225
42
+ ],
43
+ [
44
+ 0.5,
45
+ 0.5,
46
+ 0.5
47
+ ]
48
+ ],
49
+ "tvf_crop_params": [
50
+ {
51
+ "output_size": [
52
+ 224,
53
+ 224
54
+ ]
55
+ },
56
+ {
57
+ "output_size": [
58
+ 224,
59
+ 224
60
+ ]
61
+ }
62
+ ],
63
+ "tvf_do_letterbox": false,
64
+ "tvf_letterbox_fill": null,
65
+ "tvf_normalize_params": [
66
+ {
67
+ "inplace": false,
68
+ "mean": [
69
+ 0.484375,
70
+ 0.455078125,
71
+ 0.40625
72
+ ],
73
+ "std": [
74
+ 0.228515625,
75
+ 0.2236328125,
76
+ 0.224609375
77
+ ]
78
+ },
79
+ {
80
+ "inplace": false,
81
+ "mean": [
82
+ 0.5,
83
+ 0.5,
84
+ 0.5
85
+ ],
86
+ "std": [
87
+ 0.5,
88
+ 0.5,
89
+ 0.5
90
+ ]
91
+ }
92
+ ],
93
+ "tvf_resize_params": [
94
+ {
95
+ "antialias": true,
96
+ "interpolation": 3,
97
+ "max_size": null,
98
+ "size": [
99
+ 224,
100
+ 224
101
+ ]
102
+ },
103
+ {
104
+ "antialias": true,
105
+ "interpolation": 3,
106
+ "max_size": null,
107
+ "size": [
108
+ 224,
109
+ 224
110
+ ]
111
+ }
112
+ ],
113
+ "use_fused_vision_backbone": true
114
+ }
piper_pick_can_10k/processing_prismatic.py ADDED
@@ -0,0 +1,257 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """
2
+ processing_prismatic.py
3
+
4
+ HuggingFace-style preprocessor definitions for Prismatic VLMs, inheriting from `ProcessorMixin`. Default configuration
5
+ specifies `siglip-224px+7b`.
6
+ """
7
+
8
+ from typing import Any, ClassVar, List, Optional, Tuple, Union
9
+
10
+ import timm.data
11
+ import torch
12
+ import torchvision.transforms.functional as TVF
13
+ from PIL import Image
14
+ from torchvision.transforms import CenterCrop, Compose, Normalize, Resize, ToTensor
15
+ from transformers import PreTrainedTokenizerBase
16
+ from transformers.image_processing_utils import BatchFeature, ImageProcessingMixin
17
+ from transformers.processing_utils import ProcessorMixin
18
+ from transformers.tokenization_utils import PaddingStrategy, PreTokenizedInput, TextInput, TruncationStrategy
19
+ from transformers.utils import TensorType
20
+
21
+
22
+ # === Image Processing ===
23
+ def letterbox_pad_transform(image: Image.Image, padding_fill_value: Tuple[int, int, int]) -> Image.Image:
24
+ """Given a PIL.Image, pad to square by adding a symmetric border around the height/width."""
25
+ (w, h), max_wh = image.size, max(image.size)
26
+ horizontal_pad, vertical_pad = int((max_wh - w) / 2), int((max_wh - h) / 2)
27
+ padding = (horizontal_pad, vertical_pad, horizontal_pad, vertical_pad)
28
+
29
+ return TVF.pad(image, padding, fill=padding_fill_value, padding_mode="constant")
30
+
31
+
32
+ class PrismaticImageProcessor(ImageProcessingMixin):
33
+ model_input_names: ClassVar[List[str]] = ["pixel_values"]
34
+
35
+ def __init__(
36
+ self,
37
+ use_fused_vision_backbone: bool = False,
38
+ image_resize_strategy: str = "letterbox",
39
+ input_sizes: Optional[List[Tuple[int, int, int]]] = None,
40
+ interpolations: Optional[List[str]] = None,
41
+ means: Optional[List[Tuple[float, float, float]]] = None,
42
+ stds: Optional[List[Tuple[float, float, float]]] = None,
43
+ **kwargs: str,
44
+ ) -> None:
45
+ """
46
+ Initialize a PrismaticImageProcessor as a wrapper around a torchvision transform; this transform will be
47
+ created by TIMM, and edited to follow our custom `image_resize_strategy` logic.
48
+
49
+ @param use_fused_vision_backbone: Boolean indicating single or fused (dual) vision backbone
50
+ @param image_resize_strategy: Prismatic image resize strategy in < resize-naive | resize-crop | letterbox >
51
+ @param input_size: [TIMM :: `data_cfg`] Input image size as tuple (channels, width, height)
52
+ @param interpolation: [TIMM :: `data_cfg`] Interpolation as string (default: "bicubic")
53
+ @param mean: [TIMM :: `data_cfg`] Normalization mean as float tuple (or two-tuple if `fused_backbone`)
54
+ @param std: [TIMM :: `data_cfg`] Normalization std as float tuple (or two-tuple if `fused_backbone`)
55
+ """
56
+ self.use_fused_vision_backbone = use_fused_vision_backbone
57
+ self.image_resize_strategy = image_resize_strategy
58
+
59
+ # Handle `None` default values
60
+ input_sizes = [(3, 224, 224)] if input_sizes is None else input_sizes
61
+ means = [(0.5, 0.5, 0.5)] if means is None else means
62
+ stds = [(0.5, 0.5, 0.5)] if stds is None else stds
63
+
64
+ # TIMM `data_cfg` Parameters
65
+ self.input_sizes, self.interpolations, self.means, self.stds = input_sizes, interpolations, means, stds
66
+
67
+ # Grab torchvision transforms via TIMM =>> need to parse for specific "functional" transform values!
68
+ self.tvf_resize_params, self.tvf_crop_params, self.tvf_normalize_params = [], [], []
69
+ self.tvf_do_letterbox, self.tvf_letterbox_fill = False, None
70
+
71
+ for idx in range(len(input_sizes)):
72
+ transform = timm.data.create_transform(
73
+ input_size=self.input_sizes[idx],
74
+ interpolation=self.interpolations[idx],
75
+ mean=self.means[idx],
76
+ std=self.stds[idx],
77
+ crop_pct=1.0, # Set to 1.0 to ignore cropping (initial Resize sets `input_size`)
78
+ crop_mode="center", # Default crop mode -- no-op when `crop_pct == 1.0`
79
+ is_training=False, # No image augmentations when loading the transform!
80
+ )
81
+
82
+ # [Validation] Ensure appropriate transform structure, expected sizes
83
+ if not (
84
+ isinstance(transform, Compose)
85
+ and (len(transform.transforms) == 4)
86
+ and isinstance(transform.transforms[0], Resize)
87
+ and isinstance(transform.transforms[1], CenterCrop)
88
+ and isinstance(transform.transforms[2], ToTensor)
89
+ and isinstance(transform.transforms[3], Normalize)
90
+ and (transform.transforms[0].size == self.input_sizes[idx][-1])
91
+ and (transform.transforms[1].size == self.input_sizes[idx][-2:])
92
+ ):
93
+ raise ValueError(f"Unexpected TIMM image transformation structure/sizes: `{transform}`")
94
+
95
+ # HF Image Processors *must* be JSON-serializable; as such, cannot have torchvision. as an attribute.
96
+ # => Instead, we're going to parse the transform and call "torchvision.transforms.functional" (`tvf`)
97
+ resize_t, crop_t, norm_t = transform.transforms[0], transform.transforms[1], transform.transforms[3]
98
+ self.tvf_resize_params.append(
99
+ {
100
+ "size": resize_t.size,
101
+ "interpolation": TVF.pil_modes_mapping[resize_t.interpolation],
102
+ "max_size": None,
103
+ "antialias": True,
104
+ }
105
+ )
106
+ self.tvf_crop_params.append({"output_size": crop_t.size})
107
+ self.tvf_normalize_params.append(
108
+ {
109
+ "mean": norm_t.mean.float().numpy().tolist(),
110
+ "std": norm_t.std.float().numpy().tolist(),
111
+ "inplace": False,
112
+ }
113
+ )
114
+ self.tvf_do_letterbox, self.tvf_letterbox_fill = False, None
115
+
116
+ # Handle Prismatic `image_resize_strategy`
117
+ if self.image_resize_strategy == "resize-naive":
118
+ self.tvf_resize_params[idx]["size"] = (resize_t.size, resize_t.size)
119
+ elif self.image_resize_strategy == "letterbox":
120
+ self.tvf_do_letterbox, self.tvf_letterbox_fill = True, tuple([int(x * 255) for x in self.means[idx]])
121
+ elif self.image_resize_strategy == "resize-crop":
122
+ pass
123
+ else:
124
+ raise ValueError(f"Image resize strategy `{self.image_resize_strategy}` is not supported!")
125
+
126
+ # Dispatch **kwargs to super()
127
+ super().__init__(**kwargs)
128
+
129
+ def apply_transform(self, img: Image.Image) -> torch.Tensor:
130
+ """Apply `functional` variant of TIMM's Transform = Compose([Resize -> CenterCrop -> ToTensor -> Normalize])"""
131
+ if self.tvf_do_letterbox:
132
+ img = letterbox_pad_transform(img, self.tvf_letterbox_fill)
133
+
134
+ # [Contract] Fused Backbones expect "channel-stacked" inputs; we'll unpack on the model side!
135
+ imgs_t = []
136
+ for idx in range(len(self.input_sizes)):
137
+ img_idx = TVF.resize(img, **self.tvf_resize_params[idx])
138
+ img_idx = TVF.center_crop(img_idx, **self.tvf_crop_params[idx])
139
+ img_idx_t = TVF.to_tensor(img_idx)
140
+ img_idx_t = TVF.normalize(img_idx_t, **self.tvf_normalize_params[idx])
141
+ imgs_t.append(img_idx_t)
142
+
143
+ # [Contract] `imgs_t` is a list of Tensors of shape [3, input_size, input_size]; stack along dim = 0
144
+ img_t = torch.vstack(imgs_t)
145
+
146
+ return img_t
147
+
148
+ def preprocess(
149
+ self,
150
+ images: Union[Image.Image, List[Image.Image]],
151
+ return_tensors: Optional[Union[str, TensorType]] = None,
152
+ **_: str,
153
+ ) -> BatchFeature:
154
+ """
155
+ Preprocess an image (or batch of images); note that unlike the `transformers :: BaseImageProcessor` we
156
+ explicitly only handle PIL.Image.Image instances for simplicity.
157
+
158
+ @param images: A (batch of) PIL.Image.Image instance(s) to preprocess.
159
+ @param return_tensors: BatchFeature default Tensor format (e.g., "pt" for torch); if None, returns np.ndarray
160
+
161
+ @return: Instance of `transformers :: BatchFeature` with a single key "pixel_values"
162
+ """
163
+ if not isinstance(images, list):
164
+ images = [images]
165
+
166
+ # Apply `self.img_transform` to each image (will return list of torch.Tensors); stack into "batched" Tensor
167
+ pixel_values = torch.stack([self.apply_transform(img.convert("RGB")) for img in images])
168
+
169
+ # Return BatchFeature =>> note that for compatibility, constructor expects Dict[str, np.ndarray], so we convert
170
+ return BatchFeature(data={"pixel_values": pixel_values.float().numpy()}, tensor_type=return_tensors)
171
+
172
+ def __call__(self, images: Union[Image.Image, List[Image.Image]], **kwargs) -> BatchFeature:
173
+ return self.preprocess(images, **kwargs)
174
+
175
+
176
+ # === PrismaticProcessor =>> Wraps both ImageProcessor and Tokenizer ===
177
+ # =>> https://github.com/huggingface/transformers/blob/main/src/transformers/models/llava/processing_llava.py
178
+ class PrismaticProcessor(ProcessorMixin):
179
+ attributes: ClassVar[List[str]] = ["image_processor", "tokenizer"]
180
+ image_processor_class: str = "AutoImageProcessor"
181
+ tokenizer_class: str = "AutoTokenizer"
182
+
183
+ def __init__(
184
+ self,
185
+ image_processor: Optional[ImageProcessingMixin] = None,
186
+ tokenizer: Optional[PreTrainedTokenizerBase] = None,
187
+ ) -> None:
188
+ super().__init__(image_processor, tokenizer)
189
+
190
+ def __call__(
191
+ self,
192
+ text: Union[TextInput, PreTokenizedInput, List[TextInput], List[PreTokenizedInput]],
193
+ images: Union[Image.Image, List[Image.Image]],
194
+ padding: Union[bool, str, PaddingStrategy] = False,
195
+ truncation: Optional[Union[bool, str, TruncationStrategy]] = None,
196
+ max_length: Optional[int] = None,
197
+ return_tensors: Optional[Union[str, TensorType]] = TensorType.PYTORCH,
198
+ ) -> BatchFeature:
199
+ """
200
+ Preprocess a given (batch) of text/images for a Prismatic VLM; forwards text to the underlying LLM's tokenizer,
201
+ forwards images to PrismaticImageProcessor.
202
+
203
+ @param text: The (batch) of text to encode; must be a string or list of strings.
204
+ @param images: A (batch of) PIL.Image.Image instance(s) to preprocess.
205
+ @param padding: Sequence padding strategy (if multiple specified) in < True = "longest" | "max_length" | False >
206
+ @param truncation: Truncation strategy for the output sequences; requires `max_length` to be specified
207
+ @param max_length: Maximum length (in tokens) to truncate
208
+ @param return_tensors: Type of return tensors (usually "pt" or TensorType.PYTORCH)
209
+
210
+ @return: BatchFeature with keys for `input_ids`, `attention_mask` and `pixel_values`.
211
+ """
212
+ pixel_values = self.image_processor(images, return_tensors=return_tensors)["pixel_values"]
213
+ text_inputs = self.tokenizer(
214
+ text, return_tensors=return_tensors, padding=padding, truncation=truncation, max_length=max_length
215
+ )
216
+
217
+ # [Validate] Need same number of images and text inputs!
218
+ if pixel_values.shape[0] != text_inputs.input_ids.shape[0]:
219
+ raise ValueError("Batch is malformed; expected same number of images and text inputs!")
220
+
221
+ return BatchFeature(data={**text_inputs, "pixel_values": pixel_values})
222
+
223
+ # === Tokenizer Dispatch Utilities =>> check `PreTrainedTokenizerBase` for documentation ===
224
+ def batch_decode(
225
+ self,
226
+ sequences: Union[List[int], List[List[int]], torch.Tensor, Any], # `Any` = np.ndarray | tf.Tensor
227
+ skip_special_tokens: bool = False,
228
+ clean_up_tokenization_spaces: Optional[bool] = None,
229
+ **kwargs: str,
230
+ ) -> List[str]:
231
+ return self.tokenizer.batch_decode(
232
+ sequences=sequences,
233
+ skip_special_tokens=skip_special_tokens,
234
+ clean_up_tokenization_spaces=clean_up_tokenization_spaces,
235
+ **kwargs,
236
+ )
237
+
238
+ def decode(
239
+ self,
240
+ token_ids: Union[int, List[int], torch.Tensor, Any], # `Any` = np.ndarray | tf.Tensor
241
+ skip_special_tokens: bool = False,
242
+ clean_up_tokenization_spaces: Optional[bool] = None,
243
+ **kwargs: str,
244
+ ) -> str:
245
+ return self.tokenizer.decode(
246
+ token_ids=token_ids,
247
+ skip_special_tokens=skip_special_tokens,
248
+ clean_up_tokenization_spaces=clean_up_tokenization_spaces,
249
+ **kwargs,
250
+ )
251
+
252
+ @property
253
+ def model_input_names(self) -> List[str]:
254
+ tokenizer_input_names = self.tokenizer.model_input_names
255
+ image_processor_input_names = self.image_processor.model_input_names
256
+
257
+ return list(dict.fromkeys(tokenizer_input_names + image_processor_input_names))
piper_pick_can_10k/processor_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "auto_map": {
3
+ "AutoProcessor": "processing_prismatic.PrismaticProcessor"
4
+ },
5
+ "processor_class": "PrismaticProcessor"
6
+ }
piper_pick_can_10k/special_tokens_map.json ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ {
4
+ "content": "<ACT_0>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false
9
+ },
10
+ {
11
+ "content": "<ACT_1>",
12
+ "lstrip": false,
13
+ "normalized": false,
14
+ "rstrip": false,
15
+ "single_word": false
16
+ },
17
+ {
18
+ "content": "<ACT_2>",
19
+ "lstrip": false,
20
+ "normalized": false,
21
+ "rstrip": false,
22
+ "single_word": false
23
+ },
24
+ {
25
+ "content": "<ACT_3>",
26
+ "lstrip": false,
27
+ "normalized": false,
28
+ "rstrip": false,
29
+ "single_word": false
30
+ }
31
+ ],
32
+ "bos_token": {
33
+ "content": "<s>",
34
+ "lstrip": false,
35
+ "normalized": false,
36
+ "rstrip": false,
37
+ "single_word": false
38
+ },
39
+ "eos_token": {
40
+ "content": "</s>",
41
+ "lstrip": false,
42
+ "normalized": false,
43
+ "rstrip": false,
44
+ "single_word": false
45
+ },
46
+ "pad_token": {
47
+ "content": "<PAD>",
48
+ "lstrip": false,
49
+ "normalized": false,
50
+ "rstrip": false,
51
+ "single_word": false
52
+ },
53
+ "unk_token": {
54
+ "content": "<unk>",
55
+ "lstrip": false,
56
+ "normalized": false,
57
+ "rstrip": false,
58
+ "single_word": false
59
+ }
60
+ }
piper_pick_can_10k/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1219be150dec6ecbf36f507377e7df878602b89df9388db708d2dc3356aa43be
3
+ size 1843708
piper_pick_can_10k/tokenizer.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e556afd44213b6bd1be2b850ebbbd98f5481437a8021afaf58ee7fb1818d347
3
+ size 499723
piper_pick_can_10k/tokenizer_config.json ADDED
@@ -0,0 +1,91 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": true,
3
+ "add_eos_token": false,
4
+ "added_tokens_decoder": {
5
+ "0": {
6
+ "content": "<unk>",
7
+ "lstrip": false,
8
+ "normalized": false,
9
+ "rstrip": false,
10
+ "single_word": false,
11
+ "special": true
12
+ },
13
+ "1": {
14
+ "content": "<s>",
15
+ "lstrip": false,
16
+ "normalized": false,
17
+ "rstrip": false,
18
+ "single_word": false,
19
+ "special": true
20
+ },
21
+ "2": {
22
+ "content": "</s>",
23
+ "lstrip": false,
24
+ "normalized": false,
25
+ "rstrip": false,
26
+ "single_word": false,
27
+ "special": true
28
+ },
29
+ "32000": {
30
+ "content": "<PAD>",
31
+ "lstrip": false,
32
+ "normalized": false,
33
+ "rstrip": false,
34
+ "single_word": false,
35
+ "special": true
36
+ },
37
+ "32001": {
38
+ "content": "<ACT_0>",
39
+ "lstrip": false,
40
+ "normalized": false,
41
+ "rstrip": false,
42
+ "single_word": false,
43
+ "special": true
44
+ },
45
+ "32002": {
46
+ "content": "<ACT_1>",
47
+ "lstrip": false,
48
+ "normalized": false,
49
+ "rstrip": false,
50
+ "single_word": false,
51
+ "special": true
52
+ },
53
+ "32003": {
54
+ "content": "<ACT_2>",
55
+ "lstrip": false,
56
+ "normalized": false,
57
+ "rstrip": false,
58
+ "single_word": false,
59
+ "special": true
60
+ },
61
+ "32004": {
62
+ "content": "<ACT_3>",
63
+ "lstrip": false,
64
+ "normalized": false,
65
+ "rstrip": false,
66
+ "single_word": false,
67
+ "special": true
68
+ }
69
+ },
70
+ "additional_special_tokens": [
71
+ "<ACT_0>",
72
+ "<ACT_1>",
73
+ "<ACT_2>",
74
+ "<ACT_3>"
75
+ ],
76
+ "auto_map": {
77
+ "AutoProcessor": "processing_prismatic.PrismaticProcessor"
78
+ },
79
+ "bos_token": "<s>",
80
+ "clean_up_tokenization_spaces": false,
81
+ "eos_token": "</s>",
82
+ "legacy": false,
83
+ "model_max_length": 2048,
84
+ "pad_token": "<PAD>",
85
+ "padding_side": "right",
86
+ "processor_class": "PrismaticProcessor",
87
+ "sp_model_kwargs": {},
88
+ "tokenizer_class": "LlamaTokenizer",
89
+ "unk_token": "<unk>",
90
+ "use_default_system_prompt": false
91
+ }
piper_pick_can_10k/train_log.jsonl ADDED
@@ -0,0 +1,400 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"step": 25, "latent_r2": 0.24713469803333282, "latent_loss": 1.615971332192421, "action_l1_1step": 0.40671144872903825, "p_oracle": 0.996, "lr": 0.00035, "seconds": 55}
2
+ {"step": 50, "latent_r2": 0.46883842766284944, "latent_loss": 1.2833967173099519, "action_l1_1step": 0.2002254507690668, "p_oracle": 0.9918333333333333, "lr": 0.00035, "seconds": 115}
3
+ {"step": 75, "latent_r2": 0.5538767677545547, "latent_loss": 1.1087893772125244, "action_l1_1step": 0.18017434261739254, "p_oracle": 0.9876666666666667, "lr": 0.00035, "seconds": 171}
4
+ {"step": 100, "latent_r2": 0.575291457772255, "latent_loss": 1.0247715955972672, "action_l1_1step": 0.16412587314844132, "p_oracle": 0.9835, "lr": 0.00035, "seconds": 230}
5
+ {"step": 125, "latent_r2": 0.6025295549631119, "latent_loss": 0.9703685522079468, "action_l1_1step": 0.16291079856455326, "p_oracle": 0.9793333333333334, "lr": 0.00035, "seconds": 286}
6
+ {"step": 150, "latent_r2": 0.6249476134777069, "latent_loss": 0.9327966624498367, "action_l1_1step": 0.15170668110251426, "p_oracle": 0.9751666666666666, "lr": 0.00035, "seconds": 341}
7
+ {"step": 175, "latent_r2": 0.636535724401474, "latent_loss": 0.9104489487409592, "action_l1_1step": 0.1505958116799593, "p_oracle": 0.971, "lr": 0.00035, "seconds": 397}
8
+ {"step": 200, "latent_r2": 0.641583902835846, "latent_loss": 0.8743717193603515, "action_l1_1step": 0.14482259191572666, "p_oracle": 0.9668333333333333, "lr": 0.00035, "seconds": 452}
9
+ {"step": 225, "latent_r2": 0.689115264415741, "latent_loss": 0.8670964306592941, "action_l1_1step": 0.13643484883010387, "p_oracle": 0.9626666666666667, "lr": 0.00035, "seconds": 507}
10
+ {"step": 250, "latent_r2": 0.6708480215072632, "latent_loss": 0.829840463399887, "action_l1_1step": 0.13745581559836864, "p_oracle": 0.9585, "lr": 0.00035, "seconds": 563}
11
+ {"step": 275, "latent_r2": 0.6925591218471527, "latent_loss": 0.8346778613328933, "action_l1_1step": 0.12502445071935653, "p_oracle": 0.9543333333333334, "lr": 0.00035, "seconds": 625}
12
+ {"step": 300, "latent_r2": 0.6973156893253326, "latent_loss": 0.8087809151411056, "action_l1_1step": 0.1258083061128855, "p_oracle": 0.9501666666666667, "lr": 0.00035, "seconds": 681}
13
+ {"step": 325, "latent_r2": 0.6968361270427704, "latent_loss": 0.800451266169548, "action_l1_1step": 0.12429429933428765, "p_oracle": 0.946, "lr": 0.00035, "seconds": 742}
14
+ {"step": 350, "latent_r2": 0.7110520666837692, "latent_loss": 0.8078470271825791, "action_l1_1step": 0.12415348201990127, "p_oracle": 0.9418333333333333, "lr": 0.00035, "seconds": 797}
15
+ {"step": 375, "latent_r2": 0.6982990020513534, "latent_loss": 0.7922458958625793, "action_l1_1step": 0.1234408339112997, "p_oracle": 0.9376666666666666, "lr": 0.00035, "seconds": 852}
16
+ {"step": 400, "latent_r2": 0.7018646204471588, "latent_loss": 0.7847280240058899, "action_l1_1step": 0.11676639568060637, "p_oracle": 0.9335, "lr": 0.00035, "seconds": 907}
17
+ {"step": 425, "latent_r2": 0.7017230200767517, "latent_loss": 0.779858992099762, "action_l1_1step": 0.11434786409139633, "p_oracle": 0.9293333333333333, "lr": 0.00035, "seconds": 962}
18
+ {"step": 450, "latent_r2": 0.7160915243625641, "latent_loss": 0.7784245210886002, "action_l1_1step": 0.11210632972419261, "p_oracle": 0.9251666666666667, "lr": 0.00035, "seconds": 1018}
19
+ {"step": 475, "latent_r2": 0.7362280380725861, "latent_loss": 0.7810861390829086, "action_l1_1step": 0.11177432075142861, "p_oracle": 0.921, "lr": 0.00035, "seconds": 1073}
20
+ {"step": 500, "latent_r2": 0.7284565401077271, "latent_loss": 0.7623290604352951, "action_l1_1step": 0.11011491857469081, "p_oracle": 0.9168333333333334, "lr": 0.00035, "seconds": 1135}
21
+ {"step": 525, "latent_r2": 0.7056619197130203, "latent_loss": 0.7481558215618134, "action_l1_1step": 0.11473990336060524, "p_oracle": 0.9126666666666666, "lr": 0.00035, "seconds": 1191}
22
+ {"step": 550, "latent_r2": 0.7178732591867447, "latent_loss": 0.7448985117673874, "action_l1_1step": 0.10779515981674194, "p_oracle": 0.9085, "lr": 0.00035, "seconds": 1251}
23
+ {"step": 575, "latent_r2": 0.7235252791643143, "latent_loss": 0.7386216515302658, "action_l1_1step": 0.10841914128512144, "p_oracle": 0.9043333333333333, "lr": 0.00035, "seconds": 1306}
24
+ {"step": 600, "latent_r2": 0.7253024655580521, "latent_loss": 0.7425705862045288, "action_l1_1step": 0.10349898800253868, "p_oracle": 0.9001666666666667, "lr": 0.00035, "seconds": 1364}
25
+ {"step": 625, "latent_r2": 0.7227128291130066, "latent_loss": 0.7258235102891922, "action_l1_1step": 0.10686212312430143, "p_oracle": 0.896, "lr": 0.00035, "seconds": 1419}
26
+ {"step": 650, "latent_r2": 0.7307551425695419, "latent_loss": 0.7330788719654083, "action_l1_1step": 0.10279592886567115, "p_oracle": 0.8918333333333334, "lr": 0.00035, "seconds": 1477}
27
+ {"step": 675, "latent_r2": 0.7294180119037628, "latent_loss": 0.729849084019661, "action_l1_1step": 0.10357192151248455, "p_oracle": 0.8876666666666667, "lr": 0.00035, "seconds": 1532}
28
+ {"step": 700, "latent_r2": 0.7367560917139053, "latent_loss": 0.7225811678171158, "action_l1_1step": 0.10459008187055588, "p_oracle": 0.8835, "lr": 0.00035, "seconds": 1589}
29
+ {"step": 725, "latent_r2": 0.7433342093229294, "latent_loss": 0.7337736225128174, "action_l1_1step": 0.1004676290601492, "p_oracle": 0.8793333333333333, "lr": 0.00035, "seconds": 1649}
30
+ {"step": 750, "latent_r2": 0.7336837434768677, "latent_loss": 0.7335594481229782, "action_l1_1step": 0.10108281932771206, "p_oracle": 0.8751666666666666, "lr": 0.00035, "seconds": 1705}
31
+ {"step": 775, "latent_r2": 0.7269551938772202, "latent_loss": 0.7263218641281128, "action_l1_1step": 0.09992009818553925, "p_oracle": 0.871, "lr": 0.00035, "seconds": 1766}
32
+ {"step": 800, "latent_r2": 0.7353724724054337, "latent_loss": 0.7298247587680816, "action_l1_1step": 0.10071688197553158, "p_oracle": 0.8668333333333333, "lr": 0.00035, "seconds": 1821}
33
+ {"step": 825, "latent_r2": 0.7363368713855744, "latent_loss": 0.7094754004478454, "action_l1_1step": 0.09890947617590427, "p_oracle": 0.8626666666666667, "lr": 0.00035, "seconds": 1877}
34
+ {"step": 850, "latent_r2": 0.7436561715602875, "latent_loss": 0.7201018619537354, "action_l1_1step": 0.09931190025061369, "p_oracle": 0.8585, "lr": 0.00035, "seconds": 1932}
35
+ {"step": 875, "latent_r2": 0.7568805140256881, "latent_loss": 0.7281739687919617, "action_l1_1step": 0.0966040888801217, "p_oracle": 0.8543333333333334, "lr": 0.00035, "seconds": 1987}
36
+ {"step": 900, "latent_r2": 0.7217999130487442, "latent_loss": 0.7100288897752762, "action_l1_1step": 0.10066183518618345, "p_oracle": 0.8501666666666666, "lr": 0.00035, "seconds": 2042}
37
+ {"step": 925, "latent_r2": 0.7363890355825424, "latent_loss": 0.7158119207620621, "action_l1_1step": 0.098316021412611, "p_oracle": 0.846, "lr": 0.00035, "seconds": 2099}
38
+ {"step": 950, "latent_r2": 0.7232470685243606, "latent_loss": 0.7144586515426635, "action_l1_1step": 0.09637989360839129, "p_oracle": 0.8418333333333333, "lr": 0.00035, "seconds": 2161}
39
+ {"step": 975, "latent_r2": 0.737501220703125, "latent_loss": 0.7098477911949158, "action_l1_1step": 0.09605750877410174, "p_oracle": 0.8376666666666667, "lr": 0.00035, "seconds": 2217}
40
+ {"step": 1000, "latent_r2": 0.7460629743337631, "latent_loss": 0.7111299568414688, "action_l1_1step": 0.09237641107290984, "p_oracle": 0.8335, "lr": 0.00035, "seconds": 2275}
41
+ {"step": 1025, "latent_r2": 0.7482826524972915, "latent_loss": 0.7025927442312241, "action_l1_1step": 0.0922438506782055, "p_oracle": 0.8293333333333334, "lr": 0.00035, "seconds": 2339}
42
+ {"step": 1050, "latent_r2": 0.7491305178403854, "latent_loss": 0.7109052538871765, "action_l1_1step": 0.0906205989420414, "p_oracle": 0.8251666666666666, "lr": 0.00035, "seconds": 2394}
43
+ {"step": 1075, "latent_r2": 0.7495535892248154, "latent_loss": 0.7101803582906723, "action_l1_1step": 0.087930116802454, "p_oracle": 0.821, "lr": 0.00035, "seconds": 2450}
44
+ {"step": 1100, "latent_r2": 0.7436224323511124, "latent_loss": 0.7081825149059295, "action_l1_1step": 0.0935745806619525, "p_oracle": 0.8168333333333333, "lr": 0.00035, "seconds": 2505}
45
+ {"step": 1125, "latent_r2": 0.7237441748380661, "latent_loss": 0.6854030275344849, "action_l1_1step": 0.09326360858976841, "p_oracle": 0.8126666666666666, "lr": 0.00035, "seconds": 2560}
46
+ {"step": 1150, "latent_r2": 0.7563421243429184, "latent_loss": 0.7137868130207061, "action_l1_1step": 0.08775397799909115, "p_oracle": 0.8085, "lr": 0.00035, "seconds": 2618}
47
+ {"step": 1175, "latent_r2": 0.7546551370620728, "latent_loss": 0.6848169094324112, "action_l1_1step": 0.08923269487917423, "p_oracle": 0.8043333333333333, "lr": 0.00035, "seconds": 2677}
48
+ {"step": 1200, "latent_r2": 0.7612514859437942, "latent_loss": 0.6959750655293465, "action_l1_1step": 0.08873735319823027, "p_oracle": 0.8001666666666667, "lr": 0.00035, "seconds": 2733}
49
+ {"step": 1225, "latent_r2": 0.7463935017585754, "latent_loss": 0.6787053886055946, "action_l1_1step": 0.08947099417448044, "p_oracle": 0.796, "lr": 0.00035, "seconds": 2794}
50
+ {"step": 1250, "latent_r2": 0.7402651888132096, "latent_loss": 0.6705477130413056, "action_l1_1step": 0.08993307124823331, "p_oracle": 0.7918333333333334, "lr": 0.00035, "seconds": 2849}
51
+ {"step": 1275, "latent_r2": 0.744080080986023, "latent_loss": 0.6868360328674317, "action_l1_1step": 0.08965229149907827, "p_oracle": 0.7876666666666666, "lr": 0.00035, "seconds": 2909}
52
+ {"step": 1300, "latent_r2": 0.7505458056926727, "latent_loss": 0.6927961295843125, "action_l1_1step": 0.08756936393678189, "p_oracle": 0.7835, "lr": 0.00035, "seconds": 2964}
53
+ {"step": 1325, "latent_r2": 0.7444037181138993, "latent_loss": 0.682880003452301, "action_l1_1step": 0.08737980905920267, "p_oracle": 0.7793333333333333, "lr": 0.00035, "seconds": 3019}
54
+ {"step": 1350, "latent_r2": 0.7489987587928773, "latent_loss": 0.6813306576013565, "action_l1_1step": 0.08389454495161772, "p_oracle": 0.7751666666666667, "lr": 0.00035, "seconds": 3078}
55
+ {"step": 1375, "latent_r2": 0.753590224981308, "latent_loss": 0.6951387393474578, "action_l1_1step": 0.0855080385133624, "p_oracle": 0.771, "lr": 0.00035, "seconds": 3137}
56
+ {"step": 1400, "latent_r2": 0.7548593533039093, "latent_loss": 0.6816092067956925, "action_l1_1step": 0.08627999398857356, "p_oracle": 0.7668333333333334, "lr": 0.00035, "seconds": 3193}
57
+ {"step": 1425, "latent_r2": 0.7510571283102035, "latent_loss": 0.6594125026464462, "action_l1_1step": 0.08884270042181015, "p_oracle": 0.7626666666666666, "lr": 0.00035, "seconds": 3255}
58
+ {"step": 1450, "latent_r2": 0.7686213183403016, "latent_loss": 0.7030737921595573, "action_l1_1step": 0.08381440356373787, "p_oracle": 0.7585, "lr": 0.00035, "seconds": 3311}
59
+ {"step": 1475, "latent_r2": 0.7580589300394058, "latent_loss": 0.6752447244524956, "action_l1_1step": 0.08764339692890644, "p_oracle": 0.7543333333333333, "lr": 0.00035, "seconds": 3366}
60
+ {"step": 1500, "latent_r2": 0.7564413297176361, "latent_loss": 0.6708270865678787, "action_l1_1step": 0.08500146821141243, "p_oracle": 0.7501666666666666, "lr": 0.00035, "seconds": 3421}
61
+ {"step": 1525, "latent_r2": 0.7412615674734115, "latent_loss": 0.677560310959816, "action_l1_1step": 0.08631958063691854, "p_oracle": 0.746, "lr": 0.00035, "seconds": 3477}
62
+ {"step": 1550, "latent_r2": 0.7526605272293091, "latent_loss": 0.6642361062765122, "action_l1_1step": 0.08386842831969261, "p_oracle": 0.7418333333333333, "lr": 0.00035, "seconds": 3533}
63
+ {"step": 1575, "latent_r2": 0.7508710706233979, "latent_loss": 0.6611780121922493, "action_l1_1step": 0.08491994675248861, "p_oracle": 0.7376666666666667, "lr": 0.00035, "seconds": 3594}
64
+ {"step": 1600, "latent_r2": 0.7597137457132339, "latent_loss": 0.6748465126752854, "action_l1_1step": 0.07999477811157703, "p_oracle": 0.7335, "lr": 0.00035, "seconds": 3650}
65
+ {"step": 1625, "latent_r2": 0.7550743579864502, "latent_loss": 0.6747476279735565, "action_l1_1step": 0.08072443444281817, "p_oracle": 0.7293333333333334, "lr": 0.00035, "seconds": 3709}
66
+ {"step": 1650, "latent_r2": 0.7621232634782791, "latent_loss": 0.6743027177453041, "action_l1_1step": 0.08180644016712904, "p_oracle": 0.7251666666666667, "lr": 0.00035, "seconds": 3766}
67
+ {"step": 1675, "latent_r2": 0.758370361328125, "latent_loss": 0.6757417419552803, "action_l1_1step": 0.08531426213681698, "p_oracle": 0.721, "lr": 0.00035, "seconds": 3822}
68
+ {"step": 1700, "latent_r2": 0.7562271898984909, "latent_loss": 0.6709678140282631, "action_l1_1step": 0.08218072574585676, "p_oracle": 0.7168333333333333, "lr": 0.00035, "seconds": 3877}
69
+ {"step": 1725, "latent_r2": 0.7582158982753754, "latent_loss": 0.6604804193973541, "action_l1_1step": 0.08028558161109686, "p_oracle": 0.7126666666666667, "lr": 0.00035, "seconds": 3933}
70
+ {"step": 1750, "latent_r2": 0.757350721359253, "latent_loss": 0.6646727174520493, "action_l1_1step": 0.08059627607464791, "p_oracle": 0.7085, "lr": 0.00035, "seconds": 3988}
71
+ {"step": 1775, "latent_r2": 0.767531823515892, "latent_loss": 0.6601039373874664, "action_l1_1step": 0.08111606623977423, "p_oracle": 0.7043333333333333, "lr": 0.00035, "seconds": 4049}
72
+ {"step": 1800, "latent_r2": 0.7684644621610641, "latent_loss": 0.6804415440559387, "action_l1_1step": 0.07939812209457159, "p_oracle": 0.7001666666666666, "lr": 0.00035, "seconds": 4106}
73
+ {"step": 1825, "latent_r2": 0.7666828793287277, "latent_loss": 0.66221888422966, "action_l1_1step": 0.077962540499866, "p_oracle": 0.696, "lr": 0.00035, "seconds": 4164}
74
+ {"step": 1850, "latent_r2": 0.7606836545467377, "latent_loss": 0.6558243620395661, "action_l1_1step": 0.08250693257898092, "p_oracle": 0.6918333333333333, "lr": 0.00035, "seconds": 4222}
75
+ {"step": 1875, "latent_r2": 0.7694813656806946, "latent_loss": 0.6643183541297912, "action_l1_1step": 0.0748392141982913, "p_oracle": 0.6876666666666666, "lr": 0.00035, "seconds": 4277}
76
+ {"step": 1900, "latent_r2": 0.7782694560289383, "latent_loss": 0.6504885831475258, "action_l1_1step": 0.07633225653320551, "p_oracle": 0.6835, "lr": 0.00035, "seconds": 4335}
77
+ {"step": 1925, "latent_r2": 0.7452460318803787, "latent_loss": 0.659137174487114, "action_l1_1step": 0.08385283775627612, "p_oracle": 0.6793333333333333, "lr": 0.00035, "seconds": 4391}
78
+ {"step": 1950, "latent_r2": 0.7783204835653305, "latent_loss": 0.6557733935117721, "action_l1_1step": 0.07736110106110573, "p_oracle": 0.6751666666666667, "lr": 0.00035, "seconds": 4446}
79
+ {"step": 1975, "latent_r2": 0.7598499399423599, "latent_loss": 0.6517538145184517, "action_l1_1step": 0.07807388685643674, "p_oracle": 0.671, "lr": 0.00035, "seconds": 4501}
80
+ {"step": 2000, "latent_r2": 0.7623269504308701, "latent_loss": 0.6559752142429351, "action_l1_1step": 0.08188733112066984, "p_oracle": 0.6668333333333334, "lr": 0.00035, "seconds": 4557}
81
+ {"step": 2025, "latent_r2": 0.7835645055770875, "latent_loss": 0.6564526668190956, "action_l1_1step": 0.07597641076892614, "p_oracle": 0.6626666666666667, "lr": 0.00035, "seconds": 4630}
82
+ {"step": 2050, "latent_r2": 0.7610481876134872, "latent_loss": 0.6479675632715225, "action_l1_1step": 0.07643775910139083, "p_oracle": 0.6585, "lr": 0.00035, "seconds": 4688}
83
+ {"step": 2075, "latent_r2": 0.7787607556581497, "latent_loss": 0.6550792720913887, "action_l1_1step": 0.0762852582335472, "p_oracle": 0.6543333333333333, "lr": 0.00035, "seconds": 4746}
84
+ {"step": 2100, "latent_r2": 0.7681032800674439, "latent_loss": 0.6382042369246483, "action_l1_1step": 0.07613888043910265, "p_oracle": 0.6501666666666667, "lr": 0.00035, "seconds": 4807}
85
+ {"step": 2125, "latent_r2": 0.7658608025312423, "latent_loss": 0.6603466394543648, "action_l1_1step": 0.07864309936761855, "p_oracle": 0.646, "lr": 0.00035, "seconds": 4862}
86
+ {"step": 2150, "latent_r2": 0.7767838364839554, "latent_loss": 0.658401175737381, "action_l1_1step": 0.07472824539989233, "p_oracle": 0.6418333333333333, "lr": 0.00035, "seconds": 4918}
87
+ {"step": 2175, "latent_r2": 0.7737958604097366, "latent_loss": 0.6492047929763793, "action_l1_1step": 0.07481635496020317, "p_oracle": 0.6376666666666666, "lr": 0.00035, "seconds": 4973}
88
+ {"step": 2200, "latent_r2": 0.7739313572645188, "latent_loss": 0.6442709663510322, "action_l1_1step": 0.07764185316860676, "p_oracle": 0.6335, "lr": 0.00035, "seconds": 5029}
89
+ {"step": 2225, "latent_r2": 0.7750390654802323, "latent_loss": 0.649660162627697, "action_l1_1step": 0.07459895700216293, "p_oracle": 0.6293333333333333, "lr": 0.00035, "seconds": 5086}
90
+ {"step": 2250, "latent_r2": 0.7741895055770874, "latent_loss": 0.6552875527739525, "action_l1_1step": 0.07652805462479591, "p_oracle": 0.6251666666666666, "lr": 0.00035, "seconds": 5146}
91
+ {"step": 2275, "latent_r2": 0.7729463237524032, "latent_loss": 0.6438614740967751, "action_l1_1step": 0.07451078675687313, "p_oracle": 0.621, "lr": 0.00035, "seconds": 5204}
92
+ {"step": 2300, "latent_r2": 0.7683783155679703, "latent_loss": 0.6393237638473511, "action_l1_1step": 0.07530065260827541, "p_oracle": 0.6168333333333333, "lr": 0.00035, "seconds": 5263}
93
+ {"step": 2325, "latent_r2": 0.7775632643699646, "latent_loss": 0.6534097430109977, "action_l1_1step": 0.07148197028785944, "p_oracle": 0.6126666666666667, "lr": 0.00035, "seconds": 5318}
94
+ {"step": 2350, "latent_r2": 0.7669839471578598, "latent_loss": 0.6443203848600387, "action_l1_1step": 0.07759860463440418, "p_oracle": 0.6085, "lr": 0.00035, "seconds": 5373}
95
+ {"step": 2375, "latent_r2": 0.7602559918165207, "latent_loss": 0.6349053752422332, "action_l1_1step": 0.0748582535237074, "p_oracle": 0.6043333333333334, "lr": 0.00035, "seconds": 5429}
96
+ {"step": 2400, "latent_r2": 0.7668520998954773, "latent_loss": 0.6406247156858444, "action_l1_1step": 0.07371264718472957, "p_oracle": 0.6001666666666667, "lr": 0.00035, "seconds": 5484}
97
+ {"step": 2425, "latent_r2": 0.7852633160352707, "latent_loss": 0.6306700283288955, "action_l1_1step": 0.07063896335661411, "p_oracle": 0.596, "lr": 0.00035, "seconds": 5542}
98
+ {"step": 2450, "latent_r2": 0.7922415721416474, "latent_loss": 0.6587877678871155, "action_l1_1step": 0.07236864950507879, "p_oracle": 0.5918333333333333, "lr": 0.00035, "seconds": 5603}
99
+ {"step": 2475, "latent_r2": 0.7880981868505478, "latent_loss": 0.6542456820607185, "action_l1_1step": 0.07345686953514814, "p_oracle": 0.5876666666666667, "lr": 0.00035, "seconds": 5659}
100
+ {"step": 2500, "latent_r2": 0.7838786751031875, "latent_loss": 0.6333817279338837, "action_l1_1step": 0.07437116410583258, "p_oracle": 0.5835, "lr": 0.00035, "seconds": 5719}
101
+ {"step": 2525, "latent_r2": 0.7855376994609833, "latent_loss": 0.6510905155539513, "action_l1_1step": 0.07288071315735578, "p_oracle": 0.5793333333333333, "lr": 0.00035, "seconds": 5778}
102
+ {"step": 2550, "latent_r2": 0.7699252456426621, "latent_loss": 0.6342764437198639, "action_l1_1step": 0.07573056980967521, "p_oracle": 0.5751666666666666, "lr": 0.00035, "seconds": 5833}
103
+ {"step": 2575, "latent_r2": 0.767706453204155, "latent_loss": 0.6379271054267883, "action_l1_1step": 0.07548930041491986, "p_oracle": 0.571, "lr": 0.00035, "seconds": 5888}
104
+ {"step": 2600, "latent_r2": 0.7781894361972809, "latent_loss": 0.6466183465719223, "action_l1_1step": 0.07402126424014568, "p_oracle": 0.5668333333333333, "lr": 0.00035, "seconds": 5944}
105
+ {"step": 2625, "latent_r2": 0.7776057547330857, "latent_loss": 0.6298833265900612, "action_l1_1step": 0.07291012983769178, "p_oracle": 0.5626666666666666, "lr": 0.00035, "seconds": 5999}
106
+ {"step": 2650, "latent_r2": 0.7647379446029663, "latent_loss": 0.631395702958107, "action_l1_1step": 0.07600073013454675, "p_oracle": 0.5585, "lr": 0.00035, "seconds": 6061}
107
+ {"step": 2675, "latent_r2": 0.7661408597230911, "latent_loss": 0.6365236556529998, "action_l1_1step": 0.07410657081753015, "p_oracle": 0.5543333333333333, "lr": 0.00035, "seconds": 6117}
108
+ {"step": 2700, "latent_r2": 0.7763503080606461, "latent_loss": 0.6376482540369034, "action_l1_1step": 0.07398439329117537, "p_oracle": 0.5501666666666667, "lr": 0.00035, "seconds": 6176}
109
+ {"step": 2725, "latent_r2": 0.7802535873651505, "latent_loss": 0.6338701191544532, "action_l1_1step": 0.07000153061002493, "p_oracle": 0.546, "lr": 0.00035, "seconds": 6234}
110
+ {"step": 2750, "latent_r2": 0.7830642366409302, "latent_loss": 0.6316836780309677, "action_l1_1step": 0.07227260433137417, "p_oracle": 0.5418333333333334, "lr": 0.00035, "seconds": 6289}
111
+ {"step": 2775, "latent_r2": 0.7846486240625381, "latent_loss": 0.6410473597049713, "action_l1_1step": 0.07235703412443399, "p_oracle": 0.5376666666666667, "lr": 0.00035, "seconds": 6345}
112
+ {"step": 2800, "latent_r2": 0.7826536411046982, "latent_loss": 0.6351696050167084, "action_l1_1step": 0.07237887024879455, "p_oracle": 0.5335, "lr": 0.00035, "seconds": 6400}
113
+ {"step": 2825, "latent_r2": 0.7764726126194, "latent_loss": 0.6277770963311196, "action_l1_1step": 0.06957579001784325, "p_oracle": 0.5293333333333333, "lr": 0.00035, "seconds": 6455}
114
+ {"step": 2850, "latent_r2": 0.7794821953773499, "latent_loss": 0.62989177942276, "action_l1_1step": 0.07099344719201327, "p_oracle": 0.5251666666666667, "lr": 0.00035, "seconds": 6511}
115
+ {"step": 2875, "latent_r2": 0.7782496446371079, "latent_loss": 0.6408387166261673, "action_l1_1step": 0.07201239682734012, "p_oracle": 0.521, "lr": 0.00035, "seconds": 6573}
116
+ {"step": 2900, "latent_r2": 0.7781941044330597, "latent_loss": 0.6215573963522911, "action_l1_1step": 0.07334222935140133, "p_oracle": 0.5168333333333333, "lr": 0.00035, "seconds": 6629}
117
+ {"step": 2925, "latent_r2": 0.7798014485836029, "latent_loss": 0.640693930387497, "action_l1_1step": 0.07104950901120902, "p_oracle": 0.5126666666666666, "lr": 0.00035, "seconds": 6690}
118
+ {"step": 2950, "latent_r2": 0.7839416617155075, "latent_loss": 0.6259585466980934, "action_l1_1step": 0.06865016423165798, "p_oracle": 0.5085, "lr": 0.00035, "seconds": 6746}
119
+ {"step": 2975, "latent_r2": 0.7852713531255722, "latent_loss": 0.6342732441425324, "action_l1_1step": 0.06982143599539996, "p_oracle": 0.5043333333333333, "lr": 0.00035, "seconds": 6801}
120
+ {"step": 3000, "latent_r2": 0.7855996417999268, "latent_loss": 0.633515621125698, "action_l1_1step": 0.0680725659057498, "p_oracle": 0.5001666666666666, "lr": 0.00035, "seconds": 6856}
121
+ {"step": 3025, "latent_r2": 0.7806027674674988, "latent_loss": 0.6354331347346306, "action_l1_1step": 0.07178554866462945, "p_oracle": 0.496, "lr": 0.00035, "seconds": 6922}
122
+ {"step": 3050, "latent_r2": 0.7808019280433655, "latent_loss": 0.6305890011787415, "action_l1_1step": 0.07051472336053849, "p_oracle": 0.49183333333333334, "lr": 0.00035, "seconds": 6978}
123
+ {"step": 3075, "latent_r2": 0.7899183320999146, "latent_loss": 0.6221693843603134, "action_l1_1step": 0.06943029403686524, "p_oracle": 0.4876666666666667, "lr": 0.00035, "seconds": 7033}
124
+ {"step": 3100, "latent_r2": 0.7958403295278549, "latent_loss": 0.6353306916356086, "action_l1_1step": 0.06721409156918526, "p_oracle": 0.48350000000000004, "lr": 0.00035, "seconds": 7094}
125
+ {"step": 3125, "latent_r2": 0.7741015434265137, "latent_loss": 0.6200106245279312, "action_l1_1step": 0.07152404692023992, "p_oracle": 0.4793333333333333, "lr": 0.00035, "seconds": 7152}
126
+ {"step": 3150, "latent_r2": 0.7886409986019135, "latent_loss": 0.6268642300367355, "action_l1_1step": 0.0696645138412714, "p_oracle": 0.4751666666666666, "lr": 0.00035, "seconds": 7212}
127
+ {"step": 3175, "latent_r2": 0.7917615985870361, "latent_loss": 0.6292522326111794, "action_l1_1step": 0.06691477220505476, "p_oracle": 0.471, "lr": 0.00035, "seconds": 7275}
128
+ {"step": 3200, "latent_r2": 0.7846657717227936, "latent_loss": 0.6297276946902275, "action_l1_1step": 0.07019910115748644, "p_oracle": 0.4668333333333333, "lr": 0.00035, "seconds": 7331}
129
+ {"step": 3225, "latent_r2": 0.7878886592388153, "latent_loss": 0.621647390127182, "action_l1_1step": 0.06861495215445756, "p_oracle": 0.46266666666666667, "lr": 0.00035, "seconds": 7386}
130
+ {"step": 3250, "latent_r2": 0.7860119450092315, "latent_loss": 0.6147918966412544, "action_l1_1step": 0.06911690298467875, "p_oracle": 0.4585, "lr": 0.00035, "seconds": 7441}
131
+ {"step": 3275, "latent_r2": 0.7874009335041046, "latent_loss": 0.624572194814682, "action_l1_1step": 0.06749346978962421, "p_oracle": 0.45433333333333337, "lr": 0.00035, "seconds": 7497}
132
+ {"step": 3300, "latent_r2": 0.7795550566911698, "latent_loss": 0.6230722892284394, "action_l1_1step": 0.06978401470929384, "p_oracle": 0.4501666666666667, "lr": 0.00035, "seconds": 7554}
133
+ {"step": 3325, "latent_r2": 0.7873090553283691, "latent_loss": 0.6267603307962417, "action_l1_1step": 0.06975494962185622, "p_oracle": 0.44599999999999995, "lr": 0.00035, "seconds": 7615}
134
+ {"step": 3350, "latent_r2": 0.768581936955452, "latent_loss": 0.6320060360431671, "action_l1_1step": 0.06957406230270863, "p_oracle": 0.4418333333333333, "lr": 0.00035, "seconds": 7670}
135
+ {"step": 3375, "latent_r2": 0.7870222896337509, "latent_loss": 0.6469762939214706, "action_l1_1step": 0.06945525493472815, "p_oracle": 0.43766666666666665, "lr": 0.00035, "seconds": 7732}
136
+ {"step": 3400, "latent_r2": 0.7761796987056733, "latent_loss": 0.6111948069930077, "action_l1_1step": 0.06964601352810859, "p_oracle": 0.4335, "lr": 0.00035, "seconds": 7787}
137
+ {"step": 3425, "latent_r2": 0.7820429944992066, "latent_loss": 0.6223228195309639, "action_l1_1step": 0.0706011563912034, "p_oracle": 0.42933333333333334, "lr": 0.00035, "seconds": 7842}
138
+ {"step": 3450, "latent_r2": 0.7797098636627198, "latent_loss": 0.6203371939063073, "action_l1_1step": 0.07205027900636196, "p_oracle": 0.4251666666666667, "lr": 0.00035, "seconds": 7897}
139
+ {"step": 3475, "latent_r2": 0.7855118703842163, "latent_loss": 0.6181580382585525, "action_l1_1step": 0.06919050041586161, "p_oracle": 0.42100000000000004, "lr": 0.00035, "seconds": 7953}
140
+ {"step": 3500, "latent_r2": 0.7928458964824676, "latent_loss": 0.6369060999155045, "action_l1_1step": 0.06608406916260719, "p_oracle": 0.4168333333333333, "lr": 0.00035, "seconds": 8009}
141
+ {"step": 3525, "latent_r2": 0.7683884835243225, "latent_loss": 0.6139556097984314, "action_l1_1step": 0.07105350229889154, "p_oracle": 0.4126666666666666, "lr": 0.00035, "seconds": 8071}
142
+ {"step": 3550, "latent_r2": 0.7833969140052796, "latent_loss": 0.6058767399191857, "action_l1_1step": 0.06696346387267113, "p_oracle": 0.4085, "lr": 0.00035, "seconds": 8127}
143
+ {"step": 3575, "latent_r2": 0.7911898422241211, "latent_loss": 0.6233505922555923, "action_l1_1step": 0.06619370233267546, "p_oracle": 0.4043333333333333, "lr": 0.00035, "seconds": 8188}
144
+ {"step": 3600, "latent_r2": 0.7876256030797958, "latent_loss": 0.6103871098160744, "action_l1_1step": 0.06809253782033921, "p_oracle": 0.40016666666666667, "lr": 0.00035, "seconds": 8245}
145
+ {"step": 3625, "latent_r2": 0.799996537566185, "latent_loss": 0.6482096749544144, "action_l1_1step": 0.06457525726407766, "p_oracle": 0.396, "lr": 0.00035, "seconds": 8300}
146
+ {"step": 3650, "latent_r2": 0.781835258603096, "latent_loss": 0.6084101837873459, "action_l1_1step": 0.06984655100852251, "p_oracle": 0.39183333333333337, "lr": 0.00035, "seconds": 8356}
147
+ {"step": 3675, "latent_r2": 0.7868277239799499, "latent_loss": 0.5993633410334587, "action_l1_1step": 0.06779305078089237, "p_oracle": 0.3876666666666667, "lr": 0.00035, "seconds": 8411}
148
+ {"step": 3700, "latent_r2": 0.7723805373907089, "latent_loss": 0.6142375504970551, "action_l1_1step": 0.07088966436684131, "p_oracle": 0.38349999999999995, "lr": 0.00035, "seconds": 8467}
149
+ {"step": 3725, "latent_r2": 0.7888615012168885, "latent_loss": 0.6201304364204406, "action_l1_1step": 0.06841443188488483, "p_oracle": 0.3793333333333333, "lr": 0.00035, "seconds": 8529}
150
+ {"step": 3750, "latent_r2": 0.7811824399232864, "latent_loss": 0.6167449948191642, "action_l1_1step": 0.06932285899296403, "p_oracle": 0.37516666666666665, "lr": 0.00035, "seconds": 8585}
151
+ {"step": 3775, "latent_r2": 0.7839460986852645, "latent_loss": 0.6215417069196701, "action_l1_1step": 0.06792689047753811, "p_oracle": 0.371, "lr": 0.00035, "seconds": 8647}
152
+ {"step": 3800, "latent_r2": 0.7928369909524917, "latent_loss": 0.6178420808911324, "action_l1_1step": 0.06561156805604697, "p_oracle": 0.36683333333333334, "lr": 0.00035, "seconds": 8705}
153
+ {"step": 3825, "latent_r2": 0.7967340183258057, "latent_loss": 0.615997394323349, "action_l1_1step": 0.06361336871981621, "p_oracle": 0.3626666666666667, "lr": 0.00035, "seconds": 8760}
154
+ {"step": 3850, "latent_r2": 0.7846212261915206, "latent_loss": 0.6122805014252662, "action_l1_1step": 0.06858722556382418, "p_oracle": 0.35850000000000004, "lr": 0.00035, "seconds": 8816}
155
+ {"step": 3875, "latent_r2": 0.7915326994657517, "latent_loss": 0.6036852422356606, "action_l1_1step": 0.06622193332761526, "p_oracle": 0.3543333333333333, "lr": 0.00035, "seconds": 8871}
156
+ {"step": 3900, "latent_r2": 0.7783168464899063, "latent_loss": 0.5936454403400421, "action_l1_1step": 0.06871210407465696, "p_oracle": 0.3501666666666666, "lr": 0.00035, "seconds": 8927}
157
+ {"step": 3925, "latent_r2": 0.7919314485788346, "latent_loss": 0.6185066986083985, "action_l1_1step": 0.06534332875162363, "p_oracle": 0.346, "lr": 0.00035, "seconds": 8988}
158
+ {"step": 3950, "latent_r2": 0.7907845759391785, "latent_loss": 0.6237410080432891, "action_l1_1step": 0.06523677162826061, "p_oracle": 0.3418333333333333, "lr": 0.00035, "seconds": 9048}
159
+ {"step": 3975, "latent_r2": 0.7886963409185409, "latent_loss": 0.612959044277668, "action_l1_1step": 0.06574399385601282, "p_oracle": 0.33766666666666667, "lr": 0.00035, "seconds": 9106}
160
+ {"step": 4000, "latent_r2": 0.786857264637947, "latent_loss": 0.6136607968807221, "action_l1_1step": 0.06566437348723411, "p_oracle": 0.3335, "lr": 0.00035, "seconds": 9161}
161
+ {"step": 4025, "latent_r2": 0.7863984274864196, "latent_loss": 0.6093593782186508, "action_l1_1step": 0.06729479447007179, "p_oracle": 0.32933333333333337, "lr": 0.00035, "seconds": 9231}
162
+ {"step": 4050, "latent_r2": 0.7916694939136505, "latent_loss": 0.6206255793571472, "action_l1_1step": 0.06608744889497757, "p_oracle": 0.3251666666666667, "lr": 0.00035, "seconds": 9286}
163
+ {"step": 4075, "latent_r2": 0.7843539577722549, "latent_loss": 0.5972763589024543, "action_l1_1step": 0.06773824289441109, "p_oracle": 0.32099999999999995, "lr": 0.00035, "seconds": 9345}
164
+ {"step": 4100, "latent_r2": 0.788744061589241, "latent_loss": 0.6162457552552223, "action_l1_1step": 0.06544768191874027, "p_oracle": 0.3168333333333333, "lr": 0.00035, "seconds": 9405}
165
+ {"step": 4125, "latent_r2": 0.8005705195665359, "latent_loss": 0.6223501682281494, "action_l1_1step": 0.06419703111052513, "p_oracle": 0.31266666666666665, "lr": 0.00035, "seconds": 9463}
166
+ {"step": 4150, "latent_r2": 0.7965727865695953, "latent_loss": 0.5921277669072151, "action_l1_1step": 0.0650469383224845, "p_oracle": 0.3085, "lr": 0.00035, "seconds": 9524}
167
+ {"step": 4175, "latent_r2": 0.8019447380304336, "latent_loss": 0.6097994121909142, "action_l1_1step": 0.06512975346297026, "p_oracle": 0.30433333333333334, "lr": 0.00035, "seconds": 9579}
168
+ {"step": 4200, "latent_r2": 0.7850868183374405, "latent_loss": 0.604014972448349, "action_l1_1step": 0.0659199906885624, "p_oracle": 0.3001666666666667, "lr": 0.00035, "seconds": 9634}
169
+ {"step": 4225, "latent_r2": 0.7995544427633285, "latent_loss": 0.6233579522371292, "action_l1_1step": 0.06406984660774469, "p_oracle": 0.29600000000000004, "lr": 0.00035, "seconds": 9692}
170
+ {"step": 4250, "latent_r2": 0.7868197596073151, "latent_loss": 0.6122223791480065, "action_l1_1step": 0.06876928877085448, "p_oracle": 0.2918333333333333, "lr": 0.00035, "seconds": 9752}
171
+ {"step": 4275, "latent_r2": 0.7910827642679215, "latent_loss": 0.6090673652291297, "action_l1_1step": 0.06658284209668636, "p_oracle": 0.2876666666666666, "lr": 0.00035, "seconds": 9808}
172
+ {"step": 4300, "latent_r2": 0.7898437976837158, "latent_loss": 0.6113437715172768, "action_l1_1step": 0.06526374649256468, "p_oracle": 0.2835, "lr": 0.00035, "seconds": 9870}
173
+ {"step": 4325, "latent_r2": 0.7993466484546662, "latent_loss": 0.5972581666707992, "action_l1_1step": 0.06515017788857222, "p_oracle": 0.2793333333333333, "lr": 0.00035, "seconds": 9925}
174
+ {"step": 4350, "latent_r2": 0.783736668229103, "latent_loss": 0.6034885793924332, "action_l1_1step": 0.06344170056283474, "p_oracle": 0.27516666666666667, "lr": 0.00035, "seconds": 9980}
175
+ {"step": 4375, "latent_r2": 0.7989558833837509, "latent_loss": 0.6056970554590225, "action_l1_1step": 0.061347454264760015, "p_oracle": 0.271, "lr": 0.00035, "seconds": 10036}
176
+ {"step": 4400, "latent_r2": 0.7995797306299209, "latent_loss": 0.6088775718212127, "action_l1_1step": 0.0630873128771782, "p_oracle": 0.26683333333333337, "lr": 0.00035, "seconds": 10103}
177
+ {"step": 4425, "latent_r2": 0.7920908385515213, "latent_loss": 0.5982371482253075, "action_l1_1step": 0.06397123087197543, "p_oracle": 0.2626666666666667, "lr": 0.00035, "seconds": 10159}
178
+ {"step": 4450, "latent_r2": 0.7894573783874512, "latent_loss": 0.6088614663481713, "action_l1_1step": 0.06524037774652243, "p_oracle": 0.25849999999999995, "lr": 0.00035, "seconds": 10220}
179
+ {"step": 4475, "latent_r2": 0.804866008758545, "latent_loss": 0.5954887980222702, "action_l1_1step": 0.0621812492236495, "p_oracle": 0.2543333333333333, "lr": 0.00035, "seconds": 10276}
180
+ {"step": 4500, "latent_r2": 0.7955138731002808, "latent_loss": 0.6128098902106285, "action_l1_1step": 0.06443659901618957, "p_oracle": 0.25016666666666665, "lr": 0.00035, "seconds": 10331}
181
+ {"step": 4525, "latent_r2": 0.7869938570261001, "latent_loss": 0.6009871089458465, "action_l1_1step": 0.06628871433436871, "p_oracle": 0.246, "lr": 0.00035, "seconds": 10387}
182
+ {"step": 4550, "latent_r2": 0.7941971069574356, "latent_loss": 0.6067679682374001, "action_l1_1step": 0.06393603842705488, "p_oracle": 0.24183333333333334, "lr": 0.00035, "seconds": 10450}
183
+ {"step": 4575, "latent_r2": 0.7839489871263504, "latent_loss": 0.6030887255072593, "action_l1_1step": 0.06620860140770674, "p_oracle": 0.2376666666666667, "lr": 0.00035, "seconds": 10510}
184
+ {"step": 4600, "latent_r2": 0.808998071551323, "latent_loss": 0.6224246713519096, "action_l1_1step": 0.061441829279065135, "p_oracle": 0.23350000000000004, "lr": 0.00035, "seconds": 10567}
185
+ {"step": 4625, "latent_r2": 0.7829880946874619, "latent_loss": 0.594524253308773, "action_l1_1step": 0.0625146509334445, "p_oracle": 0.22933333333333328, "lr": 0.00035, "seconds": 10622}
186
+ {"step": 4650, "latent_r2": 0.796906436085701, "latent_loss": 0.6172113284468651, "action_l1_1step": 0.06325606860220433, "p_oracle": 0.22516666666666663, "lr": 0.00035, "seconds": 10678}
187
+ {"step": 4675, "latent_r2": 0.7953584182262421, "latent_loss": 0.5997614005208015, "action_l1_1step": 0.0628564426675439, "p_oracle": 0.22099999999999997, "lr": 0.00035, "seconds": 10739}
188
+ {"step": 4700, "latent_r2": 0.793770101070404, "latent_loss": 0.6090272122621536, "action_l1_1step": 0.0648041020706296, "p_oracle": 0.21683333333333332, "lr": 0.00035, "seconds": 10796}
189
+ {"step": 4725, "latent_r2": 0.7929830479621888, "latent_loss": 0.5985545599460602, "action_l1_1step": 0.06519355230033398, "p_oracle": 0.21266666666666667, "lr": 0.00035, "seconds": 10857}
190
+ {"step": 4750, "latent_r2": 0.7944458770751953, "latent_loss": 0.5853330144286155, "action_l1_1step": 0.06521461550146342, "p_oracle": 0.20850000000000002, "lr": 0.00035, "seconds": 10913}
191
+ {"step": 4775, "latent_r2": 0.8004570335149765, "latent_loss": 0.6083453515172005, "action_l1_1step": 0.06322007467970252, "p_oracle": 0.20433333333333337, "lr": 0.00035, "seconds": 10968}
192
+ {"step": 4800, "latent_r2": 0.7883030241727829, "latent_loss": 0.6162597900629043, "action_l1_1step": 0.06689661581069231, "p_oracle": 0.20016666666666671, "lr": 0.00035, "seconds": 11024}
193
+ {"step": 4825, "latent_r2": 0.7972704613208771, "latent_loss": 0.6058529096841813, "action_l1_1step": 0.06314778011292219, "p_oracle": 0.19599999999999995, "lr": 0.00035, "seconds": 11079}
194
+ {"step": 4850, "latent_r2": 0.80690181016922, "latent_loss": 0.6175889819860458, "action_l1_1step": 0.06226144291460514, "p_oracle": 0.1918333333333333, "lr": 0.00035, "seconds": 11139}
195
+ {"step": 4875, "latent_r2": 0.7862539172172547, "latent_loss": 0.5992867958545685, "action_l1_1step": 0.06405656885355711, "p_oracle": 0.18766666666666665, "lr": 0.00035, "seconds": 11200}
196
+ {"step": 4900, "latent_r2": 0.7983426052331924, "latent_loss": 0.6091624310612679, "action_l1_1step": 0.06021495368331671, "p_oracle": 0.1835, "lr": 0.00035, "seconds": 11259}
197
+ {"step": 4925, "latent_r2": 0.7884333610534668, "latent_loss": 0.5919309094548225, "action_l1_1step": 0.06277744177728892, "p_oracle": 0.17933333333333334, "lr": 0.00035, "seconds": 11318}
198
+ {"step": 4950, "latent_r2": 0.7957078045606614, "latent_loss": 0.6352707517147064, "action_l1_1step": 0.06287265665829182, "p_oracle": 0.1751666666666667, "lr": 0.00035, "seconds": 11373}
199
+ {"step": 4975, "latent_r2": 0.7936656421422958, "latent_loss": 0.6131727659702301, "action_l1_1step": 0.06425862725824118, "p_oracle": 0.17100000000000004, "lr": 0.00035, "seconds": 11429}
200
+ {"step": 5000, "latent_r2": 0.8014556419849396, "latent_loss": 0.6169186267256737, "action_l1_1step": 0.06248650528490543, "p_oracle": 0.16683333333333328, "lr": 0.00035, "seconds": 11491}
201
+ {"step": 5025, "latent_r2": 0.7918646150827408, "latent_loss": 0.6025285392999649, "action_l1_1step": 0.06364741411060094, "p_oracle": 0.16266666666666663, "lr": 0.00035, "seconds": 11562}
202
+ {"step": 5050, "latent_r2": 0.7851763111352921, "latent_loss": 0.5899858191609383, "action_l1_1step": 0.0669641488417983, "p_oracle": 0.15849999999999997, "lr": 0.00035, "seconds": 11622}
203
+ {"step": 5075, "latent_r2": 0.7996404993534089, "latent_loss": 0.5919415602087974, "action_l1_1step": 0.06343118827790022, "p_oracle": 0.15433333333333332, "lr": 0.00035, "seconds": 11678}
204
+ {"step": 5100, "latent_r2": 0.7969009274244309, "latent_loss": 0.6135800668597221, "action_l1_1step": 0.06112902414053679, "p_oracle": 0.15016666666666667, "lr": 0.00035, "seconds": 11733}
205
+ {"step": 5125, "latent_r2": 0.7926060390472413, "latent_loss": 0.6018412598967552, "action_l1_1step": 0.0639836410433054, "p_oracle": 0.14600000000000002, "lr": 0.00035, "seconds": 11794}
206
+ {"step": 5150, "latent_r2": 0.7883092153072357, "latent_loss": 0.5828200304508209, "action_l1_1step": 0.06455569937825203, "p_oracle": 0.14183333333333337, "lr": 0.00035, "seconds": 11851}
207
+ {"step": 5175, "latent_r2": 0.7888491427898408, "latent_loss": 0.6089055088162422, "action_l1_1step": 0.06362905118614436, "p_oracle": 0.13766666666666671, "lr": 0.00035, "seconds": 11910}
208
+ {"step": 5200, "latent_r2": 0.7942286765575409, "latent_loss": 0.5932871896028519, "action_l1_1step": 0.06266421414911746, "p_oracle": 0.13349999999999995, "lr": 0.00035, "seconds": 11969}
209
+ {"step": 5225, "latent_r2": 0.7961902624368667, "latent_loss": 0.5871735671162606, "action_l1_1step": 0.06390733931213617, "p_oracle": 0.1293333333333333, "lr": 0.00035, "seconds": 12024}
210
+ {"step": 5250, "latent_r2": 0.7955133110284806, "latent_loss": 0.5822257074713707, "action_l1_1step": 0.06267236541956663, "p_oracle": 0.12516666666666665, "lr": 0.00035, "seconds": 12079}
211
+ {"step": 5275, "latent_r2": 0.7972299301624298, "latent_loss": 0.5951688241958618, "action_l1_1step": 0.06249260757118463, "p_oracle": 0.121, "lr": 0.00035, "seconds": 12138}
212
+ {"step": 5300, "latent_r2": 0.7884190744161605, "latent_loss": 0.5874868324398994, "action_l1_1step": 0.06380679506808519, "p_oracle": 0.11683333333333334, "lr": 0.00035, "seconds": 12199}
213
+ {"step": 5325, "latent_r2": 0.7943149560689926, "latent_loss": 0.5838056620955467, "action_l1_1step": 0.06318315472453832, "p_oracle": 0.11266666666666669, "lr": 0.00035, "seconds": 12259}
214
+ {"step": 5350, "latent_r2": 0.7850027644634247, "latent_loss": 0.5846791037917137, "action_l1_1step": 0.06277959946542978, "p_oracle": 0.10850000000000004, "lr": 0.00035, "seconds": 12314}
215
+ {"step": 5375, "latent_r2": 0.7956074130535126, "latent_loss": 0.5853462043404579, "action_l1_1step": 0.06349413327872754, "p_oracle": 0.10433333333333328, "lr": 0.00035, "seconds": 12369}
216
+ {"step": 5400, "latent_r2": 0.8042709946632385, "latent_loss": 0.607983073592186, "action_l1_1step": 0.05911547977477312, "p_oracle": 0.10016666666666663, "lr": 0.00035, "seconds": 12431}
217
+ {"step": 5425, "latent_r2": 0.7978731560707092, "latent_loss": 0.5991579231619835, "action_l1_1step": 0.06327659502625466, "p_oracle": 0.09599999999999997, "lr": 0.00035, "seconds": 12492}
218
+ {"step": 5450, "latent_r2": 0.792947815656662, "latent_loss": 0.5997323015332222, "action_l1_1step": 0.06124070696532726, "p_oracle": 0.09183333333333332, "lr": 0.00035, "seconds": 12550}
219
+ {"step": 5475, "latent_r2": 0.8032525372505188, "latent_loss": 0.6103581768274308, "action_l1_1step": 0.060030469428747894, "p_oracle": 0.08766666666666667, "lr": 0.00035, "seconds": 12605}
220
+ {"step": 5500, "latent_r2": 0.7885139542818069, "latent_loss": 0.5967560276389122, "action_l1_1step": 0.06113441940397024, "p_oracle": 0.08350000000000002, "lr": 0.00035, "seconds": 12660}
221
+ {"step": 5525, "latent_r2": 0.8046418642997741, "latent_loss": 0.5959299603104591, "action_l1_1step": 0.06003856405615807, "p_oracle": 0.07933333333333337, "lr": 0.00035, "seconds": 12715}
222
+ {"step": 5550, "latent_r2": 0.7945068275928497, "latent_loss": 0.5951017966866493, "action_l1_1step": 0.06375149685889482, "p_oracle": 0.07516666666666671, "lr": 0.00035, "seconds": 12770}
223
+ {"step": 5575, "latent_r2": 0.8036445432901382, "latent_loss": 0.5873084768652916, "action_l1_1step": 0.05912935551255941, "p_oracle": 0.07099999999999995, "lr": 0.00035, "seconds": 12825}
224
+ {"step": 5600, "latent_r2": 0.7912298947572708, "latent_loss": 0.5819542869925499, "action_l1_1step": 0.06259454280138016, "p_oracle": 0.0668333333333333, "lr": 0.00035, "seconds": 12880}
225
+ {"step": 5625, "latent_r2": 0.8002019602060318, "latent_loss": 0.6090871727466584, "action_l1_1step": 0.06093582190573216, "p_oracle": 0.06266666666666665, "lr": 0.00035, "seconds": 12936}
226
+ {"step": 5650, "latent_r2": 0.7922103393077851, "latent_loss": 0.5854140824079513, "action_l1_1step": 0.06161205071955919, "p_oracle": 0.058499999999999996, "lr": 0.00035, "seconds": 12994}
227
+ {"step": 5675, "latent_r2": 0.7942759555578232, "latent_loss": 0.5887929859757424, "action_l1_1step": 0.06126743827015162, "p_oracle": 0.054333333333333345, "lr": 0.00035, "seconds": 13049}
228
+ {"step": 5700, "latent_r2": 0.8028414046764374, "latent_loss": 0.5885099595785142, "action_l1_1step": 0.06087498962879181, "p_oracle": 0.05016666666666669, "lr": 0.00035, "seconds": 13104}
229
+ {"step": 5725, "latent_r2": 0.8010178631544114, "latent_loss": 0.5873246350884438, "action_l1_1step": 0.06297566886991263, "p_oracle": 0.04600000000000004, "lr": 0.00035, "seconds": 13159}
230
+ {"step": 5750, "latent_r2": 0.8022740286588669, "latent_loss": 0.5815244722366333, "action_l1_1step": 0.06023975141346455, "p_oracle": 0.04183333333333328, "lr": 0.00035, "seconds": 13214}
231
+ {"step": 5775, "latent_r2": 0.803556889295578, "latent_loss": 0.5986636537313461, "action_l1_1step": 0.06192242845892906, "p_oracle": 0.037666666666666626, "lr": 0.00035, "seconds": 13270}
232
+ {"step": 5800, "latent_r2": 0.8102494430541992, "latent_loss": 0.5986594572663307, "action_l1_1step": 0.06114773288369179, "p_oracle": 0.033499999999999974, "lr": 0.00035, "seconds": 13325}
233
+ {"step": 5825, "latent_r2": 0.8070320910215378, "latent_loss": 0.5824254193902015, "action_l1_1step": 0.06048277843743563, "p_oracle": 0.029333333333333322, "lr": 0.00035, "seconds": 13380}
234
+ {"step": 5850, "latent_r2": 0.7881750237941741, "latent_loss": 0.5957807025313377, "action_l1_1step": 0.06122187323868275, "p_oracle": 0.02516666666666667, "lr": 0.00035, "seconds": 13435}
235
+ {"step": 5875, "latent_r2": 0.8049837100505829, "latent_loss": 0.5904800093173981, "action_l1_1step": 0.058744539860636, "p_oracle": 0.02100000000000002, "lr": 0.00035, "seconds": 13490}
236
+ {"step": 5900, "latent_r2": 0.8079044955968857, "latent_loss": 0.589237841963768, "action_l1_1step": 0.06010661032050848, "p_oracle": 0.016833333333333367, "lr": 0.00035, "seconds": 13545}
237
+ {"step": 5925, "latent_r2": 0.7904721957445144, "latent_loss": 0.5872637352347374, "action_l1_1step": 0.06366982631385326, "p_oracle": 0.012666666666666715, "lr": 0.00035, "seconds": 13601}
238
+ {"step": 5950, "latent_r2": 0.7966641372442246, "latent_loss": 0.5966090878844261, "action_l1_1step": 0.05999276127666235, "p_oracle": 0.008499999999999952, "lr": 0.00035, "seconds": 13657}
239
+ {"step": 5975, "latent_r2": 0.7936114120483398, "latent_loss": 0.5741517236828804, "action_l1_1step": 0.06035854388028383, "p_oracle": 0.0043333333333333, "lr": 0.00035, "seconds": 13713}
240
+ {"step": 6000, "latent_r2": 0.7881263190507889, "latent_loss": 0.5773379844427109, "action_l1_1step": 0.06289160534739495, "p_oracle": 0.0001666666666666483, "lr": 0.00035, "seconds": 13768}
241
+ {"step": 6025, "latent_r2": 0.8022233378887177, "latent_loss": 0.5753428742289544, "action_l1_1step": 0.060623404718935486, "p_oracle": 0.0, "lr": 0.00035, "seconds": 13836}
242
+ {"step": 6050, "latent_r2": 0.8077007853984832, "latent_loss": 0.5992327037453652, "action_l1_1step": 0.05914801090955734, "p_oracle": 0.0, "lr": 0.00035, "seconds": 13899}
243
+ {"step": 6075, "latent_r2": 0.8150954157114029, "latent_loss": 0.6059704011678696, "action_l1_1step": 0.05521045984700322, "p_oracle": 0.0, "lr": 0.00035, "seconds": 13962}
244
+ {"step": 6100, "latent_r2": 0.8040972304344177, "latent_loss": 0.5864256432652474, "action_l1_1step": 0.05846527807414532, "p_oracle": 0.0, "lr": 0.00035, "seconds": 14024}
245
+ {"step": 6125, "latent_r2": 0.8038929730653763, "latent_loss": 0.5719444161653519, "action_l1_1step": 0.05723807662725448, "p_oracle": 0.0, "lr": 0.00035, "seconds": 14086}
246
+ {"step": 6150, "latent_r2": 0.8074007397890091, "latent_loss": 0.5793657919764519, "action_l1_1step": 0.058100131899118425, "p_oracle": 0.0, "lr": 0.00035, "seconds": 14149}
247
+ {"step": 6175, "latent_r2": 0.8132085877656937, "latent_loss": 0.6012835240364075, "action_l1_1step": 0.05933292835950851, "p_oracle": 0.0, "lr": 0.00035, "seconds": 14205}
248
+ {"step": 6200, "latent_r2": 0.7974764376878738, "latent_loss": 0.6048667371273041, "action_l1_1step": 0.060365595016628505, "p_oracle": 0.0, "lr": 0.00035, "seconds": 14261}
249
+ {"step": 6225, "latent_r2": 0.8045081555843353, "latent_loss": 0.6038799372315407, "action_l1_1step": 0.06018066298216582, "p_oracle": 0.0, "lr": 0.00035, "seconds": 14316}
250
+ {"step": 6250, "latent_r2": 0.797462962269783, "latent_loss": 0.5747735950350762, "action_l1_1step": 0.0603871551901102, "p_oracle": 0.0, "lr": 0.00035, "seconds": 14372}
251
+ {"step": 6275, "latent_r2": 0.8021433997154236, "latent_loss": 0.5930037885904312, "action_l1_1step": 0.058883981481194496, "p_oracle": 0.0, "lr": 0.00035, "seconds": 14434}
252
+ {"step": 6300, "latent_r2": 0.8032728958129883, "latent_loss": 0.5959905225038529, "action_l1_1step": 0.058413064628839495, "p_oracle": 0.0, "lr": 0.00035, "seconds": 14497}
253
+ {"step": 6325, "latent_r2": 0.8025972265005111, "latent_loss": 0.5903766071796417, "action_l1_1step": 0.05841266792267561, "p_oracle": 0.0, "lr": 0.00035, "seconds": 14560}
254
+ {"step": 6350, "latent_r2": 0.7876634460687637, "latent_loss": 0.5772294735908509, "action_l1_1step": 0.06102369412779808, "p_oracle": 0.0, "lr": 0.00035, "seconds": 14622}
255
+ {"step": 6375, "latent_r2": 0.8054523992538453, "latent_loss": 0.6024782225489617, "action_l1_1step": 0.05912519484758377, "p_oracle": 0.0, "lr": 0.00035, "seconds": 14685}
256
+ {"step": 6400, "latent_r2": 0.803035392165184, "latent_loss": 0.5911752673983574, "action_l1_1step": 0.056522738989442586, "p_oracle": 0.0, "lr": 0.00035, "seconds": 14747}
257
+ {"step": 6425, "latent_r2": 0.800202654004097, "latent_loss": 0.5937372735142707, "action_l1_1step": 0.05972881864756346, "p_oracle": 0.0, "lr": 0.00035, "seconds": 14803}
258
+ {"step": 6450, "latent_r2": 0.7943079257011414, "latent_loss": 0.5813764464855194, "action_l1_1step": 0.06078251946717501, "p_oracle": 0.0, "lr": 0.00035, "seconds": 14858}
259
+ {"step": 6475, "latent_r2": 0.7951201397180557, "latent_loss": 0.5854175439476967, "action_l1_1step": 0.06033561319112778, "p_oracle": 0.0, "lr": 0.00035, "seconds": 14914}
260
+ {"step": 6500, "latent_r2": 0.804198772907257, "latent_loss": 0.5761619660258294, "action_l1_1step": 0.0589245225302875, "p_oracle": 0.0, "lr": 0.00035, "seconds": 14970}
261
+ {"step": 6525, "latent_r2": 0.8079917526245117, "latent_loss": 0.5930726009607316, "action_l1_1step": 0.059818534813821315, "p_oracle": 0.0, "lr": 0.00035, "seconds": 15032}
262
+ {"step": 6550, "latent_r2": 0.8031141483783721, "latent_loss": 0.5887448444962502, "action_l1_1step": 0.0608715495467186, "p_oracle": 0.0, "lr": 0.00035, "seconds": 15094}
263
+ {"step": 6575, "latent_r2": 0.8088588482141494, "latent_loss": 0.5995187282562255, "action_l1_1step": 0.05939747393131256, "p_oracle": 0.0, "lr": 0.00035, "seconds": 15157}
264
+ {"step": 6600, "latent_r2": 0.8011348402500152, "latent_loss": 0.5783641520142555, "action_l1_1step": 0.0585862547904253, "p_oracle": 0.0, "lr": 0.00035, "seconds": 15220}
265
+ {"step": 6625, "latent_r2": 0.7987886452674866, "latent_loss": 0.576362911760807, "action_l1_1step": 0.058962227627635005, "p_oracle": 0.0, "lr": 0.00035, "seconds": 15283}
266
+ {"step": 6650, "latent_r2": 0.7974434286355973, "latent_loss": 0.5752878540754318, "action_l1_1step": 0.05932116009294987, "p_oracle": 0.0, "lr": 0.00035, "seconds": 15343}
267
+ {"step": 6675, "latent_r2": 0.8073092728853226, "latent_loss": 0.5846043381094933, "action_l1_1step": 0.05597912482917309, "p_oracle": 0.0, "lr": 0.00035, "seconds": 15400}
268
+ {"step": 6700, "latent_r2": 0.8136991131305694, "latent_loss": 0.5892412468791008, "action_l1_1step": 0.057227915413677694, "p_oracle": 0.0, "lr": 0.00035, "seconds": 15456}
269
+ {"step": 6725, "latent_r2": 0.800301103591919, "latent_loss": 0.5817480161786079, "action_l1_1step": 0.0575292032584548, "p_oracle": 0.0, "lr": 0.00035, "seconds": 15512}
270
+ {"step": 6750, "latent_r2": 0.8108054709434509, "latent_loss": 0.5788264897465706, "action_l1_1step": 0.057511185593903065, "p_oracle": 0.0, "lr": 0.00035, "seconds": 15568}
271
+ {"step": 6775, "latent_r2": 0.8121240186691284, "latent_loss": 0.5940721189975738, "action_l1_1step": 0.05766133047640323, "p_oracle": 0.0, "lr": 0.00035, "seconds": 15642}
272
+ {"step": 6800, "latent_r2": 0.8135707223415375, "latent_loss": 0.5839469322562217, "action_l1_1step": 0.05583127018064261, "p_oracle": 0.0, "lr": 0.00035, "seconds": 15708}
273
+ {"step": 6825, "latent_r2": 0.8052540397644044, "latent_loss": 0.5809315878152848, "action_l1_1step": 0.05731205180287361, "p_oracle": 0.0, "lr": 0.00035, "seconds": 15781}
274
+ {"step": 6850, "latent_r2": 0.7975669920444488, "latent_loss": 0.570467183291912, "action_l1_1step": 0.05810625921934843, "p_oracle": 0.0, "lr": 0.00035, "seconds": 15848}
275
+ {"step": 6875, "latent_r2": 0.8151797872781753, "latent_loss": 0.5865707507729531, "action_l1_1step": 0.05951883723959327, "p_oracle": 0.0, "lr": 0.00035, "seconds": 15922}
276
+ {"step": 6900, "latent_r2": 0.8124734485149383, "latent_loss": 0.5794597750902176, "action_l1_1step": 0.05653207175433636, "p_oracle": 0.0, "lr": 0.00035, "seconds": 15992}
277
+ {"step": 6925, "latent_r2": 0.810906040072441, "latent_loss": 0.588465250134468, "action_l1_1step": 0.055767975114285946, "p_oracle": 0.0, "lr": 0.00035, "seconds": 16062}
278
+ {"step": 6950, "latent_r2": 0.8049701911211014, "latent_loss": 0.5976803091168403, "action_l1_1step": 0.0578295286372304, "p_oracle": 0.0, "lr": 0.00035, "seconds": 16130}
279
+ {"step": 6975, "latent_r2": 0.798417090177536, "latent_loss": 0.5669166225194932, "action_l1_1step": 0.05753637012094259, "p_oracle": 0.0, "lr": 0.00035, "seconds": 16192}
280
+ {"step": 7000, "latent_r2": 0.8049777030944825, "latent_loss": 0.5777972647547722, "action_l1_1step": 0.057033692635595795, "p_oracle": 0.0, "lr": 0.00035, "seconds": 16248}
281
+ {"step": 7025, "latent_r2": 0.8164282470941544, "latent_loss": 0.5990325152873993, "action_l1_1step": 0.0556571739539504, "p_oracle": 0.0, "lr": 0.00035, "seconds": 16316}
282
+ {"step": 7050, "latent_r2": 0.8001994919776917, "latent_loss": 0.5689744198322296, "action_l1_1step": 0.05597706411033869, "p_oracle": 0.0, "lr": 0.00035, "seconds": 16378}
283
+ {"step": 7075, "latent_r2": 0.802187573313713, "latent_loss": 0.5790019929409027, "action_l1_1step": 0.059628762453794476, "p_oracle": 0.0, "lr": 0.00035, "seconds": 16441}
284
+ {"step": 7100, "latent_r2": 0.8090643411874772, "latent_loss": 0.5909631362557412, "action_l1_1step": 0.056084377765655516, "p_oracle": 0.0, "lr": 0.00035, "seconds": 16503}
285
+ {"step": 7125, "latent_r2": 0.8111424720287324, "latent_loss": 0.5780019038915634, "action_l1_1step": 0.05579276025295257, "p_oracle": 0.0, "lr": 0.00035, "seconds": 16565}
286
+ {"step": 7150, "latent_r2": 0.7987326043844223, "latent_loss": 0.5777959442138672, "action_l1_1step": 0.05945072624832392, "p_oracle": 0.0, "lr": 0.00035, "seconds": 16623}
287
+ {"step": 7175, "latent_r2": 0.8229126405715942, "latent_loss": 0.5788129696249962, "action_l1_1step": 0.05796510517597198, "p_oracle": 0.0, "lr": 0.00035, "seconds": 16678}
288
+ {"step": 7200, "latent_r2": 0.8012265247106553, "latent_loss": 0.5705946886539459, "action_l1_1step": 0.057277859188616276, "p_oracle": 0.0, "lr": 0.00035, "seconds": 16734}
289
+ {"step": 7225, "latent_r2": 0.8078897333145142, "latent_loss": 0.5870501807332039, "action_l1_1step": 0.05527037724852562, "p_oracle": 0.0, "lr": 0.00035, "seconds": 16789}
290
+ {"step": 7250, "latent_r2": 0.7996218174695968, "latent_loss": 0.5777967569231987, "action_l1_1step": 0.05781579826027155, "p_oracle": 0.0, "lr": 0.00035, "seconds": 16844}
291
+ {"step": 7275, "latent_r2": 0.8091505175828934, "latent_loss": 0.5811550334095955, "action_l1_1step": 0.056729919835925104, "p_oracle": 0.0, "lr": 0.00035, "seconds": 16899}
292
+ {"step": 7300, "latent_r2": 0.7939720642566681, "latent_loss": 0.583445979654789, "action_l1_1step": 0.05704528346657753, "p_oracle": 0.0, "lr": 0.00035, "seconds": 16955}
293
+ {"step": 7325, "latent_r2": 0.7977289205789566, "latent_loss": 0.5625314554572105, "action_l1_1step": 0.05892928579822183, "p_oracle": 0.0, "lr": 0.00035, "seconds": 17010}
294
+ {"step": 7350, "latent_r2": 0.8078835541009903, "latent_loss": 0.5722463154792785, "action_l1_1step": 0.06008763741701841, "p_oracle": 0.0, "lr": 0.00035, "seconds": 17065}
295
+ {"step": 7375, "latent_r2": 0.809339205622673, "latent_loss": 0.5822340101003647, "action_l1_1step": 0.05550679724663496, "p_oracle": 0.0, "lr": 0.00035, "seconds": 17120}
296
+ {"step": 7400, "latent_r2": 0.8215412867069244, "latent_loss": 0.5872210904955864, "action_l1_1step": 0.057242787592113015, "p_oracle": 0.0, "lr": 0.00035, "seconds": 17175}
297
+ {"step": 7425, "latent_r2": 0.8166622883081436, "latent_loss": 0.5767869901657104, "action_l1_1step": 0.05499388501048088, "p_oracle": 0.0, "lr": 0.00035, "seconds": 17230}
298
+ {"step": 7450, "latent_r2": 0.8156259751319885, "latent_loss": 0.5797564995288849, "action_l1_1step": 0.0555664562061429, "p_oracle": 0.0, "lr": 0.00035, "seconds": 17286}
299
+ {"step": 7475, "latent_r2": 0.8058321142196655, "latent_loss": 0.5666031813621522, "action_l1_1step": 0.05536375615745783, "p_oracle": 0.0, "lr": 0.00035, "seconds": 17341}
300
+ {"step": 7500, "latent_r2": 0.8038906985521317, "latent_loss": 0.5673196360468864, "action_l1_1step": 0.05736745350062847, "p_oracle": 0.0, "lr": 0.00035, "seconds": 17400}
301
+ {"step": 7525, "latent_r2": 0.8144581341743469, "latent_loss": 0.57291609197855, "action_l1_1step": 0.055020355992019175, "p_oracle": 0.0, "lr": 0.00035, "seconds": 17459}
302
+ {"step": 7550, "latent_r2": 0.7907074862718582, "latent_loss": 0.5703436440229416, "action_l1_1step": 0.05847824078053236, "p_oracle": 0.0, "lr": 0.00035, "seconds": 17517}
303
+ {"step": 7575, "latent_r2": 0.8073119759559632, "latent_loss": 0.5797794279456139, "action_l1_1step": 0.05662359280511737, "p_oracle": 0.0, "lr": 0.00035, "seconds": 17575}
304
+ {"step": 7600, "latent_r2": 0.7932431018352508, "latent_loss": 0.5668445393443108, "action_l1_1step": 0.058680830616503955, "p_oracle": 0.0, "lr": 0.00035, "seconds": 17632}
305
+ {"step": 7625, "latent_r2": 0.8176644372940064, "latent_loss": 0.5830393159389495, "action_l1_1step": 0.05652373794466257, "p_oracle": 0.0, "lr": 0.00035, "seconds": 17690}
306
+ {"step": 7650, "latent_r2": 0.8112465327978134, "latent_loss": 0.5987412047386169, "action_l1_1step": 0.058959836214780806, "p_oracle": 0.0, "lr": 0.00035, "seconds": 17748}
307
+ {"step": 7675, "latent_r2": 0.8162561702728272, "latent_loss": 0.5688181197643281, "action_l1_1step": 0.05344455774873495, "p_oracle": 0.0, "lr": 0.00035, "seconds": 17806}
308
+ {"step": 7700, "latent_r2": 0.800448140501976, "latent_loss": 0.5917912071943283, "action_l1_1step": 0.05766189368441701, "p_oracle": 0.0, "lr": 0.00035, "seconds": 17864}
309
+ {"step": 7725, "latent_r2": 0.8192344009876251, "latent_loss": 0.5813369926810265, "action_l1_1step": 0.05380120266228914, "p_oracle": 0.0, "lr": 0.00035, "seconds": 17922}
310
+ {"step": 7750, "latent_r2": 0.8152891045808792, "latent_loss": 0.5635042408108711, "action_l1_1step": 0.05622835021466017, "p_oracle": 0.0, "lr": 0.00035, "seconds": 17979}
311
+ {"step": 7775, "latent_r2": 0.8060800611972809, "latent_loss": 0.5654846549034118, "action_l1_1step": 0.05531188864260912, "p_oracle": 0.0, "lr": 0.00035, "seconds": 18037}
312
+ {"step": 7800, "latent_r2": 0.8100943183898925, "latent_loss": 0.5661415529251098, "action_l1_1step": 0.054524486660957334, "p_oracle": 0.0, "lr": 0.00035, "seconds": 18095}
313
+ {"step": 7825, "latent_r2": 0.8084219431877137, "latent_loss": 0.5921964502334595, "action_l1_1step": 0.05601335298269987, "p_oracle": 0.0, "lr": 0.00035, "seconds": 18153}
314
+ {"step": 7850, "latent_r2": 0.8098576819896698, "latent_loss": 0.5770860975980758, "action_l1_1step": 0.05622516632080078, "p_oracle": 0.0, "lr": 0.00035, "seconds": 18211}
315
+ {"step": 7875, "latent_r2": 0.811309466958046, "latent_loss": 0.5755506843328476, "action_l1_1step": 0.054447191432118416, "p_oracle": 0.0, "lr": 0.00035, "seconds": 18269}
316
+ {"step": 7900, "latent_r2": 0.8187490224838256, "latent_loss": 0.5715221232175827, "action_l1_1step": 0.055211045425385234, "p_oracle": 0.0, "lr": 0.00035, "seconds": 18327}
317
+ {"step": 7925, "latent_r2": 0.8165089237689972, "latent_loss": 0.5838001933693886, "action_l1_1step": 0.0539763356000185, "p_oracle": 0.0, "lr": 0.00035, "seconds": 18386}
318
+ {"step": 7950, "latent_r2": 0.8127135509252548, "latent_loss": 0.5748524415493012, "action_l1_1step": 0.054592755381017925, "p_oracle": 0.0, "lr": 0.00035, "seconds": 18444}
319
+ {"step": 7975, "latent_r2": 0.8246252244710922, "latent_loss": 0.5863270482420921, "action_l1_1step": 0.052450739350169896, "p_oracle": 0.0, "lr": 0.00035, "seconds": 18501}
320
+ {"step": 8000, "latent_r2": 0.8123499572277069, "latent_loss": 0.581271855533123, "action_l1_1step": 0.05673086538910866, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 18559}
321
+ {"step": 8025, "latent_r2": 0.8055578684806823, "latent_loss": 0.5747324591875076, "action_l1_1step": 0.05365774609148502, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 18627}
322
+ {"step": 8050, "latent_r2": 0.8024893754720688, "latent_loss": 0.527537003159523, "action_l1_1step": 0.05077098364010453, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 18685}
323
+ {"step": 8075, "latent_r2": 0.8221979308128357, "latent_loss": 0.5500173225998879, "action_l1_1step": 0.045087579321116206, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 18743}
324
+ {"step": 8100, "latent_r2": 0.8101543140411377, "latent_loss": 0.537232109606266, "action_l1_1step": 0.04791993012651801, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 18801}
325
+ {"step": 8125, "latent_r2": 0.8237059158086777, "latent_loss": 0.5524992075562477, "action_l1_1step": 0.04584508709609508, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 18859}
326
+ {"step": 8150, "latent_r2": 0.82177090883255, "latent_loss": 0.5527615353465081, "action_l1_1step": 0.04549948846921325, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 18919}
327
+ {"step": 8175, "latent_r2": 0.824982241988182, "latent_loss": 0.5548144784569741, "action_l1_1step": 0.04429756946861744, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 18977}
328
+ {"step": 8200, "latent_r2": 0.8150716853141785, "latent_loss": 0.5458153402805328, "action_l1_1step": 0.04591913195326924, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 19036}
329
+ {"step": 8225, "latent_r2": 0.8242568123340607, "latent_loss": 0.5677298182249069, "action_l1_1step": 0.04521731324493885, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 19094}
330
+ {"step": 8250, "latent_r2": 0.8246245151758194, "latent_loss": 0.5418720069527626, "action_l1_1step": 0.04499660540372133, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 19152}
331
+ {"step": 8275, "latent_r2": 0.8194357442855835, "latent_loss": 0.5428470680117607, "action_l1_1step": 0.04418135337531567, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 19210}
332
+ {"step": 8300, "latent_r2": 0.8222680985927582, "latent_loss": 0.539847317636013, "action_l1_1step": 0.04489687703549862, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 19268}
333
+ {"step": 8325, "latent_r2": 0.8134865009784699, "latent_loss": 0.5478072369098663, "action_l1_1step": 0.044595440588891505, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 19326}
334
+ {"step": 8350, "latent_r2": 0.8152775031328201, "latent_loss": 0.5382382977008819, "action_l1_1step": 0.045299050994217396, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 19384}
335
+ {"step": 8375, "latent_r2": 0.8354583930969238, "latent_loss": 0.5383406946063042, "action_l1_1step": 0.0417259725369513, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 19442}
336
+ {"step": 8400, "latent_r2": 0.8260195100307465, "latent_loss": 0.5350059434771538, "action_l1_1step": 0.04347769407555461, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 19500}
337
+ {"step": 8425, "latent_r2": 0.8211366635560989, "latent_loss": 0.5491492238640785, "action_l1_1step": 0.04231045056134462, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 19559}
338
+ {"step": 8450, "latent_r2": 0.8262718260288239, "latent_loss": 0.5473422518372536, "action_l1_1step": 0.04324500195682049, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 19617}
339
+ {"step": 8475, "latent_r2": 0.8170499300956726, "latent_loss": 0.5340044225752354, "action_l1_1step": 0.045029176957905294, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 19675}
340
+ {"step": 8500, "latent_r2": 0.824539869427681, "latent_loss": 0.5262641298770905, "action_l1_1step": 0.0421377157792449, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 19734}
341
+ {"step": 8525, "latent_r2": 0.8189191341400146, "latent_loss": 0.5455016830563545, "action_l1_1step": 0.04321740165352821, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 19794}
342
+ {"step": 8550, "latent_r2": 0.830089830160141, "latent_loss": 0.5227985447645187, "action_l1_1step": 0.043398691490292546, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 19856}
343
+ {"step": 8575, "latent_r2": 0.8334721779823303, "latent_loss": 0.5401770859956742, "action_l1_1step": 0.04251384912058711, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 19918}
344
+ {"step": 8600, "latent_r2": 0.821257296204567, "latent_loss": 0.5527034097909928, "action_l1_1step": 0.04451778443530202, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 19979}
345
+ {"step": 8625, "latent_r2": 0.8169907850027084, "latent_loss": 0.5429641491174698, "action_l1_1step": 0.04469767337664962, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 20042}
346
+ {"step": 8650, "latent_r2": 0.834189355969429, "latent_loss": 0.5222624260187149, "action_l1_1step": 0.04269539779052138, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 20105}
347
+ {"step": 8675, "latent_r2": 0.8121450680494309, "latent_loss": 0.5308627614378929, "action_l1_1step": 0.04479842133820057, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 20164}
348
+ {"step": 8700, "latent_r2": 0.8284182280302048, "latent_loss": 0.5277395576238633, "action_l1_1step": 0.04285585321485996, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 20225}
349
+ {"step": 8725, "latent_r2": 0.8307010114192963, "latent_loss": 0.542295032441616, "action_l1_1step": 0.04394227407872677, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 20287}
350
+ {"step": 8750, "latent_r2": 0.8246931850910186, "latent_loss": 0.5413480433821678, "action_l1_1step": 0.04190945316106081, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 20347}
351
+ {"step": 8775, "latent_r2": 0.8341583395004273, "latent_loss": 0.5545538583397865, "action_l1_1step": 0.04064751731231809, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 20412}
352
+ {"step": 8800, "latent_r2": 0.820474426150322, "latent_loss": 0.538009772002697, "action_l1_1step": 0.04198850078508258, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 20473}
353
+ {"step": 8825, "latent_r2": 0.8210324656963348, "latent_loss": 0.5328013888001442, "action_l1_1step": 0.04334873342886567, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 20536}
354
+ {"step": 8850, "latent_r2": 0.8271902930736542, "latent_loss": 0.544119111597538, "action_l1_1step": 0.043635978121310474, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 20594}
355
+ {"step": 8875, "latent_r2": 0.817283826470375, "latent_loss": 0.540978969335556, "action_l1_1step": 0.04310807960107923, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 20653}
356
+ {"step": 8900, "latent_r2": 0.8270735722780228, "latent_loss": 0.5426767668128014, "action_l1_1step": 0.0426302501372993, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 20711}
357
+ {"step": 8925, "latent_r2": 0.8206800377368927, "latent_loss": 0.5357730436325073, "action_l1_1step": 0.04283970199525356, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 20770}
358
+ {"step": 8950, "latent_r2": 0.8332424300909043, "latent_loss": 0.5433872002363205, "action_l1_1step": 0.0424224847368896, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 20831}
359
+ {"step": 8975, "latent_r2": 0.8171015882492065, "latent_loss": 0.5352970844507218, "action_l1_1step": 0.04524309672415257, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 20893}
360
+ {"step": 9000, "latent_r2": 0.8238437849283219, "latent_loss": 0.5332616299390793, "action_l1_1step": 0.043290225807577375, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 20955}
361
+ {"step": 9025, "latent_r2": 0.8264462745189667, "latent_loss": 0.5391098135709762, "action_l1_1step": 0.04148501120507717, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 21025}
362
+ {"step": 9050, "latent_r2": 0.8337821835279464, "latent_loss": 0.5480040457844734, "action_l1_1step": 0.04314354522153735, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 21088}
363
+ {"step": 9075, "latent_r2": 0.8240514606237411, "latent_loss": 0.5295933926105499, "action_l1_1step": 0.043377626091241836, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 21152}
364
+ {"step": 9100, "latent_r2": 0.8253245848417282, "latent_loss": 0.5395922234654427, "action_l1_1step": 0.042524919472634794, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 21211}
365
+ {"step": 9125, "latent_r2": 0.8311049914360047, "latent_loss": 0.535415962934494, "action_l1_1step": 0.04360181853175163, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 21273}
366
+ {"step": 9150, "latent_r2": 0.8285695803165436, "latent_loss": 0.5188460767269134, "action_l1_1step": 0.04146268771961331, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 21336}
367
+ {"step": 9175, "latent_r2": 0.8236466920375825, "latent_loss": 0.5399910044670105, "action_l1_1step": 0.04288518441841006, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 21398}
368
+ {"step": 9200, "latent_r2": 0.8332341873645782, "latent_loss": 0.5340766912698746, "action_l1_1step": 0.04090960819274187, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 21460}
369
+ {"step": 9225, "latent_r2": 0.8256381165981292, "latent_loss": 0.5291493651270867, "action_l1_1step": 0.041884247586131094, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 21520}
370
+ {"step": 9250, "latent_r2": 0.8169253659248352, "latent_loss": 0.5421398881077767, "action_l1_1step": 0.04344782888889313, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 21582}
371
+ {"step": 9275, "latent_r2": 0.8149932813644409, "latent_loss": 0.5334717598557472, "action_l1_1step": 0.04373153241351247, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 21641}
372
+ {"step": 9300, "latent_r2": 0.8381965965032577, "latent_loss": 0.5481440603733063, "action_l1_1step": 0.04179883275181055, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 21700}
373
+ {"step": 9325, "latent_r2": 0.8193444770574569, "latent_loss": 0.5359067192673683, "action_l1_1step": 0.044203849378973245, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 21759}
374
+ {"step": 9350, "latent_r2": 0.8241340613365173, "latent_loss": 0.5319571861624718, "action_l1_1step": 0.04214767627418041, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 21818}
375
+ {"step": 9375, "latent_r2": 0.8320129883289337, "latent_loss": 0.5292999604344368, "action_l1_1step": 0.04161937110126018, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 21877}
376
+ {"step": 9400, "latent_r2": 0.8385528403520585, "latent_loss": 0.5355642569065094, "action_l1_1step": 0.0399441047757864, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 21943}
377
+ {"step": 9425, "latent_r2": 0.8258424884080887, "latent_loss": 0.5290612769126892, "action_l1_1step": 0.041705040857195855, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 22002}
378
+ {"step": 9450, "latent_r2": 0.8302540552616119, "latent_loss": 0.5396887230873108, "action_l1_1step": 0.04101906761527061, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 22066}
379
+ {"step": 9475, "latent_r2": 0.8320059037208557, "latent_loss": 0.5391575881838798, "action_l1_1step": 0.041734024360775945, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 22129}
380
+ {"step": 9500, "latent_r2": 0.8286759781837464, "latent_loss": 0.5320176929235458, "action_l1_1step": 0.041094321869313714, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 22192}
381
+ {"step": 9525, "latent_r2": 0.8249393463134765, "latent_loss": 0.5300733464956283, "action_l1_1step": 0.04188535435125232, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 22251}
382
+ {"step": 9550, "latent_r2": 0.8306220448017121, "latent_loss": 0.5268280053138733, "action_l1_1step": 0.038932679146528246, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 22314}
383
+ {"step": 9575, "latent_r2": 0.824765185713768, "latent_loss": 0.5232857641577721, "action_l1_1step": 0.04113214612007141, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 22375}
384
+ {"step": 9600, "latent_r2": 0.8286150521039963, "latent_loss": 0.5431966665387153, "action_l1_1step": 0.040318315513432026, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 22437}
385
+ {"step": 9625, "latent_r2": 0.8315667927265167, "latent_loss": 0.5434672603011131, "action_l1_1step": 0.041938033290207385, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 22497}
386
+ {"step": 9650, "latent_r2": 0.8164663910865784, "latent_loss": 0.5199505618214607, "action_l1_1step": 0.041472932305186984, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 22561}
387
+ {"step": 9675, "latent_r2": 0.8327529710531235, "latent_loss": 0.5346806579828263, "action_l1_1step": 0.04006625635549426, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 22620}
388
+ {"step": 9700, "latent_r2": 0.8341779398918152, "latent_loss": 0.5306623789668083, "action_l1_1step": 0.040460790898650886, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 22683}
389
+ {"step": 9725, "latent_r2": 0.8282845240831375, "latent_loss": 0.5374186620116234, "action_l1_1step": 0.041506569851189855, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 22742}
390
+ {"step": 9750, "latent_r2": 0.8213041281700134, "latent_loss": 0.5288753354549408, "action_l1_1step": 0.04187092261388898, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 22800}
391
+ {"step": 9775, "latent_r2": 0.8152276819944382, "latent_loss": 0.5380428764224052, "action_l1_1step": 0.04201696269214153, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 22866}
392
+ {"step": 9800, "latent_r2": 0.825583072900772, "latent_loss": 0.5285345676541329, "action_l1_1step": 0.04043401977047324, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 22933}
393
+ {"step": 9825, "latent_r2": 0.8234961420297623, "latent_loss": 0.5232138520479203, "action_l1_1step": 0.04115278070792556, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 22997}
394
+ {"step": 9850, "latent_r2": 0.8349818509817123, "latent_loss": 0.5497750249505043, "action_l1_1step": 0.03908168291673064, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 23064}
395
+ {"step": 9875, "latent_r2": 0.8339760458469391, "latent_loss": 0.5558808514475823, "action_l1_1step": 0.04000969419255853, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 23125}
396
+ {"step": 9900, "latent_r2": 0.8323103731870651, "latent_loss": 0.5431602171063423, "action_l1_1step": 0.04016922663897276, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 23186}
397
+ {"step": 9925, "latent_r2": 0.8357971948385239, "latent_loss": 0.5370484206080437, "action_l1_1step": 0.040945048294961456, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 23249}
398
+ {"step": 9950, "latent_r2": 0.8409292668104171, "latent_loss": 0.534945749938488, "action_l1_1step": 0.03885136604309082, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 23312}
399
+ {"step": 9975, "latent_r2": 0.8228735142946243, "latent_loss": 0.5271611213684082, "action_l1_1step": 0.04137781772762537, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 23372}
400
+ {"step": 10000, "latent_r2": 0.835334210395813, "latent_loss": 0.5364477586746216, "action_l1_1step": 0.04114575617015362, "p_oracle": 0.0, "lr": 3.5000000000000004e-05, "seconds": 23435}