| { |
| "type": "recap_value", |
| "n_obs_steps": 1, |
| "input_features": {}, |
| "output_features": {}, |
| "device": "cuda", |
| "use_amp": false, |
| "use_peft": false, |
| "push_to_hub": true, |
| "repo_id": "reece-omahoney/value-steps-gemma", |
| "private": null, |
| "tags": null, |
| "license": null, |
| "pretrained_path": null, |
| "paligemma_variant": "gemma_2b", |
| "precision": "bfloat16", |
| "image_size": 224, |
| "tokenizer_name": "google/paligemma-3b-pt-224", |
| "hidden_dim": 768, |
| "num_value_bins": 50, |
| "v_min": -1.0, |
| "v_max": 0.0, |
| "freeze_vision_encoder": true, |
| "freeze_backbone": true, |
| "num_unfrozen_backbone_layers": 3, |
| "num_vlm_layers": 10, |
| "value_head_depth": 1, |
| "dropout": 0.1, |
| "vlm_pretrained_path": "lerobot/pi05_base", |
| "normalization_mapping": { |
| "VISUAL": "IDENTITY", |
| "STATE": "IDENTITY", |
| "ACTION": "IDENTITY" |
| } |
| } |
