{
  "data_root_dir": "/mnt/pentagon/dataset/maniskill/data/scene_datasets/replica_cad_dataset/rearrange-dataset",
  "hf_token": ".hf_token",
  "image_aug": false,
  "is_resume": false,
  "model_type": "img_only_vla",
  "pretrained_checkpoint": "openvla/openvla-7b-prismatic",
  "resume_epoch": null,
  "resume_step": null,
  "run_id": "img_only_vla_last_layer",
  "run_id_note": null,
  "run_root_dir": "mshab_exp",
  "save_interval": 1000,
  "seed": 7,
  "trackers": [
    "jsonl",
    "wandb"
  ],
  "vla": {
    "base_vlm": "prism-dinosiglip-224px+7b",
    "data_mix": "bridge",
    "enable_gradient_checkpointing": true,
    "enable_mixed_precision_training": true,
    "epochs": 5,
    "expected_world_size": 8,
    "freeze_llm_backbone": true,
    "freeze_vision_backbone": true,
    "global_batch_size": 512,
    "learning_rate": 0.0005,
    "lr_scheduler_type": "constant",
    "max_grad_norm": 1.0,
    "max_steps": null,
    "per_device_batch_size": 64,
    "reduce_in_full_precision": true,
    "shuffle_buffer_size": 256000,
    "train_strategy": "fsdp-full-shard",
    "type": "prism-dinosiglip-224px+mx-bridge",
    "unfreeze_last_llm_layer": true,
    "vla_id": "prism-dinosiglip-224px+mx-bridge",
    "warmup_ratio": 0.0,
    "weight_decay": 0.0
  },
  "wandb_entity": "traysen879-uc-san-diego",
  "wandb_project": "mshab_vla"
}