BruceYuan commited on
Commit
4711add
·
verified ·
1 Parent(s): 10353e2

Upload folder using huggingface_hub

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. .gitattributes +2 -0
  2. soup_L_100M_rerank2_work_dir/ckpt/0.pt +3 -0
  3. soup_L_100M_rerank2_work_dir/ckpt/100_000_000.pt +3 -0
  4. soup_L_100M_rerank2_work_dir/ckpt/10_000_000.pt +3 -0
  5. soup_L_100M_rerank2_work_dir/ckpt/15_000_000.pt +3 -0
  6. soup_L_100M_rerank2_work_dir/ckpt/20_000_000.pt +3 -0
  7. soup_L_100M_rerank2_work_dir/ckpt/25_000_000.pt +3 -0
  8. soup_L_100M_rerank2_work_dir/ckpt/30_000_000.pt +3 -0
  9. soup_L_100M_rerank2_work_dir/ckpt/35_000_000.pt +3 -0
  10. soup_L_100M_rerank2_work_dir/ckpt/40_000_000.pt +3 -0
  11. soup_L_100M_rerank2_work_dir/ckpt/45_000_000.pt +3 -0
  12. soup_L_100M_rerank2_work_dir/ckpt/50_000_000.pt +3 -0
  13. soup_L_100M_rerank2_work_dir/ckpt/55_000_000.pt +3 -0
  14. soup_L_100M_rerank2_work_dir/ckpt/5_000_000.pt +3 -0
  15. soup_L_100M_rerank2_work_dir/ckpt/60_000_000.pt +3 -0
  16. soup_L_100M_rerank2_work_dir/ckpt/65_000_000.pt +3 -0
  17. soup_L_100M_rerank2_work_dir/ckpt/70_000_000.pt +3 -0
  18. soup_L_100M_rerank2_work_dir/ckpt/75_000_000.pt +3 -0
  19. soup_L_100M_rerank2_work_dir/ckpt/80_000_000.pt +3 -0
  20. soup_L_100M_rerank2_work_dir/ckpt/85_000_000.pt +3 -0
  21. soup_L_100M_rerank2_work_dir/ckpt/90_000_000.pt +3 -0
  22. soup_L_100M_rerank2_work_dir/ckpt/95_000_000.pt +3 -0
  23. soup_L_100M_rerank2_work_dir/ckpt/final.pt +3 -0
  24. soup_L_100M_rerank2_work_dir/score_dataset/metadata.json +297 -0
  25. soup_L_100M_rerank2_work_dir/score_dataset/rank_000/metadata_rank000.json +308 -0
  26. soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000000.npz +3 -0
  27. soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000001.npz +3 -0
  28. soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000002.npz +3 -0
  29. soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000003.npz +3 -0
  30. soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000004.npz +3 -0
  31. soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000005.npz +3 -0
  32. soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000006.npz +3 -0
  33. soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000007.npz +3 -0
  34. soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000008.npz +3 -0
  35. soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000009.npz +3 -0
  36. soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000010.npz +3 -0
  37. soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000011.npz +3 -0
  38. soup_L_100M_rerank2_work_dir/score_dataset/rank_001/metadata_rank001.json +308 -0
  39. soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000000.npz +3 -0
  40. soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000001.npz +3 -0
  41. soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000002.npz +3 -0
  42. soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000003.npz +3 -0
  43. soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000004.npz +3 -0
  44. soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000005.npz +3 -0
  45. soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000006.npz +3 -0
  46. soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000007.npz +3 -0
  47. soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000008.npz +3 -0
  48. soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000009.npz +3 -0
  49. soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000010.npz +3 -0
  50. soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000011.npz +3 -0
.gitattributes CHANGED
@@ -43,3 +43,5 @@ soup_L_100M_rerank_work_dir/wandb/run-20260625_233207-xoqubnnq/run-xoqubnnq.wand
43
  soup_L_100M_rerank_work_dir/wandb/run-20260625_235046-uduzidoi/run-uduzidoi.wandb filter=lfs diff=lfs merge=lfs -text
44
  soup_L_100M_rerank_work_dir/wandb/run-20260626_003936-5b2se9vf/files/output.log filter=lfs diff=lfs merge=lfs -text
45
  soup_L_100M_rerank_work_dir/wandb/run-20260626_003936-5b2se9vf/run-5b2se9vf.wandb filter=lfs diff=lfs merge=lfs -text
 
 
 
43
  soup_L_100M_rerank_work_dir/wandb/run-20260625_235046-uduzidoi/run-uduzidoi.wandb filter=lfs diff=lfs merge=lfs -text
44
  soup_L_100M_rerank_work_dir/wandb/run-20260626_003936-5b2se9vf/files/output.log filter=lfs diff=lfs merge=lfs -text
45
  soup_L_100M_rerank_work_dir/wandb/run-20260626_003936-5b2se9vf/run-5b2se9vf.wandb filter=lfs diff=lfs merge=lfs -text
