{ "dataset": { "repo_id": "Basr88/Merged_800_2", "root": null, "episodes": [ 444, 731, 6, 601, 586, 725, 416, 67, 496, 392, 342, 300, 785, 434, 710, 108, 177, 415, 619, 171, 658, 560, 763, 764, 635, 758, 548, 448, 137, 59, 344, 335, 253, 657, 593, 71, 21, 669, 231, 293, 433, 277, 131, 303, 715, 523, 732, 81, 693, 738, 32, 99, 324, 720, 698, 643, 609, 594, 76, 84, 296, 632, 306, 75, 311, 753, 114, 289, 430, 672, 435, 535, 298, 280, 723, 552, 459, 272, 29, 684, 572, 227, 441, 760, 189, 573, 405, 489, 425, 15, 349, 506, 183, 125, 510, 610, 505, 39, 508, 642, 624, 403, 676, 395, 9, 238, 126, 668, 597, 770, 302, 418, 682, 718, 616, 475, 608, 603, 143, 185, 182, 389, 413, 336, 359, 366, 480, 540, 122, 730, 796, 134, 757, 148, 542, 12, 559, 60, 229, 90, 176, 74, 320, 761, 279, 13, 467, 788, 239, 562, 736, 541, 144, 391, 631, 769, 246, 561, 504, 742, 53, 719, 3, 604, 717, 46, 707, 22, 347, 283, 452, 153, 368, 97, 644, 49, 524, 690, 666, 269, 274, 107, 312, 149, 607, 95, 514, 266, 522, 756, 410, 61, 388, 353, 590, 217, 193, 94, 294, 529, 611, 281, 352, 721, 665, 722, 778, 356, 646, 711, 579, 50, 640, 321, 614, 583, 408, 629, 376, 432, 204, 650, 453, 194, 119, 737, 549, 196, 315, 599, 132, 145, 591, 54, 292, 175, 7, 243, 337, 772, 654, 129, 438, 613, 265, 139, 458, 304, 398, 553, 313, 261, 257, 254, 663, 301, 38, 558, 35, 89, 792, 568, 206, 248, 495, 533, 521, 240, 534, 709, 498, 492, 714, 621, 188, 531, 497, 526, 437, 235, 519, 746, 214, 641, 158, 166, 701, 207, 726, 645, 197, 617, 637, 93, 318, 400, 449, 402, 515, 502, 140, 488, 361, 96, 685, 520, 588, 428, 799, 110, 308, 716, 696, 705, 345, 439, 307, 695, 73, 636, 86, 466, 369, 462, 782, 85, 278, 426, 771, 530, 592, 270, 331, 759, 674, 712, 370, 355, 332, 170, 199, 547, 589, 169, 781, 151, 4, 380, 406, 536, 454, 252, 208, 679, 659, 503, 576, 124, 156, 442, 186, 681, 440, 223, 457, 363, 136, 484, 485, 299, 443, 284, 688, 27, 138, 202, 386, 546, 686, 135, 655, 638, 739, 436, 451, 555, 83, 351, 456, 314, 622, 282, 20, 66, 211, 768, 264, 791, 178, 8, 776, 255, 330, 752, 774, 678, 578, 116, 1, 476, 371, 326, 790, 639, 163, 664, 507, 581, 411, 630, 319, 226, 794, 338, 130, 218, 346, 662, 241, 165, 48, 626, 728, 286, 113, 215, 747, 128, 111, 787, 587, 627, 551, 112, 699, 56, 564, 494, 98, 734, 571, 667, 779, 343, 585, 234, 651, 472, 31, 192, 103, 478, 58, 735, 23, 754, 382, 80, 618, 57, 364, 584, 180, 26, 201, 276, 713, 350, 288, 647, 419, 538, 154, 620, 367, 577, 450, 748, 40, 147, 16, 675, 471, 729, 291, 677, 393, 115, 205, 82 ], "image_transforms": { "enable": false, "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": null, "use_imagenet_stats": true, "video_backend": "pyav", "streaming": false }, "env": null, "policy": { "type": "act", "n_obs_steps": 1, "input_features": { "observation.state": { "type": "STATE", "shape": [ 6 ] }, "observation.images.front": { "type": "VISUAL", "shape": [ 3, 480, 640 ] } }, "output_features": { "action": { "type": "ACTION", "shape": [ 6 ] } }, "device": "cuda", "use_amp": false, "use_peft": false, "push_to_hub": true, "repo_id": "Basr88/Merged_dataset_4", "private": null, "tags": null, "license": null, "pretrained_path": null, "chunk_size": 100, "n_action_steps": 100, "normalization_mapping": { "VISUAL": "MEAN_STD", "STATE": "MEAN_STD", "ACTION": "MEAN_STD" }, "vision_backbone": "resnet18", "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1", "replace_final_stride_with_dilation": false, "pre_norm": false, "dim_model": 512, "n_heads": 8, "dim_feedforward": 3200, "feedforward_activation": "relu", "n_encoder_layers": 4, "n_decoder_layers": 1, "use_vae": true, "latent_dim": 32, "n_vae_encoder_layers": 4, "temporal_ensemble_coeff": null, "dropout": 0.1, "kl_weight": 10.0, "optimizer_lr": 1e-05, "optimizer_weight_decay": 0.0001, "optimizer_lr_backbone": 1e-05 }, "output_dir": "/home/basr88/data/outputs/train/act_gluestick_merged_500", "job_name": "act_gluestick", "resume": false, "seed": 1000, "cudnn_deterministic": false, "num_workers": 4, "batch_size": 8, "steps": 150000, "eval_freq": 20000, "log_freq": 200, "tolerance_s": 0.0001, "save_checkpoint": true, "save_freq": 20000, "use_policy_training_preset": true, "optimizer": { "type": "adamw", "lr": 1e-05, "weight_decay": 0.0001, "grad_clip_norm": 10.0, "betas": [ 0.9, 0.999 ], "eps": 1e-08 }, "scheduler": null, "eval": { "n_episodes": 50, "batch_size": 50, "use_async_envs": false }, "wandb": { "enable": false, "disable_artifact": false, "project": "lerobot", "entity": null, "notes": null, "run_id": null, "mode": null, "add_tags": true }, "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 }