File size: 1,959 Bytes
7f0618c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 | {
"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"
}
|