| { |
| "dataset": { |
| "repo_id": "roboeval_ee_delta_20hz_v3_merged", |
| "root": "/weka/helenw/data/huggingface/lerobot/roboeval_ee_delta_20hz_v3_merged", |
| "episodes": [ |
| 1127, |
| 1128, |
| 1129, |
| 1130, |
| 1131, |
| 1132, |
| 1133, |
| 1134, |
| 1135, |
| 1136, |
| 1137, |
| 1138, |
| 1139, |
| 1140, |
| 1141, |
| 1142, |
| 1143, |
| 1144, |
| 1145, |
| 1146, |
| 1147, |
| 1148, |
| 1149, |
| 1150, |
| 1151, |
| 1152, |
| 1153, |
| 1154, |
| 1155, |
| 1156, |
| 1157, |
| 1158, |
| 1159, |
| 1160, |
| 1161, |
| 1162, |
| 1163, |
| 1164, |
| 1165, |
| 1166, |
| 1167, |
| 1168, |
| 1169, |
| 1170, |
| 1171, |
| 1172, |
| 1173, |
| 1174, |
| 1175, |
| 1176, |
| 1177, |
| 1178, |
| 1179, |
| 1180, |
| 1181, |
| 1182, |
| 1183, |
| 1184, |
| 1185, |
| 1186, |
| 1187, |
| 1188, |
| 1189, |
| 1190, |
| 1191, |
| 1192, |
| 1193, |
| 1194, |
| 1195, |
| 1196, |
| 1197, |
| 1198, |
| 1199, |
| 1200, |
| 1201, |
| 1202, |
| 1203, |
| 1204, |
| 1205, |
| 1206, |
| 1207, |
| 1208, |
| 1209, |
| 1210, |
| 1211, |
| 1212, |
| 1213, |
| 1214, |
| 1215, |
| 1216, |
| 1217, |
| 1218, |
| 1219, |
| 1220, |
| 1221, |
| 1222, |
| 1223, |
| 1224, |
| 1225, |
| 1226, |
| 1227, |
| 1228, |
| 1229, |
| 1230, |
| 1231, |
| 1232, |
| 1233, |
| 1234, |
| 1235, |
| 1236, |
| 1237, |
| 1238, |
| 1239, |
| 1240, |
| 1241, |
| 1242, |
| 1243, |
| 1244, |
| 1245, |
| 1246, |
| 1247, |
| 1248, |
| 1249, |
| 1250, |
| 1251, |
| 1252, |
| 1253, |
| 1254, |
| 1255, |
| 1256, |
| 1257, |
| 1258, |
| 1259, |
| 1260, |
| 1261, |
| 1262, |
| 1263, |
| 1264, |
| 1265, |
| 1266, |
| 1267, |
| 1268, |
| 1269, |
| 1270, |
| 1271, |
| 1272, |
| 1273, |
| 1274, |
| 1275, |
| 1276, |
| 1277, |
| 1278, |
| 1279, |
| 1280, |
| 1281, |
| 1282, |
| 1283, |
| 1284, |
| 1285, |
| 1286, |
| 1287, |
| 1288, |
| 1289, |
| 1290, |
| 1291, |
| 1292, |
| 1293, |
| 1294, |
| 1295, |
| 1296, |
| 1297, |
| 1298, |
| 1299, |
| 1300, |
| 1301, |
| 1302, |
| 1303, |
| 1304, |
| 1305, |
| 1306, |
| 1307, |
| 1308, |
| 1309, |
| 1310, |
| 1311, |
| 1312, |
| 1313, |
| 1314, |
| 1315, |
| 1316, |
| 1317, |
| 1318, |
| 1319, |
| 1320, |
| 1321, |
| 1322, |
| 1323, |
| 1324, |
| 1325, |
| 1326, |
| 1327, |
| 1328, |
| 1329, |
| 1330, |
| 1331, |
| 1332, |
| 1333, |
| 1334, |
| 1335, |
| 1336, |
| 1337, |
| 1338, |
| 1339, |
| 1340, |
| 1341, |
| 1342, |
| 1343, |
| 1344, |
| 1345, |
| 1346, |
| 1347, |
| 1348, |
| 1349, |
| 1350, |
| 1351, |
| 1352, |
| 1353, |
| 1354, |
| 1355, |
| 1356, |
| 1357, |
| 1358, |
| 1359, |
| 1360, |
| 1361, |
| 1362, |
| 1363, |
| 1364, |
| 1365, |
| 1366, |
| 1367, |
| 1368, |
| 1369, |
| 1370, |
| 1371, |
| 1372, |
| 1373, |
| 1374, |
| 1375, |
| 1376, |
| 1377, |
| 1378, |
| 1379, |
| 1380, |
| 1381, |
| 1382, |
| 1383, |
| 1384, |
| 1385, |
| 1386, |
| 1387, |
| 1388, |
| 1389, |
| 1390, |
| 1391, |
| 1392, |
| 1393, |
| 1394, |
| 1395, |
| 1396, |
| 1397, |
| 1398, |
| 1399, |
| 1400, |
| 1401, |
| 1402, |
| 1403, |
| 1404, |
| 1405, |
| 1406, |
| 1407, |
| 1408, |
| 1409, |
| 1410, |
| 1411, |
| 1412, |
| 1413, |
| 1414, |
| 1415, |
| 1416, |
| 1417, |
| 1418, |
| 1419, |
| 1420, |
| 1421, |
| 1422, |
| 1423, |
| 1424, |
| 1425, |
| 1426, |
| 1427, |
| 1428, |
| 1429, |
| 1430, |
| 1431, |
| 1432, |
| 1433, |
| 1434, |
| 1435, |
| 1436, |
| 1437, |
| 1438, |
| 1439, |
| 1440, |
| 1441, |
| 1442, |
| 1443, |
| 1444, |
| 1445, |
| 1446, |
| 1447, |
| 1448, |
| 1449, |
| 1450, |
| 1451, |
| 1452, |
| 1453, |
| 1454, |
| 1455, |
| 1456, |
| 1457, |
| 1458, |
| 1459, |
| 1460, |
| 1461, |
| 1462, |
| 1463, |
| 1464, |
| 1465, |
| 1466, |
| 1467, |
| 1468, |
| 1469, |
| 1470, |
| 1471, |
| 1472, |
| 1473, |
| 1474, |
| 1475, |
| 1476, |
| 1477, |
| 1478, |
| 1479, |
| 1480, |
| 1481, |
| 1482, |
| 1483, |
| 1484, |
| 1485, |
| 1486, |
| 1487, |
| 1488, |
| 1489, |
| 1490, |
| 1491, |
| 1492, |
| 1493, |
| 1494, |
| 1495, |
| 1496, |
| 1497, |
| 1498, |
| 1499, |
| 1500, |
| 1501, |
| 1502, |
| 1503, |
| 1504, |
| 1505, |
| 1506, |
| 1507, |
| 1508, |
| 1509, |
| 1510, |
| 1511, |
| 1512, |
| 1513, |
| 1514, |
| 1515, |
| 1516, |
| 1517, |
| 1518, |
| 1519, |
| 1520 |
| ], |
| "image_transforms": { |
| "enable": true, |
| "max_num_transforms": 3, |
| "random_order": false, |
| "tfs": { |
| "brightness": { |
| "weight": 1.0, |
| "type": "ColorJitter", |
| "kwargs": { |
| "brightness": [ |
| 0.8, |
| 1.2 |
| ] |
| } |
| }, |
| "contrast": { |
| "weight": 1.0, |
| "type": "ColorJitter", |
| "kwargs": { |
| "contrast": [ |
| 0.8, |
| 1.2 |
| ] |
| } |
| }, |
| "saturation": { |
| "weight": 1.0, |
| "type": "ColorJitter", |
| "kwargs": { |
| "saturation": [ |
| 0.5, |
| 1.5 |
| ] |
| } |
| }, |
| "hue": { |
| "weight": 1.0, |
| "type": "ColorJitter", |
| "kwargs": { |
| "hue": [ |
| -0.05, |
| 0.05 |
| ] |
| } |
| }, |
| "sharpness": { |
| "weight": 1.0, |
| "type": "SharpnessJitter", |
| "kwargs": { |
| "sharpness": [ |
| 0.5, |
| 1.5 |
| ] |
| } |
| }, |
| "affine": { |
| "weight": 1.0, |
| "type": "RandomAffine", |
| "kwargs": { |
| "degrees": [ |
| -5.0, |
| 5.0 |
| ], |
| "translate": [ |
| 0.05, |
| 0.05 |
| ] |
| } |
| } |
| } |
| }, |
| "revision": "local", |
| "use_imagenet_stats": true, |
| "video_backend": "torchcodec", |
| "streaming": false |
| }, |
| "env": null, |
| "policy": { |
| "type": "diffusion", |
| "n_obs_steps": 2, |
| "input_features": { |
| "observation.images.cam_head": { |
| "type": "VISUAL", |
| "shape": [ |
| 3, |
| 256, |
| 256 |
| ] |
| }, |
| "observation.images.cam_left_wrist": { |
| "type": "VISUAL", |
| "shape": [ |
| 3, |
| 256, |
| 256 |
| ] |
| }, |
| "observation.images.cam_right_wrist": { |
| "type": "VISUAL", |
| "shape": [ |
| 3, |
| 256, |
| 256 |
| ] |
| }, |
| "observation.state": { |
| "type": "STATE", |
| "shape": [ |
| 16 |
| ] |
| } |
| }, |
| "output_features": { |
| "action": { |
| "type": "ACTION", |
| "shape": [ |
| 14 |
| ] |
| } |
| }, |
| "device": "cuda", |
| "use_amp": false, |
| "use_peft": false, |
| "push_to_hub": false, |
| "repo_id": null, |
| "private": null, |
| "tags": null, |
| "license": null, |
| "pretrained_path": null, |
| "horizon": 16, |
| "n_action_steps": 8, |
| "normalization_mapping": { |
| "VISUAL": "MEAN_STD", |
| "STATE": "MIN_MAX", |
| "ACTION": "MIN_MAX" |
| }, |
| "drop_n_last_frames": 7, |
| "vision_backbone": "resnet18", |
| "crop_shape": [ |
| 84, |
| 84 |
| ], |
| "crop_is_random": true, |
| "pretrained_backbone_weights": null, |
| "use_group_norm": true, |
| "spatial_softmax_num_keypoints": 32, |
| "use_separate_rgb_encoder_per_camera": false, |
| "down_dims": [ |
| 512, |
| 1024, |
| 2048 |
| ], |
| "kernel_size": 5, |
| "n_groups": 8, |
| "diffusion_step_embed_dim": 128, |
| "use_film_scale_modulation": true, |
| "noise_scheduler_type": "DDPM", |
| "num_train_timesteps": 100, |
| "beta_schedule": "squaredcos_cap_v2", |
| "beta_start": 0.0001, |
| "beta_end": 0.02, |
| "prediction_type": "epsilon", |
| "clip_sample": true, |
| "clip_sample_range": 1.0, |
| "num_inference_steps": null, |
| "do_mask_loss_for_padding": false, |
| "optimizer_lr": 0.0001, |
| "optimizer_betas": [ |
| 0.95, |
| 0.999 |
| ], |
| "optimizer_eps": 1e-08, |
| "optimizer_weight_decay": 1e-06, |
| "scheduler_name": "cosine", |
| "scheduler_warmup_steps": 500 |
| }, |
| "output_dir": "/weka/helenw/outputs/diffusion_PackBox", |
| "job_name": "diffusion_roboeval_v3_merged", |
| "resume": false, |
| "seed": 1000, |
| "num_workers": 2, |
| "batch_size": 256, |
| "steps": 5000, |
| "eval_freq": 20000, |
| "log_freq": 10, |
| "tolerance_s": 0.0001, |
| "save_checkpoint": true, |
| "save_freq": 1000, |
| "use_policy_training_preset": true, |
| "optimizer": { |
| "type": "adam", |
| "lr": 0.0001, |
| "weight_decay": 1e-06, |
| "grad_clip_norm": 10.0, |
| "betas": [ |
| 0.95, |
| 0.999 |
| ], |
| "eps": 1e-08 |
| }, |
| "scheduler": { |
| "type": "diffuser", |
| "num_warmup_steps": 500, |
| "name": "cosine" |
| }, |
| "eval": { |
| "n_episodes": 50, |
| "batch_size": 50, |
| "use_async_envs": false |
| }, |
| "wandb": { |
| "enable": true, |
| "disable_artifact": true, |
| "project": "lerobot", |
| "entity": null, |
| "notes": null, |
| "run_id": "taz7jcbk", |
| "mode": null |
| }, |
| "peft": null, |
| "use_rabc": false, |
| "rabc_progress_path": null, |
| "rabc_kappa": 0.01, |
| "rabc_epsilon": 1e-06, |
| "rabc_head_mode": "sparse", |
| "rename_map": {}, |
| "checkpoint_path": null |
| } |
