| { |
| "data_root_dir": null, |
| "dataset_statistics_map": null, |
| "global_subset_fraction": null, |
| "hf_token": null, |
| "image_aug": false, |
| "is_resume": true, |
| "non_action_datasets": null, |
| "normalize_data": true, |
| "pretrained_checkpoint": null, |
| "random_llm_weights": false, |
| "resume_epoch": null, |
| "resume_step": null, |
| "run_id": "base--mg_turn_on_sink_lite", |
| "run_id_note": "mg_turn_on_sink_lite", |
| "run_root_dir": null, |
| "save_interval": 5000, |
| "seed": 7, |
| "subset_percentages": null, |
| "trackers": [ |
| "jsonl", |
| "wandb" |
| ], |
| "vla": { |
| "action_tokenizer": "mg_turn_on_sink_lite_vq_extra_action_tokenizer", |
| "base_vlm": "ajaysri/prism-qwen25-extra-dinosiglip-224px-0_5b-stage-finetune-x7", |
| "data_mix": "mg_turn_on_sink_lite", |
| "ee_pose_2D_stride": 8, |
| "enable_gradient_checkpointing": true, |
| "enable_mixed_precision_training": true, |
| "epochs": 1000, |
| "expected_world_size": 8, |
| "freeze_llm_backbone": false, |
| "freeze_vision_backbone": false, |
| "future_2D_trace_window_size": 40, |
| "future_obj_pose_window_size": 40, |
| "global_batch_size": 256, |
| "image_sequence_len": 1, |
| "learning_rate": 2e-05, |
| "lr_scheduler_type": "constant", |
| "max_grad_norm": 1.0, |
| "max_steps": 15000, |
| "normalize_data": true, |
| "obj_pose_stride": 8, |
| "past_2D_trace_window_size": 0, |
| "past_obj_pose_window_size": 0, |
| "per_device_batch_size": 32, |
| "reduce_in_full_precision": true, |
| "save_every_n_steps": 25000, |
| "shuffle_buffer_size": 256000, |
| "train_strategy": "fsdp-full-shard", |
| "transform_types": "action", |
| "transform_weights": null, |
| "type": "prism-qwen25-dinosiglip-224px+0_5b+mx-xembod-robocasa-full", |
| "unfreeze_last_llm_layer": false, |
| "use_wrist_image": false, |
| "vla_id": "prism-qwen25-dinosiglip-224px+0_5b+mx-xembod-robocasa-full", |
| "warmup_ratio": 0.0, |
| "weight_decay": 0.0 |
| }, |
| "wandb_entity": "jensen_team", |
| "wandb_project": "prismatic-aux" |
| } |
|
|
