IffYuan commited on
Commit
089490f
·
verified ·
1 Parent(s): d501729

Upload config.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.yaml +67 -0
config.yaml ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ datasets:
2
+ vla_data:
3
+ CoT_prompt: Your task is {instruction}. To identify the key objects for your task.
4
+ Locate their bounding boxes in [x1,y1,x2,y2] format.
5
+ data_mix: libero_all
6
+ data_root_dir: ./playground/Datasets/LEROBOT_LIBERO_DATA
7
+ dataset_py: lerobot_datasets
8
+ per_device_batch_size: 8
9
+ video_backend: torchvision_av
10
+ framework:
11
+ action_model:
12
+ action_dim: 7
13
+ action_horizon: 8
14
+ action_model_type: DiT-B
15
+ add_pos_embed: true
16
+ diffusion_model_cfg:
17
+ cross_attention_dim: 4096
18
+ dropout: 0.2
19
+ final_dropout: true
20
+ interleave_self_attention: true
21
+ norm_type: ada_norm
22
+ num_layers: 16
23
+ output_dim: 1024
24
+ positional_embeddings: null
25
+ future_action_window_size: 7
26
+ hidden_size: 1024
27
+ max_seq_len: 1024
28
+ noise_beta_alpha: 1.5
29
+ noise_beta_beta: 1.0
30
+ noise_s: 0.999
31
+ num_inference_timesteps: 4
32
+ num_target_vision_tokens: 32
33
+ num_timestep_buckets: 1000
34
+ past_action_window_size: 0
35
+ state_dim: 7
36
+ name: QwenGR00T
37
+ qwenvl:
38
+ base_vlm: /mnt/18T/starVLAproject/Qwen3-VL-8B-Instruct
39
+ output_dir: /starvla/Checkpoints/libero4in1_QwenGR00T_2node_0201_1721
40
+ run_id: libero4in1_QwenGR00T_2node_0201_1721
41
+ run_root_dir: /starvla/Checkpoints
42
+ seed: 42
43
+ trainer:
44
+ eval_interval: 100
45
+ freeze_modules: true
46
+ gradient_accumulation_steps: 4
47
+ gradient_clipping: 1.0
48
+ is_resume: false
49
+ learning_rate:
50
+ action_model: 0.0001
51
+ base: 2.5e-05
52
+ qwen_vl_interface: 1.0e-05
53
+ logging_frequency: 100
54
+ lr_scheduler_type: cosine_with_min_lr
55
+ max_train_steps: 80000
56
+ num_warmup_steps: 5000
57
+ optimizer:
58
+ betas:
59
+ - 0.9
60
+ - 0.95
61
+ eps: 1.0e-08
62
+ weight_decay: 1.0e-08
63
+ save_interval: 10000
64
+ scheduler_specific_kwargs:
65
+ min_lr: 1.0e-06
66
+ wandb_entity: xiguapi
67
+ wandb_project: Qwen3VL_libero_all_QwenGR00T_2node