vkamenski/groot-stacking-blocks
05
1{2 "dataset": {3 "repo_id": "vkamenski/stacking-blocks-v5",4 "root": null,5 "episodes": null,6 "image_transforms": {7 "enable": false,8 "max_num_transforms": 3,9 "random_order": false,10 "tfs": {11 "brightness": {12 "weight": 1.0,13 "type": "ColorJitter",14 "kwargs": {15 "brightness": [16 0.8,17 1.218 ]19 }20 },21 "contrast": {22 "weight": 1.0,23 "type": "ColorJitter",24 "kwargs": {25 "contrast": [26 0.8,27 1.228 ]29 }30 },31 "saturation": {32 "weight": 1.0,33 "type": "ColorJitter",34 "kwargs": {35 "saturation": [36 0.5,37 1.538 ]39 }40 },41 "hue": {42 "weight": 1.0,43 "type": "ColorJitter",44 "kwargs": {45 "hue": [46 -0.05,47 0.0548 ]49 }50 },51 "sharpness": {52 "weight": 1.0,53 "type": "SharpnessJitter",54 "kwargs": {55 "sharpness": [56 0.5,57 1.558 ]59 }60 },61 "affine": {62 "weight": 1.0,63 "type": "RandomAffine",64 "kwargs": {65 "degrees": [66 -5.0,67 5.068 ],69 "translate": [70 0.05,71 0.0572 ]73 }74 }75 }76 },77 "revision": null,78 "use_imagenet_stats": true,79 "video_backend": "pyav",80 "streaming": false81 },82 "env": null,83 "policy": {84 "type": "groot",85 "n_obs_steps": 1,86 "input_features": {87 "observation.state": {88 "type": "STATE",89 "shape": [90 691 ]92 },93 "observation.images.front": {94 "type": "VISUAL",95 "shape": [96 3,97 480,98 64099 ]100 },101 "observation.images.wrist": {102 "type": "VISUAL",103 "shape": [104 3,105 480,106 640107 ]108 }109 },110 "output_features": {111 "action": {112 "type": "ACTION",113 "shape": [114 6115 ]116 }117 },118 "device": "cuda",119 "use_amp": false,120 "push_to_hub": true,121 "repo_id": "vkamenski/groot-stacking-blocks",122 "private": null,123 "tags": null,124 "license": null,125 "pretrained_path": null,126 "chunk_size": 50,127 "n_action_steps": 50,128 "max_state_dim": 64,129 "max_action_dim": 32,130 "normalization_mapping": {131 "VISUAL": "IDENTITY",132 "STATE": "MEAN_STD",133 "ACTION": "MEAN_STD"134 },135 "image_size": [136 224,137 224138 ],139 "base_model_path": "nvidia/GR00T-N1.5-3B",140 "tokenizer_assets_repo": "lerobot/eagle2hg-processor-groot-n1p5",141 "embodiment_tag": "new_embodiment",142 "tune_llm": false,143 "tune_visual": false,144 "tune_projector": true,145 "tune_diffusion_model": false,146 "lora_rank": 0,147 "lora_alpha": 16,148 "lora_dropout": 0.1,149 "lora_full_model": false,150 "optimizer_lr": 0.0001,151 "optimizer_betas": [152 0.95,153 0.999154 ],155 "optimizer_eps": 1e-08,156 "optimizer_weight_decay": 1e-05,157 "warmup_ratio": 0.05,158 "use_bf16": true,159 "video_backend": "decord",160 "balance_dataset_weights": true,161 "balance_trajectory_weights": true,162 "dataset_paths": null,163 "output_dir": "./tmp/gr00t",164 "save_steps": 1000,165 "max_steps": 10000,166 "batch_size": 32,167 "dataloader_num_workers": 8,168 "report_to": "wandb",169 "resume": false170 },171 "output_dir": "outputs/train/my_groot",172 "job_name": "groot_training",173 "resume": false,174 "seed": 1000,175 "num_workers": 4,176 "batch_size": 64,177 "steps": 30000,178 "eval_freq": 20000,179 "log_freq": 200,180 "save_checkpoint": true,181 "save_freq": 10000,182 "use_policy_training_preset": true,183 "optimizer": {184 "type": "adamw",185 "lr": 0.0001,186 "weight_decay": 1e-05,187 "grad_clip_norm": 10.0,188 "betas": [189 0.95,190 0.999191 ],192 "eps": 1e-08193 },194 "scheduler": {195 "type": "cosine_decay_with_warmup",196 "num_warmup_steps": 500,197 "num_decay_steps": 10000,198 "peak_lr": 0.0001,199 "decay_lr": 1e-05200 },201 "eval": {202 "n_episodes": 50,203 "batch_size": 50,204 "use_async_envs": false205 },206 "wandb": {207 "enable": true,208 "disable_artifact": true,209 "project": "lerobot",210 "entity": null,211 "notes": null,212 "run_id": "7w8lohyv",213 "mode": null214 },215 "checkpoint_path": null,216 "rename_map": {}217}