szang18 commited on
Commit
0297372
·
verified ·
1 Parent(s): f53950b

Upload prism-qwen25-dinosiglip-224px+0_5b+mx-libero-90-vq7-vggt-cross-freeze_v-new_cat-camera-head/config.json with huggingface_hub

Browse files
prism-qwen25-dinosiglip-224px+0_5b+mx-libero-90-vq7-vggt-cross-freeze_v-new_cat-camera-head/config.json ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_root_dir": [
3
+ "libero-90/1.0.0"
4
+ ],
5
+ "expected_world_size": 16,
6
+ "global_batch_size": 512,
7
+ "hf_token": ".hf_token",
8
+ "image_aug": false,
9
+ "is_resume": true,
10
+ "per_device_batch_size": 32,
11
+ "pretrained_checkpoint": null,
12
+ "resume_epoch": null,
13
+ "resume_step": null,
14
+ "run_id": "prism-qwen25-dinosiglip-224px+0_5b+mx-libero-90-vq7-vggt-cross-freeze_v-new_cat-camera-head",
15
+ "run_id_note": null,
16
+ "run_root_dir": "logs",
17
+ "save_interval": 2500,
18
+ "seed": 7,
19
+ "trackers": [
20
+ "jsonl",
21
+ "wandb"
22
+ ],
23
+ "vla": {
24
+ "action_tokenizer": "libero-90_vq_extra_action_tokenizer",
25
+ "base_vlm": "prism-qwen25-extra-dinosiglip-224px+0_5b",
26
+ "data_mix": "libero_90",
27
+ "enable_gradient_checkpointing": true,
28
+ "enable_mixed_precision_training": true,
29
+ "epochs": 50,
30
+ "expected_world_size": 16,
31
+ "freeze_llm_backbone": false,
32
+ "freeze_vision_backbone": true,
33
+ "global_batch_size": 512,
34
+ "image_sequence_len": 1,
35
+ "learning_rate": 2e-05,
36
+ "lr_scheduler_type": "constant",
37
+ "max_grad_norm": 1.0,
38
+ "max_steps": null,
39
+ "per_device_batch_size": 32,
40
+ "reduce_in_full_precision": true,
41
+ "save_every_n_steps": 25000,
42
+ "shuffle_buffer_size": 256000,
43
+ "train_strategy": "fsdp-full-shard",
44
+ "type": "prism-qwen25-dinosiglip-224px+0_5b+mx-libero-90",
45
+ "unfreeze_last_llm_layer": false,
46
+ "use_wrist_image": false,
47
+ "vla_id": "prism-qwen25-dinosiglip-224px+0_5b+mx-libero-90",
48
+ "warmup_ratio": 0.0,
49
+ "weight_decay": 0.0
50
+ },
51
+ "wandb_entity": "szang",
52
+ "wandb_project": "openvla"
53
+ }