{ "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": "rs__ext__delta__osc__rotvec", "state_derivation_implementation": "rs__panda__current", "normalization_stats": { "observation.state": { "mean": [ 0.010927826461886586, -0.11101506060948091, 0.9274730413732397, 0.7497428323639496, 1.0645374215748182, 0.06514694207382514, 0.025863758656798477, -0.025972364711242486, 0.017526891187463006, -0.11250957163830645, 0.9279108377180693, 0.6515663296136903, 0.8407249809379918, 0.05847413601680504, 0.0839868758002561 ], "std": [ 0.13625355214133486, 0.06556713013701973, 0.05885601617154858, 1.660531214280759, 2.2292484419876866, 0.1377698731190702, 0.011327420885424582, 0.01139249930698528, 0.13905677029133814, 0.0638221814525735, 0.0600715067798994, 1.6892137051635012, 2.3268053626589844, 0.14215820255821376, 0.9964768288182814 ] }, "action": { "mean": [ 0.001430536423258196, 5.983774069280805e-05, 9.523755249480281e-05, -5.2735714710504744e-05, -8.54343881777062e-06, 0.0022963504568156534, 0.08360275288092177 ], "std": [ 0.0033059263914112244, 0.002065465304466641, 0.003003305718389858, 0.012557244241380657, 0.012832757063846172, 0.026635096131961848, 0.9964993612854287 ] } }, "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": true, "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 }