samithva/stack_cups
06
1{2 "dataset": {3 "repo_id": "stack_cups_3cams",4 "repo_type": "dataset",5 "root": "./datasets/stack_cups_3cams",6 "episodes": null,7 "image_transforms": {8 "enable": false,9 "max_num_transforms": 3,10 "random_order": false,11 "tfs": {12 "brightness": {13 "weight": 1.0,14 "type": "ColorJitter",15 "kwargs": {16 "brightness": [17 0.8,18 1.219 ]20 }21 },22 "contrast": {23 "weight": 1.0,24 "type": "ColorJitter",25 "kwargs": {26 "contrast": [27 0.8,28 1.229 ]30 }31 },32 "saturation": {33 "weight": 1.0,34 "type": "ColorJitter",35 "kwargs": {36 "saturation": [37 0.5,38 1.539 ]40 }41 },42 "hue": {43 "weight": 1.0,44 "type": "ColorJitter",45 "kwargs": {46 "hue": [47 -0.05,48 0.0549 ]50 }51 },52 "sharpness": {53 "weight": 1.0,54 "type": "SharpnessJitter",55 "kwargs": {56 "sharpness": [57 0.5,58 1.559 ]60 }61 },62 "affine": {63 "weight": 1.0,64 "type": "RandomAffine",65 "kwargs": {66 "degrees": [67 -5.0,68 5.069 ],70 "translate": [71 0.05,72 0.0573 ]74 }75 }76 }77 },78 "revision": null,79 "use_imagenet_stats": true,80 "video_backend": "pyav",81 "return_uint8": false,82 "depth_output_unit": "mm",83 "streaming": false,84 "eval_split": 0.0585 },86 "env": null,87 "policy": {88 "type": "pi05",89 "n_obs_steps": 1,90 "input_features": {91 "observation.state": {92 "type": "STATE",93 "shape": [94 795 ]96 },97 "observation.images.r_wrist": {98 "type": "VISUAL",99 "shape": [100 3,101 480,102 640103 ]104 },105 "observation.images.top": {106 "type": "VISUAL",107 "shape": [108 3,109 480,110 640111 ]112 },113 "observation.images.realsense": {114 "type": "VISUAL",115 "shape": [116 3,117 480,118 640119 ]120 }121 },122 "output_features": {123 "action": {124 "type": "ACTION",125 "shape": [126 7127 ]128 }129 },130 "device": "cuda",131 "use_amp": false,132 "use_peft": false,133 "push_to_hub": true,134 "repo_id": "samithva/pi05_stack_cups_3cams",135 "private": null,136 "tags": null,137 "license": null,138 "pretrained_path": "/data4/wanshan/workspace/lerobot/checkpoints/lerobot--pi05-base",139 "pretrained_revision": null,140 "paligemma_variant": "gemma_2b",141 "action_expert_variant": "gemma_300m",142 "dtype": "bfloat16",143 "chunk_size": 50,144 "n_action_steps": 50,145 "max_state_dim": 32,146 "max_action_dim": 32,147 "num_inference_steps": 10,148 "time_sampling_beta_alpha": 1.5,149 "time_sampling_beta_beta": 1.0,150 "time_sampling_scale": 0.999,151 "time_sampling_offset": 0.001,152 "min_period": 0.004,153 "max_period": 4.0,154 "use_relative_actions": false,155 "relative_exclude_joints": [156 "gripper"157 ],158 "action_feature_names": [159 "joint_1.pos",160 "joint_2.pos",161 "joint_3.pos",162 "joint_4.pos",163 "joint_5.pos",164 "joint_6.pos",165 "gripper.pos"166 ],167 "rtc_config": null,168 "image_resolution": [169 224,170 224171 ],172 "empty_cameras": 0,173 "tokenizer_max_length": 200,174 "normalization_mapping": {175 "VISUAL": "IDENTITY",176 "STATE": "QUANTILES",177 "ACTION": "QUANTILES"178 },179 "gradient_checkpointing": true,180 "compile_model": true,181 "compile_mode": "max-autotune",182 "freeze_vision_encoder": false,183 "train_expert_only": false,184 "optimizer_lr": 2.5e-05,185 "optimizer_betas": [186 0.9,187 0.95188 ],189 "optimizer_eps": 1e-08,190 "optimizer_weight_decay": 0.01,191 "optimizer_grad_clip_norm": 1.0,192 "scheduler_warmup_steps": 1000,193 "scheduler_decay_steps": 30000,194 "scheduler_decay_lr": 2.5e-06195 },196 "reward_model": null,197 "output_dir": "/data4/wanshan/workspace/lerobot/outputs/train/pi05_stack_cups_3cams",198 "job_name": "pi05_stack_cups_3cams",199 "resume": false,200 "seed": 1000,201 "cudnn_deterministic": false,202 "num_workers": 8,203 "batch_size": 8,204 "prefetch_factor": 4,205 "persistent_workers": true,206 "dataloader_multiprocessing_context": "spawn",207 "steps": 5000,208 "env_eval_freq": 20000,209 "log_freq": 10,210 "eval_steps": 200,211 "max_eval_samples": 0,212 "tolerance_s": 0.0001,213 "save_checkpoint": true,214 "save_freq": 200,215 "use_policy_training_preset": true,216 "optimizer": {217 "type": "adamw",218 "lr": 2.5e-05,219 "weight_decay": 0.01,220 "grad_clip_norm": 1.0,221 "betas": [222 0.9,223 0.95224 ],225 "eps": 1e-08226 },227 "scheduler": {228 "type": "cosine_decay_with_warmup",229 "num_warmup_steps": 1000,230 "num_decay_steps": 30000,231 "peak_lr": 2.5e-05,232 "decay_lr": 2.5e-06233 },234 "eval": {235 "n_episodes": 50,236 "batch_size": 50,237 "use_async_envs": true,238 "recording": false,239 "recording_repo_id": null,240 "recording_private": false241 },242 "wandb": {243 "enable": true,244 "disable_artifact": false,245 "project": "lerobot",246 "entity": null,247 "notes": null,248 "run_id": "ioc7s007",249 "mode": null,250 "add_tags": true251 },252 "peft": null,253 "job": {254 "target": null,255 "image": "huggingface/lerobot-gpu:latest",256 "timeout": "2d",257 "detach": false,258 "tags": []259 },260 "save_checkpoint_to_hub": false,261 "sample_weighting": null,262 "rename_map": {},263 "checkpoint_path": null264}