Hugging Face's logo

The11One
/
diffusion_policy_test

Safetensors
Model card Files
xet
 Community
The11One commited on
Commit
05c277d
·
verified ·
1 Parent(s): 49cdf51

Upload full pretrained model and eval_info.json

Browse files
Files changed (8) hide show
  1. config.json +82 -0
  2. eval_info.json +1568 -0
  3. model.safetensors +3 -0
  4. policy_postprocessor.json +32 -0
  5. policy_postprocessor_step_0_unnormalizer_processor.safetensors +3 -0
  6. policy_preprocessor.json +56 -0
  7. policy_preprocessor_step_3_normalizer_processor.safetensors +3 -0
  8. train_config.json +250 -0
config.json ADDED
@@ -0,0 +1,82 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "type": "diffusion",
3
+ "n_obs_steps": 2,
4
+ "input_features": {
5
+ "observation.image": {
6
+ "type": "VISUAL",
7
+ "shape": [
8
+ 3,
9
+ 96,
10
+ 96
11
+ ]
12
+ },
13
+ "observation.state": {
14
+ "type": "STATE",
15
+ "shape": [
16
+ 2
17
+ ]
18
+ }
19
+ },
20
+ "output_features": {
21
+ "action": {
22
+ "type": "ACTION",
23
+ "shape": [
24
+ 2
25
+ ]
26
+ }
27
+ },
28
+ "device": "cuda",
29
+ "use_amp": false,
30
+ "push_to_hub": true,
31
+ "repo_id": "The11One/diffusion_policy_test",
32
+ "private": null,
33
+ "tags": null,
34
+ "license": null,
35
+ "pretrained_path": null,
36
+ "horizon": 16,
37
+ "n_action_steps": 8,
38
+ "normalization_mapping": {
39
+ "VISUAL": "MEAN_STD",
40
+ "STATE": "MIN_MAX",
41
+ "ACTION": "MIN_MAX"
42
+ },
43
+ "drop_n_last_frames": 7,
44
+ "vision_backbone": "resnet18",
45
+ "crop_shape": [
46
+ 84,
47
+ 84
48
+ ],
49
+ "crop_is_random": true,
50
+ "pretrained_backbone_weights": null,
51
+ "use_group_norm": true,
52
+ "spatial_softmax_num_keypoints": 32,
53
+ "use_separate_rgb_encoder_per_camera": false,
54
+ "down_dims": [
55
+ 512,
56
+ 1024,
57
+ 2048
58
+ ],
59
+ "kernel_size": 5,
60
+ "n_groups": 8,
61
+ "diffusion_step_embed_dim": 128,
62
+ "use_film_scale_modulation": true,
63
+ "noise_scheduler_type": "DDPM",
64
+ "num_train_timesteps": 100,
65
+ "beta_schedule": "squaredcos_cap_v2",
66
+ "beta_start": 0.0001,
67
+ "beta_end": 0.02,
68
+ "prediction_type": "epsilon",
69
+ "clip_sample": true,
70
+ "clip_sample_range": 1.0,
71
+ "num_inference_steps": null,
72
+ "do_mask_loss_for_padding": false,
73
+ "optimizer_lr": 0.0001,
74
+ "optimizer_betas": [
75
+ 0.95,
76
+ 0.999
77
+ ],
78
+ "optimizer_eps": 1e-08,
79
+ "optimizer_weight_decay": 1e-06,
80
+ "scheduler_name": "cosine",
81
+ "scheduler_warmup_steps": 500
82
+ }
eval_info.json ADDED
@@ -0,0 +1,1568 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "per_task": [
3
+ {
4
+ "task_group": "pusht",
5
+ "task_id": 0,
6
+ "metrics": {
7
+ "sum_rewards": [
8
+ 111.56404697410203,
9
+ 193.61793571804296,
10
+ 103.83318941995799,
11
+ 45.46605427527467,
12
+ 66.12045047844286,
13
+ 89.63558011264348,
14
+ 36.265570111315704,
15
+ 78.00541255278657,
16
+ 169.64589028378845,
17
+ 171.57484023580787,
18
+ 231.0467489085503,
19
+ 56.77865916273342,
20
+ 197.60199922637196,
21
+ 145.49729640446955,
22
+ 110.45184191002448,
23
+ 37.56729591507454,
24
+ 104.58098270543314,
25
+ 171.66839653196325,
26
+ 205.8792158365526,
27
+ 162.9799839747621,
28
+ 38.55032407001241,
29
+ 154.2663389337842,
30
+ 28.796632682509777,
31
+ 204.34801602523393,
32
+ 131.68095477811266,
33
+ 32.303619635140976,
34
+ 101.0520763956312,
35
+ 163.05983167814782,
36
+ 32.761231945352066,
37
+ 135.42394011302622,
38
+ 63.09736549439866,
39
+ 169.1267890304343,
40
+ 73.09232506981851,
41
+ 135.3532101910123,
42
+ 45.454132032069495,
43
+ 165.27378061014605,
44
+ 109.43746741839766,
45
+ 187.70777735663603,
46
+ 83.25975451964379,
47
+ 115.74788071342518,
48
+ 53.07836173631563,
49
+ 194.15075376257371,
50
+ 52.22725579148266,
51
+ 39.19938790407086,
52
+ 42.69423309265051,
53
+ 36.39515267311374,
54
+ 28.75094109742784,
55
+ 212.38492489019006,
56
+ 152.6723197931053,
57
+ 209.25791412256802,
58
+ 55.50162962095789,
59
+ 123.83306250859104,
60
+ 37.28037257208147,
61
+ 148.72313640049288,
62
+ 200.06693045547226,
63
+ 78.3216490970312,
64
+ 64.99344350157583,
65
+ 62.432547979412526,
66
+ 63.6357497324675,
67
+ 68.6253373682313,
68
+ 186.43209179496645,
69
+ 59.64827363969116,
70
+ 170.22990819092215,
71
+ 140.5593812265399,
72
+ 42.45979426845416,
73
+ 63.68997551487708,
74
+ 23.851256428973013,
75
+ 65.8595295790024,
76
+ 133.63024721194645,
77
+ 236.34449022282578,
78
+ 169.9212047508988,
79
+ 80.70919884626818,
80
+ 126.35304632411194,
81
+ 34.18952488762126,
82
+ 115.01944378464012,
83
+ 108.07180853727519,
84
+ 159.0933965222437,
85
+ 66.19945217498287,
86
+ 109.96909993769074,
87
+ 85.71511874384564,
88
+ 132.54459046208154,
89
+ 82.67135672385137,
90
+ 59.19470828542782,
91
+ 200.40517829326052,
92
+ 85.10068547583097,
93
+ 103.1089730495155,
94
+ 209.45175177778535,
95
+ 31.173051572690543,
96
+ 130.8613617822068,
97
+ 208.00691574080076,
98
+ 197.5999112836621,
99
+ 99.98699711361758,
100
+ 214.94568704365835,
101
+ 201.36026188810177,
102
+ 71.785188610893,
103
+ 215.40076724595673,
104
+ 47.363206786808846,
105
+ 88.24387817694891,
106
+ 185.55155993977516,
107
+ 28.24432681169033,
108
+ 174.7647941197012,
109
+ 187.27786474779433,
110
+ 165.78742963535737,
111
+ 189.5895635490848,
112
+ 30.391873646648662,
113
+ 269.0107726002071,
114
+ 102.95933180221982,
115
+ 43.45829063207225,
116
+ 228.88735465353415,
117
+ 202.12059343868754,
118
+ 39.118449615199154,
119
+ 25.528781420850745,
120
+ 30.080614966948524,
121
+ 169.02148382298003,
122
+ 91.25255854068516,
123
+ 93.80481038171948,
124
+ 124.43288747908875,
125
+ 75.14116800062294,
126
+ 61.98210967420777,
127
+ 157.9577643467993,
128
+ 213.9502007266178,
129
+ 99.47607648090386,
130
+ 232.40342352045909,
131
+ 109.44170227192319,
132
+ 38.71187609853576,
133
+ 48.63765029655586,
134
+ 140.3766123935633,
135
+ 240.17248285723537,
136
+ 17.636172715859857,
137
+ 102.04260315039033,
138
+ 29.042911774525273,
139
+ 107.28625539182542,
140
+ 152.33715844904606,
141
+ 209.43047308075148,
142
+ 82.11216834275156,
143
+ 104.44902112027631,
144
+ 120.31575729224733,
145
+ 191.98373948788435,
146
+ 174.24193971105888,
147
+ 160.32819767240557,
148
+ 139.76250571554382,
149
+ 152.79673094577305,
150
+ 34.572787713251806,
151
+ 77.20099604432235,
152
+ 140.77349500848604,
153
+ 47.826697012734144,
154
+ 245.0941931934138,
155
+ 112.4866959012123,
156
+ 207.88710465082494,
157
+ 32.785318147056486,
158
+ 117.01641876264101,
159
+ 85.88157488546005,
160
+ 111.21654327483805,
161
+ 34.81246693041669,
162
+ 57.363093912431786,
163
+ 185.8514391307881,
164
+ 69.95160427694442,
165
+ 115.18634832856124,
166
+ 55.37841782430496,
167
+ 179.00047267017084,
168
+ 101.0507906793432,
169
+ 200.18336535749572,
170
+ 23.565394889608555,
171
+ 183.93893540116343,
172
+ 22.535404337430776,
173
+ 39.02153267901673,
174
+ 38.159927614661825,
175
+ 98.03458373613715,
176
+ 199.2069679784024,
177
+ 220.22131277096628,
178
+ 49.3015008142878,
179
+ 138.09768070892798,
180
+ 121.4745645112836,
181
+ 180.65989902089947,
182
+ 175.25465140706058,
183
+ 149.54857440560585,
184
+ 125.42158400564405,
185
+ 119.6733501215633,
186
+ 144.4445827830086,
187
+ 225.6417130439008,
188
+ 60.72385970789064,
189
+ 94.17168620700785,
190
+ 49.13012188761342,
191
+ 49.69752044941207,
192
+ 79.24059603181004,
193
+ 76.4922892407886,
194
+ 123.0978560314125,
195
+ 52.622588127872845,
196
+ 130.3036266939617,
197
+ 169.14286758106897,
198
+ 70.43514147555052,
199
+ 68.92096672287109,
200
+ 204.56008463892064,
201
+ 82.07503595439618,
202
+ 46.017504761615186,
203
+ 123.03010568129005,
204
+ 39.49925890025408,
205
+ 165.44819455398147,
206
+ 34.817264127175335,
207
+ 102.60501483058566,
208
+ 81.19482688629739,
209
+ 74.78763183295473,
210
+ 139.49087067654966,
211
+ 187.29726491420695,
212
+ 142.45095176576046,
213
+ 99.3993481182584,
214
+ 101.20462947796068,
215
+ 171.26287732328865,
216
+ 160.58078159746407,
217
+ 120.3154328065141,
218
+ 51.704982749914585,
219
+ 45.45710754528844,
220
+ 16.530065465840853,
221
+ 195.76937405123624,
222
+ 49.31599940660915,
223
+ 59.324077192116086,
224
+ 133.88306145837421,
225
+ 91.45882478782535,
226
+ 47.07139072926795,
227
+ 61.534708379042925,
228
+ 65.37856013308087,
229
+ 55.31581170678312,
230
+ 95.27285448475506,
231
+ 85.99955960415332,
232
+ 62.36249475198547,
233
+ 84.06926416361584,
234
+ 104.77523908742218,
235
+ 219.53041342081286,
236
+ 156.60377716431583,
237
+ 137.18187243068246,
238
+ 219.52019230115752,
239
+ 116.09550257588623,
240
+ 50.75487529720983,
241
+ 239.78993660489718,
242
+ 82.73334663724155,
243
+ 130.0232574314436,
244
+ 197.61385738962014,
245
+ 36.39124248408886,
246
+ 192.11894486518102,
247
+ 203.15111137260223,
248
+ 75.00138567318278,
249
+ 21.060361968543823,
250
+ 115.09417253945027,
251
+ 120.5534045311965,
252
+ 123.29428938107377,
253
+ 151.47352735853278,
254
+ 34.7341290258795,
255
+ 106.01151446070801,
256
+ 177.76339687708182,
257
+ 32.655869035865926,
258
+ 101.70074612736336,
259
+ 186.68058276185607,
260
+ 227.14366866331187,
261
+ 88.64476386762946,
262
+ 48.52599392092667,
263
+ 125.49746719881423,
264
+ 108.13485594028184,
265
+ 207.8429875063205,
266
+ 164.05851097732835,
267
+ 67.14450182773092,
268
+ 98.97333498665972,
269
+ 90.76366780341345,
270
+ 90.97542378486395,
271
+ 28.151781147875706,
272
+ 187.25485717913256,
273
+ 194.20599795022264,
274
+ 226.84640629410978,
275
+ 167.6188274719189,
276
+ 193.64453487599573,
277
+ 105.86140580170834,
278
+ 84.97285441624578,
279
+ 176.94764109938862,
280
+ 107.53114350406818,
281
+ 95.21145748211428,
282
+ 118.35223006644739,
283
+ 36.69134969813447,
284
+ 160.26711480522937,
285
+ 84.99926320637317,
286
+ 239.03582803889665,
287
+ 58.39321483814956,
288
+ 101.2940643411973,
289
+ 60.58637636801387,
290
+ 79.98731767389475,
291
+ 109.28503334911161,
292
+ 58.88966726141709,
293
+ 208.96969598784705,
294
+ 43.42544299175565,
295
+ 58.949415088340245,
296
+ 36.78546115385822,
297
+ 216.63549466981496,
298
+ 72.20073624399859,
299
+ 186.2376843346133,
300
+ 68.7333724998454,
301
+ 191.2550178131566,
302
+ 85.16545661684641,
303
+ 163.1359793692307,
304
+ 51.45094950041769,
305
+ 91.5772878888893,
306
+ 149.29047525683964,
307
+ 111.58860266646658,
308
+ 126.95624726214561,
309
+ 206.97787912204404,
310
+ 69.86678910385899,
311
+ 64.74013110078837,
312
+ 76.94592640360628,
313
+ 201.88503485555867,
314
+ 43.162456352205155,
315
+ 52.0759359432382,
316
+ 193.7296616084568,
317
+ 193.94817940442317,
318
+ 207.75276771742492,
319
+ 194.636170010645,
320
+ 62.982349601062865,
321
+ 203.88485896461992,
322
+ 132.22067800905398,
323
+ 37.572917039570434,
324
+ 162.42871721604098,
325
+ 120.6439839206654,
326
+ 60.55183138276764,
327
+ 116.11477283785706,
328
+ 51.48708223189002,
329
+ 80.00481829132951,
330
+ 67.0028507821228,
331
+ 198.0973803765421,
332
+ 113.59230809676903,
333
+ 35.98369901934831,
334
+ 20.20486278864099,
335
+ 96.80549448301161,
336
+ 149.48725332368412,
337
+ 113.52031574990096,
338
+ 43.17738228325593,
339
+ 51.06307359755249,
340
+ 156.08277679633696,
341
+ 220.11481081466985,
342
+ 82.20215198076622,
343
+ 150.40271711722212,
344
+ 71.1366730517958,
345
+ 36.40019922474994,
346
+ 165.10991650451507,
347
+ 116.65766691076081,
348
+ 49.078997457761794,
349
+ 134.40304814484995,
350
+ 31.578195131803326,
351
+ 233.40700070795359,
352
+ 204.99023228981707,
353
+ 148.164555352293,
354
+ 241.255343315738,
355
+ 60.98627433720393,
356
+ 127.82702499396333,
357
+ 227.3571183358508,
358
+ 28.794257780138793,
359
+ 43.36820094121288,
360
+ 105.59820132602282,
361
+ 214.81226651339438,
362
+ 139.2565693389518,
363
+ 178.5460110857902,
364
+ 87.53345744807285,
365
+ 34.434950733077756,
366
+ 135.23980074115838,
367
+ 168.04013514728274,
368
+ 28.201672600291605,
369
+ 69.96534746658308,
370
+ 45.58187021493082,
371
+ 35.5387300073765,
372
+ 16.973338774510978,
373
+ 210.16477132266812,
374
+ 102.4042685068836,
375
+ 106.01124749960096,
376
+ 190.14158959638695,
377
+ 144.62303093402545,
378
+ 125.2240710170933,
379
+ 190.0749444357752,
380
+ 103.09615945587663,
381
+ 161.75168579233815,
382
+ 169.74271902995747,
383
+ 115.86131021436381,
384
+ 15.078243194909296,
385
+ 42.53971669663175,
386
+ 104.76916069103427,
387
+ 66.77227207853159,
388
+ 80.6390124332766,
389
+ 54.280131723248616,
390
+ 54.354810934329045,
391
+ 141.99714731296956,
392
+ 223.21857109330466,
393
+ 53.41647982669441,
394
+ 53.9526540410693,
395
+ 41.33155078780479,
396
+ 76.35960423682977,
397
+ 54.127134893240154,
398
+ 131.16683181511945,
399
+ 118.8151341016286,
400
+ 140.2121491518282,
401
+ 34.52694405915425,
402
+ 44.39972620009464,
403
+ 115.70620400005343,
404
+ 110.31838573103192,
405
+ 84.2919020885869,
406
+ 70.1122357696515,
407
+ 133.69556505874968,
408
+ 34.48359797940053,
409
+ 166.53546557560423,
410
+ 39.10985931377145,
411
+ 55.20978061495294,
412
+ 161.07317105021872,
413
+ 206.07079331308364,
414
+ 37.1235773407208,
415
+ 41.09821224289214,
416
+ 35.96059240011851,
417
+ 46.12202160898013,
418
+ 51.97431514222171,
419
+ 101.35704194794289,
420
+ 174.58934081042116,
421
+ 87.85944248980151,
422
+ 187.0023753300431,
423
+ 42.092136847875985,
424
+ 155.3891294741054,
425
+ 136.48813285128838,
426
+ 69.35579132018812,
427
+ 19.04613604523439,
428
+ 104.23800346313868,
429
+ 34.86384787336185,
430
+ 39.87969612373443,
431
+ 48.62042085561159,
432
+ 120.03993825622578,
433
+ 140.05388842998417,
434
+ 156.0267984976452,
435
+ 166.99009960202582,
436
+ 176.2542070159243,
437
+ 56.14856603910687,
438
+ 85.39386807188717,
439
+ 112.207527334164,
440
+ 202.5555611501983,
441
+ 80.94270098118211,
442
+ 21.55578238368766,
443
+ 18.741187773979306,
444
+ 23.199014381924954,
445
+ 61.36289899605849,
446
+ 64.90109395257267,
447
+ 39.458803471310304,
448
+ 228.44406951761482,
449
+ 207.2239585052551,
450
+ 194.8085930168615,
451
+ 88.40544175533638,
452
+ 48.24841341999062,
453
+ 42.90559461447896,
454
+ 72.65554298652555,
455
+ 251.6951069032997,
456
+ 34.35437683552371,
457
+ 95.41948639602394,
458
+ 106.69185575929208,
459
+ 14.140771392653251,
460
+ 159.37132405670806,
461
+ 161.04160476279213,
462
+ 149.5762009508232,
463
+ 120.79046264790637,
464
+ 58.0605032693238,
465
+ 65.81178621246481,
466
+ 37.667683988386614,
467
+ 191.8230048837001,
468
+ 66.91246201678099,
469
+ 121.76815334374626,
470
+ 44.603814639574836,
471
+ 118.06011304053749,
472
+ 191.48131366809986,
473
+ 66.39826870120233,
474
+ 97.48451493813636,
475
+ 46.88175955607326,
476
+ 31.660153752685822,
477
+ 204.75351122444388,
478
+ 206.3672392826985,
479
+ 190.81564023588604,
480
+ 42.05826095435562,
481
+ 135.02571932985205,
482
+ 208.67788527118475,
483
+ 60.7756085595493,
484
+ 65.01837855827523,
485
+ 50.3671107216645,
486
+ 116.86714764794857,
487
+ 32.53569098419721,
488
+ 0.0,
489
+ 91.38345454194744,
490
+ 56.923888438583354,
491
+ 84.13361414431891,
492
+ 136.60953602590456,
493
+ 137.70219392528344,
494
+ 132.3691164661548,
495
+ 43.41577070048635,
496
+ 37.19597869564389,
497
+ 91.14623752655982,
498
+ 57.22771476452916,
499
+ 75.78399437663693,
500
+ 176.43850547120044,
501
+ 104.508279675951,
502
+ 53.58291035011527,
503
+ 38.14028626331286,
504
+ 116.6810320354511,
505
+ 196.80912998521103,
506
+ 65.30644389280707,
507
+ 230.92975180119703
508
+ ],
509
+ "max_rewards": [
510
+ 1.0,
511
+ 0.9931066694958849,
512
+ 1.0,
513
+ 1.0,
514
+ 1.0,
515
+ 1.0,
516
+ 1.0,
517
+ 1.0,
518
+ 1.0,
519
+ 0.9730314648785043,
520
+ 0.9539243849518481,
521
+ 1.0,
522
+ 0.9888783556031512,
523
+ 0.9953896784835351,
524
+ 1.0,
525
+ 1.0,
526
+ 1.0,
527
+ 0.9815438641054443,
528
+ 1.0,
529
+ 1.0,
530
+ 1.0,
531
+ 0.6207976681772706,
532
+ 1.0,
533
+ 0.9913176593843067,
534
+ 1.0,
535
+ 1.0,
536
+ 0.9837987088689247,
537
+ 0.9946086271075304,
538
+ 0.4834204861114093,
539
+ 1.0,
540
+ 1.0,
541
+ 0.9441208295601562,
542
+ 1.0,
543
+ 0.8118380454070521,
544
+ 1.0,
545
+ 1.0,
546
+ 1.0,
547
+ 0.9902060832997713,
548
+ 1.0,
549
+ 1.0,
550
+ 1.0,
551
+ 0.9908777352101551,
552
+ 1.0,
553
+ 1.0,
554
+ 1.0,
555
+ 1.0,
556
+ 1.0,
557
+ 0.9583942272574208,
558
+ 1.0,
559
+ 0.9805563172627707,
560
+ 1.0,
561
+ 0.9899661647676871,
562
+ 1.0,
563
+ 0.9701902203547514,
564
+ 0.9894593700686894,
565
+ 1.0,
566
+ 1.0,
567
+ 1.0,
568
+ 1.0,
569
+ 0.9829329205014047,
570
+ 0.9921247102127568,
571
+ 1.0,
572
+ 0.9121012551698987,
573
+ 1.0,
574
+ 1.0,
575
+ 1.0,
576
+ 1.0,
577
+ 1.0,
578
+ 1.0,
579
+ 0.9945814415498622,
580
+ 0.633518603708999,
581
+ 1.0,
582
+ 0.982174304073799,
583
+ 1.0,
584
+ 0.6801087762615218,
585
+ 1.0,
586
+ 0.9433058513611151,
587
+ 1.0,
588
+ 1.0,
589
+ 1.0,
590
+ 1.0,
591
+ 1.0,
592
+ 1.0,
593
+ 0.9940882601292587,
594
+ 0.3107914407274139,
595
+ 1.0,
596
+ 1.0,
597
+ 1.0,
598
+ 1.0,
599
+ 0.9901286898741118,
600
+ 0.9875376626351711,
601
+ 1.0,
602
+ 0.9979458376926759,
603
+ 0.9975274369304119,
604
+ 1.0,
605
+ 0.9627283602132803,
606
+ 1.0,
607
+ 1.0,
608
+ 1.0,
609
+ 1.0,
610
+ 0.999582372462522,
611
+ 0.956330850662828,
612
+ 1.0,
613
+ 0.9889799078387158,
614
+ 1.0,
615
+ 0.994602617864113,
616
+ 1.0,
617
+ 1.0,
618
+ 0.9678613987001068,
619
+ 0.979891903449822,
620
+ 1.0,
621
+ 1.0,
622
+ 1.0,
623
+ 1.0,
624
+ 1.0,
625
+ 1.0,
626
+ 1.0,
627
+ 1.0,
628
+ 1.0,
629
+ 0.9494298023548124,
630
+ 0.9693258276387066,
631
+ 1.0,
632
+ 0.9841909640403446,
633
+ 0.9981104613944847,
634
+ 1.0,
635
+ 1.0,
636
+ 0.991956533549058,
637
+ 0.9902396968708657,
638
+ 1.0,
639
+ 1.0,
640
+ 1.0,
641
+ 1.0,
642
+ 0.9932292945809599,
643
+ 0.9612991994791861,
644
+ 1.0,
645
+ 1.0,
646
+ 0.7096266571564678,
647
+ 0.992882991687916,
648
+ 1.0,
649
+ 0.9492713153850485,
650
+ 0.9938247138531937,
651
+ 0.9909258857775339,
652
+ 1.0,
653
+ 0.9708803350276102,
654
+ 1.0,
655
+ 1.0,
656
+ 0.978852280364711,
657
+ 0.7199332401876607,
658
+ 0.9820487971629225,
659
+ 1.0,
660
+ 1.0,
661
+ 1.0,
662
+ 1.0,
663
+ 1.0,
664
+ 1.0,
665
+ 0.9663946826946822,
666
+ 1.0,
667
+ 0.9937850838629276,
668
+ 1.0,
669
+ 1.0,
670
+ 0.6566603847779996,
671
+ 0.962156059059947,
672
+ 1.0,
673
+ 0.9998171384175382,
674
+ 1.0,
675
+ 1.0,
676
+ 1.0,
677
+ 1.0,
678
+ 0.9907364928824642,
679
+ 0.9815599839917968,
680
+ 0.9671171062746892,
681
+ 1.0,
682
+ 1.0,
683
+ 0.972616959301054,
684
+ 1.0,
685
+ 0.966075057371971,
686
+ 1.0,
687
+ 0.999358526502677,
688
+ 1.0,
689
+ 0.9962582985310074,
690
+ 1.0,
691
+ 1.0,
692
+ 1.0,
693
+ 1.0,
694
+ 1.0,
695
+ 0.9877690174510755,
696
+ 1.0,
697
+ 1.0,
698
+ 1.0,
699
+ 1.0,
700
+ 1.0,
701
+ 1.0,
702
+ 0.9905010102191842,
703
+ 1.0,
704
+ 1.0,
705
+ 0.9924197268562222,
706
+ 1.0,
707
+ 0.983461831738217,
708
+ 1.0,
709
+ 0.8985700249187338,
710
+ 0.9903337979619757,
711
+ 1.0,
712
+ 0.9922163734864086,
713
+ 1.0,
714
+ 0.95945710282012,
715
+ 1.0,
716
+ 0.9821804912630292,
717
+ 1.0,
718
+ 0.7093536981700919,
719
+ 0.9916793078207446,
720
+ 1.0,
721
+ 1.0,
722
+ 1.0,
723
+ 0.9835867569883742,
724
+ 1.0,
725
+ 1.0,
726
+ 0.93464192851786,
727
+ 1.0,
728
+ 1.0,
729
+ 1.0,
730
+ 1.0,
731
+ 1.0,
732
+ 1.0,
733
+ 0.6647828948812543,
734
+ 1.0,
735
+ 1.0,
736
+ 1.0,
737
+ 0.9811302744215781,
738
+ 1.0,
739
+ 1.0,
740
+ 0.9887526357495251,
741
+ 1.0,
742
+ 0.2859541156406985,
743
+ 0.9912610212613476,
744
+ 1.0,
745
+ 1.0,
746
+ 0.9736225492782914,
747
+ 1.0,
748
+ 0.961766248753535,
749
+ 0.9897000225830414,
750
+ 1.0,
751
+ 1.0,
752
+ 0.697417932320667,
753
+ 1.0,
754
+ 0.9594881574575931,
755
+ 1.0,
756
+ 1.0,
757
+ 1.0,
758
+ 0.9907635985106943,
759
+ 1.0,
760
+ 1.0,
761
+ 0.9737380311057238,
762
+ 0.9968505314775506,
763
+ 1.0,
764
+ 1.0,
765
+ 1.0,
766
+ 1.0,
767
+ 0.9994543724769545,
768
+ 1.0,
769
+ 1.0,
770
+ 0.9583096202259439,
771
+ 1.0,
772
+ 1.0,
773
+ 1.0,
774
+ 0.9766591049754804,
775
+ 0.9817073138565053,
776
+ 0.9887522510953808,
777
+ 0.9835882738276024,
778
+ 0.9980179701638549,
779
+ 1.0,
780
+ 1.0,
781
+ 0.944164523614217,
782
+ 1.0,
783
+ 1.0,
784
+ 0.9634871647006592,
785
+ 1.0,
786
+ 0.9809443447888401,
787
+ 1.0,
788
+ 0.9999138315601348,
789
+ 1.0,
790
+ 1.0,
791
+ 1.0,
792
+ 1.0,
793
+ 1.0,
794
+ 1.0,
795
+ 0.983707858017457,
796
+ 1.0,
797
+ 1.0,
798
+ 1.0,
799
+ 0.9858604990007879,
800
+ 1.0,
801
+ 0.9673436827288483,
802
+ 0.49862027098531114,
803
+ 0.9830639345124099,
804
+ 1.0,
805
+ 0.9909715460048645,
806
+ 1.0,
807
+ 1.0,
808
+ 0.9845377736698859,
809
+ 1.0,
810
+ 0.9668703761943288,
811
+ 0.9921517947934557,
812
+ 1.0,
813
+ 1.0,
814
+ 1.0,
815
+ 0.9885145227673452,
816
+ 1.0,
817
+ 1.0,
818
+ 0.9414887361399953,
819
+ 0.9884917944827368,
820
+ 0.9952556204677784,
821
+ 1.0,
822
+ 1.0,
823
+ 0.9982735026049552,
824
+ 1.0,
825
+ 1.0,
826
+ 0.9747601172687081,
827
+ 1.0,
828
+ 0.9714827142067812,
829
+ 0.9737904590160222,
830
+ 1.0,
831
+ 1.0,
832
+ 1.0,
833
+ 0.9625593610178751,
834
+ 0.9989764501044405,
835
+ 1.0,
836
+ 1.0,
837
+ 1.0,
838
+ 0.9809117684107029,
839
+ 0.396226012272729,
840
+ 1.0,
841
+ 1.0,
842
+ 0.9452371598682366,
843
+ 0.9213801238057235,
844
+ 1.0,
845
+ 0.9991345893959517,
846
+ 1.0,
847
+ 1.0,
848
+ 0.984527118605828,
849
+ 1.0,
850
+ 1.0,
851
+ 0.9903401691248781,
852
+ 1.0,
853
+ 0.9827331864118363,
854
+ 0.996782622135209,
855
+ 1.0,
856
+ 0.9690010510060082,
857
+ 1.0,
858
+ 1.0,
859
+ 0.9855564733680211,
860
+ 1.0,
861
+ 1.0,
862
+ 1.0,
863
+ 0.9970004692359444,
864
+ 1.0,
865
+ 0.6921713920287654,
866
+ 1.0,
867
+ 1.0,
868
+ 1.0,
869
+ 1.0,
870
+ 1.0,
871
+ 1.0,
872
+ 1.0,
873
+ 1.0,
874
+ 1.0,
875
+ 0.9605978195212138,
876
+ 1.0,
877
+ 0.4714533888859396,
878
+ 0.9903633396531539,
879
+ 1.0,
880
+ 0.7234537282128585,
881
+ 0.9720619700358029,
882
+ 1.0,
883
+ 0.9227946641809074,
884
+ 0.7207106018584281,
885
+ 1.0,
886
+ 1.0,
887
+ 1.0,
888
+ 1.0,
889
+ 1.0,
890
+ 1.0,
891
+ 1.0,
892
+ 1.0,
893
+ 0.9468403653211902,
894
+ 0.9962252740190406,
895
+ 1.0,
896
+ 0.9198736994925141,
897
+ 1.0,
898
+ 1.0,
899
+ 1.0,
900
+ 1.0,
901
+ 1.0,
902
+ 0.9983862283140997,
903
+ 1.0,
904
+ 1.0,
905
+ 1.0,
906
+ 1.0,
907
+ 1.0,
908
+ 1.0,
909
+ 0.9736781133938246,
910
+ 1.0,
911
+ 0.9877162013935828,
912
+ 1.0,
913
+ 1.0,
914
+ 1.0,
915
+ 0.9746721960746527,
916
+ 1.0,
917
+ 1.0,
918
+ 1.0,
919
+ 1.0,
920
+ 1.0,
921
+ 1.0,
922
+ 1.0,
923
+ 1.0,
924
+ 1.0,
925
+ 1.0,
926
+ 0.9319387845664227,
927
+ 0.7183413464184357,
928
+ 1.0,
929
+ 1.0,
930
+ 1.0,
931
+ 1.0,
932
+ 0.8592849987023707,
933
+ 0.44067578942279484,
934
+ 1.0,
935
+ 0.9860322019984729,
936
+ 0.9923727153554253,
937
+ 0.9251241499129295,
938
+ 0.9953228000983328,
939
+ 1.0,
940
+ 1.0,
941
+ 1.0,
942
+ 0.990844864423546,
943
+ 1.0,
944
+ 1.0,
945
+ 1.0,
946
+ 1.0,
947
+ 0.6233805041479217,
948
+ 1.0,
949
+ 1.0,
950
+ 0.9254754282360146,
951
+ 0.991617467868987,
952
+ 1.0,
953
+ 1.0,
954
+ 1.0,
955
+ 1.0,
956
+ 1.0,
957
+ 0.9720041064154187,
958
+ 1.0,
959
+ 0.983313660468886,
960
+ 1.0,
961
+ 1.0,
962
+ 0.972921121804848,
963
+ 0.9609339628859167,
964
+ 0.9635560735428206,
965
+ 1.0,
966
+ 1.0,
967
+ 1.0,
968
+ 1.0,
969
+ 1.0,
970
+ 1.0,
971
+ 1.0,
972
+ 1.0,
973
+ 0.9778894505953262,
974
+ 0.9971831468168342,
975
+ 1.0,
976
+ 0.9777099679149392,
977
+ 1.0,
978
+ 1.0,
979
+ 0.9910242210538476,
980
+ 0.9776351019929075,
981
+ 1.0,
982
+ 1.0,
983
+ 1.0,
984
+ 0.9886155586298592,
985
+ 0.9406849255916336,
986
+ 1.0,
987
+ 1.0,
988
+ 1.0,
989
+ 1.0,
990
+ 0.0,
991
+ 1.0,
992
+ 1.0,
993
+ 1.0,
994
+ 1.0,
995
+ 0.9762122816290172,
996
+ 1.0,
997
+ 1.0,
998
+ 1.0,
999
+ 0.9524850685935963,
1000
+ 0.5174889651469767,
1001
+ 1.0,
1002
+ 0.9436587439275765,
1003
+ 1.0,
1004
+ 1.0,
1005
+ 1.0,
1006
+ 0.5541429545332189,
1007
+ 0.9876401158046316,
1008
+ 1.0,
1009
+ 0.9897962089913616
1010
+ ],
1011
+ "successes": [
1012
+ true,
1013
+ false,
1014
+ true,
1015
+ true,
1016
+ true,
1017
+ true,
1018
+ true,
1019
+ true,
1020
+ true,
1021
+ false,
1022
+ false,
1023
+ true,
1024
+ false,
1025
+ false,
1026
+ true,
1027
+ true,
1028
+ true,
1029
+ false,
1030
+ true,
1031
+ true,
1032
+ true,
1033
+ false,
1034
+ true,
1035
+ false,
1036
+ true,
1037
+ true,
1038
+ false,
1039
+ false,
1040
+ false,
1041
+ true,
1042
+ true,
1043
+ false,
1044
+ true,
1045
+ false,
1046
+ true,
1047
+ true,
1048
+ true,
1049
+ false,
1050
+ true,
1051
+ true,
1052
+ true,
1053
+ false,
1054
+ true,
1055
+ true,
1056
+ true,
1057
+ true,
1058
+ true,
1059
+ false,
1060
+ true,
1061
+ false,
1062
+ true,
1063
+ false,
1064
+ true,
1065
+ false,
1066
+ false,
1067
+ true,
1068
+ true,
1069
+ true,
1070
+ true,
1071
+ false,
1072
+ false,
1073
+ true,
1074
+ false,
1075
+ true,
1076
+ true,
1077
+ true,
1078
+ true,
1079
+ true,
1080
+ true,
1081
+ false,
1082
+ false,
1083
+ true,
1084
+ false,
1085
+ true,
1086
+ false,
1087
+ true,
1088
+ false,
1089
+ true,
1090
+ true,
1091
+ true,
1092
+ true,
1093
+ true,
1094
+ true,
1095
+ false,
1096
+ false,
1097
+ true,
1098
+ true,
1099
+ true,
1100
+ true,
1101
+ false,
1102
+ false,
1103
+ true,
1104
+ false,
1105
+ false,
1106
+ true,
1107
+ false,
1108
+ true,
1109
+ true,
1110
+ true,
1111
+ true,
1112
+ false,
1113
+ false,
1114
+ true,
1115
+ false,
1116
+ true,
1117
+ false,
1118
+ true,
1119
+ true,
1120
+ false,
1121
+ false,
1122
+ true,
1123
+ true,
1124
+ true,
1125
+ true,
1126
+ true,
1127
+ true,
1128
+ true,
1129
+ true,
1130
+ true,
1131
+ false,
1132
+ false,
1133
+ true,
1134
+ false,
1135
+ false,
1136
+ true,
1137
+ true,
1138
+ false,
1139
+ false,
1140
+ true,
1141
+ true,
1142
+ true,
1143
+ true,
1144
+ false,
1145
+ false,
1146
+ true,
1147
+ true,
1148
+ false,
1149
+ false,
1150
+ true,
1151
+ false,
1152
+ false,
1153
+ false,
1154
+ true,
1155
+ false,
1156
+ true,
1157
+ true,
1158
+ false,
1159
+ false,
1160
+ false,
1161
+ true,
1162
+ true,
1163
+ true,
1164
+ true,
1165
+ true,
1166
+ true,
1167
+ false,
1168
+ true,
1169
+ false,
1170
+ true,
1171
+ true,
1172
+ false,
1173
+ false,
1174
+ true,
1175
+ false,
1176
+ true,
1177
+ true,
1178
+ true,
1179
+ true,
1180
+ false,
1181
+ false,
1182
+ false,
1183
+ true,
1184
+ true,
1185
+ false,
1186
+ true,
1187
+ false,
1188
+ true,
1189
+ false,
1190
+ true,
1191
+ false,
1192
+ true,
1193
+ true,
1194
+ true,
1195
+ true,
1196
+ true,
1197
+ false,
1198
+ true,
1199
+ true,
1200
+ true,
1201
+ true,
1202
+ true,
1203
+ true,
1204
+ false,
1205
+ true,
1206
+ true,
1207
+ false,
1208
+ true,
1209
+ false,
1210
+ true,
1211
+ false,
1212
+ false,
1213
+ true,
1214
+ false,
1215
+ true,
1216
+ false,
1217
+ true,
1218
+ false,
1219
+ true,
1220
+ false,
1221
+ false,
1222
+ true,
1223
+ true,
1224
+ true,
1225
+ false,
1226
+ true,
1227
+ true,
1228
+ false,
1229
+ true,
1230
+ true,
1231
+ true,
1232
+ true,
1233
+ true,
1234
+ true,
1235
+ false,
1236
+ true,
1237
+ true,
1238
+ true,
1239
+ false,
1240
+ true,
1241
+ true,
1242
+ false,
1243
+ true,
1244
+ false,
1245
+ false,
1246
+ true,
1247
+ true,
1248
+ false,
1249
+ true,
1250
+ false,
1251
+ false,
1252
+ true,
1253
+ true,
1254
+ false,
1255
+ true,
1256
+ false,
1257
+ true,
1258
+ true,
1259
+ true,
1260
+ false,
1261
+ true,
1262
+ true,
1263
+ false,
1264
+ false,
1265
+ true,
1266
+ true,
1267
+ true,
1268
+ true,
1269
+ false,
1270
+ true,
1271
+ true,
1272
+ false,
1273
+ true,
1274
+ true,
1275
+ true,
1276
+ false,
1277
+ false,
1278
+ false,
1279
+ false,
1280
+ false,
1281
+ true,
1282
+ true,
1283
+ false,
1284
+ true,
1285
+ true,
1286
+ false,
1287
+ true,
1288
+ false,
1289
+ true,
1290
+ false,
1291
+ true,
1292
+ true,
1293
+ true,
1294
+ true,
1295
+ true,
1296
+ true,
1297
+ false,
1298
+ true,
1299
+ true,
1300
+ true,
1301
+ false,
1302
+ true,
1303
+ false,
1304
+ false,
1305
+ false,
1306
+ true,
1307
+ false,
1308
+ true,
1309
+ true,
1310
+ false,
1311
+ true,
1312
+ false,
1313
+ false,
1314
+ true,
1315
+ true,
1316
+ true,
1317
+ false,
1318
+ true,
1319
+ true,
1320
+ false,
1321
+ false,
1322
+ false,
1323
+ true,
1324
+ true,
1325
+ false,
1326
+ true,
1327
+ true,
1328
+ false,
1329
+ true,
1330
+ false,
1331
+ false,
1332
+ true,
1333
+ true,
1334
+ true,
1335
+ false,
1336
+ false,
1337
+ true,
1338
+ true,
1339
+ true,
1340
+ false,
1341
+ false,
1342
+ true,
1343
+ true,
1344
+ false,
1345
+ false,
1346
+ true,
1347
+ false,
1348
+ true,
1349
+ true,
1350
+ false,
1351
+ true,
1352
+ true,
1353
+ false,
1354
+ true,
1355
+ false,
1356
+ false,
1357
+ true,
1358
+ false,
1359
+ true,
1360
+ true,
1361
+ false,
1362
+ true,
1363
+ true,
1364
+ true,
1365
+ false,
1366
+ true,
1367
+ false,
1368
+ true,
1369
+ true,
1370
+ true,
1371
+ true,
1372
+ true,
1373
+ true,
1374
+ true,
1375
+ true,
1376
+ true,
1377
+ false,
1378
+ true,
1379
+ false,
1380
+ false,
1381
+ true,
1382
+ false,
1383
+ false,
1384
+ true,
1385
+ false,
1386
+ false,
1387
+ true,
1388
+ true,
1389
+ true,
1390
+ true,
1391
+ true,
1392
+ true,
1393
+ true,
1394
+ true,
1395
+ false,
1396
+ false,
1397
+ true,
1398
+ false,
1399
+ true,
1400
+ true,
1401
+ true,
1402
+ true,
1403
+ true,
1404
+ false,
1405
+ true,
1406
+ true,
1407
+ true,
1408
+ true,
1409
+ true,
1410
+ true,
1411
+ false,
1412
+ true,
1413
+ false,
1414
+ true,
1415
+ true,
1416
+ true,
1417
+ false,
1418
+ true,
1419
+ true,
1420
+ true,
1421
+ true,
1422
+ true,
1423
+ true,
1424
+ true,
1425
+ true,
1426
+ true,
1427
+ true,
1428
+ false,
1429
+ false,
1430
+ true,
1431
+ true,
1432
+ true,
1433
+ true,
1434
+ false,
1435
+ false,
1436
+ true,
1437
+ false,
1438
+ false,
1439
+ false,
1440
+ false,
1441
+ true,
1442
+ true,
1443
+ true,
1444
+ false,
1445
+ true,
1446
+ true,
1447
+ true,
1448
+ true,
1449
+ false,
1450
+ true,
1451
+ true,
1452
+ false,
1453
+ false,
1454
+ true,
1455
+ true,
1456
+ true,
1457
+ true,
1458
+ true,
1459
+ false,
1460
+ true,
1461
+ false,
1462
+ true,
1463
+ true,
1464
+ false,
1465
+ false,
1466
+ false,
1467
+ true,
1468
+ true,
1469
+ true,
1470
+ true,
1471
+ true,
1472
+ true,
1473
+ true,
1474
+ true,
1475
+ false,
1476
+ false,
1477
+ true,
1478
+ false,
1479
+ true,
1480
+ true,
1481
+ false,
1482
+ false,
1483
+ true,
1484
+ true,
1485
+ true,
1486
+ false,
1487
+ false,
1488
+ true,
1489
+ true,
1490
+ true,
1491
+ true,
1492
+ false,
1493
+ true,
1494
+ true,
1495
+ true,
1496
+ true,
1497
+ false,
1498
+ true,
1499
+ true,
1500
+ true,
1501
+ false,
1502
+ false,
1503
+ true,
1504
+ false,
1505
+ true,
1506
+ true,
1507
+ true,
1508
+ false,
1509
+ false,
1510
+ true,
1511
+ false
1512
+ ],
1513
+ "video_paths": [
1514
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_0.mp4",
1515
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_1.mp4",
1516
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_2.mp4",
1517
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_3.mp4",
1518
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_4.mp4",
1519
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_5.mp4",
1520
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_6.mp4",
1521
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_7.mp4",
1522
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_8.mp4",
1523
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_9.mp4"
1524
+ ]
1525
+ }
1526
+ }
1527
+ ],
1528
+ "per_group": {
1529
+ "pusht": {
1530
+ "avg_sum_reward": 110.94182091492434,
1531
+ "avg_max_reward": 0.9711821804259954,
1532
+ "pc_success": 63.0,
1533
+ "n_episodes": 500,
1534
+ "video_paths": [
1535
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_0.mp4",
1536
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_1.mp4",
1537
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_2.mp4",
1538
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_3.mp4",
1539
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_4.mp4",
1540
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_5.mp4",
1541
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_6.mp4",
1542
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_7.mp4",
1543
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_8.mp4",
1544
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_9.mp4"
1545
+ ]
1546
+ }
1547
+ },
1548
+ "overall": {
1549
+ "avg_sum_reward": 110.94182091492434,
1550
+ "avg_max_reward": 0.9711821804259954,
1551
+ "pc_success": 63.0,
1552
+ "n_episodes": 500,
1553
+ "eval_s": 1183.6755743026733,
1554
+ "eval_ep_s": 2.3673511490821837,
1555
+ "video_paths": [
1556
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_0.mp4",
1557
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_1.mp4",
1558
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_2.mp4",
1559
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_3.mp4",
1560
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_4.mp4",
1561
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_5.mp4",
1562
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_6.mp4",
1563
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_7.mp4",
1564
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_8.mp4",
1565
+ "outputs/eval/diffusion_pusht/175000/videos/pusht_0/eval_episode_9.mp4"
1566
+ ]
1567
+ }
1568
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:06b0565f60464ef798c0fdf3fd3864925db07d261fd15487d3fde45262ee623a
3
+ size 1050861448
policy_postprocessor.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "policy_postprocessor",
3
+ "steps": [
4
+ {
5
+ "registry_name": "unnormalizer_processor",
6
+ "config": {
7
+ "eps": 1e-08,
8
+ "features": {
9
+ "action": {
10
+ "type": "ACTION",
11
+ "shape": [
12
+ 2
13
+ ]
14
+ }
15
+ },
16
+ "norm_map": {
17
+ "VISUAL": "MEAN_STD",
18
+ "STATE": "MIN_MAX",
19
+ "ACTION": "MIN_MAX"
20
+ }
21
+ },
22
+ "state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors"
23
+ },
24
+ {
25
+ "registry_name": "device_processor",
26
+ "config": {
27
+ "device": "cpu",
28
+ "float_dtype": null
29
+ }
30
+ }
31
+ ]
32
+ }
policy_postprocessor_step_0_unnormalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bc174a7436ff5075efebe7be7cece5e5d8430658cc9b5314fb8a652eab912d96
3
+ size 4204
policy_preprocessor.json ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "policy_preprocessor",
3
+ "steps": [
4
+ {
5
+ "registry_name": "rename_observations_processor",
6
+ "config": {
7
+ "rename_map": {}
8
+ }
9
+ },
10
+ {
11
+ "registry_name": "to_batch_processor",
12
+ "config": {}
13
+ },
14
+ {
15
+ "registry_name": "device_processor",
16
+ "config": {
17
+ "device": "cuda",
18
+ "float_dtype": null
19
+ }
20
+ },
21
+ {
22
+ "registry_name": "normalizer_processor",
23
+ "config": {
24
+ "eps": 1e-08,
25
+ "features": {
26
+ "observation.image": {
27
+ "type": "VISUAL",
28
+ "shape": [
29
+ 3,
30
+ 96,
31
+ 96
32
+ ]
33
+ },
34
+ "observation.state": {
35
+ "type": "STATE",
36
+ "shape": [
37
+ 2
38
+ ]
39
+ },
40
+ "action": {
41
+ "type": "ACTION",
42
+ "shape": [
43
+ 2
44
+ ]
45
+ }
46
+ },
47
+ "norm_map": {
48
+ "VISUAL": "MEAN_STD",
49
+ "STATE": "MIN_MAX",
50
+ "ACTION": "MIN_MAX"
51
+ }
52
+ },
53
+ "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
54
+ }
55
+ ]
56
+ }
policy_preprocessor_step_3_normalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bc174a7436ff5075efebe7be7cece5e5d8430658cc9b5314fb8a652eab912d96
3
+ size 4204
train_config.json ADDED
@@ -0,0 +1,250 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": {
3
+ "repo_id": "lerobot/pusht",
4
+ "root": null,
5
+ "episodes": null,
6
+ "image_transforms": {
7
+ "enable": false,
8
+ "max_num_transforms": 3,
9
+ "random_order": false,
10
+ "tfs": {
11
+ "brightness": {
12
+ "weight": 1.0,
13
+ "type": "ColorJitter",
14
+ "kwargs": {
15
+ "brightness": [
16
+ 0.8,
17
+ 1.2
18
+ ]
19
+ }
20
+ },
21
+ "contrast": {
22
+ "weight": 1.0,
23
+ "type": "ColorJitter",
24
+ "kwargs": {
25
+ "contrast": [
26
+ 0.8,
27
+ 1.2
28
+ ]
29
+ }
30
+ },
31
+ "saturation": {
32
+ "weight": 1.0,
33
+ "type": "ColorJitter",
34
+ "kwargs": {
35
+ "saturation": [
36
+ 0.5,
37
+ 1.5
38
+ ]
39
+ }
40
+ },
41
+ "hue": {
42
+ "weight": 1.0,
43
+ "type": "ColorJitter",
44
+ "kwargs": {
45
+ "hue": [
46
+ -0.05,
47
+ 0.05
48
+ ]
49
+ }
50
+ },
51
+ "sharpness": {
52
+ "weight": 1.0,
53
+ "type": "SharpnessJitter",
54
+ "kwargs": {
55
+ "sharpness": [
56
+ 0.5,
57
+ 1.5
58
+ ]
59
+ }
60
+ },
61
+ "affine": {
62
+ "weight": 1.0,
63
+ "type": "RandomAffine",
64
+ "kwargs": {
65
+ "degrees": [
66
+ -5.0,
67
+ 5.0
68
+ ],
69
+ "translate": [
70
+ 0.05,
71
+ 0.05
72
+ ]
73
+ }
74
+ }
75
+ }
76
+ },
77
+ "revision": null,
78
+ "use_imagenet_stats": true,
79
+ "video_backend": "torchcodec",
80
+ "streaming": false
81
+ },
82
+ "env": {
83
+ "type": "pusht",
84
+ "task": "PushT-v0",
85
+ "fps": 10,
86
+ "features": {
87
+ "action": {
88
+ "type": "ACTION",
89
+ "shape": [
90
+ 2
91
+ ]
92
+ },
93
+ "agent_pos": {
94
+ "type": "STATE",
95
+ "shape": [
96
+ 2
97
+ ]
98
+ },
99
+ "pixels": {
100
+ "type": "VISUAL",
101
+ "shape": [
102
+ 384,
103
+ 384,
104
+ 3
105
+ ]
106
+ }
107
+ },
108
+ "features_map": {
109
+ "action": "action",
110
+ "agent_pos": "observation.state",
111
+ "environment_state": "observation.environment_state",
112
+ "pixels": "observation.image"
113
+ },
114
+ "max_parallel_tasks": 1,
115
+ "disable_env_checker": true,
116
+ "episode_length": 300,
117
+ "obs_type": "pixels_agent_pos",
118
+ "render_mode": "rgb_array",
119
+ "visualization_width": 384,
120
+ "visualization_height": 384,
121
+ "observation_height": 384,
122
+ "observation_width": 384
123
+ },
124
+ "policy": {
125
+ "type": "diffusion",
126
+ "n_obs_steps": 2,
127
+ "input_features": {
128
+ "observation.image": {
129
+ "type": "VISUAL",
130
+ "shape": [
131
+ 3,
132
+ 96,
133
+ 96
134
+ ]
135
+ },
136
+ "observation.state": {
137
+ "type": "STATE",
138
+ "shape": [
139
+ 2
140
+ ]
141
+ }
142
+ },
143
+ "output_features": {
144
+ "action": {
145
+ "type": "ACTION",
146
+ "shape": [
147
+ 2
148
+ ]
149
+ }
150
+ },
151
+ "device": "cuda",
152
+ "use_amp": false,
153
+ "push_to_hub": true,
154
+ "repo_id": "The11One/diffusion_policy_test",
155
+ "private": null,
156
+ "tags": null,
157
+ "license": null,
158
+ "pretrained_path": null,
159
+ "horizon": 16,
160
+ "n_action_steps": 8,
161
+ "normalization_mapping": {
162
+ "VISUAL": "MEAN_STD",
163
+ "STATE": "MIN_MAX",
164
+ "ACTION": "MIN_MAX"
165
+ },
166
+ "drop_n_last_frames": 7,
167
+ "vision_backbone": "resnet18",
168
+ "crop_shape": [
169
+ 84,
170
+ 84
171
+ ],
172
+ "crop_is_random": true,
173
+ "pretrained_backbone_weights": null,
174
+ "use_group_norm": true,
175
+ "spatial_softmax_num_keypoints": 32,
176
+ "use_separate_rgb_encoder_per_camera": false,
177
+ "down_dims": [
178
+ 512,
179
+ 1024,
180
+ 2048
181
+ ],
182
+ "kernel_size": 5,
183
+ "n_groups": 8,
184
+ "diffusion_step_embed_dim": 128,
185
+ "use_film_scale_modulation": true,
186
+ "noise_scheduler_type": "DDPM",
187
+ "num_train_timesteps": 100,
188
+ "beta_schedule": "squaredcos_cap_v2",
189
+ "beta_start": 0.0001,
190
+ "beta_end": 0.02,
191
+ "prediction_type": "epsilon",
192
+ "clip_sample": true,
193
+ "clip_sample_range": 1.0,
194
+ "num_inference_steps": null,
195
+ "do_mask_loss_for_padding": false,
196
+ "optimizer_lr": 0.0001,
197
+ "optimizer_betas": [
198
+ 0.95,
199
+ 0.999
200
+ ],
201
+ "optimizer_eps": 1e-08,
202
+ "optimizer_weight_decay": 1e-06,
203
+ "scheduler_name": "cosine",
204
+ "scheduler_warmup_steps": 500
205
+ },
206
+ "output_dir": "outputs/train/diffusion_pusht",
207
+ "job_name": "pusht_diffusion",
208
+ "resume": false,
209
+ "seed": 100000,
210
+ "num_workers": 4,
211
+ "batch_size": 64,
212
+ "steps": 200000,
213
+ "eval_freq": 25000,
214
+ "log_freq": 200,
215
+ "save_checkpoint": true,
216
+ "save_freq": 25000,
217
+ "use_policy_training_preset": true,
218
+ "optimizer": {
219
+ "type": "adam",
220
+ "lr": 0.0001,
221
+ "weight_decay": 1e-06,
222
+ "grad_clip_norm": 10.0,
223
+ "betas": [
224
+ 0.95,
225
+ 0.999
226
+ ],
227
+ "eps": 1e-08
228
+ },
229
+ "scheduler": {
230
+ "type": "diffuser",
231
+ "num_warmup_steps": 500,
232
+ "name": "cosine"
233
+ },
234
+ "eval": {
235
+ "n_episodes": 50,
236
+ "batch_size": 50,
237
+ "use_async_envs": false
238
+ },
239
+ "wandb": {
240
+ "enable": false,
241
+ "disable_artifact": false,
242
+ "project": "lerobot",
243
+ "entity": null,
244
+ "notes": null,
245
+ "run_id": null,
246
+ "mode": null
247
+ },
248
+ "checkpoint_path": null,
249
+ "rename_map": {}
250
+ }