| { | |
| "job_name": "value-step-remove-ethernet", | |
| "dataset_repo_id": "reece-omahoney/rollout_remove-ethernet", | |
| "train_steps": 20000, | |
| "batch_size": 64, | |
| "num_workers": 4, | |
| "learning_rate": 2.5e-05, | |
| "val_split_ratio": 0.1, | |
| "seed": 42, | |
| "device": "cuda", | |
| "log_every_n_steps": 100, | |
| "plot_every_n_train_steps": 200, | |
| "save_every_n_steps": 1000, | |
| "max_val_steps": 20, | |
| "early_stopping_patience": 20, | |
| "early_stopping_min_delta": 0.001, | |
| "c_fail": 500.0, | |
| "reward": { | |
| "type": "steps", | |
| "cache": true | |
| }, | |
| "model": { | |
| "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": null, | |
| "private": null, | |
| "tags": null, | |
| "license": null, | |
| "pretrained_path": null, | |
| "text_backbone": "google/gemma-3-270m", | |
| "vision_tower": "google/siglip-so400m-patch14-224", | |
| "precision": "bfloat16", | |
| "image_size": 224, | |
| "hidden_dim": 640, | |
| "num_value_bins": 201, | |
| "v_min": -1.0, | |
| "v_max": 0.0, | |
| "value_head_depth": 1, | |
| "gradient_checkpointing": false, | |
| "compile_model": true, | |
| "normalization_mapping": { | |
| "VISUAL": "IDENTITY", | |
| "STATE": "IDENTITY", | |
| "ACTION": "IDENTITY" | |
| } | |
| }, | |
| "hub_user": "reece-omahoney", | |
| "push_to_hub": true, | |
| "wandb_project": "distal-value", | |
| "wandb_entity": null | |
| } |