{ "type": "custom_pi0", "n_obs_steps": 1, "input_features": {}, "output_features": {}, "device": "cuda", "use_amp": false, "use_peft": false, "push_to_hub": true, "repo_id": null, "private": null, "tags": null, "license": null, "pretrained_path": "lerobot/pi0_base", "paligemma_variant": "gemma_2b", "action_expert_variant": "gemma_300m", "dtype": "bfloat16", "image_keys": [ "robot0_robotview_image", "robot0_eye_in_hand_image" ], "action_derivation_implementation": "ext__delta__osc__rotvec", "state_derivation_implementation": "rs_default", "normalization_stats": { "observation.state": { "mean": [ 0.011385115435902654, -0.11066487852350426, 0.9275649457826504, 0.7494435731164166, 1.069831168567631, 0.06506713687963374, 0.025805657331220012, -0.02590401388618914, 0.01803541620692451, -0.11212986440453838, 0.9280827677356757, 0.6500980088799709, 0.8341998775202242, 0.05805249210678411, 0.08762804097311136 ], "std": [ 0.1364878357112203, 0.06521976306006763, 0.05895699006536035, 1.6648833415711914, 2.2236785714685396, 0.13710193958271333, 0.011328591949326798, 0.011394605880609308, 0.13924010834100095, 0.06349817660760358, 0.06014770884144378, 1.694195263111645, 2.3261636263757737, 0.14145846058575146, 0.9961632294015995 ] }, "action": { "mean": [ 0.0014429540247476425, 5.855477872899049e-05, 0.00010094234202338543, -5.514898634610806e-05, -1.1709633593695834e-05, 0.0023358867284308323, 0.08610835467349552 ], "std": [ 0.0033159534044946504, 0.0020661345832887618, 0.003004563348536115, 0.012571091883774563, 0.012866583562733206, 0.026665803880115392, 0.9962859772258239 ] } }, "action_dim": 7, "chunk_size": 50, "max_state_dim": 32, "max_action_dim": 32, "num_inference_steps": 10, "time_sampling_beta_alpha": 1.5, "time_sampling_beta_beta": 1.0, "time_sampling_scale": 0.999, "time_sampling_offset": 0.001, "min_period": 0.004, "max_period": 4.0, "rtc_config": null, "image_resolution": [ 224, 224 ], "gradient_checkpointing": true, "compile_model": true, "compile_mode": "max-autotune", "freeze_vision_encoder": false, "train_expert_only": false, "optimizer_lr": 2.5e-05, "optimizer_betas": [ 0.9, 0.95 ], "optimizer_eps": 1e-08, "optimizer_weight_decay": 0.01, "optimizer_grad_clip_norm": 1.0, "scheduler_warmup_steps": 1000, "scheduler_decay_steps": 30000, "scheduler_decay_lr": 2.5e-06, "tokenizer_max_length": 48 }