config.json

{
    "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
}