46
+ soup_L_100M_rerank2_work_dir/wandb/run-20260627_000046-z51ici4q/files/output.log filter=lfs diff=lfs merge=lfs -text
47
+ soup_L_100M_rerank2_work_dir/wandb/run-20260627_000046-z51ici4q/run-z51ici4q.wandb filter=lfs diff=lfs merge=lfs -text
soup_L_100M_rerank2_work_dir/ckpt/0.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:88d09bea7fdc43be85d7c441e7eecea5636b657c6ceb9f0999bbebe7e1a9b1eb
3
+ size 314731275
soup_L_100M_rerank2_work_dir/ckpt/100_000_000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:055af4b0d5596dd249ebfdf69b0a3301581297440604c7c69913b287310166cb
3
+ size 331862407
soup_L_100M_rerank2_work_dir/ckpt/10_000_000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8b169b1a6e8d70acde53842d73e10381c97964d83e165a083f78676980a56ca8
3
+ size 331861897
soup_L_100M_rerank2_work_dir/ckpt/15_000_000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:de3d1874de7f89ffab47f05c0a8329c34d6637420132121c67921ebadb153adf
3
+ size 331861897
soup_L_100M_rerank2_work_dir/ckpt/20_000_000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e68ecd5e66e906932758efabd18eb50d42464ff8aac8ee26e63948226c6d72f
3
+ size 331861897
soup_L_100M_rerank2_work_dir/ckpt/25_000_000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0dd078ba1544acd15a7143486de5798b504472d6f4a088a201560a13f875d749
3
+ size 331861897
soup_L_100M_rerank2_work_dir/ckpt/30_000_000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f7281d5af7872a2ef81146b46bb81c4bf49cc30a0d22bb5002a602d555af1a9b
3
+ size 331861897
soup_L_100M_rerank2_work_dir/ckpt/35_000_000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e2c7a0c9ff3c03a8dca233704ebeaa6ace3d5b48ad75c733c6242510366aa6e
3
+ size 331861897
soup_L_100M_rerank2_work_dir/ckpt/40_000_000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e491017c7e8b416ad679da8183e95f9970dc22714037571772b6a9fa23c5bf48
3
+ size 331861897
soup_L_100M_rerank2_work_dir/ckpt/45_000_000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:df3496f82e482ee069e7a70469aa2a4c6770fbfbe2303a05f146ba79016662d1
3
+ size 331861897
soup_L_100M_rerank2_work_dir/ckpt/50_000_000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:48eee605b6d2fdbe7f99c20dc7b2f0b08202ef5fd78f362b5f325cf1a4b12849
3
+ size 331861897
soup_L_100M_rerank2_work_dir/ckpt/55_000_000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:761a0c01f609517b61c4ab8962fe40f479f89110ff6b3dd4d329dbb249339111
3
+ size 331861897
soup_L_100M_rerank2_work_dir/ckpt/5_000_000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:861ef6c1a7df3b99b8e27e7cf1d926ede8ea5202d3aabb6a951bfefade13d24c
3
+ size 331861387
soup_L_100M_rerank2_work_dir/ckpt/60_000_000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:631699d0f1d2818e629709027ea1f2d5bb8fec44c4c4707287837a019c9aedeb
3
+ size 331861897
soup_L_100M_rerank2_work_dir/ckpt/65_000_000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:56e2f55226d95a2a5d655f1bfe71de54c9b008e806da97d55ccbb7a3d1d3f1ba
3
+ size 331861897
soup_L_100M_rerank2_work_dir/ckpt/70_000_000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8071ea53d2f9843b24403f7588e4d1dc1f531caa42cf5bd3129dd49f2fb1e154
3
+ size 331861897
soup_L_100M_rerank2_work_dir/ckpt/75_000_000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1fca1dfa1a8b820c6e97aebf3139d8bc89935a07a0f7cd831b5538643129ee28
3
+ size 331861897
soup_L_100M_rerank2_work_dir/ckpt/80_000_000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c4d0defa3c4fd70f0b68b3e1b9df79f53c1b025641efb640a59bfd0d250c92e5
3
+ size 331861897
soup_L_100M_rerank2_work_dir/ckpt/85_000_000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ff18649ef635c13bef985898b846ffe60a6c92316edcfa467734f520dd0b5d27
3
+ size 331861897
soup_L_100M_rerank2_work_dir/ckpt/90_000_000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8d2588f41fa5befee702a2bd93002b237a8b7ae3c43887da852f8c622901b9c7
3
+ size 331861897
soup_L_100M_rerank2_work_dir/ckpt/95_000_000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:31cff42afd12d42e2a34bb5e7d41a3060190ed404ba14bf52f2bff0402c8a1f8
3
+ size 331861897
soup_L_100M_rerank2_work_dir/ckpt/final.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:840cec185f0f192495ae15a705d81c8e4dbcfcdeb91541b964b488b15e1e32f0
3
+ size 331836435
soup_L_100M_rerank2_work_dir/score_dataset/metadata.json ADDED
@@ -0,0 +1,297 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "tasks": [
3
+ "walker-stand",
4
+ "walker-walk",
5
+ "walker-run",
6
+ "cheetah-run",
7
+ "reacher-easy",
8
+ "reacher-hard",
9
+ "acrobot-swingup",
10
+ "pendulum-swingup",
11
+ "cartpole-balance",
12
+ "cartpole-balance-sparse",
13
+ "cartpole-swingup",
14
+ "cartpole-swingup-sparse",
15
+ "cup-catch",
16
+ "finger-spin",
17
+ "finger-turn-easy",
18
+ "finger-turn-hard",
19
+ "fish-swim",
20
+ "hopper-stand",
21
+ "hopper-hop",
22
+ "quadruped-walk",
23
+ "quadruped-run",
24
+ "walker-walk-backward",
25
+ "walker-run-backward",
26
+ "cheetah-run-backward",
27
+ "cheetah-run-front",
28
+ "cheetah-run-back",
29
+ "cheetah-jump",
30
+ "hopper-hop-backward",
31
+ "reacher-three-easy",
32
+ "reacher-three-hard",
33
+ "cup-spin",
34
+ "pendulum-spin",
35
+ "jumper-jump",
36
+ "spinner-spin",
37
+ "spinner-spin-backward",
38
+ "spinner-jump",
39
+ "giraffe-run",
40
+ "mw-assembly",
41
+ "mw-basketball",
42
+ "mw-button-press-topdown",
43
+ "mw-button-press-topdown-wall",
44
+ "mw-button-press",
45
+ "mw-button-press-wall",
46
+ "mw-coffee-button",
47
+ "mw-coffee-pull",
48
+ "mw-coffee-push",
49
+ "mw-dial-turn",
50
+ "mw-disassemble",
51
+ "mw-door-open",
52
+ "mw-door-close",
53
+ "mw-drawer-close",
54
+ "mw-drawer-open",
55
+ "mw-faucet-open",
56
+ "mw-faucet-close",
57
+ "mw-hammer",
58
+ "mw-handle-press-side",
59
+ "mw-handle-press",
60
+ "mw-handle-pull-side",
61
+ "mw-handle-pull",
62
+ "mw-lever-pull",
63
+ "mw-peg-insert-side",
64
+ "mw-peg-unplug-side",
65
+ "mw-pick-out-of-hole",
66
+ "mw-pick-place",
67
+ "mw-pick-place-wall",
68
+ "mw-plate-slide",
69
+ "mw-plate-slide-side",
70
+ "mw-plate-slide-back",
71
+ "mw-plate-slide-back-side",
72
+ "mw-push-back",
73
+ "mw-push",
74
+ "mw-push-wall",
75
+ "mw-reach",
76
+ "mw-reach-wall",
77
+ "mw-soccer",
78
+ "mw-stick-push",
79
+ "mw-stick-pull",
80
+ "mw-sweep-into",
81
+ "mw-sweep",
82
+ "mw-window-open",
83
+ "mw-window-close",
84
+ "mw-bin-picking",
85
+ "mw-box-close",
86
+ "mw-door-lock",
87
+ "mw-door-unlock",
88
+ "mw-hand-insert",
89
+ "ms-ant-walk",
90
+ "ms-ant-run",
91
+ "ms-cartpole-balance",
92
+ "ms-cartpole-swingup",
93
+ "ms-hopper-stand",
94
+ "ms-hopper-hop",
95
+ "ms-pick-cube",
96
+ "ms-pick-cube-eepose",
97
+ "ms-pick-cube-so",
98
+ "ms-poke-cube",
99
+ "ms-push-cube",
100
+ "ms-pull-cube",
101
+ "ms-pull-cube-tool",
102
+ "ms-stack-cube",
103
+ "ms-place-sphere",
104
+ "ms-lift-peg",
105
+ "ms-pick-apple",
106
+ "ms-pick-banana",
107
+ "ms-pick-can",
108
+ "ms-pick-hammer",
109
+ "ms-pick-fork",
110
+ "ms-pick-knife",
111
+ "ms-pick-mug",
112
+ "ms-pick-orange",
113
+ "ms-pick-screwdriver",
114
+ "ms-pick-spoon",
115
+ "ms-pick-tennis-ball",
116
+ "ms-pick-baseball",
117
+ "ms-pick-cube-xarm6",
118
+ "ms-pick-sponge",
119
+ "ms-anymal-reach",
120
+ "ms-reach",
121
+ "ms-reach-eepose",
122
+ "ms-reach-xarm6",
123
+ "ms-cartpole-balance-sparse",
124
+ "ms-cartpole-swingup-sparse",
125
+ "mujoco-ant",
126
+ "mujoco-halfcheetah",
127
+ "mujoco-hopper",
128
+ "mujoco-inverted-pendulum",
129
+ "mujoco-reacher",
130
+ "mujoco-walker",
131
+ "bipedal-walker-flat",
132
+ "bipedal-walker-uneven",
133
+ "bipedal-walker-rugged",
134
+ "bipedal-walker-hills",
135
+ "bipedal-walker-obstacles",
136
+ "lunarlander-land",
137
+ "lunarlander-hover",
138
+ "lunarlander-takeoff",
139
+ "rd-push-red",
140
+ "rd-push-green",
141
+ "rd-push-blue",
142
+ "rd-open-slide",
143
+ "rd-open-drawer",
144
+ "rd-flat-block-in-bin",
145
+ "og-ant",
146
+ "og-antball",
147
+ "og-point-arena",
148
+ "og-point-maze",
149
+ "og-point-bottleneck",
150
+ "og-point-circle",
151
+ "og-point-spiral",
152
+ "og-ant-arena",
153
+ "og-ant-maze",
154
+ "og-ant-bottleneck",
155
+ "og-ant-circle",
156
+ "og-ant-spiral",
157
+ "pygame-cowboy",
158
+ "pygame-coinrun",
159
+ "pygame-spaceship",
160
+ "pygame-pong",
161
+ "pygame-bird-attack",
162
+ "pygame-highway",
163
+ "pygame-landing",
164
+ "pygame-air-hockey",
165
+ "pygame-rocket-collect",
166
+ "pygame-chase-evade",
167
+ "pygame-coconut-dodge",
168
+ "pygame-cartpole-balance",
169
+ "pygame-cartpole-swingup",
170
+ "pygame-cartpole-balance-sparse",
171
+ "pygame-cartpole-swingup-sparse",
172
+ "pygame-cartpole-tremor",
173
+ "pygame-point-maze-var1",
174
+ "pygame-point-maze-var2",
175
+ "pygame-point-maze-var3",
176
+ "atari-alien",
177
+ "atari-assault",
178
+ "atari-asterix",
179
+ "atari-atlantis",
180
+ "atari-bank-heist",
181
+ "atari-battle-zone",
182
+ "atari-beamrider",
183
+ "atari-boxing",
184
+ "atari-chopper-command",
185
+ "atari-crazy-climber",
186
+ "atari-double-dunk",
187
+ "atari-gopher",
188
+ "atari-ice-hockey",
189
+ "atari-jamesbond",
190
+ "atari-kangaroo",
191
+ "atari-krull",
192
+ "atari-ms-pacman",
193
+ "atari-name-this-game",
194
+ "atari-phoenix",
195
+ "atari-pong",
196
+ "atari-road-runner",
197
+ "atari-robotank",
198
+ "atari-seaquest",
199
+ "atari-space-invaders",
200
+ "atari-tutankham",
201
+ "atari-upndown",
202
+ "atari-yars-revenge"
203
+ ],
204
+ "domains": [
205
+ "dmcontrol",
206
+ "dmcontrol-ext",
207
+ "metaworld",
208
+ "maniskill",
209
+ "mujoco",
210
+ "box2d",
211
+ "robodesk",
212
+ "ogbench",
213
+ "pygame",
214
+ "atari"
215
+ ],
216
+ "action_dim": 16,
217
+ "horizon": 3,
218
+ "obs_state_dim": 128,
219
+ "z_dim": 512,
220
+ "lang_emb_dim": 512,
221
+ "dataset_size": 190000,
222
+ "num_condition_states": 10000,
223
+ "shard_size": 4096,
224
+ "num_shards": 48,
225
+ "shards": [
226
+ "rank_000/shard_000000.npz",
227
+ "rank_000/shard_000001.npz",
228
+ "rank_000/shard_000002.npz",
229
+ "rank_000/shard_000003.npz",
230
+ "rank_000/shard_000004.npz",
231
+ "rank_000/shard_000005.npz",
232
+ "rank_000/shard_000006.npz",
233
+ "rank_000/shard_000007.npz",
234
+ "rank_000/shard_000008.npz",
235
+ "rank_000/shard_000009.npz",
236
+ "rank_000/shard_000010.npz",
237
+ "rank_000/shard_000011.npz",
238
+ "rank_001/shard_000000.npz",
239
+ "rank_001/shard_000001.npz",
240
+ "rank_001/shard_000002.npz",
241
+ "rank_001/shard_000003.npz",
242
+ "rank_001/shard_000004.npz",
243
+ "rank_001/shard_000005.npz",
244
+ "rank_001/shard_000006.npz",
245
+ "rank_001/shard_000007.npz",
246
+ "rank_001/shard_000008.npz",
247
+ "rank_001/shard_000009.npz",
248
+ "rank_001/shard_000010.npz",
249
+ "rank_001/shard_000011.npz",
250
+ "rank_002/shard_000000.npz",
251
+ "rank_002/shard_000001.npz",
252
+ "rank_002/shard_000002.npz",
253
+ "rank_002/shard_000003.npz",
254
+ "rank_002/shard_000004.npz",
255
+ "rank_002/shard_000005.npz",
256
+ "rank_002/shard_000006.npz",
257
+ "rank_002/shard_000007.npz",
258
+ "rank_002/shard_000008.npz",
259
+ "rank_002/shard_000009.npz",
260
+ "rank_002/shard_000010.npz",
261
+ "rank_002/shard_000011.npz",
262
+ "rank_003/shard_000000.npz",
263
+ "rank_003/shard_000001.npz",
264
+ "rank_003/shard_000002.npz",
265
+ "rank_003/shard_000003.npz",
266
+ "rank_003/shard_000004.npz",
267
+ "rank_003/shard_000005.npz",
268
+ "rank_003/shard_000006.npz",
269
+ "rank_003/shard_000007.npz",
270
+ "rank_003/shard_000008.npz",
271
+ "rank_003/shard_000009.npz",
272
+ "rank_003/shard_000010.npz",
273
+ "rank_003/shard_000011.npz"
274
+ ],
275
+ "target_type": "score_total",
276
+ "normalization_mode": "task",
277
+ "include_demos": true,
278
+ "has_score_reward": false,
279
+ "has_score_prior": false,
280
+ "checkpoint": "None",
281
+ "checkpoint_step": 0,
282
+ "planner_type": "diffusion",
283
+ "score_source": "analytic",
284
+ "world_size": 4,
285
+ "fallbacks": {
286
+ "score_reward": "zero arrays; current analytic diffusion planner does not expose reward-score decomposition",
287
+ "score_prior": "zero arrays; current analytic diffusion planner does not expose prior-score decomposition",
288
+ "terminal_Q": "zero; trajectory evaluator currently returns only aggregate reward+bootstrap value",
289
+ "episode_step": "0; replay sampler returns subsequences without original step indices"
290
+ },
291
+ "rank_metadata": [
292
+ "rank_000/metadata_rank000.json",
293
+ "rank_001/metadata_rank001.json",
294
+ "rank_002/metadata_rank002.json",
295
+ "rank_003/metadata_rank003.json"
296
+ ]
297
+ }
soup_L_100M_rerank2_work_dir/score_dataset/rank_000/metadata_rank000.json ADDED
@@ -0,0 +1,308 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "tasks": [
3
+ "walker-stand",
4
+ "walker-walk",
5
+ "walker-run",
6
+ "cheetah-run",
7
+ "reacher-easy",
8
+ "reacher-hard",
9
+ "acrobot-swingup",
10
+ "pendulum-swingup",
11
+ "cartpole-balance",
12
+ "cartpole-balance-sparse",
13
+ "cartpole-swingup",
14
+ "cartpole-swingup-sparse",
15
+ "cup-catch",
16
+ "finger-spin",
17
+ "finger-turn-easy",
18
+ "finger-turn-hard",
19
+ "fish-swim",
20
+ "hopper-stand",
21
+ "hopper-hop",
22
+ "quadruped-walk",
23
+ "quadruped-run",
24
+ "walker-walk-backward",
25
+ "walker-run-backward",
26
+ "cheetah-run-backward",
27
+ "cheetah-run-front",
28
+ "cheetah-run-back",
29
+ "cheetah-jump",
30
+ "hopper-hop-backward",
31
+ "reacher-three-easy",
32
+ "reacher-three-hard",
33
+ "cup-spin",
34
+ "pendulum-spin",
35
+ "jumper-jump",
36
+ "spinner-spin",
37
+ "spinner-spin-backward",
38
+ "spinner-jump",
39
+ "giraffe-run",
40
+ "mw-assembly",
41
+ "mw-basketball",
42
+ "mw-button-press-topdown",
43
+ "mw-button-press-topdown-wall",
44
+ "mw-button-press",
45
+ "mw-button-press-wall",
46
+ "mw-coffee-button",
47
+ "mw-coffee-pull",
48
+ "mw-coffee-push",
49
+ "mw-dial-turn",
50
+ "mw-disassemble",
51
+ "mw-door-open",
52
+ "mw-door-close",
53
+ "mw-drawer-close",
54
+ "mw-drawer-open",
55
+ "mw-faucet-open",
56
+ "mw-faucet-close",
57
+ "mw-hammer",
58
+ "mw-handle-press-side",
59
+ "mw-handle-press",
60
+ "mw-handle-pull-side",
61
+ "mw-handle-pull",
62
+ "mw-lever-pull",
63
+ "mw-peg-insert-side",
64
+ "mw-peg-unplug-side",
65
+ "mw-pick-out-of-hole",
66
+ "mw-pick-place",
67
+ "mw-pick-place-wall",
68
+ "mw-plate-slide",
69
+ "mw-plate-slide-side",
70
+ "mw-plate-slide-back",
71
+ "mw-plate-slide-back-side",
72
+ "mw-push-back",
73
+ "mw-push",
74
+ "mw-push-wall",
75
+ "mw-reach",
76
+ "mw-reach-wall",
77
+ "mw-soccer",
78
+ "mw-stick-push",
79
+ "mw-stick-pull",
80
+ "mw-sweep-into",
81
+ "mw-sweep",
82
+ "mw-window-open",
83
+ "mw-window-close",
84
+ "mw-bin-picking",
85
+ "mw-box-close",
86
+ "mw-door-lock",
87
+ "mw-door-unlock",
88
+ "mw-hand-insert",
89
+ "ms-ant-walk",
90
+ "ms-ant-run",
91
+ "ms-cartpole-balance",
92
+ "ms-cartpole-swingup",
93
+ "ms-hopper-stand",
94
+ "ms-hopper-hop",
95
+ "ms-pick-cube",
96
+ "ms-pick-cube-eepose",
97
+ "ms-pick-cube-so",
98
+ "ms-poke-cube",
99
+ "ms-push-cube",
100
+ "ms-pull-cube",
101
+ "ms-pull-cube-tool",
102
+ "ms-stack-cube",
103
+ "ms-place-sphere",
104
+ "ms-lift-peg",
105
+ "ms-pick-apple",
106
+ "ms-pick-banana",
107
+ "ms-pick-can",
108
+ "ms-pick-hammer",
109
+ "ms-pick-fork",
110
+ "ms-pick-knife",
111
+ "ms-pick-mug",
112
+ "ms-pick-orange",
113
+ "ms-pick-screwdriver",
114
+ "ms-pick-spoon",
115
+ "ms-pick-tennis-ball",
116
+ "ms-pick-baseball",
117
+ "ms-pick-cube-xarm6",
118
+ "ms-pick-sponge",
119
+ "ms-anymal-reach",
120
+ "ms-reach",
121
+ "ms-reach-eepose",
122
+ "ms-reach-xarm6",
123
+ "ms-cartpole-balance-sparse",
124
+ "ms-cartpole-swingup-sparse",
125
+ "mujoco-ant",
126
+ "mujoco-halfcheetah",
127
+ "mujoco-hopper",
128
+ "mujoco-inverted-pendulum",
129
+ "mujoco-reacher",
130
+ "mujoco-walker",
131
+ "bipedal-walker-flat",
132
+ "bipedal-walker-uneven",
133
+ "bipedal-walker-rugged",
134
+ "bipedal-walker-hills",
135
+ "bipedal-walker-obstacles",
136
+ "lunarlander-land",
137
+ "lunarlander-hover",
138
+ "lunarlander-takeoff",
139
+ "rd-push-red",
140
+ "rd-push-green",
141
+ "rd-push-blue",
142
+ "rd-open-slide",
143
+ "rd-open-drawer",
144
+ "rd-flat-block-in-bin",
145
+ "og-ant",
146
+ "og-antball",
147
+ "og-point-arena",
148
+ "og-point-maze",
149
+ "og-point-bottleneck",
150
+ "og-point-circle",
151
+ "og-point-spiral",
152
+ "og-ant-arena",
153
+ "og-ant-maze",
154
+ "og-ant-bottleneck",
155
+ "og-ant-circle",
156
+ "og-ant-spiral",
157
+ "pygame-cowboy",
158
+ "pygame-coinrun",
159
+ "pygame-spaceship",
160
+ "pygame-pong",
161
+ "pygame-bird-attack",
162
+ "pygame-highway",
163
+ "pygame-landing",
164
+ "pygame-air-hockey",
165
+ "pygame-rocket-collect",
166
+ "pygame-chase-evade",
167
+ "pygame-coconut-dodge",
168
+ "pygame-cartpole-balance",
169
+ "pygame-cartpole-swingup",
170
+ "pygame-cartpole-balance-sparse",
171
+ "pygame-cartpole-swingup-sparse",
172
+ "pygame-cartpole-tremor",
173
+ "pygame-point-maze-var1",
174
+ "pygame-point-maze-var2",
175
+ "pygame-point-maze-var3",
176
+ "atari-alien",
177
+ "atari-assault",
178
+ "atari-asterix",
179
+ "atari-atlantis",
180
+ "atari-bank-heist",
181
+ "atari-battle-zone",
182
+ "atari-beamrider",
183
+ "atari-boxing",
184
+ "atari-chopper-command",
185
+ "atari-crazy-climber",
186
+ "atari-double-dunk",
187
+ "atari-gopher",
188
+ "atari-ice-hockey",
189
+ "atari-jamesbond",
190
+ "atari-kangaroo",
191
+ "atari-krull",
192
+ "atari-ms-pacman",
193
+ "atari-name-this-game",
194
+ "atari-phoenix",
195
+ "atari-pong",
196
+ "atari-road-runner",
197
+ "atari-robotank",
198
+ "atari-seaquest",
199
+ "atari-space-invaders",
200
+ "atari-tutankham",
201
+ "atari-upndown",
202
+ "atari-yars-revenge"
203
+ ],
204
+ "domains": [
205
+ "dmcontrol",
206
+ "dmcontrol-ext",
207
+ "metaworld",
208
+ "maniskill",
209
+ "mujoco",
210
+ "box2d",
211
+ "robodesk",
212
+ "ogbench",
213
+ "pygame",
214
+ "atari"
215
+ ],
216
+ "action_dim": 16,
217
+ "horizon": 3,
218
+ "obs_state_dim": 128,
219
+ "z_dim": 512,
220
+ "lang_emb_dim": 512,
221
+ "dataset_size": 47500,
222
+ "num_condition_states": 2500,
223
+ "shard_size": 4096,
224
+ "num_shards": 12,
225
+ "shards": [
226
+ "rank_000/shard_000000.npz",
227
+ "rank_000/shard_000001.npz",
228
+ "rank_000/shard_000002.npz",
229
+ "rank_000/shard_000003.npz",
230
+ "rank_000/shard_000004.npz",
231
+ "rank_000/shard_000005.npz",
232
+ "rank_000/shard_000006.npz",
233
+ "rank_000/shard_000007.npz",
234
+ "rank_000/shard_000008.npz",
235
+ "rank_000/shard_000009.npz",
236
+ "rank_000/shard_000010.npz",
237
+ "rank_000/shard_000011.npz"
238
+ ],
239
+ "target_type": "score_total",
240
+ "normalization_mode": "task",
241
+ "include_demos": true,
242
+ "has_score_reward": false,
243
+ "has_score_prior": false,
244
+ "checkpoint": "None",
245
+ "checkpoint_step": 0,
246
+ "planner_type": "diffusion",
247
+ "score_source": "analytic",
248
+ "rank": 0,
249
+ "world_size": 4,
250
+ "local_task_ids": [
251
+ 0,
252
+ 4,
253
+ 8,
254
+ 12,
255
+ 16,
256
+ 20,
257
+ 24,
258
+ 28,
259
+ 32,
260
+ 36,
261
+ 40,
262
+ 44,
263
+ 48,
264
+ 52,
265
+ 56,
266
+ 60,
267
+ 64,
268
+ 68,
269
+ 72,
270
+ 76,
271
+ 80,
272
+ 84,
273
+ 88,
274
+ 92,
275
+ 96,
276
+ 100,
277
+ 104,
278
+ 108,
279
+ 112,
280
+ 116,
281
+ 120,
282
+ 124,
283
+ 128,
284
+ 132,
285
+ 136,
286
+ 140,
287
+ 144,
288
+ 148,
289
+ 152,
290
+ 156,
291
+ 160,
292
+ 164,
293
+ 168,
294
+ 172,
295
+ 176,
296
+ 180,
297
+ 184,
298
+ 188,
299
+ 192,
300
+ 196
301
+ ],
302
+ "fallbacks": {
303
+ "score_reward": "zero arrays; current analytic diffusion planner does not expose reward-score decomposition",
304
+ "score_prior": "zero arrays; current analytic diffusion planner does not expose prior-score decomposition",
305
+ "terminal_Q": "zero; trajectory evaluator currently returns only aggregate reward+bootstrap value",
306
+ "episode_step": "0; replay sampler returns subsequences without original step indices"
307
+ }
308
+ }
soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000000.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:79e2645097c0a697d075d0356e5e375c30264265fb2ced702099812178af3ba2
3
+ size 1631942
soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000001.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8bc188c8221ee283c178ebd656588d8555dff574275aa58b143625d10d84bce5
3
+ size 1641051
soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000002.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:14764160a353be7d8aa7c7038d6b1df41a307dec0f316a717be17ba637b74995
3
+ size 1619902
soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000003.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2882023779c807a575f0652d55866538ab7589a14e54bb7bf99e1f02db88a11a
3
+ size 1634863
soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000004.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e81bdc775cbe574455c96df142fc0c077e661d357f9d1b539f2a6f811db81c37
3
+ size 1631691
soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000005.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:836ffa58c90633ea1ceb04fcacdfe7d0181f2df82e84a60618e5d6dae48a975b
3
+ size 1631228
soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000006.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1282719f5693ab421f51fd5452c2b26beb538eab5c55bada7ee5c469245c7a35
3
+ size 1639847
soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000007.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fa12803dd00e224f089730a7e23d308af4612f5b95eadf4abf0a19e924e785a5
3
+ size 1684910
soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000008.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0a92fa8eb057bd7cea5347ccc639e6a7271368df9739a10585808bd0e5d826a5
3
+ size 1732633
soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000009.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6d8b82b2062f364736169fb6055c14e2b5939efc0692071c42c34493917b9146
3
+ size 1825480
soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000010.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5fcba01f9013b018581b76b43e87c80c24a278deeeaadec83ca27aaf735f78d2
3
+ size 1900951
soup_L_100M_rerank2_work_dir/score_dataset/rank_000/shard_000011.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61f9fa9433bc751ae5980bf4f884592bcdd2acb8c49c280395254385c6e09f50
3
+ size 1084053
soup_L_100M_rerank2_work_dir/score_dataset/rank_001/metadata_rank001.json ADDED
@@ -0,0 +1,308 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "tasks": [
3
+ "walker-stand",
4
+ "walker-walk",
5
+ "walker-run",
6
+ "cheetah-run",
7
+ "reacher-easy",
8
+ "reacher-hard",
9
+ "acrobot-swingup",
10
+ "pendulum-swingup",
11
+ "cartpole-balance",
12
+ "cartpole-balance-sparse",
13
+ "cartpole-swingup",
14
+ "cartpole-swingup-sparse",
15
+ "cup-catch",
16
+ "finger-spin",
17
+ "finger-turn-easy",
18
+ "finger-turn-hard",
19
+ "fish-swim",
20
+ "hopper-stand",
21
+ "hopper-hop",
22
+ "quadruped-walk",
23
+ "quadruped-run",
24
+ "walker-walk-backward",
25
+ "walker-run-backward",
26
+ "cheetah-run-backward",
27
+ "cheetah-run-front",
28
+ "cheetah-run-back",
29
+ "cheetah-jump",
30
+ "hopper-hop-backward",
31
+ "reacher-three-easy",
32
+ "reacher-three-hard",
33
+ "cup-spin",
34
+ "pendulum-spin",
35
+ "jumper-jump",
36
+ "spinner-spin",
37
+ "spinner-spin-backward",
38
+ "spinner-jump",
39
+ "giraffe-run",
40
+ "mw-assembly",
41
+ "mw-basketball",
42
+ "mw-button-press-topdown",
43
+ "mw-button-press-topdown-wall",
44
+ "mw-button-press",
45
+ "mw-button-press-wall",
46
+ "mw-coffee-button",
47
+ "mw-coffee-pull",
48
+ "mw-coffee-push",
49
+ "mw-dial-turn",
50
+ "mw-disassemble",
51
+ "mw-door-open",
52
+ "mw-door-close",
53
+ "mw-drawer-close",
54
+ "mw-drawer-open",
55
+ "mw-faucet-open",
56
+ "mw-faucet-close",
57
+ "mw-hammer",
58
+ "mw-handle-press-side",
59
+ "mw-handle-press",
60
+ "mw-handle-pull-side",
61
+ "mw-handle-pull",
62
+ "mw-lever-pull",
63
+ "mw-peg-insert-side",
64
+ "mw-peg-unplug-side",
65
+ "mw-pick-out-of-hole",
66
+ "mw-pick-place",
67
+ "mw-pick-place-wall",
68
+ "mw-plate-slide",
69
+ "mw-plate-slide-side",
70
+ "mw-plate-slide-back",
71
+ "mw-plate-slide-back-side",
72
+ "mw-push-back",
73
+ "mw-push",
74
+ "mw-push-wall",
75
+ "mw-reach",
76
+ "mw-reach-wall",
77
+ "mw-soccer",
78
+ "mw-stick-push",
79
+ "mw-stick-pull",
80
+ "mw-sweep-into",
81
+ "mw-sweep",
82
+ "mw-window-open",
83
+ "mw-window-close",
84
+ "mw-bin-picking",
85
+ "mw-box-close",
86
+ "mw-door-lock",
87
+ "mw-door-unlock",
88
+ "mw-hand-insert",
89
+ "ms-ant-walk",
90
+ "ms-ant-run",
91
+ "ms-cartpole-balance",
92
+ "ms-cartpole-swingup",
93
+ "ms-hopper-stand",
94
+ "ms-hopper-hop",
95
+ "ms-pick-cube",
96
+ "ms-pick-cube-eepose",
97
+ "ms-pick-cube-so",
98
+ "ms-poke-cube",
99
+ "ms-push-cube",
100
+ "ms-pull-cube",
101
+ "ms-pull-cube-tool",
102
+ "ms-stack-cube",
103
+ "ms-place-sphere",
104
+ "ms-lift-peg",
105
+ "ms-pick-apple",
106
+ "ms-pick-banana",
107
+ "ms-pick-can",
108
+ "ms-pick-hammer",
109
+ "ms-pick-fork",
110
+ "ms-pick-knife",
111
+ "ms-pick-mug",
112
+ "ms-pick-orange",
113
+ "ms-pick-screwdriver",
114
+ "ms-pick-spoon",
115
+ "ms-pick-tennis-ball",
116
+ "ms-pick-baseball",
117
+ "ms-pick-cube-xarm6",
118
+ "ms-pick-sponge",
119
+ "ms-anymal-reach",
120
+ "ms-reach",
121
+ "ms-reach-eepose",
122
+ "ms-reach-xarm6",
123
+ "ms-cartpole-balance-sparse",
124
+ "ms-cartpole-swingup-sparse",
125
+ "mujoco-ant",
126
+ "mujoco-halfcheetah",
127
+ "mujoco-hopper",
128
+ "mujoco-inverted-pendulum",
129
+ "mujoco-reacher",
130
+ "mujoco-walker",
131
+ "bipedal-walker-flat",
132
+ "bipedal-walker-uneven",
133
+ "bipedal-walker-rugged",
134
+ "bipedal-walker-hills",
135
+ "bipedal-walker-obstacles",
136
+ "lunarlander-land",
137
+ "lunarlander-hover",
138
+ "lunarlander-takeoff",
139
+ "rd-push-red",
140
+ "rd-push-green",
141
+ "rd-push-blue",
142
+ "rd-open-slide",
143
+ "rd-open-drawer",
144
+ "rd-flat-block-in-bin",
145
+ "og-ant",
146
+ "og-antball",
147
+ "og-point-arena",
148
+ "og-point-maze",
149
+ "og-point-bottleneck",
150
+ "og-point-circle",
151
+ "og-point-spiral",
152
+ "og-ant-arena",
153
+ "og-ant-maze",
154
+ "og-ant-bottleneck",
155
+ "og-ant-circle",
156
+ "og-ant-spiral",
157
+ "pygame-cowboy",
158
+ "pygame-coinrun",
159
+ "pygame-spaceship",
160
+ "pygame-pong",
161
+ "pygame-bird-attack",
162
+ "pygame-highway",
163
+ "pygame-landing",
164
+ "pygame-air-hockey",
165
+ "pygame-rocket-collect",
166
+ "pygame-chase-evade",
167
+ "pygame-coconut-dodge",
168
+ "pygame-cartpole-balance",
169
+ "pygame-cartpole-swingup",
170
+ "pygame-cartpole-balance-sparse",
171
+ "pygame-cartpole-swingup-sparse",
172
+ "pygame-cartpole-tremor",
173
+ "pygame-point-maze-var1",
174
+ "pygame-point-maze-var2",
175
+ "pygame-point-maze-var3",
176
+ "atari-alien",
177
+ "atari-assault",
178
+ "atari-asterix",
179
+ "atari-atlantis",
180
+ "atari-bank-heist",
181
+ "atari-battle-zone",
182
+ "atari-beamrider",
183
+ "atari-boxing",
184
+ "atari-chopper-command",
185
+ "atari-crazy-climber",
186
+ "atari-double-dunk",
187
+ "atari-gopher",
188
+ "atari-ice-hockey",
189
+ "atari-jamesbond",
190
+ "atari-kangaroo",
191
+ "atari-krull",
192
+ "atari-ms-pacman",
193
+ "atari-name-this-game",
194
+ "atari-phoenix",
195
+ "atari-pong",
196
+ "atari-road-runner",
197
+ "atari-robotank",
198
+ "atari-seaquest",
199
+ "atari-space-invaders",
200
+ "atari-tutankham",
201
+ "atari-upndown",
202
+ "atari-yars-revenge"
203
+ ],
204
+ "domains": [
205
+ "dmcontrol",
206
+ "dmcontrol-ext",
207
+ "metaworld",
208
+ "maniskill",
209
+ "mujoco",
210
+ "box2d",
211
+ "robodesk",
212
+ "ogbench",
213
+ "pygame",
214
+ "atari"
215
+ ],
216
+ "action_dim": 16,
217
+ "horizon": 3,
218
+ "obs_state_dim": 128,
219
+ "z_dim": 512,
220
+ "lang_emb_dim": 512,
221
+ "dataset_size": 47500,
222
+ "num_condition_states": 2500,
223
+ "shard_size": 4096,
224
+ "num_shards": 12,
225
+ "shards": [
226
+ "rank_001/shard_000000.npz",
227
+ "rank_001/shard_000001.npz",
228
+ "rank_001/shard_000002.npz",
229
+ "rank_001/shard_000003.npz",
230
+ "rank_001/shard_000004.npz",
231
+ "rank_001/shard_000005.npz",
232
+ "rank_001/shard_000006.npz",
233
+ "rank_001/shard_000007.npz",
234
+ "rank_001/shard_000008.npz",
235
+ "rank_001/shard_000009.npz",
236
+ "rank_001/shard_000010.npz",
237
+ "rank_001/shard_000011.npz"
238
+ ],
239
+ "target_type": "score_total",
240
+ "normalization_mode": "task",
241
+ "include_demos": true,
242
+ "has_score_reward": false,
243
+ "has_score_prior": false,
244
+ "checkpoint": "None",
245
+ "checkpoint_step": 0,
246
+ "planner_type": "diffusion",
247
+ "score_source": "analytic",
248
+ "rank": 1,
249
+ "world_size": 4,
250
+ "local_task_ids": [
251
+ 1,
252
+ 5,
253
+ 9,
254
+ 13,
255
+ 17,
256
+ 21,
257
+ 25,
258
+ 29,
259
+ 33,
260
+ 37,
261
+ 41,
262
+ 45,
263
+ 49,
264
+ 53,
265
+ 57,
266
+ 61,
267
+ 65,
268
+ 69,
269
+ 73,
270
+ 77,
271
+ 81,
272
+ 85,
273
+ 89,
274
+ 93,
275
+ 97,
276
+ 101,
277
+ 105,
278
+ 109,
279
+ 113,
280
+ 117,
281
+ 121,
282
+ 125,
283
+ 129,
284
+ 133,
285
+ 137,
286
+ 141,
287
+ 145,
288
+ 149,
289
+ 153,
290
+ 157,
291
+ 161,
292
+ 165,
293
+ 169,
294
+ 173,
295
+ 177,
296
+ 181,
297
+ 185,
298
+ 189,
299
+ 193,
300
+ 197
301
+ ],
302
+ "fallbacks": {
303
+ "score_reward": "zero arrays; current analytic diffusion planner does not expose reward-score decomposition",
304
+ "score_prior": "zero arrays; current analytic diffusion planner does not expose prior-score decomposition",
305
+ "terminal_Q": "zero; trajectory evaluator currently returns only aggregate reward+bootstrap value",
306
+ "episode_step": "0; replay sampler returns subsequences without original step indices"
307
+ }
308
+ }
soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000000.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7072a9e743c8db9e814ba846596f124c1ac3b9918f763ef8e676edc3bd6d53f7
3
+ size 1608854
soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000001.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b0ca53cac08c439ee709dc92aab5b3055b82f01a694352d4520246228ea8e614
3
+ size 1621138
soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000002.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3954dd3c5d46ca8bc8b7b2489417042bc6bd3e46240b2a9eeb4a94706ccb0cf7
3
+ size 1602837
soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000003.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:33c4b923cc8fa16c9bb62b517baf09e0f11f192bba22111bda71f89433f5f397
3
+ size 1612564
soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000004.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6a02a33e780c9125d560206ca7e35aaea903481d817566673feedc0be3415c91
3
+ size 1617334
soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000005.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1d9e239547dd97d71cca1e6817a0b3ded4c98ed980d10f7bf3ee18ed994776ea
3
+ size 1609926
soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000006.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5fc8abebb8972866667a67712141b72495fcd63fd0a615fe1c5f7357011ecb2d
3
+ size 1640435
soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000007.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fc96d1e5ddbec0bae1260fe8b69669edc2e697341e889c93dec1914c69a37217
3
+ size 1725889
soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000008.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7d58037ecf3848866654337c8ff0629d72cd5b2eb549857e1e50f79c6b258cca
3
+ size 1742141
soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000009.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c78576b6d0f2515ba46961f7b3833b9fadac4f521385d03c077cac045e19ae9c
3
+ size 1900665
soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000010.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8cfafe6dffa7b76622af01dd715b31be8321eada35b334aace1105f535522c71
3
+ size 1971435
soup_L_100M_rerank2_work_dir/score_dataset/rank_001/shard_000011.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:64abb5f77b81445f0ee7af58a3d969d7f1f208a5dec5beb2d7df02accf10e89f
3
+ size 1152696