RobotisSW/Task_255_PickUP_Jelly_10K_MCAP

1{2    "dataset": {3        "repo_id": "local/Task_000255_Pick_Jelly_WhiteShelf_Lift30_lerobot_v3_filtered",4        "root": "/home/robotis-ai/cyclo_intelligence/docker/workspace/lerobot/Task_000255_Pick_Jelly_WhiteShelf_Lift30_lerobot_v3_filtered",5        "episodes": null,6        "image_transforms": {7            "enable": true,8            "max_num_transforms": 2,9            "random_order": false,10            "tfs": {11                "brightness": {12                    "weight": 1.0,13                    "type": "ColorJitter",14                    "kwargs": {15                        "brightness": [16                            0.95,17                            1.0518                        ]19                    }20                },21                "affine": {22                    "weight": 1.0,23                    "type": "RandomAffine",24                    "kwargs": {25                        "degrees": [26                            -2,27                            228                        ],29                        "translate": [30                            0.01,31                            0.0132                        ]33                    }34                }35            }36        },37        "revision": null,38        "use_imagenet_stats": true,39        "video_backend": "torchcodec",40        "return_uint8": false,41        "depth_output_unit": "mm",42        "streaming": false,43        "eval_split": 0.044    },45    "env": null,46    "policy": {47        "type": "act",48        "n_obs_steps": 1,49        "input_features": {50            "observation.state": {51                "type": "STATE",52                "shape": [53                    1654                ]55            },56            "observation.images.rgb.cam_left_head": {57                "type": "VISUAL",58                "shape": [59                    3,60                    720,61                    128062                ]63            },64            "observation.images.rgb.cam_left_wrist": {65                "type": "VISUAL",66                "shape": [67                    3,68                    640,69                    48070                ]71            },72            "observation.images.rgb.cam_right_wrist": {73                "type": "VISUAL",74                "shape": [75                    3,76                    640,77                    48078                ]79            }80        },81        "output_features": {82            "action": {83                "type": "ACTION",84                "shape": [85                    1686                ]87            }88        },89        "device": "cuda",90        "use_amp": false,91        "use_peft": false,92        "push_to_hub": false,93        "repo_id": null,94        "private": null,95        "tags": null,96        "license": null,97        "pretrained_path": null,98        "pretrained_revision": null,99        "chunk_size": 30,100        "n_action_steps": 30,101        "normalization_mapping": {102            "VISUAL": "MEAN_STD",103            "STATE": "MEAN_STD",104            "ACTION": "MEAN_STD"105        },106        "vision_backbone": "resnet18",107        "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",108        "replace_final_stride_with_dilation": false,109        "pre_norm": false,110        "dim_model": 512,111        "n_heads": 8,112        "dim_feedforward": 3200,113        "feedforward_activation": "relu",114        "n_encoder_layers": 4,115        "n_decoder_layers": 1,116        "use_vae": true,117        "latent_dim": 32,118        "n_vae_encoder_layers": 4,119        "temporal_ensemble_coeff": null,120        "dropout": 0.1,121        "kl_weight": 10.0,122        "optimizer_lr": 1e-05,123        "optimizer_weight_decay": 0.0001,124        "optimizer_lr_backbone": 1e-05125    },126    "reward_model": null,127    "output_dir": "/home/robotis-ai/cyclo_intelligence/docker/workspace/model/lerobot/Task_000255_Pick_Jelly_WhiteShelf_Lift30_lerobot_v3_batch16",128    "job_name": "Task_000255_Pick_Jelly_WhiteShelf_Lift30_lerobot_v3_filtered",129    "resume": false,130    "seed": 1000,131    "cudnn_deterministic": false,132    "num_workers": 4,133    "batch_size": 16,134    "prefetch_factor": 4,135    "persistent_workers": true,136    "steps": 50000,137    "env_eval_freq": 0,138    "log_freq": 100,139    "eval_steps": 0,140    "max_eval_samples": 0,141    "tolerance_s": 0.0001,142    "save_checkpoint": true,143    "save_freq": 10000,144    "use_policy_training_preset": true,145    "optimizer": {146        "type": "adamw",147        "lr": 1e-05,148        "weight_decay": 0.0001,149        "grad_clip_norm": 10.0,150        "betas": [151            0.9,152            0.999153        ],154        "eps": 1e-08155    },156    "scheduler": null,157    "eval": {158        "n_episodes": 50,159        "batch_size": 16,160        "use_async_envs": true,161        "recording": false,162        "recording_repo_id": null,163        "recording_private": false164    },165    "wandb": {166        "enable": false,167        "disable_artifact": false,168        "project": "lerobot",169        "entity": null,170        "notes": null,171        "run_id": null,172        "mode": null,173        "add_tags": true174    },175    "peft": null,176    "job": {177        "target": null,178        "image": "huggingface/lerobot-gpu:latest",179        "timeout": "2d",180        "detach": false,181        "tags": []182    },183    "save_checkpoint_to_hub": false,184    "sample_weighting": null,185    "rename_map": {},186    "checkpoint_path": null187}