Rayao/pi05_g1_blockstacking
015
1{2 "dataset": {3 "repo_id": "unitreerobotics/G1_Dex3_BlockStacking_Dataset",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": "torchcodec",80 "streaming": false81 },82 "env": null,83 "policy": {84 "type": "pi05",85 "n_obs_steps": 1,86 "input_features": {87 "observation.state": {88 "type": "STATE",89 "shape": [90 2891 ]92 },93 "observation.images.cam_left_high": {94 "type": "VISUAL",95 "shape": [96 3,97 480,98 64099 ]100 },101 "observation.images.cam_right_high": {102 "type": "VISUAL",103 "shape": [104 3,105 480,106 640107 ]108 },109 "observation.images.cam_left_wrist": {110 "type": "VISUAL",111 "shape": [112 3,113 480,114 640115 ]116 },117 "observation.images.cam_right_wrist": {118 "type": "VISUAL",119 "shape": [120 3,121 480,122 640123 ]124 }125 },126 "output_features": {127 "action": {128 "type": "ACTION",129 "shape": [130 28131 ]132 }133 },134 "device": "cuda",135 "use_amp": false,136 "push_to_hub": false,137 "repo_id": null,138 "private": null,139 "tags": null,140 "license": null,141 "pretrained_path": "lerobot/pi05_base",142 "paligemma_variant": "gemma_2b",143 "action_expert_variant": "gemma_300m",144 "dtype": "bfloat16",145 "chunk_size": 50,146 "n_action_steps": 50,147 "max_state_dim": 32,148 "max_action_dim": 32,149 "num_inference_steps": 10,150 "time_sampling_beta_alpha": 1.5,151 "time_sampling_beta_beta": 1.0,152 "time_sampling_scale": 0.999,153 "time_sampling_offset": 0.001,154 "min_period": 0.004,155 "max_period": 4.0,156 "image_resolution": [157 224,158 224159 ],160 "empty_cameras": 0,161 "tokenizer_max_length": 200,162 "normalization_mapping": {163 "VISUAL": "IDENTITY",164 "STATE": "QUANTILES",165 "ACTION": "QUANTILES"166 },167 "gradient_checkpointing": true,168 "compile_model": true,169 "compile_mode": "max-autotune",170 "optimizer_lr": 2.5e-05,171 "optimizer_betas": [172 0.9,173 0.95174 ],175 "optimizer_eps": 1e-08,176 "optimizer_weight_decay": 0.01,177 "optimizer_grad_clip_norm": 1.0,178 "scheduler_warmup_steps": 1000,179 "scheduler_decay_steps": 30000,180 "scheduler_decay_lr": 2.5e-06181 },182 "output_dir": "/workspace/outputs/pi05_g1_blockstacking",183 "job_name": "pi05_g1_blockstacking",184 "resume": false,185 "seed": 1000,186 "num_workers": 4,187 "batch_size": 32,188 "steps": 8000,189 "eval_freq": 20000,190 "log_freq": 200,191 "save_checkpoint": true,192 "save_freq": 8000,193 "use_policy_training_preset": true,194 "optimizer": {195 "type": "adamw",196 "lr": 2.5e-05,197 "weight_decay": 0.01,198 "grad_clip_norm": 1.0,199 "betas": [200 0.9,201 0.95202 ],203 "eps": 1e-08204 },205 "scheduler": {206 "type": "cosine_decay_with_warmup",207 "num_warmup_steps": 1000,208 "num_decay_steps": 30000,209 "peak_lr": 2.5e-05,210 "decay_lr": 2.5e-06211 },212 "eval": {213 "n_episodes": 50,214 "batch_size": 50,215 "use_async_envs": false216 },217 "wandb": {218 "enable": false,219 "disable_artifact": false,220 "project": "lerobot",221 "entity": null,222 "notes": null,223 "run_id": null,224 "mode": null225 },226 "checkpoint_path": null,227 "rename_map": {}228}