Hugging Face's logo

RoboEval
/
roboeval-diffusion

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

RotateValve: step 005000 weights

Browse files
Files changed (7) hide show
  1. RotateValve/config.json +99 -0
  2. RotateValve/model.safetensors +3 -0
  3. RotateValve/policy_postprocessor.json +32 -0
  4. RotateValve/policy_postprocessor_step_0_unnormalizer_processor.safetensors +3 -0
  5. RotateValve/policy_preprocessor.json +72 -0
  6. RotateValve/policy_preprocessor_step_3_normalizer_processor.safetensors +3 -0
  7. RotateValve/train_config.json +583 -0
RotateValve/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
+ }
RotateValve/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e0228463df33bcbea22ce16bf979c03336da7fcdc16b7a69b1f1ab65937ab147
3
+ size 1083604936
RotateValve/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
+ }
RotateValve/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
RotateValve/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
+ }
RotateValve/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
RotateValve/train_config.json ADDED
@@ -0,0 +1,583 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ 1887,
7
+ 1888,
8
+ 1889,
9
+ 1890,
10
+ 1891,
11
+ 1892,
12
+ 1893,
13
+ 1894,
14
+ 1895,
15
+ 1896,
16
+ 1897,
17
+ 1898,
18
+ 1899,
19
+ 1900,
20
+ 1901,
21
+ 1902,
22
+ 1903,
23
+ 1904,
24
+ 1905,
25
+ 1906,
26
+ 1907,
27
+ 1908,
28
+ 1909,
29
+ 1910,
30
+ 1911,
31
+ 1912,
32
+ 1913,
33
+ 1914,
34
+ 1915,
35
+ 1916,
36
+ 1917,
37
+ 1918,
38
+ 1919,
39
+ 1920,
40
+ 1921,
41
+ 1922,
42
+ 1923,
43
+ 1924,
44
+ 1925,
45
+ 1926,
46
+ 1927,
47
+ 1928,
48
+ 1929,
49
+ 1930,
50
+ 1931,
51
+ 1932,
52
+ 1933,
53
+ 1934,
54
+ 1935,
55
+ 1936,
56
+ 1937,
57
+ 1938,
58
+ 1939,
59
+ 1940,
60
+ 1941,
61
+ 1942,
62
+ 1943,
63
+ 1944,
64
+ 1945,
65
+ 1946,
66
+ 1947,
67
+ 1948,
68
+ 1949,
69
+ 1950,
70
+ 1951,
71
+ 1952,
72
+ 1953,
73
+ 1954,
74
+ 1955,
75
+ 1956,
76
+ 1957,
77
+ 1958,
78
+ 1959,
79
+ 1960,
80
+ 1961,
81
+ 1962,
82
+ 1963,
83
+ 1964,
84
+ 1965,
85
+ 1966,
86
+ 1967,
87
+ 1968,
88
+ 1969,
89
+ 1970,
90
+ 1971,
91
+ 1972,
92
+ 1973,
93
+ 1974,
94
+ 1975,
95
+ 1976,
96
+ 1977,
97
+ 1978,
98
+ 1979,
99
+ 1980,
100
+ 1981,
101
+ 1982,
102
+ 1983,
103
+ 1984,
104
+ 1985,
105
+ 1986,
106
+ 1987,
107
+ 1988,
108
+ 1989,
109
+ 1990,
110
+ 1991,
111
+ 1992,
112
+ 1993,
113
+ 1994,
114
+ 1995,
115
+ 1996,
116
+ 1997,
117
+ 1998,
118
+ 1999,
119
+ 2000,
120
+ 2001,
121
+ 2002,
122
+ 2003,
123
+ 2004,
124
+ 2005,
125
+ 2006,
126
+ 2007,
127
+ 2008,
128
+ 2009,
129
+ 2010,
130
+ 2011,
131
+ 2012,
132
+ 2013,
133
+ 2014,
134
+ 2015,
135
+ 2016,
136
+ 2017,
137
+ 2018,
138
+ 2019,
139
+ 2020,
140
+ 2021,
141
+ 2022,
142
+ 2023,
143
+ 2024,
144
+ 2025,
145
+ 2026,
146
+ 2027,
147
+ 2028,
148
+ 2029,
149
+ 2030,
150
+ 2031,
151
+ 2032,
152
+ 2033,
153
+ 2034,
154
+ 2035,
155
+ 2036,
156
+ 2037,
157
+ 2038,
158
+ 2039,
159
+ 2040,
160
+ 2041,
161
+ 2042,
162
+ 2043,
163
+ 2044,
164
+ 2045,
165
+ 2046,
166
+ 2047,
167
+ 2048,
168
+ 2049,
169
+ 2050,
170
+ 2051,
171
+ 2052,
172
+ 2053,
173
+ 2054,
174
+ 2055,
175
+ 2056,
176
+ 2057,
177
+ 2058,
178
+ 2059,
179
+ 2060,
180
+ 2061,
181
+ 2062,
182
+ 2063,
183
+ 2064,
184
+ 2065,
185
+ 2066,
186
+ 2067,
187
+ 2068,
188
+ 2069,
189
+ 2070,
190
+ 2071,
191
+ 2072,
192
+ 2073,
193
+ 2074,
194
+ 2075,
195
+ 2076,
196
+ 2077,
197
+ 2078,
198
+ 2079,
199
+ 2080,
200
+ 2081,
201
+ 2082,
202
+ 2083,
203
+ 2084,
204
+ 2085,
205
+ 2086,
206
+ 2087,
207
+ 2088,
208
+ 2089,
209
+ 2090,
210
+ 2091,
211
+ 2092,
212
+ 2093,
213
+ 2094,
214
+ 2095,
215
+ 2096,
216
+ 2097,
217
+ 2098,
218
+ 2099,
219
+ 2100,
220
+ 2101,
221
+ 2102,
222
+ 2103,
223
+ 2104,
224
+ 2105,
225
+ 2106,
226
+ 2107,
227
+ 2108,
228
+ 2109,
229
+ 2110,
230
+ 2111,
231
+ 2112,
232
+ 2113,
233
+ 2114,
234
+ 2115,
235
+ 2116,
236
+ 2117,
237
+ 2118,
238
+ 2119,
239
+ 2120,
240
+ 2121,
241
+ 2122,
242
+ 2123,
243
+ 2124,
244
+ 2125,
245
+ 2126,
246
+ 2127,
247
+ 2128,
248
+ 2129,
249
+ 2130,
250
+ 2131,
251
+ 2132,
252
+ 2133,
253
+ 2134,
254
+ 2135,
255
+ 2136,
256
+ 2137,
257
+ 2138,
258
+ 2139,
259
+ 2140,
260
+ 2141,
261
+ 2142,
262
+ 2143,
263
+ 2144,
264
+ 2145,
265
+ 2146,
266
+ 2147,
267
+ 2148,
268
+ 2149,
269
+ 2150,
270
+ 2151,
271
+ 2152,
272
+ 2153,
273
+ 2154,
274
+ 2155,
275
+ 2156,
276
+ 2157,
277
+ 2158,
278
+ 2159,
279
+ 2160,
280
+ 2161,
281
+ 2162,
282
+ 2163,
283
+ 2164,
284
+ 2165,
285
+ 2166,
286
+ 2167,
287
+ 2168,
288
+ 2169,
289
+ 2170,
290
+ 2171,
291
+ 2172,
292
+ 2173,
293
+ 2174,
294
+ 2175,
295
+ 2176,
296
+ 2177,
297
+ 2178,
298
+ 2179,
299
+ 2180,
300
+ 2181,
301
+ 2182,
302
+ 2183,
303
+ 2184,
304
+ 2185,
305
+ 2186,
306
+ 2187,
307
+ 2188,
308
+ 2189,
309
+ 2190,
310
+ 2191,
311
+ 2192,
312
+ 2193,
313
+ 2194,
314
+ 2195,
315
+ 2196,
316
+ 2197,
317
+ 2198,
318
+ 2199,
319
+ 2200,
320
+ 2201,
321
+ 2202,
322
+ 2203,
323
+ 2204,
324
+ 2205,
325
+ 2206,
326
+ 2207,
327
+ 2208,
328
+ 2209,
329
+ 2210,
330
+ 2211,
331
+ 2212,
332
+ 2213,
333
+ 2214,
334
+ 2215,
335
+ 2216,
336
+ 2217,
337
+ 2218,
338
+ 2219,
339
+ 2220,
340
+ 2221,
341
+ 2222,
342
+ 2223,
343
+ 2224,
344
+ 2225,
345
+ 2226,
346
+ 2227,
347
+ 2228,
348
+ 2229,
349
+ 2230,
350
+ 2231,
351
+ 2232,
352
+ 2233,
353
+ 2234,
354
+ 2235
355
+ ],
356
+ "image_transforms": {
357
+ "enable": true,
358
+ "max_num_transforms": 3,
359
+ "random_order": false,
360
+ "tfs": {
361
+ "brightness": {
362
+ "weight": 1.0,
363
+ "type": "ColorJitter",
364
+ "kwargs": {
365
+ "brightness": [
366
+ 0.8,
367
+ 1.2
368
+ ]
369
+ }
370
+ },
371
+ "contrast": {
372
+ "weight": 1.0,
373
+ "type": "ColorJitter",
374
+ "kwargs": {
375
+ "contrast": [
376
+ 0.8,
377
+ 1.2
378
+ ]
379
+ }
380
+ },
381
+ "saturation": {
382
+ "weight": 1.0,
383
+ "type": "ColorJitter",
384
+ "kwargs": {
385
+ "saturation": [
386
+ 0.5,
387
+ 1.5
388
+ ]
389
+ }
390
+ },
391
+ "hue": {
392
+ "weight": 1.0,
393
+ "type": "ColorJitter",
394
+ "kwargs": {
395
+ "hue": [
396
+ -0.05,
397
+ 0.05
398
+ ]
399
+ }
400
+ },
401
+ "sharpness": {
402
+ "weight": 1.0,
403
+ "type": "SharpnessJitter",
404
+ "kwargs": {
405
+ "sharpness": [
406
+ 0.5,
407
+ 1.5
408
+ ]
409
+ }
410
+ },
411
+ "affine": {
412
+ "weight": 1.0,
413
+ "type": "RandomAffine",
414
+ "kwargs": {
415
+ "degrees": [
416
+ -5.0,
417
+ 5.0
418
+ ],
419
+ "translate": [
420
+ 0.05,
421
+ 0.05
422
+ ]
423
+ }
424
+ }
425
+ }
426
+ },
427
+ "revision": "local",
428
+ "use_imagenet_stats": true,
429
+ "video_backend": "torchcodec",
430
+ "streaming": false
431
+ },
432
+ "env": null,
433
+ "policy": {
434
+ "type": "diffusion",
435
+ "n_obs_steps": 2,
436
+ "input_features": {
437
+ "observation.images.cam_head": {
438
+ "type": "VISUAL",
439
+ "shape": [
440
+ 3,
441
+ 256,
442
+ 256
443
+ ]
444
+ },
445
+ "observation.images.cam_left_wrist": {
446
+ "type": "VISUAL",
447
+ "shape": [
448
+ 3,
449
+ 256,
450
+ 256
451
+ ]
452
+ },
453
+ "observation.images.cam_right_wrist": {
454
+ "type": "VISUAL",
455
+ "shape": [
456
+ 3,
457
+ 256,
458
+ 256
459
+ ]
460
+ },
461
+ "observation.state": {
462
+ "type": "STATE",
463
+ "shape": [
464
+ 16
465
+ ]
466
+ }
467
+ },
468
+ "output_features": {
469
+ "action": {
470
+ "type": "ACTION",
471
+ "shape": [
472
+ 14
473
+ ]
474
+ }
475
+ },
476
+ "device": "cuda",
477
+ "use_amp": false,
478
+ "use_peft": false,
479
+ "push_to_hub": false,
480
+ "repo_id": null,
481
+ "private": null,
482
+ "tags": null,
483
+ "license": null,
484
+ "pretrained_path": null,
485
+ "horizon": 16,
486
+ "n_action_steps": 8,
487
+ "normalization_mapping": {
488
+ "VISUAL": "MEAN_STD",
489
+ "STATE": "MIN_MAX",
490
+ "ACTION": "MIN_MAX"
491
+ },
492
+ "drop_n_last_frames": 7,
493
+ "vision_backbone": "resnet18",
494
+ "crop_shape": [
495
+ 84,
496
+ 84
497
+ ],
498
+ "crop_is_random": true,
499
+ "pretrained_backbone_weights": null,
500
+ "use_group_norm": true,
501
+ "spatial_softmax_num_keypoints": 32,
502
+ "use_separate_rgb_encoder_per_camera": false,
503
+ "down_dims": [
504
+ 512,
505
+ 1024,
506
+ 2048
507
+ ],
508
+ "kernel_size": 5,
509
+ "n_groups": 8,
510
+ "diffusion_step_embed_dim": 128,
511
+ "use_film_scale_modulation": true,
512
+ "noise_scheduler_type": "DDPM",
513
+ "num_train_timesteps": 100,
514
+ "beta_schedule": "squaredcos_cap_v2",
515
+ "beta_start": 0.0001,
516
+ "beta_end": 0.02,
517
+ "prediction_type": "epsilon",
518
+ "clip_sample": true,
519
+ "clip_sample_range": 1.0,
520
+ "num_inference_steps": null,
521
+ "do_mask_loss_for_padding": false,
522
+ "optimizer_lr": 0.0001,
523
+ "optimizer_betas": [
524
+ 0.95,
525
+ 0.999
526
+ ],
527
+ "optimizer_eps": 1e-08,
528
+ "optimizer_weight_decay": 1e-06,
529
+ "scheduler_name": "cosine",
530
+ "scheduler_warmup_steps": 500
531
+ },
532
+ "output_dir": "/weka/helenw/outputs/diffusion_RotateValve",
533
+ "job_name": "diffusion_roboeval_v3_merged",
534
+ "resume": false,
535
+ "seed": 1000,
536
+ "num_workers": 2,
537
+ "batch_size": 256,
538
+ "steps": 5000,
539
+ "eval_freq": 20000,
540
+ "log_freq": 10,
541
+ "tolerance_s": 0.0001,
542
+ "save_checkpoint": true,
543
+ "save_freq": 1000,
544
+ "use_policy_training_preset": true,
545
+ "optimizer": {
546
+ "type": "adam",
547
+ "lr": 0.0001,
548
+ "weight_decay": 1e-06,
549
+ "grad_clip_norm": 10.0,
550
+ "betas": [
551
+ 0.95,
552
+ 0.999
553
+ ],
554
+ "eps": 1e-08
555
+ },
556
+ "scheduler": {
557
+ "type": "diffuser",
558
+ "num_warmup_steps": 500,
559
+ "name": "cosine"
560
+ },
561
+ "eval": {
562
+ "n_episodes": 50,
563
+ "batch_size": 50,
564
+ "use_async_envs": false
565
+ },
566
+ "wandb": {
567
+ "enable": true,
568
+ "disable_artifact": true,
569
+ "project": "lerobot",
570
+ "entity": null,
571
+ "notes": null,
572
+ "run_id": "x4tpu0pm",
573
+ "mode": null
574
+ },
575
+ "peft": null,
576
+ "use_rabc": false,
577
+ "rabc_progress_path": null,
578
+ "rabc_kappa": 0.01,
579
+ "rabc_epsilon": 1e-06,
580
+ "rabc_head_mode": "sparse",
581
+ "rename_map": {},
582
+ "checkpoint_path": null
583
+ }