Hugging Face's logo

RoboEval
/
roboeval-diffusion

Robotics
Safetensors
roboeval
diffusion
Model card Files
xet
 Community
helen9975 commited on
Commit
e5a11fe
·
verified ·
1 Parent(s): 85d9d65

PickSingleBookFromTable: step 005000 weights

Browse files
Files changed (7) hide show
  1. PickSingleBookFromTable/config.json +99 -0
  2. PickSingleBookFromTable/model.safetensors +3 -0
  3. PickSingleBookFromTable/policy_postprocessor.json +32 -0
  4. PickSingleBookFromTable/policy_postprocessor_step_0_unnormalizer_processor.safetensors +3 -0
  5. PickSingleBookFromTable/policy_preprocessor.json +72 -0
  6. PickSingleBookFromTable/policy_preprocessor_step_3_normalizer_processor.safetensors +3 -0
  7. PickSingleBookFromTable/train_config.json +600 -0
PickSingleBookFromTable/config.json ADDED
@@ -0,0 +1,99 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "type": "diffusion",
3
+ "n_obs_steps": 2,
4
+ "input_features": {
5
+ "observation.images.cam_head": {
6
+ "type": "VISUAL",
7
+ "shape": [
8
+ 3,
9
+ 256,
10
+ 256
11
+ ]
12
+ },
13
+ "observation.images.cam_left_wrist": {
14
+ "type": "VISUAL",
15
+ "shape": [
16
+ 3,
17
+ 256,
18
+ 256
19
+ ]
20
+ },
21
+ "observation.images.cam_right_wrist": {
22
+ "type": "VISUAL",
23
+ "shape": [
24
+ 3,
25
+ 256,
26
+ 256
27
+ ]
28
+ },
29
+ "observation.state": {
30
+ "type": "STATE",
31
+ "shape": [
32
+ 16
33
+ ]
34
+ }
35
+ },
36
+ "output_features": {
37
+ "action": {
38
+ "type": "ACTION",
39
+ "shape": [
40
+ 14
41
+ ]
42
+ }
43
+ },
44
+ "device": "cuda",
45
+ "use_amp": false,
46
+ "use_peft": false,
47
+ "push_to_hub": false,
48
+ "repo_id": null,
49
+ "private": null,
50
+ "tags": null,
51
+ "license": null,
52
+ "pretrained_path": null,
53
+ "horizon": 16,
54
+ "n_action_steps": 8,
55
+ "normalization_mapping": {
56
+ "VISUAL": "MEAN_STD",
57
+ "STATE": "MIN_MAX",
58
+ "ACTION": "MIN_MAX"
59
+ },
60
+ "drop_n_last_frames": 7,
61
+ "vision_backbone": "resnet18",
62
+ "crop_shape": [
63
+ 84,
64
+ 84
65
+ ],
66
+ "crop_is_random": true,
67
+ "pretrained_backbone_weights": null,
68
+ "use_group_norm": true,
69
+ "spatial_softmax_num_keypoints": 32,
70
+ "use_separate_rgb_encoder_per_camera": false,
71
+ "down_dims": [
72
+ 512,
73
+ 1024,
74
+ 2048
75
+ ],
76
+ "kernel_size": 5,
77
+ "n_groups": 8,
78
+ "diffusion_step_embed_dim": 128,
79
+ "use_film_scale_modulation": true,
80
+ "noise_scheduler_type": "DDPM",
81
+ "num_train_timesteps": 100,
82
+ "beta_schedule": "squaredcos_cap_v2",
83
+ "beta_start": 0.0001,
84
+ "beta_end": 0.02,
85
+ "prediction_type": "epsilon",
86
+ "clip_sample": true,
87
+ "clip_sample_range": 1.0,
88
+ "num_inference_steps": null,
89
+ "do_mask_loss_for_padding": false,
90
+ "optimizer_lr": 0.0001,
91
+ "optimizer_betas": [
92
+ 0.95,
93
+ 0.999
94
+ ],
95
+ "optimizer_eps": 1e-08,
96
+ "optimizer_weight_decay": 1e-06,
97
+ "scheduler_name": "cosine",
98
+ "scheduler_warmup_steps": 500
99
+ }
PickSingleBookFromTable/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4c318dfbfd006494c4484148806a0926ba3432bfe78c17b7848746a3a96f7b4c
3
+ size 1083604936
PickSingleBookFromTable/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
+ 14
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
+ }
PickSingleBookFromTable/policy_postprocessor_step_0_unnormalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce357734e4942a37891e12d1d1d546efa14f70d156003dd33255010a73c28c0c
3
+ size 16036
PickSingleBookFromTable/policy_preprocessor.json ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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.images.cam_head": {
27
+ "type": "VISUAL",
28
+ "shape": [
29
+ 3,
30
+ 256,
31
+ 256
32
+ ]
33
+ },
34
+ "observation.images.cam_left_wrist": {
35
+ "type": "VISUAL",
36
+ "shape": [
37
+ 3,
38
+ 256,
39
+ 256
40
+ ]
41
+ },
42
+ "observation.images.cam_right_wrist": {
43
+ "type": "VISUAL",
44
+ "shape": [
45
+ 3,
46
+ 256,
47
+ 256
48
+ ]
49
+ },
50
+ "observation.state": {
51
+ "type": "STATE",
52
+ "shape": [
53
+ 16
54
+ ]
55
+ },
56
+ "action": {
57
+ "type": "ACTION",
58
+ "shape": [
59
+ 14
60
+ ]
61
+ }
62
+ },
63
+ "norm_map": {
64
+ "VISUAL": "MEAN_STD",
65
+ "STATE": "MIN_MAX",
66
+ "ACTION": "MIN_MAX"
67
+ }
68
+ },
69
+ "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
70
+ }
71
+ ]
72
+ }
PickSingleBookFromTable/policy_preprocessor_step_3_normalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce357734e4942a37891e12d1d1d546efa14f70d156003dd33255010a73c28c0c
3
+ size 16036
PickSingleBookFromTable/train_config.json ADDED
@@ -0,0 +1,600 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": {
3
+ "repo_id": "roboeval_ee_delta_20hz_v3_merged",
4
+ "root": "/weka/helenw/data/huggingface/lerobot/roboeval_ee_delta_20hz_v3_merged",
5
+ "episodes": [
6
+ 1521,
7
+ 1522,
8
+ 1523,
9
+ 1524,
10
+ 1525,
11
+ 1526,
12
+ 1527,
13
+ 1528,
14
+ 1529,
15
+ 1530,
16
+ 1531,
17
+ 1532,
18
+ 1533,
19
+ 1534,
20
+ 1535,
21
+ 1536,
22
+ 1537,
23
+ 1538,
24
+ 1539,
25
+ 1540,
26
+ 1541,
27
+ 1542,
28
+ 1543,
29
+ 1544,
30
+ 1545,
31
+ 1546,
32
+ 1547,
33
+ 1548,
34
+ 1549,
35
+ 1550,
36
+ 1551,
37
+ 1552,
38
+ 1553,
39
+ 1554,
40
+ 1555,
41
+ 1556,
42
+ 1557,
43
+ 1558,
44
+ 1559,
45
+ 1560,
46
+ 1561,
47
+ 1562,
48
+ 1563,
49
+ 1564,
50
+ 1565,
51
+ 1566,
52
+ 1567,
53
+ 1568,
54
+ 1569,
55
+ 1570,
56
+ 1571,
57
+ 1572,
58
+ 1573,
59
+ 1574,
60
+ 1575,
61
+ 1576,
62
+ 1577,
63
+ 1578,
64
+ 1579,
65
+ 1580,
66
+ 1581,
67
+ 1582,
68
+ 1583,
69
+ 1584,
70
+ 1585,
71
+ 1586,
72
+ 1587,
73
+ 1588,
74
+ 1589,
75
+ 1590,
76
+ 1591,
77
+ 1592,
78
+ 1593,
79
+ 1594,
80
+ 1595,
81
+ 1596,
82
+ 1597,
83
+ 1598,
84
+ 1599,
85
+ 1600,
86
+ 1601,
87
+ 1602,
88
+ 1603,
89
+ 1604,
90
+ 1605,
91
+ 1606,
92
+ 1607,
93
+ 1608,
94
+ 1609,
95
+ 1610,
96
+ 1611,
97
+ 1612,
98
+ 1613,
99
+ 1614,
100
+ 1615,
101
+ 1616,
102
+ 1617,
103
+ 1618,
104
+ 1619,
105
+ 1620,
106
+ 1621,
107
+ 1622,
108
+ 1623,
109
+ 1624,
110
+ 1625,
111
+ 1626,
112
+ 1627,
113
+ 1628,
114
+ 1629,
115
+ 1630,
116
+ 1631,
117
+ 1632,
118
+ 1633,
119
+ 1634,
120
+ 1635,
121
+ 1636,
122
+ 1637,
123
+ 1638,
124
+ 1639,
125
+ 1640,
126
+ 1641,
127
+ 1642,
128
+ 1643,
129
+ 1644,
130
+ 1645,
131
+ 1646,
132
+ 1647,
133
+ 1648,
134
+ 1649,
135
+ 1650,
136
+ 1651,
137
+ 1652,
138
+ 1653,
139
+ 1654,
140
+ 1655,
141
+ 1656,
142
+ 1657,
143
+ 1658,
144
+ 1659,
145
+ 1660,
146
+ 1661,
147
+ 1662,
148
+ 1663,
149
+ 1664,
150
+ 1665,
151
+ 1666,
152
+ 1667,
153
+ 1668,
154
+ 1669,
155
+ 1670,
156
+ 1671,
157
+ 1672,
158
+ 1673,
159
+ 1674,
160
+ 1675,
161
+ 1676,
162
+ 1677,
163
+ 1678,
164
+ 1679,
165
+ 1680,
166
+ 1681,
167
+ 1682,
168
+ 1683,
169
+ 1684,
170
+ 1685,
171
+ 1686,
172
+ 1687,
173
+ 1688,
174
+ 1689,
175
+ 1690,
176
+ 1691,
177
+ 1692,
178
+ 1693,
179
+ 1694,
180
+ 1695,
181
+ 1696,
182
+ 1697,
183
+ 1698,
184
+ 1699,
185
+ 1700,
186
+ 1701,
187
+ 1702,
188
+ 1703,
189
+ 1704,
190
+ 1705,
191
+ 1706,
192
+ 1707,
193
+ 1708,
194
+ 1709,
195
+ 1710,
196
+ 1711,
197
+ 1712,
198
+ 1713,
199
+ 1714,
200
+ 1715,
201
+ 1716,
202
+ 1717,
203
+ 1718,
204
+ 1719,
205
+ 1720,
206
+ 1721,
207
+ 1722,
208
+ 1723,
209
+ 1724,
210
+ 1725,
211
+ 1726,
212
+ 1727,
213
+ 1728,
214
+ 1729,
215
+ 1730,
216
+ 1731,
217
+ 1732,
218
+ 1733,
219
+ 1734,
220
+ 1735,
221
+ 1736,
222
+ 1737,
223
+ 1738,
224
+ 1739,
225
+ 1740,
226
+ 1741,
227
+ 1742,
228
+ 1743,
229
+ 1744,
230
+ 1745,
231
+ 1746,
232
+ 1747,
233
+ 1748,
234
+ 1749,
235
+ 1750,
236
+ 1751,
237
+ 1752,
238
+ 1753,
239
+ 1754,
240
+ 1755,
241
+ 1756,
242
+ 1757,
243
+ 1758,
244
+ 1759,
245
+ 1760,
246
+ 1761,
247
+ 1762,
248
+ 1763,
249
+ 1764,
250
+ 1765,
251
+ 1766,
252
+ 1767,
253
+ 1768,
254
+ 1769,
255
+ 1770,
256
+ 1771,
257
+ 1772,
258
+ 1773,
259
+ 1774,
260
+ 1775,
261
+ 1776,
262
+ 1777,
263
+ 1778,
264
+ 1779,
265
+ 1780,
266
+ 1781,
267
+ 1782,
268
+ 1783,
269
+ 1784,
270
+ 1785,
271
+ 1786,
272
+ 1787,
273
+ 1788,
274
+ 1789,
275
+ 1790,
276
+ 1791,
277
+ 1792,
278
+ 1793,
279
+ 1794,
280
+ 1795,
281
+ 1796,
282
+ 1797,
283
+ 1798,
284
+ 1799,
285
+ 1800,
286
+ 1801,
287
+ 1802,
288
+ 1803,
289
+ 1804,
290
+ 1805,
291
+ 1806,
292
+ 1807,
293
+ 1808,
294
+ 1809,
295
+ 1810,
296
+ 1811,
297
+ 1812,
298
+ 1813,
299
+ 1814,
300
+ 1815,
301
+ 1816,
302
+ 1817,
303
+ 1818,
304
+ 1819,
305
+ 1820,
306
+ 1821,
307
+ 1822,
308
+ 1823,
309
+ 1824,
310
+ 1825,
311
+ 1826,
312
+ 1827,
313
+ 1828,
314
+ 1829,
315
+ 1830,
316
+ 1831,
317
+ 1832,
318
+ 1833,
319
+ 1834,
320
+ 1835,
321
+ 1836,
322
+ 1837,
323
+ 1838,
324
+ 1839,
325
+ 1840,
326
+ 1841,
327
+ 1842,
328
+ 1843,
329
+ 1844,
330
+ 1845,
331
+ 1846,
332
+ 1847,
333
+ 1848,
334
+ 1849,
335
+ 1850,
336
+ 1851,
337
+ 1852,
338
+ 1853,
339
+ 1854,
340
+ 1855,
341
+ 1856,
342
+ 1857,
343
+ 1858,
344
+ 1859,
345
+ 1860,
346
+ 1861,
347
+ 1862,
348
+ 1863,
349
+ 1864,
350
+ 1865,
351
+ 1866,
352
+ 1867,
353
+ 1868,
354
+ 1869,
355
+ 1870,
356
+ 1871,
357
+ 1872,
358
+ 1873,
359
+ 1874,
360
+ 1875,
361
+ 1876,
362
+ 1877,
363
+ 1878,
364
+ 1879,
365
+ 1880,
366
+ 1881,
367
+ 1882,
368
+ 1883,
369
+ 1884,
370
+ 1885,
371
+ 1886
372
+ ],
373
+ "image_transforms": {
374
+ "enable": true,
375
+ "max_num_transforms": 3,
376
+ "random_order": false,
377
+ "tfs": {
378
+ "brightness": {
379
+ "weight": 1.0,
380
+ "type": "ColorJitter",
381
+ "kwargs": {
382
+ "brightness": [
383
+ 0.8,
384
+ 1.2
385
+ ]
386
+ }
387
+ },
388
+ "contrast": {
389
+ "weight": 1.0,
390
+ "type": "ColorJitter",
391
+ "kwargs": {
392
+ "contrast": [
393
+ 0.8,
394
+ 1.2
395
+ ]
396
+ }
397
+ },
398
+ "saturation": {
399
+ "weight": 1.0,
400
+ "type": "ColorJitter",
401
+ "kwargs": {
402
+ "saturation": [
403
+ 0.5,
404
+ 1.5
405
+ ]
406
+ }
407
+ },
408
+ "hue": {
409
+ "weight": 1.0,
410
+ "type": "ColorJitter",
411
+ "kwargs": {
412
+ "hue": [
413
+ -0.05,
414
+ 0.05
415
+ ]
416
+ }
417
+ },
418
+ "sharpness": {
419
+ "weight": 1.0,
420
+ "type": "SharpnessJitter",
421
+ "kwargs": {
422
+ "sharpness": [
423
+ 0.5,
424
+ 1.5
425
+ ]
426
+ }
427
+ },
428
+ "affine": {
429
+ "weight": 1.0,
430
+ "type": "RandomAffine",
431
+ "kwargs": {
432
+ "degrees": [
433
+ -5.0,
434
+ 5.0
435
+ ],
436
+ "translate": [
437
+ 0.05,
438
+ 0.05
439
+ ]
440
+ }
441
+ }
442
+ }
443
+ },
444
+ "revision": "local",
445
+ "use_imagenet_stats": true,
446
+ "video_backend": "torchcodec",
447
+ "streaming": false
448
+ },
449
+ "env": null,
450
+ "policy": {
451
+ "type": "diffusion",
452
+ "n_obs_steps": 2,
453
+ "input_features": {
454
+ "observation.images.cam_head": {
455
+ "type": "VISUAL",
456
+ "shape": [
457
+ 3,
458
+ 256,
459
+ 256
460
+ ]
461
+ },
462
+ "observation.images.cam_left_wrist": {
463
+ "type": "VISUAL",
464
+ "shape": [
465
+ 3,
466
+ 256,
467
+ 256
468
+ ]
469
+ },
470
+ "observation.images.cam_right_wrist": {
471
+ "type": "VISUAL",
472
+ "shape": [
473
+ 3,
474
+ 256,
475
+ 256
476
+ ]
477
+ },
478
+ "observation.state": {
479
+ "type": "STATE",
480
+ "shape": [
481
+ 16
482
+ ]
483
+ }
484
+ },
485
+ "output_features": {
486
+ "action": {
487
+ "type": "ACTION",
488
+ "shape": [
489
+ 14
490
+ ]
491
+ }
492
+ },
493
+ "device": "cuda",
494
+ "use_amp": false,
495
+ "use_peft": false,
496
+ "push_to_hub": false,
497
+ "repo_id": null,
498
+ "private": null,
499
+ "tags": null,
500
+ "license": null,
501
+ "pretrained_path": null,
502
+ "horizon": 16,
503
+ "n_action_steps": 8,
504
+ "normalization_mapping": {
505
+ "VISUAL": "MEAN_STD",
506
+ "STATE": "MIN_MAX",
507
+ "ACTION": "MIN_MAX"
508
+ },
509
+ "drop_n_last_frames": 7,
510
+ "vision_backbone": "resnet18",
511
+ "crop_shape": [
512
+ 84,
513
+ 84
514
+ ],
515
+ "crop_is_random": true,
516
+ "pretrained_backbone_weights": null,
517
+ "use_group_norm": true,
518
+ "spatial_softmax_num_keypoints": 32,
519
+ "use_separate_rgb_encoder_per_camera": false,
520
+ "down_dims": [
521
+ 512,
522
+ 1024,
523
+ 2048
524
+ ],
525
+ "kernel_size": 5,
526
+ "n_groups": 8,
527
+ "diffusion_step_embed_dim": 128,
528
+ "use_film_scale_modulation": true,
529
+ "noise_scheduler_type": "DDPM",
530
+ "num_train_timesteps": 100,
531
+ "beta_schedule": "squaredcos_cap_v2",
532
+ "beta_start": 0.0001,
533
+ "beta_end": 0.02,
534
+ "prediction_type": "epsilon",
535
+ "clip_sample": true,
536
+ "clip_sample_range": 1.0,
537
+ "num_inference_steps": null,
538
+ "do_mask_loss_for_padding": false,
539
+ "optimizer_lr": 0.0001,
540
+ "optimizer_betas": [
541
+ 0.95,
542
+ 0.999
543
+ ],
544
+ "optimizer_eps": 1e-08,
545
+ "optimizer_weight_decay": 1e-06,
546
+ "scheduler_name": "cosine",
547
+ "scheduler_warmup_steps": 500
548
+ },
549
+ "output_dir": "/weka/helenw/outputs/diffusion_PickSingleBookFromTable",
550
+ "job_name": "diffusion_roboeval_v3_merged",
551
+ "resume": false,
552
+ "seed": 1000,
553
+ "num_workers": 2,
554
+ "batch_size": 256,
555
+ "steps": 5000,
556
+ "eval_freq": 20000,
557
+ "log_freq": 10,
558
+ "tolerance_s": 0.0001,
559
+ "save_checkpoint": true,
560
+ "save_freq": 1000,
561
+ "use_policy_training_preset": true,
562
+ "optimizer": {
563
+ "type": "adam",
564
+ "lr": 0.0001,
565
+ "weight_decay": 1e-06,
566
+ "grad_clip_norm": 10.0,
567
+ "betas": [
568
+ 0.95,
569
+ 0.999
570
+ ],
571
+ "eps": 1e-08
572
+ },
573
+ "scheduler": {
574
+ "type": "diffuser",
575
+ "num_warmup_steps": 500,
576
+ "name": "cosine"
577
+ },
578
+ "eval": {
579
+ "n_episodes": 50,
580
+ "batch_size": 50,
581
+ "use_async_envs": false
582
+ },
583
+ "wandb": {
584
+ "enable": true,
585
+ "disable_artifact": true,
586
+ "project": "lerobot",
587
+ "entity": null,
588
+ "notes": null,
589
+ "run_id": "bu4lkwy4",
590
+ "mode": null
591
+ },
592
+ "peft": null,
593
+ "use_rabc": false,
594
+ "rabc_progress_path": null,
595
+ "rabc_kappa": 0.01,
596
+ "rabc_epsilon": 1e-06,
597
+ "rabc_head_mode": "sparse",
598
+ "rename_map": {},
599
+ "checkpoint_path": null
600
+ }