File size: 1,172 Bytes
73aa2b8 |
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 |
{
"data_root_dir": "/scr/user/data",
"hf_token": ".hf_token",
"pretrained_checkpoint": "",
"resume_epoch": null,
"resume_step": null,
"run_id": "prism-dinosiglip-224px+mx-oxe-magic-soup-plus+n8+b32+x7",
"run_id_note": null,
"run_root_dir": "./runs",
"save_interval": 2500,
"seed": 7,
"stage": "vla-full-train",
"trackers": [
"jsonl",
"wandb"
],
"vla": {
"base_vlm": "prism-dinosiglip-224px+7b",
"data_mix": "oxe_magic_soup_plus_minus",
"enable_gradient_checkpointing": true,
"enable_mixed_precision_training": true,
"epochs": 1000,
"expected_world_size": 64,
"freeze_vision_backbone": false,
"global_batch_size": 2048,
"learning_rate": 2e-05,
"lr_scheduler_type": "constant",
"max_grad_norm": 1.0,
"max_steps": null,
"per_device_batch_size": 32,
"reduce_in_full_precision": true,
"shuffle_buffer_size": 256000,
"train_strategy": "fsdp-full-shard",
"type": "prism-dinosiglip-224px+mx-oxe-magic-soup-plus",
"vla_id": "prism-dinosiglip-224px+mx-oxe-magic-soup-plus",
"warmup_ratio": 0.0,
"weight_decay": 0.0
},
"wandb_entity": "",
"wandb_project": ""
}
|