{
  "name": "pi0.5_LoRA_finetune_so100",
  "project_name": "openpi",
  "exp_name": "pi05_so100_lora_finetune",
  "model": {
    "action_dim": 32,
    "action_horizon": 50,
    "max_token_len": 200,
    "dtype": "bfloat16",
    "paligemma_variant": "gemma_2b_lora",
    "action_expert_variant": "gemma_300m_lora",
    "pi05": true,
    "discrete_state_input": true
  },
  "weight_loader": {
    "params_path": "gs://openpi-assets/checkpoints/pi05_base/params"
  },
  "pytorch_weight_path": null,
  "pytorch_training_precision": "bfloat16",
  "lr_schedule": {
    "warmup_steps": 1000,
    "peak_lr": 2.5e-05,
    "decay_steps": 30000,
    "decay_lr": 2.5e-06
  },
  "optimizer": {
    "b1": 0.9,
    "b2": 0.95,
    "eps": 1e-08,
    "weight_decay": 1e-10,
    "clip_gradient_norm": 1.0
  },
  "ema_decay": null,
  "freeze_filter": "All",
  "data": {
    "repo_id": "MrRock/fox",
    "assets": {
      "assets_dir": null,
      "asset_id": null
    },
    "base_config": {
      "repo_id": null,
      "asset_id": null,
      "norm_stats": null,
      "repack_transforms": {
        "inputs": [],
        "outputs": []
      },
      "data_transforms": {
        "inputs": [],
        "outputs": []
      },
      "model_transforms": {
        "inputs": [],
        "outputs": []
      },
      "use_quantile_norm": false,
      "action_sequence_keys": [
        "action"
      ],
      "prompt_from_task": true,
      "rlds_data_dir": null,
      "action_space": null,
      "filter_dict_path": null
    },
    "image_keys": [
      "observation.images.main",
      "observation.images.secondary_0"
    ]
  },
  "assets_base_dir": "./assets",
  "checkpoint_base_dir": "./checkpoints",
  "seed": 42,
  "batch_size": 32,
  "num_workers": 8,
  "num_train_steps": 500,
  "log_interval": 100,
  "save_interval": 100,
  "keep_period": 5000,
  "overwrite": false,
  "resume": false,
  "wandb_enabled": false,
  "policy_metadata": null,
  "fsdp_devices": 1
}