| { |
| "dataset": { |
| "repo_id": "roboeval_ee_delta_20hz_v3_merged", |
| "root": "/weka/helenw/data/huggingface/lerobot/roboeval_ee_delta_20hz_v3_merged", |
| "episodes": [ |
| 2438, |
| 2439, |
| 2440, |
| 2441, |
| 2442, |
| 2443, |
| 2444, |
| 2445, |
| 2446, |
| 2447, |
| 2448, |
| 2449, |
| 2450, |
| 2451, |
| 2452, |
| 2453, |
| 2454, |
| 2455, |
| 2456, |
| 2457, |
| 2458, |
| 2459, |
| 2460, |
| 2461, |
| 2462, |
| 2463, |
| 2464, |
| 2465, |
| 2466, |
| 2467, |
| 2468, |
| 2469, |
| 2470, |
| 2471, |
| 2472, |
| 2473, |
| 2474, |
| 2475, |
| 2476, |
| 2477, |
| 2478, |
| 2479, |
| 2480, |
| 2481, |
| 2482, |
| 2483, |
| 2484, |
| 2485, |
| 2486, |
| 2487, |
| 2488, |
| 2489, |
| 2490, |
| 2491, |
| 2492, |
| 2493, |
| 2494, |
| 2495, |
| 2496, |
| 2497, |
| 2498, |
| 2499, |
| 2500, |
| 2501, |
| 2502, |
| 2503, |
| 2504, |
| 2505, |
| 2506, |
| 2507, |
| 2508, |
| 2509, |
| 2510, |
| 2511, |
| 2512, |
| 2513, |
| 2514, |
| 2515, |
| 2516, |
| 2517, |
| 2518, |
| 2519, |
| 2520, |
| 2521, |
| 2522, |
| 2523, |
| 2524, |
| 2525, |
| 2526, |
| 2527, |
| 2528, |
| 2529, |
| 2530, |
| 2531, |
| 2532, |
| 2533, |
| 2534, |
| 2535, |
| 2536, |
| 2537, |
| 2538, |
| 2539, |
| 2540, |
| 2541, |
| 2542, |
| 2543, |
| 2544, |
| 2545, |
| 2546, |
| 2547, |
| 2548, |
| 2549, |
| 2550, |
| 2551, |
| 2552, |
| 2553, |
| 2554, |
| 2555, |
| 2556, |
| 2557, |
| 2558, |
| 2559, |
| 2560, |
| 2561, |
| 2562, |
| 2563, |
| 2564, |
| 2565, |
| 2566, |
| 2567, |
| 2568, |
| 2569, |
| 2570, |
| 2571, |
| 2572, |
| 2573, |
| 2574, |
| 2575, |
| 2576, |
| 2577, |
| 2578, |
| 2579, |
| 2580, |
| 2581, |
| 2582, |
| 2583, |
| 2584, |
| 2585, |
| 2586, |
| 2587, |
| 2588, |
| 2589, |
| 2590, |
| 2591, |
| 2592, |
| 2593, |
| 2594, |
| 2595, |
| 2596, |
| 2597, |
| 2598, |
| 2599, |
| 2600, |
| 2601, |
| 2602, |
| 2603, |
| 2604, |
| 2605, |
| 2606, |
| 2607, |
| 2608, |
| 2609, |
| 2610, |
| 2611, |
| 2612, |
| 2613, |
| 2614, |
| 2615, |
| 2616, |
| 2617, |
| 2618, |
| 2619, |
| 2620, |
| 2621, |
| 2622, |
| 2623, |
| 2624, |
| 2625, |
| 2626, |
| 2627, |
| 2628, |
| 2629, |
| 2630, |
| 2631, |
| 2632, |
| 2633, |
| 2634, |
| 2635, |
| 2636, |
| 2637, |
| 2638, |
| 2639, |
| 2640, |
| 2641, |
| 2642, |
| 2643, |
| 2644, |
| 2645, |
| 2646, |
| 2647, |
| 2648, |
| 2649, |
| 2650, |
| 2651, |
| 2652, |
| 2653, |
| 2654, |
| 2655, |
| 2656, |
| 2657, |
| 2658, |
| 2659, |
| 2660, |
| 2661, |
| 2662, |
| 2663, |
| 2664, |
| 2665, |
| 2666, |
| 2667, |
| 2668, |
| 2669, |
| 2670, |
| 2671, |
| 2672, |
| 2673, |
| 2674, |
| 2675, |
| 2676, |
| 2677, |
| 2678, |
| 2679, |
| 2680, |
| 2681, |
| 2682, |
| 2683, |
| 2684, |
| 2685, |
| 2686, |
| 2687, |
| 2688, |
| 2689, |
| 2690, |
| 2691, |
| 2692, |
| 2693, |
| 2694, |
| 2695, |
| 2696, |
| 2697, |
| 2698, |
| 2699, |
| 2700, |
| 2701, |
| 2702, |
| 2703, |
| 2704, |
| 2705, |
| 2706, |
| 2707, |
| 2708, |
| 2709, |
| 2710, |
| 2711, |
| 2712, |
| 2713, |
| 2714, |
| 2715, |
| 2716, |
| 2717, |
| 2718, |
| 2719, |
| 2720, |
| 2721, |
| 2722, |
| 2723, |
| 2724, |
| 2725, |
| 2726, |
| 2727, |
| 2728, |
| 2729, |
| 2730, |
| 2731, |
| 2732, |
| 2733, |
| 2734, |
| 2735, |
| 2736, |
| 2737, |
| 2738, |
| 2739, |
| 2740, |
| 2741, |
| 2742, |
| 2743, |
| 2744, |
| 2745, |
| 2746, |
| 2747, |
| 2748, |
| 2749, |
| 2750, |
| 2751, |
| 2752, |
| 2753, |
| 2754, |
| 2755, |
| 2756, |
| 2757, |
| 2758, |
| 2759, |
| 2760, |
| 2761, |
| 2762, |
| 2763, |
| 2764, |
| 2765, |
| 2766, |
| 2767, |
| 2768, |
| 2769, |
| 2770, |
| 2771, |
| 2772, |
| 2773, |
| 2774, |
| 2775, |
| 2776, |
| 2777, |
| 2778, |
| 2779, |
| 2780, |
| 2781, |
| 2782, |
| 2783, |
| 2784, |
| 2785, |
| 2786, |
| 2787, |
| 2788, |
| 2789, |
| 2790, |
| 2791, |
| 2792, |
| 2793, |
| 2794, |
| 2795, |
| 2796, |
| 2797, |
| 2798, |
| 2799, |
| 2800, |
| 2801, |
| 2802, |
| 2803, |
| 2804, |
| 2805, |
| 2806, |
| 2807, |
| 2808, |
| 2809, |
| 2810, |
| 2811, |
| 2812, |
| 2813, |
| 2814, |
| 2815, |
| 2816, |
| 2817, |
| 2818, |
| 2819, |
| 2820, |
| 2821, |
| 2822, |
| 2823, |
| 2824, |
| 2825, |
| 2826, |
| 2827, |
| 2828, |
| 2829, |
| 2830, |
| 2831, |
| 2832, |
| 2833, |
| 2834, |
| 2835, |
| 2836, |
| 2837, |
| 2838, |
| 2839, |
| 2840, |
| 2841, |
| 2842, |
| 2843, |
| 2844, |
| 2845, |
| 2846, |
| 2847, |
| 2848, |
| 2849, |
| 2850, |
| 2851, |
| 2852, |
| 2853, |
| 2854, |
| 2855, |
| 2856, |
| 2857, |
| 2858, |
| 2859, |
| 2860, |
| 2861, |
| 2862, |
| 2863, |
| 2864, |
| 2865, |
| 2866, |
| 2867, |
| 2868, |
| 2869, |
| 2870, |
| 2871, |
| 2872, |
| 2873, |
| 2874, |
| 2875, |
| 2876, |
| 2877, |
| 2878, |
| 2879, |
| 2880, |
| 2881, |
| 2882, |
| 2883, |
| 2884, |
| 2885, |
| 2886, |
| 2887, |
| 2888, |
| 2889, |
| 2890, |
| 2891, |
| 2892, |
| 2893, |
| 2894, |
| 2895, |
| 2896, |
| 2897, |
| 2898, |
| 2899, |
| 2900, |
| 2901, |
| 2902, |
| 2903, |
| 2904, |
| 2905, |
| 2906, |
| 2907, |
| 2908, |
| 2909, |
| 2910, |
| 2911, |
| 2912, |
| 2913, |
| 2914, |
| 2915, |
| 2916, |
| 2917, |
| 2918, |
| 2919, |
| 2920, |
| 2921, |
| 2922, |
| 2923, |
| 2924, |
| 2925, |
| 2926, |
| 2927, |
| 2928, |
| 2929 |
| ], |
| "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_StackTwoBlocks", |
| "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": "8wzpes1k", |
| "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 |
| } |
