Hugging Face's logo

RoboEval
/
roboeval-act

Robotics
Safetensors
roboeval
act
Model card Files
xet
 Community
helen9975 commited on
Commit
804394a
·
verified ·
1 Parent(s): 4709375

PackBox: step 005000 weights

Browse files
Files changed (7) hide show
  1. PackBox/config.json +79 -0
  2. PackBox/model.safetensors +3 -0
  3. PackBox/policy_postprocessor.json +32 -0
  4. PackBox/policy_postprocessor_step_0_unnormalizer_processor.safetensors +3 -0
  5. PackBox/policy_preprocessor.json +72 -0
  6. PackBox/policy_preprocessor_step_3_normalizer_processor.safetensors +3 -0
  7. PackBox/train_config.json +608 -0
PackBox/config.json ADDED
@@ -0,0 +1,79 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "type": "act",
3
+ "n_obs_steps": 1,
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
+ "chunk_size": 16,
54
+ "n_action_steps": 8,
55
+ "normalization_mapping": {
56
+ "VISUAL": "MEAN_STD",
57
+ "STATE": "QUANTILES",
58
+ "ACTION": "QUANTILES"
59
+ },
60
+ "vision_backbone": "resnet18",
61
+ "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
62
+ "replace_final_stride_with_dilation": false,
63
+ "pre_norm": false,
64
+ "dim_model": 512,
65
+ "n_heads": 8,
66
+ "dim_feedforward": 3200,
67
+ "feedforward_activation": "relu",
68
+ "n_encoder_layers": 4,
69
+ "n_decoder_layers": 1,
70
+ "use_vae": true,
71
+ "latent_dim": 32,
72
+ "n_vae_encoder_layers": 4,
73
+ "temporal_ensemble_coeff": null,
74
+ "dropout": 0.1,
75
+ "kl_weight": 0.01,
76
+ "optimizer_lr": 0.0001,
77
+ "optimizer_weight_decay": 0.0001,
78
+ "optimizer_lr_backbone": 1e-05
79
+ }
PackBox/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:db300ca88e191b2702b4284bf97afb9229c158f63ddb1f2e3af8bb6302ee7236
3
+ size 206429432
PackBox/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": "QUANTILES",
19
+ "ACTION": "QUANTILES"
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
+ }
PackBox/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
PackBox/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": "QUANTILES",
66
+ "ACTION": "QUANTILES"
67
+ }
68
+ },
69
+ "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
70
+ }
71
+ ]
72
+ }
PackBox/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
PackBox/train_config.json ADDED
@@ -0,0 +1,608 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ 1127,
7
+ 1128,
8
+ 1129,
9
+ 1130,
10
+ 1131,
11
+ 1132,
12
+ 1133,
13
+ 1134,
14
+ 1135,
15
+ 1136,
16
+ 1137,
17
+ 1138,
18
+ 1139,
19
+ 1140,
20
+ 1141,
21
+ 1142,
22
+ 1143,
23
+ 1144,
24
+ 1145,
25
+ 1146,
26
+ 1147,
27
+ 1148,
28
+ 1149,
29
+ 1150,
30
+ 1151,
31
+ 1152,
32
+ 1153,
33
+ 1154,
34
+ 1155,
35
+ 1156,
36
+ 1157,
37
+ 1158,
38
+ 1159,
39
+ 1160,
40
+ 1161,
41
+ 1162,
42
+ 1163,
43
+ 1164,
44
+ 1165,
45
+ 1166,
46
+ 1167,
47
+ 1168,
48
+ 1169,
49
+ 1170,
50
+ 1171,
51
+ 1172,
52
+ 1173,
53
+ 1174,
54
+ 1175,
55
+ 1176,
56
+ 1177,
57
+ 1178,
58
+ 1179,
59
+ 1180,
60
+ 1181,
61
+ 1182,
62
+ 1183,
63
+ 1184,
64
+ 1185,
65
+ 1186,
66
+ 1187,
67
+ 1188,
68
+ 1189,
69
+ 1190,
70
+ 1191,
71
+ 1192,
72
+ 1193,
73
+ 1194,
74
+ 1195,
75
+ 1196,
76
+ 1197,
77
+ 1198,
78
+ 1199,
79
+ 1200,
80
+ 1201,
81
+ 1202,
82
+ 1203,
83
+ 1204,
84
+ 1205,
85
+ 1206,
86
+ 1207,
87
+ 1208,
88
+ 1209,
89
+ 1210,
90
+ 1211,
91
+ 1212,
92
+ 1213,
93
+ 1214,
94
+ 1215,
95
+ 1216,
96
+ 1217,
97
+ 1218,
98
+ 1219,
99
+ 1220,
100
+ 1221,
101
+ 1222,
102
+ 1223,
103
+ 1224,
104
+ 1225,
105
+ 1226,
106
+ 1227,
107
+ 1228,
108
+ 1229,
109
+ 1230,
110
+ 1231,
111
+ 1232,
112
+ 1233,
113
+ 1234,
114
+ 1235,
115
+ 1236,
116
+ 1237,
117
+ 1238,
118
+ 1239,
119
+ 1240,
120
+ 1241,
121
+ 1242,
122
+ 1243,
123
+ 1244,
124
+ 1245,
125
+ 1246,
126
+ 1247,
127
+ 1248,
128
+ 1249,
129
+ 1250,
130
+ 1251,
131
+ 1252,
132
+ 1253,
133
+ 1254,
134
+ 1255,
135
+ 1256,
136
+ 1257,
137
+ 1258,
138
+ 1259,
139
+ 1260,
140
+ 1261,
141
+ 1262,
142
+ 1263,
143
+ 1264,
144
+ 1265,
145
+ 1266,
146
+ 1267,
147
+ 1268,
148
+ 1269,
149
+ 1270,
150
+ 1271,
151
+ 1272,
152
+ 1273,
153
+ 1274,
154
+ 1275,
155
+ 1276,
156
+ 1277,
157
+ 1278,
158
+ 1279,
159
+ 1280,
160
+ 1281,
161
+ 1282,
162
+ 1283,
163
+ 1284,
164
+ 1285,
165
+ 1286,
166
+ 1287,
167
+ 1288,
168
+ 1289,
169
+ 1290,
170
+ 1291,
171
+ 1292,
172
+ 1293,
173
+ 1294,
174
+ 1295,
175
+ 1296,
176
+ 1297,
177
+ 1298,
178
+ 1299,
179
+ 1300,
180
+ 1301,
181
+ 1302,
182
+ 1303,
183
+ 1304,
184
+ 1305,
185
+ 1306,
186
+ 1307,
187
+ 1308,
188
+ 1309,
189
+ 1310,
190
+ 1311,
191
+ 1312,
192
+ 1313,
193
+ 1314,
194
+ 1315,
195
+ 1316,
196
+ 1317,
197
+ 1318,
198
+ 1319,
199
+ 1320,
200
+ 1321,
201
+ 1322,
202
+ 1323,
203
+ 1324,
204
+ 1325,
205
+ 1326,
206
+ 1327,
207
+ 1328,
208
+ 1329,
209
+ 1330,
210
+ 1331,
211
+ 1332,
212
+ 1333,
213
+ 1334,
214
+ 1335,
215
+ 1336,
216
+ 1337,
217
+ 1338,
218
+ 1339,
219
+ 1340,
220
+ 1341,
221
+ 1342,
222
+ 1343,
223
+ 1344,
224
+ 1345,
225
+ 1346,
226
+ 1347,
227
+ 1348,
228
+ 1349,
229
+ 1350,
230
+ 1351,
231
+ 1352,
232
+ 1353,
233
+ 1354,
234
+ 1355,
235
+ 1356,
236
+ 1357,
237
+ 1358,
238
+ 1359,
239
+ 1360,
240
+ 1361,
241
+ 1362,
242
+ 1363,
243
+ 1364,
244
+ 1365,
245
+ 1366,
246
+ 1367,
247
+ 1368,
248
+ 1369,
249
+ 1370,
250
+ 1371,
251
+ 1372,
252
+ 1373,
253
+ 1374,
254
+ 1375,
255
+ 1376,
256
+ 1377,
257
+ 1378,
258
+ 1379,
259
+ 1380,
260
+ 1381,
261
+ 1382,
262
+ 1383,
263
+ 1384,
264
+ 1385,
265
+ 1386,
266
+ 1387,
267
+ 1388,
268
+ 1389,
269
+ 1390,
270
+ 1391,
271
+ 1392,
272
+ 1393,
273
+ 1394,
274
+ 1395,
275
+ 1396,
276
+ 1397,
277
+ 1398,
278
+ 1399,
279
+ 1400,
280
+ 1401,
281
+ 1402,
282
+ 1403,
283
+ 1404,
284
+ 1405,
285
+ 1406,
286
+ 1407,
287
+ 1408,
288
+ 1409,
289
+ 1410,
290
+ 1411,
291
+ 1412,
292
+ 1413,
293
+ 1414,
294
+ 1415,
295
+ 1416,
296
+ 1417,
297
+ 1418,
298
+ 1419,
299
+ 1420,
300
+ 1421,
301
+ 1422,
302
+ 1423,
303
+ 1424,
304
+ 1425,
305
+ 1426,
306
+ 1427,
307
+ 1428,
308
+ 1429,
309
+ 1430,
310
+ 1431,
311
+ 1432,
312
+ 1433,
313
+ 1434,
314
+ 1435,
315
+ 1436,
316
+ 1437,
317
+ 1438,
318
+ 1439,
319
+ 1440,
320
+ 1441,
321
+ 1442,
322
+ 1443,
323
+ 1444,
324
+ 1445,
325
+ 1446,
326
+ 1447,
327
+ 1448,
328
+ 1449,
329
+ 1450,
330
+ 1451,
331
+ 1452,
332
+ 1453,
333
+ 1454,
334
+ 1455,
335
+ 1456,
336
+ 1457,
337
+ 1458,
338
+ 1459,
339
+ 1460,
340
+ 1461,
341
+ 1462,
342
+ 1463,
343
+ 1464,
344
+ 1465,
345
+ 1466,
346
+ 1467,
347
+ 1468,
348
+ 1469,
349
+ 1470,
350
+ 1471,
351
+ 1472,
352
+ 1473,
353
+ 1474,
354
+ 1475,
355
+ 1476,
356
+ 1477,
357
+ 1478,
358
+ 1479,
359
+ 1480,
360
+ 1481,
361
+ 1482,
362
+ 1483,
363
+ 1484,
364
+ 1485,
365
+ 1486,
366
+ 1487,
367
+ 1488,
368
+ 1489,
369
+ 1490,
370
+ 1491,
371
+ 1492,
372
+ 1493,
373
+ 1494,
374
+ 1495,
375
+ 1496,
376
+ 1497,
377
+ 1498,
378
+ 1499,
379
+ 1500,
380
+ 1501,
381
+ 1502,
382
+ 1503,
383
+ 1504,
384
+ 1505,
385
+ 1506,
386
+ 1507,
387
+ 1508,
388
+ 1509,
389
+ 1510,
390
+ 1511,
391
+ 1512,
392
+ 1513,
393
+ 1514,
394
+ 1515,
395
+ 1516,
396
+ 1517,
397
+ 1518,
398
+ 1519,
399
+ 1520
400
+ ],
401
+ "image_transforms": {
402
+ "enable": true,
403
+ "max_num_transforms": 3,
404
+ "random_order": false,
405
+ "tfs": {
406
+ "brightness": {
407
+ "weight": 1.0,
408
+ "type": "ColorJitter",
409
+ "kwargs": {
410
+ "brightness": [
411
+ 0.8,
412
+ 1.2
413
+ ]
414
+ }
415
+ },
416
+ "contrast": {
417
+ "weight": 1.0,
418
+ "type": "ColorJitter",
419
+ "kwargs": {
420
+ "contrast": [
421
+ 0.8,
422
+ 1.2
423
+ ]
424
+ }
425
+ },
426
+ "saturation": {
427
+ "weight": 1.0,
428
+ "type": "ColorJitter",
429
+ "kwargs": {
430
+ "saturation": [
431
+ 0.5,
432
+ 1.5
433
+ ]
434
+ }
435
+ },
436
+ "hue": {
437
+ "weight": 1.0,
438
+ "type": "ColorJitter",
439
+ "kwargs": {
440
+ "hue": [
441
+ -0.05,
442
+ 0.05
443
+ ]
444
+ }
445
+ },
446
+ "sharpness": {
447
+ "weight": 1.0,
448
+ "type": "SharpnessJitter",
449
+ "kwargs": {
450
+ "sharpness": [
451
+ 0.5,
452
+ 1.5
453
+ ]
454
+ }
455
+ },
456
+ "affine": {
457
+ "weight": 1.0,
458
+ "type": "RandomAffine",
459
+ "kwargs": {
460
+ "degrees": [
461
+ -5.0,
462
+ 5.0
463
+ ],
464
+ "translate": [
465
+ 0.05,
466
+ 0.05
467
+ ]
468
+ }
469
+ }
470
+ }
471
+ },
472
+ "revision": "local",
473
+ "use_imagenet_stats": true,
474
+ "video_backend": "torchcodec",
475
+ "streaming": false
476
+ },
477
+ "env": null,
478
+ "policy": {
479
+ "type": "act",
480
+ "n_obs_steps": 1,
481
+ "input_features": {
482
+ "observation.images.cam_head": {
483
+ "type": "VISUAL",
484
+ "shape": [
485
+ 3,
486
+ 256,
487
+ 256
488
+ ]
489
+ },
490
+ "observation.images.cam_left_wrist": {
491
+ "type": "VISUAL",
492
+ "shape": [
493
+ 3,
494
+ 256,
495
+ 256
496
+ ]
497
+ },
498
+ "observation.images.cam_right_wrist": {
499
+ "type": "VISUAL",
500
+ "shape": [
501
+ 3,
502
+ 256,
503
+ 256
504
+ ]
505
+ },
506
+ "observation.state": {
507
+ "type": "STATE",
508
+ "shape": [
509
+ 16
510
+ ]
511
+ }
512
+ },
513
+ "output_features": {
514
+ "action": {
515
+ "type": "ACTION",
516
+ "shape": [
517
+ 14
518
+ ]
519
+ }
520
+ },
521
+ "device": "cuda",
522
+ "use_amp": false,
523
+ "use_peft": false,
524
+ "push_to_hub": false,
525
+ "repo_id": null,
526
+ "private": null,
527
+ "tags": null,
528
+ "license": null,
529
+ "pretrained_path": null,
530
+ "chunk_size": 16,
531
+ "n_action_steps": 8,
532
+ "normalization_mapping": {
533
+ "VISUAL": "MEAN_STD",
534
+ "STATE": "QUANTILES",
535
+ "ACTION": "QUANTILES"
536
+ },
537
+ "vision_backbone": "resnet18",
538
+ "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
539
+ "replace_final_stride_with_dilation": false,
540
+ "pre_norm": false,
541
+ "dim_model": 512,
542
+ "n_heads": 8,
543
+ "dim_feedforward": 3200,
544
+ "feedforward_activation": "relu",
545
+ "n_encoder_layers": 4,
546
+ "n_decoder_layers": 1,
547
+ "use_vae": true,
548
+ "latent_dim": 32,
549
+ "n_vae_encoder_layers": 4,
550
+ "temporal_ensemble_coeff": null,
551
+ "dropout": 0.1,
552
+ "kl_weight": 0.01,
553
+ "optimizer_lr": 0.0001,
554
+ "optimizer_weight_decay": 0.0001,
555
+ "optimizer_lr_backbone": 1e-05
556
+ },
557
+ "output_dir": "/weka/helenw/outputs/act_PackBox",
558
+ "job_name": "act_chunk16_roboeval_v3_merged",
559
+ "resume": false,
560
+ "seed": 1000,
561
+ "num_workers": 2,
562
+ "batch_size": 512,
563
+ "steps": 5000,
564
+ "eval_freq": 20000,
565
+ "log_freq": 10,
566
+ "tolerance_s": 0.0001,
567
+ "save_checkpoint": true,
568
+ "save_freq": 1000,
569
+ "use_policy_training_preset": true,
570
+ "optimizer": {
571
+ "type": "adamw",
572
+ "lr": 0.0001,
573
+ "weight_decay": 0.0001,
574
+ "grad_clip_norm": 10.0,
575
+ "betas": [
576
+ 0.9,
577
+ 0.999
578
+ ],
579
+ "eps": 1e-08
580
+ },
581
+ "scheduler": {
582
+ "type": "diffuser",
583
+ "num_warmup_steps": 500,
584
+ "name": "cosine"
585
+ },
586
+ "eval": {
587
+ "n_episodes": 50,
588
+ "batch_size": 50,
589
+ "use_async_envs": false
590
+ },
591
+ "wandb": {
592
+ "enable": true,
593
+ "disable_artifact": true,
594
+ "project": "lerobot",
595
+ "entity": null,
596
+ "notes": null,
597
+ "run_id": "1vzqyge3",
598
+ "mode": null
599
+ },
600
+ "peft": null,
601
+ "use_rabc": false,
602
+ "rabc_progress_path": null,
603
+ "rabc_kappa": 0.01,
604
+ "rabc_epsilon": 1e-06,
605
+ "rabc_head_mode": "sparse",
606
+ "rename_map": {},
607
+ "checkpoint_path": null
608
+ }