Caesarrr commited on
Commit
aa41c35
·
verified ·
1 Parent(s): a99b2cc

Consolidate paper experiment assets by task and run (451-556)

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2/summary.jsonl +50 -0
  2. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2_small/checkpoints/best_model.safetensors +3 -0
  3. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2_small/checkpoints/steps_1500_model.safetensors +3 -0
  4. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2_small/config.full.yaml +187 -0
  5. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2_small/config.yaml +89 -0
  6. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2_small/dataset_statistics.json +127 -0
  7. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2_small/dataset_statistics_eval.json +127 -0
  8. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2_small/hydra/.hydra/config.yaml +185 -0
  9. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2_small/hydra/.hydra/hydra.yaml +246 -0
  10. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2_small/hydra/.hydra/overrides.yaml +85 -0
  11. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2_small/hydra/train_starvla_hydra.log +0 -0
  12. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2_small/summary.jsonl +15 -0
  13. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_3_small/checkpoints/best_model.safetensors +3 -0
  14. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_3_small/checkpoints/steps_1500_model.safetensors +3 -0
  15. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_3_small/config.full.yaml +187 -0
  16. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_3_small/config.yaml +89 -0
  17. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_3_small/dataset_statistics.json +127 -0
  18. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_3_small/dataset_statistics_eval.json +127 -0
  19. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_3_small/hydra/.hydra/config.yaml +185 -0
  20. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_3_small/hydra/.hydra/hydra.yaml +246 -0
  21. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_3_small/hydra/.hydra/overrides.yaml +85 -0
  22. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_3_small/hydra/train_starvla_hydra.log +0 -0
  23. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_3_small/summary.jsonl +15 -0
  24. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_exp/checkpoints/steps_2500_model.safetensors +3 -0
  25. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_exp/checkpoints/steps_2500_pytorch_model.pt +3 -0
  26. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_exp/config.full.yaml +179 -0
  27. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_exp/config.yaml +83 -0
  28. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_exp/dataset_statistics.json +127 -0
  29. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_exp/dataset_statistics_eval.json +127 -0
  30. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_exp/hydra/.hydra/config.yaml +177 -0
  31. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_exp/hydra/.hydra/hydra.yaml +242 -0
  32. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_exp/hydra/.hydra/overrides.yaml +82 -0
  33. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_exp/hydra/train_starvla_hydra.log +0 -0
  34. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_exp/summary.jsonl +25 -0
  35. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_small/checkpoints/best_model.safetensors +3 -0
  36. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_small/checkpoints/steps_1500_model.safetensors +3 -0
  37. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_small/config.full.yaml +187 -0
  38. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_small/config.yaml +89 -0
  39. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_small/dataset_statistics.json +127 -0
  40. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_small/dataset_statistics_eval.json +127 -0
  41. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_small/hydra/.hydra/config.yaml +185 -0
  42. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_small/hydra/.hydra/hydra.yaml +246 -0
  43. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_small/hydra/.hydra/overrides.yaml +85 -0
  44. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_small/hydra/train_starvla_hydra.log +0 -0
  45. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_small/summary.jsonl +15 -0
  46. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_6/checkpoints/steps_5000_model.safetensors +3 -0
  47. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_6/checkpoints/steps_5000_pytorch_model.pt +3 -0
  48. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_6/config.full.yaml +178 -0
  49. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_6/config.yaml +79 -0
  50. latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_6/dataset_statistics.json +127 -0
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2/summary.jsonl ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"steps": 100}
2
+ {"steps": 200}
3
+ {"steps": 300}
4
+ {"steps": 400}
5
+ {"steps": 500}
6
+ {"steps": 600}
7
+ {"steps": 700}
8
+ {"steps": 800}
9
+ {"steps": 900}
10
+ {"steps": 1000}
11
+ {"steps": 1100}
12
+ {"steps": 1200}
13
+ {"steps": 1300}
14
+ {"steps": 1400}
15
+ {"steps": 1500}
16
+ {"steps": 1600}
17
+ {"steps": 1700}
18
+ {"steps": 1800}
19
+ {"steps": 1900}
20
+ {"steps": 2000}
21
+ {"steps": 2100}
22
+ {"steps": 2200}
23
+ {"steps": 2300}
24
+ {"steps": 2400}
25
+ {"steps": 2500}
26
+ {"steps": 2600}
27
+ {"steps": 2700}
28
+ {"steps": 2800}
29
+ {"steps": 2900}
30
+ {"steps": 3000}
31
+ {"steps": 3100}
32
+ {"steps": 3200}
33
+ {"steps": 3300}
34
+ {"steps": 3400}
35
+ {"steps": 3500}
36
+ {"steps": 3600}
37
+ {"steps": 3700}
38
+ {"steps": 3800}
39
+ {"steps": 3900}
40
+ {"steps": 4000}
41
+ {"steps": 4100}
42
+ {"steps": 4200}
43
+ {"steps": 4300}
44
+ {"steps": 4400}
45
+ {"steps": 4500}
46
+ {"steps": 4600}
47
+ {"steps": 4700}
48
+ {"steps": 4800}
49
+ {"steps": 4900}
50
+ {"steps": 5000}
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2_small/checkpoints/best_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ee25ed102ecbf335b536f8260de7ae96b84cfa7158024c19e6dc51d32cd9ed27
3
+ size 9138230516
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2_small/checkpoints/steps_1500_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3aa56b5e881919571003b6155387d5ea5dfe50181e8f57b788e15e94dc643c73
3
+ size 9138230516
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2_small/config.full.yaml ADDED
@@ -0,0 +1,187 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ name: QwenOFT
3
+ qwenvl:
4
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
5
+ attn_implementation: flash_attention_2
6
+ enable_gradient_checkpointing: true
7
+ action_model:
8
+ action_model_type: MLP
9
+ action_dim: 7
10
+ action_hidden_dim: 2560
11
+ future_action_window_size: 0
12
+ past_action_window_size: 0
13
+ loss_type: discrete_ce
14
+ state_dim: 7
15
+ action_horizon: 1
16
+ action_env_dim: 2
17
+ datasets:
18
+ vla_data:
19
+ dataset_py: lerobot_datasets
20
+ include_state: true
21
+ data_root_dir: /workspace/starVLA/data/flappy_fix_latency_2_small
22
+ data_mix: flappy_train__bridge
23
+ eval_data_mix: flappy_train__bridge__val
24
+ custom_mixtures_path: null
25
+ action_type: discrete
26
+ sequential_step_sampling: false
27
+ per_device_batch_size: 32
28
+ load_all_data_for_training: true
29
+ obs_image_size: null
30
+ video_backend: torchvision_av
31
+ dataset:
32
+ source_hf: ''
33
+ converted_name: flappy_train
34
+ single_source_hf: ''
35
+ mixed_source_hf: ''
36
+ single_converted_name: flappy_train
37
+ mixed_converted_name: flappy_mixed_latency_train
38
+ single_latency_filter: null
39
+ mixed_latency_filter: null
40
+ force_download: false
41
+ setup_force: false
42
+ verify_rows: 200
43
+ max_episodes: null
44
+ episodes_per_latency: null
45
+ latency_filter: null
46
+ debug_subset:
47
+ enabled: false
48
+ max_episodes: 5
49
+ suffix: debug
50
+ base_model:
51
+ repo_id: Qwen/Qwen3-VL-4B-Instruct
52
+ initialization:
53
+ checkpoint_local_dir: /workspace/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
54
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
55
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
56
+ trainer:
57
+ max_train_steps: 1500
58
+ num_warmup_steps: 100
59
+ save_interval: 100
60
+ eval_interval: 100
61
+ eval_num_batches: 20
62
+ learning_rate:
63
+ base: 2.0e-05
64
+ qwen_vl_interface: 1.0e-05
65
+ action_model: 0.0001
66
+ lr_scheduler_type: cosine_with_min_lr
67
+ scheduler_specific_kwargs:
68
+ min_lr: 1.0e-06
69
+ freeze_modules: ''
70
+ loss_scale:
71
+ vla: 1.0
72
+ vlm: 0.1
73
+ max_grad_norm: 1.0
74
+ weight_decay: 0.0
75
+ logging_frequency: 10
76
+ gradient_clipping: 1.0
77
+ gradient_accumulation_steps: 16
78
+ distributed_backend: none
79
+ is_resume: false
80
+ pretrained_checkpoint: null
81
+ resume_step: 0
82
+ reload_modules: null
83
+ optimizer:
84
+ name: AdamW
85
+ betas:
86
+ - 0.9
87
+ - 0.95
88
+ eps: 1.0e-08
89
+ weight_decay: 1.0e-08
90
+ fused: false
91
+ save_format: pt
92
+ workspace_dir: WORKSPACE_DIR
93
+ run_root_dir: /workspace/starVLA/results/Checkpoints
94
+ seed: 42
95
+ wandb_entity: dongqianyu99-zhejiang-university
96
+ wandb_project: starVLA_rl_games
97
+ auth:
98
+ env_file: null
99
+ hf_token_env: HF_TOKEN
100
+ wandb_api_key_env: WANDB_API_KEY
101
+ paths:
102
+ run_root_dir: results/Checkpoints
103
+ dataset_local_dir: playground/Datasets/rl_games
104
+ dataset_cache_dir: null
105
+ base_model_dir: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
106
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
107
+ launch:
108
+ use_accelerate: true
109
+ gpus: null
110
+ num_processes: 1
111
+ dry_run: false
112
+ conda:
113
+ enabled: true
114
+ env_name: null
115
+ rl_games:
116
+ model_alias: openvla
117
+ env_eval:
118
+ image_size: 224
119
+ frameskip: 1
120
+ seed: 42
121
+ fixed_episode_seeds: true
122
+ latency_seed_stride: 0
123
+ task_seed_stride: 0
124
+ task_description: ''
125
+ enabled: true
126
+ distributed_mode: none
127
+ vectorized:
128
+ enabled: false
129
+ batch_size: 1
130
+ latency:
131
+ prompt_map_path: /workspace/starVLA/data/flappy_fix_latency_2_small/flappy_train__bridge/latency_prompt_map.json
132
+ mode: single
133
+ values:
134
+ - 0
135
+ mid_train:
136
+ enabled: true
137
+ interval_steps: 100
138
+ latencies:
139
+ - 2
140
+ num_episodes: 5
141
+ max_steps_per_episode: 3600
142
+ post_train:
143
+ enabled: true
144
+ latencies:
145
+ - 0
146
+ - 1
147
+ - 2
148
+ - 3
149
+ - 4
150
+ num_episodes: 5
151
+ max_steps_per_episode: 3600
152
+ task: flappy
153
+ initialization_mode: bridge
154
+ action_carrier: bridge
155
+ model: openvla
156
+ env: flappy
157
+ init: bridge
158
+ bridge_base_model:
159
+ repo_id:
160
+ openvla: Qwen/Qwen3-VL-4B-Instruct
161
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
162
+ pi05: Qwen/Qwen3-VL-4B-Instruct
163
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
164
+ local_dir:
165
+ openvla: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
166
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
167
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
168
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
169
+ mode: single
170
+ checkpoint:
171
+ load: auto
172
+ hf_repo_id: null
173
+ save_best_model: true
174
+ save_pt_file: false
175
+ local:
176
+ keep_last_n: 1
177
+ sync:
178
+ enabled: false
179
+ repo_id: null
180
+ keep_last_n: 0
181
+ sync_every_n_checkpoints: 1
182
+ resume_policy: local_latest
183
+ run_id: openvla_flappy_fix_latency_2_small
184
+ output_dir: /workspace/starVLA/results/Checkpoints/openvla_flappy_fix_latency_2_small
185
+ config_yaml: null
186
+ is_debug: false
187
+ version_id: '0.21'
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2_small/config.yaml ADDED
@@ -0,0 +1,89 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ checkpoint:
2
+ local:
3
+ keep_last_n: 1
4
+ save_best_model: true
5
+ save_pt_file: false
6
+ sync:
7
+ enabled: false
8
+ keep_last_n: 0
9
+ repo_id: null
10
+ datasets:
11
+ vla_data:
12
+ data_mix: flappy_train__bridge
13
+ dataset_py: lerobot_datasets
14
+ eval_data_mix: flappy_train__bridge__val
15
+ include_state: true
16
+ obs_image_size: null
17
+ per_device_batch_size: 32
18
+ framework:
19
+ action_model:
20
+ action_dim: 7
21
+ action_env_dim: 2
22
+ action_hidden_dim: 2560
23
+ action_horizon: 1
24
+ action_model_type: MLP
25
+ loss_type: discrete_ce
26
+ state_dim: 7
27
+ name: QwenOFT
28
+ qwenvl:
29
+ attn_implementation: flash_attention_2
30
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
31
+ enable_gradient_checkpointing: true
32
+ output_dir: /workspace/starVLA/results/Checkpoints/openvla_flappy_fix_latency_2_small
33
+ rl_games:
34
+ env_eval:
35
+ distributed_mode: none
36
+ enabled: true
37
+ fixed_episode_seeds: true
38
+ frameskip: 1
39
+ image_size: 224
40
+ latency:
41
+ prompt_map_path: /workspace/starVLA/data/flappy_fix_latency_2_small/flappy_train__bridge/latency_prompt_map.json
42
+ latency_seed_stride: 0
43
+ mid_train:
44
+ enabled: true
45
+ interval_steps: 100
46
+ latencies:
47
+ - 2
48
+ max_steps_per_episode: 3600
49
+ num_episodes: 5
50
+ seed: 42
51
+ task_description: ''
52
+ task_seed_stride: 0
53
+ vectorized:
54
+ enabled: false
55
+ model_alias: openvla
56
+ task: flappy
57
+ run_id: openvla_flappy_fix_latency_2_small
58
+ run_root_dir: /workspace/starVLA/results/Checkpoints
59
+ seed: 42
60
+ trainer:
61
+ distributed_backend: none
62
+ eval_interval: 100
63
+ eval_num_batches: 20
64
+ freeze_modules: ''
65
+ gradient_accumulation_steps: 16
66
+ gradient_clipping: 1.0
67
+ is_resume: false
68
+ learning_rate:
69
+ action_model: 0.0001
70
+ base: 2.0e-05
71
+ qwen_vl_interface: 1.0e-05
72
+ logging_frequency: 10
73
+ lr_scheduler_type: cosine_with_min_lr
74
+ max_train_steps: 1500
75
+ num_warmup_steps: 100
76
+ optimizer:
77
+ betas:
78
+ - 0.9
79
+ - 0.95
80
+ eps: 1.0e-08
81
+ fused: false
82
+ weight_decay: 1.0e-08
83
+ pretrained_checkpoint: null
84
+ reload_modules: null
85
+ save_interval: 100
86
+ scheduler_specific_kwargs:
87
+ min_lr: 1.0e-06
88
+ wandb_entity: dongqianyu99-zhejiang-university
89
+ wandb_project: starVLA_rl_games
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2_small/dataset_statistics.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.7955247163772583,
6
+ 0.2044753134250641,
7
+ 0.0,
8
+ 0.0,
9
+ 0.0,
10
+ 0.0,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.4034355580806732,
15
+ 0.4034355580806732,
16
+ 0.0,
17
+ 0.0,
18
+ 0.0,
19
+ 0.0,
20
+ 0.0
21
+ ],
22
+ "max": [
23
+ 1.0,
24
+ 1.0,
25
+ 0.0,
26
+ 0.0,
27
+ 0.0,
28
+ 0.0,
29
+ 0.0
30
+ ],
31
+ "min": [
32
+ 0.0,
33
+ 0.0,
34
+ 0.0,
35
+ 0.0,
36
+ 0.0,
37
+ 0.0,
38
+ 0.0
39
+ ],
40
+ "q01": [
41
+ 0.0,
42
+ 0.0,
43
+ 0.0,
44
+ 0.0,
45
+ 0.0,
46
+ 0.0,
47
+ 0.0
48
+ ],
49
+ "q99": [
50
+ 1.0,
51
+ 1.0,
52
+ 0.0,
53
+ 0.0,
54
+ 0.0,
55
+ 0.0,
56
+ 0.0
57
+ ],
58
+ "mask": [
59
+ true,
60
+ true,
61
+ true,
62
+ true,
63
+ true,
64
+ true,
65
+ true
66
+ ]
67
+ },
68
+ "state": {
69
+ "mean": [
70
+ 0.0,
71
+ 0.0,
72
+ 0.0,
73
+ 0.0,
74
+ 0.0,
75
+ 0.0,
76
+ 0.0
77
+ ],
78
+ "std": [
79
+ 0.0,
80
+ 0.0,
81
+ 0.0,
82
+ 0.0,
83
+ 0.0,
84
+ 0.0,
85
+ 0.0
86
+ ],
87
+ "max": [
88
+ 0.0,
89
+ 0.0,
90
+ 0.0,
91
+ 0.0,
92
+ 0.0,
93
+ 0.0,
94
+ 0.0
95
+ ],
96
+ "min": [
97
+ 0.0,
98
+ 0.0,
99
+ 0.0,
100
+ 0.0,
101
+ 0.0,
102
+ 0.0,
103
+ 0.0
104
+ ],
105
+ "q01": [
106
+ 0.0,
107
+ 0.0,
108
+ 0.0,
109
+ 0.0,
110
+ 0.0,
111
+ 0.0,
112
+ 0.0
113
+ ],
114
+ "q99": [
115
+ 0.0,
116
+ 0.0,
117
+ 0.0,
118
+ 0.0,
119
+ 0.0,
120
+ 0.0,
121
+ 0.0
122
+ ]
123
+ },
124
+ "num_transitions": 97200,
125
+ "num_trajectories": 27
126
+ }
127
+ }
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2_small/dataset_statistics_eval.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.7949073910713196,
6
+ 0.20509259402751923,
7
+ 0.0,
8
+ 0.0,
9
+ 0.0,
10
+ 0.0,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.40378323197364807,
15
+ 0.40378323197364807,
16
+ 0.0,
17
+ 0.0,
18
+ 0.0,
19
+ 0.0,
20
+ 0.0
21
+ ],
22
+ "max": [
23
+ 1.0,
24
+ 1.0,
25
+ 0.0,
26
+ 0.0,
27
+ 0.0,
28
+ 0.0,
29
+ 0.0
30
+ ],
31
+ "min": [
32
+ 0.0,
33
+ 0.0,
34
+ 0.0,
35
+ 0.0,
36
+ 0.0,
37
+ 0.0,
38
+ 0.0
39
+ ],
40
+ "q01": [
41
+ 0.0,
42
+ 0.0,
43
+ 0.0,
44
+ 0.0,
45
+ 0.0,
46
+ 0.0,
47
+ 0.0
48
+ ],
49
+ "q99": [
50
+ 1.0,
51
+ 1.0,
52
+ 0.0,
53
+ 0.0,
54
+ 0.0,
55
+ 0.0,
56
+ 0.0
57
+ ],
58
+ "mask": [
59
+ true,
60
+ true,
61
+ true,
62
+ true,
63
+ true,
64
+ true,
65
+ true
66
+ ]
67
+ },
68
+ "state": {
69
+ "mean": [
70
+ 0.0,
71
+ 0.0,
72
+ 0.0,
73
+ 0.0,
74
+ 0.0,
75
+ 0.0,
76
+ 0.0
77
+ ],
78
+ "std": [
79
+ 0.0,
80
+ 0.0,
81
+ 0.0,
82
+ 0.0,
83
+ 0.0,
84
+ 0.0,
85
+ 0.0
86
+ ],
87
+ "max": [
88
+ 0.0,
89
+ 0.0,
90
+ 0.0,
91
+ 0.0,
92
+ 0.0,
93
+ 0.0,
94
+ 0.0
95
+ ],
96
+ "min": [
97
+ 0.0,
98
+ 0.0,
99
+ 0.0,
100
+ 0.0,
101
+ 0.0,
102
+ 0.0,
103
+ 0.0
104
+ ],
105
+ "q01": [
106
+ 0.0,
107
+ 0.0,
108
+ 0.0,
109
+ 0.0,
110
+ 0.0,
111
+ 0.0,
112
+ 0.0
113
+ ],
114
+ "q99": [
115
+ 0.0,
116
+ 0.0,
117
+ 0.0,
118
+ 0.0,
119
+ 0.0,
120
+ 0.0,
121
+ 0.0
122
+ ]
123
+ },
124
+ "num_transitions": 10800,
125
+ "num_trajectories": 3
126
+ }
127
+ }
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2_small/hydra/.hydra/config.yaml ADDED
@@ -0,0 +1,185 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ qwenvl:
3
+ base_vlm: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct
4
+ attn_implementation: flash_attention_2
5
+ enable_gradient_checkpointing: true
6
+ action_model:
7
+ state_dim: 7
8
+ loss_type: discrete_ce
9
+ action_horizon: 1
10
+ future_action_window_size: 0
11
+ past_action_window_size: 0
12
+ action_dim: 7
13
+ action_env_dim: 2
14
+ name: QwenOFT
15
+ datasets:
16
+ vla_data:
17
+ dataset_py: lerobot_datasets
18
+ include_state: true
19
+ data_root_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_2_small
20
+ data_mix: flappy_train__bridge
21
+ eval_data_mix: flappy_train__bridge__val
22
+ custom_mixtures_path: null
23
+ action_type: discrete
24
+ sequential_step_sampling: false
25
+ per_device_batch_size: 32
26
+ load_all_data_for_training: true
27
+ obs_image_size: null
28
+ video_backend: torchvision_av
29
+ dataset:
30
+ source_hf: ${dataset.single_source_hf}
31
+ converted_name: ${dataset.single_converted_name}
32
+ single_source_hf: ''
33
+ mixed_source_hf: ''
34
+ single_converted_name: flappy_train
35
+ mixed_converted_name: flappy_mixed_latency_train
36
+ single_latency_filter: null
37
+ mixed_latency_filter: null
38
+ force_download: false
39
+ setup_force: false
40
+ verify_rows: 200
41
+ max_episodes: null
42
+ episodes_per_latency: null
43
+ latency_filter: ${dataset.single_latency_filter}
44
+ debug_subset:
45
+ enabled: false
46
+ max_episodes: 5
47
+ suffix: debug
48
+ base_model:
49
+ repo_id: ${bridge_base_model.repo_id.${model}}
50
+ initialization:
51
+ checkpoint_local_dir: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
52
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
53
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
54
+ trainer:
55
+ max_train_steps: 1500
56
+ num_warmup_steps: 100
57
+ save_interval: 100
58
+ eval_interval: 100
59
+ eval_num_batches: 20
60
+ learning_rate:
61
+ base: 2.0e-05
62
+ qwen_vl_interface: 1.0e-05
63
+ action_model: 0.0001
64
+ lr_scheduler_type: cosine_with_min_lr
65
+ scheduler_specific_kwargs:
66
+ min_lr: 1.0e-06
67
+ freeze_modules: ''
68
+ loss_scale:
69
+ vla: 1.0
70
+ vlm: 0.1
71
+ max_grad_norm: 1.0
72
+ weight_decay: 0.0
73
+ logging_frequency: 10
74
+ gradient_clipping: 1.0
75
+ gradient_accumulation_steps: 16
76
+ distributed_backend: none
77
+ is_resume: false
78
+ pretrained_checkpoint: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
79
+ resume_step: 0
80
+ reload_modules: null
81
+ optimizer:
82
+ name: AdamW
83
+ betas:
84
+ - 0.9
85
+ - 0.95
86
+ eps: 1.0e-08
87
+ weight_decay: 1.0e-08
88
+ fused: false
89
+ save_format: pt
90
+ workspace_dir: WORKSPACE_DIR
91
+ run_root_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints
92
+ seed: 42
93
+ wandb_entity: dongqianyu99-zhejiang-university
94
+ wandb_project: starVLA_rl_games
95
+ auth:
96
+ env_file: null
97
+ hf_token_env: HF_TOKEN
98
+ wandb_api_key_env: WANDB_API_KEY
99
+ paths:
100
+ run_root_dir: results/Checkpoints
101
+ dataset_local_dir: playground/Datasets/rl_games
102
+ dataset_cache_dir: null
103
+ base_model_dir: ${bridge_base_model.local_dir.${model}}
104
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
105
+ launch:
106
+ use_accelerate: true
107
+ gpus: null
108
+ num_processes: 1
109
+ dry_run: false
110
+ conda:
111
+ enabled: true
112
+ env_name: null
113
+ rl_games:
114
+ model_alias: openvla
115
+ env_eval:
116
+ image_size: 224
117
+ frameskip: 1
118
+ seed: 42
119
+ fixed_episode_seeds: true
120
+ latency_seed_stride: 0
121
+ task_seed_stride: 0
122
+ task_description: ''
123
+ enabled: true
124
+ distributed_mode: none
125
+ vectorized:
126
+ enabled: false
127
+ batch_size: 1
128
+ latency:
129
+ prompt_map_path: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_2_small/flappy_train__bridge/latency_prompt_map.json
130
+ mode: single
131
+ values:
132
+ - 0
133
+ mid_train:
134
+ enabled: true
135
+ interval_steps: 100
136
+ latencies:
137
+ - 2
138
+ num_episodes: 5
139
+ max_steps_per_episode: 3600
140
+ post_train:
141
+ enabled: true
142
+ latencies:
143
+ - 0
144
+ - 1
145
+ - 2
146
+ - 3
147
+ - 4
148
+ num_episodes: 5
149
+ max_steps_per_episode: 3600
150
+ task: flappy
151
+ initialization_mode: bridge
152
+ action_carrier: bridge
153
+ model: openvla
154
+ env: flappy
155
+ init: bridge
156
+ bridge_base_model:
157
+ repo_id:
158
+ openvla: Qwen/Qwen3-VL-4B-Instruct
159
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
160
+ pi05: Qwen/Qwen3-VL-4B-Instruct
161
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
162
+ local_dir:
163
+ openvla: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
164
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
165
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
166
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
167
+ mode: single
168
+ checkpoint:
169
+ load: auto
170
+ hf_repo_id: null
171
+ save_best_model: true
172
+ save_pt_file: false
173
+ local:
174
+ keep_last_n: 1
175
+ sync:
176
+ enabled: false
177
+ repo_id: null
178
+ keep_last_n: 0
179
+ sync_every_n_checkpoints: 1
180
+ resume_policy: local_latest
181
+ run_id: openvla_flappy_fix_latency_2_small
182
+ output_dir: null
183
+ config_yaml: null
184
+ is_debug: false
185
+ version_id: '0.21'
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2_small/hydra/.hydra/hydra.yaml ADDED
@@ -0,0 +1,246 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ hydra:
2
+ run:
3
+ dir: ${run_root_dir}/${run_id}/hydra
4
+ sweep:
5
+ dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
6
+ subdir: ${hydra.job.num}
7
+ launcher:
8
+ _target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
9
+ sweeper:
10
+ _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
11
+ max_batch_size: null
12
+ params: null
13
+ help:
14
+ app_name: ${hydra.job.name}
15
+ header: '${hydra.help.app_name} is powered by Hydra.
16
+
17
+ '
18
+ footer: 'Powered by Hydra (https://hydra.cc)
19
+
20
+ Use --hydra-help to view Hydra specific help
21
+
22
+ '
23
+ template: '${hydra.help.header}
24
+
25
+ == Configuration groups ==
26
+
27
+ Compose your configuration from those groups (group=option)
28
+
29
+
30
+ $APP_CONFIG_GROUPS
31
+
32
+
33
+ == Config ==
34
+
35
+ Override anything in the config (foo.bar=value)
36
+
37
+
38
+ $CONFIG
39
+
40
+
41
+ ${hydra.help.footer}
42
+
43
+ '
44
+ hydra_help:
45
+ template: 'Hydra (${hydra.runtime.version})
46
+
47
+ See https://hydra.cc for more info.
48
+
49
+
50
+ == Flags ==
51
+
52
+ $FLAGS_HELP
53
+
54
+
55
+ == Configuration groups ==
56
+
57
+ Compose your configuration from those groups (For example, append hydra/job_logging=disabled
58
+ to command line)
59
+
60
+
61
+ $HYDRA_CONFIG_GROUPS
62
+
63
+
64
+ Use ''--cfg hydra'' to Show the Hydra config.
65
+
66
+ '
67
+ hydra_help: ???
68
+ hydra_logging:
69
+ version: 1
70
+ formatters:
71
+ simple:
72
+ format: '[%(asctime)s][HYDRA] %(message)s'
73
+ handlers:
74
+ console:
75
+ class: logging.StreamHandler
76
+ formatter: simple
77
+ stream: ext://sys.stdout
78
+ root:
79
+ level: INFO
80
+ handlers:
81
+ - console
82
+ loggers:
83
+ logging_example:
84
+ level: DEBUG
85
+ disable_existing_loggers: false
86
+ job_logging:
87
+ version: 1
88
+ formatters:
89
+ simple:
90
+ format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
91
+ handlers:
92
+ console:
93
+ class: logging.StreamHandler
94
+ formatter: simple
95
+ stream: ext://sys.stdout
96
+ file:
97
+ class: logging.FileHandler
98
+ formatter: simple
99
+ filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
100
+ root:
101
+ level: INFO
102
+ handlers:
103
+ - console
104
+ - file
105
+ disable_existing_loggers: false
106
+ env: {}
107
+ mode: RUN
108
+ searchpath: []
109
+ callbacks: {}
110
+ output_subdir: .hydra
111
+ overrides:
112
+ hydra:
113
+ - hydra.mode=RUN
114
+ task:
115
+ - model=openvla
116
+ - env=flappy
117
+ - init=bridge
118
+ - mode=single
119
+ - run_id=openvla_flappy_fix_latency_2_small
120
+ - run_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints
121
+ - seed=42
122
+ - wandb_entity=dongqianyu99-zhejiang-university
123
+ - wandb_project=starVLA_rl_games
124
+ - rl_games.env_eval.enabled=true
125
+ - checkpoint.sync.enabled=false
126
+ - checkpoint.sync.keep_last_n=0
127
+ - checkpoint.local.keep_last_n=1
128
+ - checkpoint.save_best_model=true
129
+ - checkpoint.save_pt_file=false
130
+ - trainer.is_resume=false
131
+ - trainer.pretrained_checkpoint=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
132
+ - trainer.resume_step=0
133
+ - trainer.max_train_steps=1500
134
+ - trainer.num_warmup_steps=100
135
+ - trainer.save_interval=100
136
+ - trainer.eval_interval=100
137
+ - trainer.eval_num_batches=20
138
+ - trainer.logging_frequency=10
139
+ - trainer.gradient_accumulation_steps=16
140
+ - trainer.distributed_backend=none
141
+ - trainer.learning_rate.base=2e-05
142
+ - trainer.learning_rate.qwen_vl_interface=1e-05
143
+ - trainer.learning_rate.action_model=0.0001
144
+ - trainer.lr_scheduler_type=cosine_with_min_lr
145
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
146
+ - trainer.loss_scale.vla=1.0
147
+ - trainer.loss_scale.vlm=0.1
148
+ - trainer.max_grad_norm=1.0
149
+ - trainer.weight_decay=0.0
150
+ - trainer.gradient_clipping=1.0
151
+ - trainer.optimizer.name=AdamW
152
+ - trainer.optimizer.betas=[0.9, 0.95]
153
+ - trainer.optimizer.eps=1e-08
154
+ - trainer.optimizer.weight_decay=1e-08
155
+ - trainer.optimizer.fused=false
156
+ - trainer.save_format=pt
157
+ - framework.name=QwenOFT
158
+ - framework.qwenvl.attn_implementation=flash_attention_2
159
+ - framework.qwenvl.enable_gradient_checkpointing=true
160
+ - framework.action_model.action_dim=7
161
+ - framework.action_model.action_env_dim=2
162
+ - framework.action_model.state_dim=7
163
+ - framework.action_model.loss_type=discrete_ce
164
+ - framework.action_model.action_horizon=1
165
+ - framework.action_model.future_action_window_size=0
166
+ - framework.action_model.past_action_window_size=0
167
+ - datasets.vla_data.include_state=true
168
+ - datasets.vla_data.action_type=discrete
169
+ - datasets.vla_data.sequential_step_sampling=false
170
+ - datasets.vla_data.per_device_batch_size=32
171
+ - datasets.vla_data.load_all_data_for_training=true
172
+ - datasets.vla_data.video_backend=torchvision_av
173
+ - datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_2_small
174
+ - datasets.vla_data.data_mix=flappy_train__bridge
175
+ - datasets.vla_data.eval_data_mix=flappy_train__bridge__val
176
+ - framework.qwenvl.base_vlm=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct
177
+ - rl_games.task=flappy
178
+ - rl_games.model_alias=openvla
179
+ - rl_games.initialization_mode=bridge
180
+ - rl_games.action_carrier=bridge
181
+ - rl_games.env_eval.distributed_mode=none
182
+ - rl_games.env_eval.latency.mode=single
183
+ - rl_games.env_eval.frameskip=1
184
+ - rl_games.env_eval.image_size=224
185
+ - rl_games.env_eval.seed=42
186
+ - rl_games.env_eval.fixed_episode_seeds=true
187
+ - rl_games.env_eval.latency_seed_stride=0
188
+ - rl_games.env_eval.task_seed_stride=0
189
+ - rl_games.env_eval.latency.values=[0]
190
+ - rl_games.env_eval.latency.prompt_map_path=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_2_small/flappy_train__bridge/latency_prompt_map.json
191
+ - rl_games.env_eval.mid_train.enabled=true
192
+ - rl_games.env_eval.mid_train.interval_steps=100
193
+ - rl_games.env_eval.mid_train.num_episodes=5
194
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
195
+ - rl_games.env_eval.mid_train.latencies=[2]
196
+ - rl_games.env_eval.post_train.enabled=true
197
+ - rl_games.env_eval.post_train.num_episodes=5
198
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
199
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
200
+ job:
201
+ name: train_starvla_hydra
202
+ chdir: false
203
+ override_dirname: checkpoint.local.keep_last_n=1,checkpoint.save_best_model=true,checkpoint.save_pt_file=false,checkpoint.sync.enabled=false,checkpoint.sync.keep_last_n=0,datasets.vla_data.action_type=discrete,datasets.vla_data.data_mix=flappy_train__bridge,datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_2_small,datasets.vla_data.eval_data_mix=flappy_train__bridge__val,datasets.vla_data.include_state=true,datasets.vla_data.load_all_data_for_training=true,datasets.vla_data.per_device_batch_size=32,datasets.vla_data.sequential_step_sampling=false,datasets.vla_data.video_backend=torchvision_av,env=flappy,framework.action_model.action_dim=7,framework.action_model.action_env_dim=2,framework.action_model.action_horizon=1,framework.action_model.future_action_window_size=0,framework.action_model.loss_type=discrete_ce,framework.action_model.past_action_window_size=0,framework.action_model.state_dim=7,framework.name=QwenOFT,framework.qwenvl.attn_implementation=flash_attention_2,framework.qwenvl.base_vlm=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct,framework.qwenvl.enable_gradient_checkpointing=true,init=bridge,mode=single,model=openvla,rl_games.action_carrier=bridge,rl_games.env_eval.distributed_mode=none,rl_games.env_eval.enabled=true,rl_games.env_eval.fixed_episode_seeds=true,rl_games.env_eval.frameskip=1,rl_games.env_eval.image_size=224,rl_games.env_eval.latency.mode=single,rl_games.env_eval.latency.prompt_map_path=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_2_small/flappy_train__bridge/latency_prompt_map.json,rl_games.env_eval.latency.values=[0],rl_games.env_eval.latency_seed_stride=0,rl_games.env_eval.mid_train.enabled=true,rl_games.env_eval.mid_train.interval_steps=100,rl_games.env_eval.mid_train.latencies=[2],rl_games.env_eval.mid_train.max_steps_per_episode=3600,rl_games.env_eval.mid_train.num_episodes=5,rl_games.env_eval.post_train.enabled=true,rl_games.env_eval.post_train.latencies=[0,1,2,3,4],rl_games.env_eval.post_train.max_steps_per_episode=3600,rl_games.env_eval.post_train.num_episodes=5,rl_games.env_eval.seed=42,rl_games.env_eval.task_seed_stride=0,rl_games.initialization_mode=bridge,rl_games.model_alias=openvla,rl_games.task=flappy,run_id=openvla_flappy_fix_latency_2_small,run_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints,seed=42,trainer.distributed_backend=none,trainer.eval_interval=100,trainer.eval_num_batches=20,trainer.gradient_accumulation_steps=16,trainer.gradient_clipping=1.0,trainer.is_resume=false,trainer.learning_rate.action_model=0.0001,trainer.learning_rate.base=2e-05,trainer.learning_rate.qwen_vl_interface=1e-05,trainer.logging_frequency=10,trainer.loss_scale.vla=1.0,trainer.loss_scale.vlm=0.1,trainer.lr_scheduler_type=cosine_with_min_lr,trainer.max_grad_norm=1.0,trainer.max_train_steps=1500,trainer.num_warmup_steps=100,trainer.optimizer.betas=[0.9,
204
+ 0.95],trainer.optimizer.eps=1e-08,trainer.optimizer.fused=false,trainer.optimizer.name=AdamW,trainer.optimizer.weight_decay=1e-08,trainer.pretrained_checkpoint=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt,trainer.resume_step=0,trainer.save_format=pt,trainer.save_interval=100,trainer.scheduler_specific_kwargs.min_lr=1e-06,trainer.weight_decay=0.0,wandb_entity=dongqianyu99-zhejiang-university,wandb_project=starVLA_rl_games
205
+ id: ???
206
+ num: ???
207
+ config_name: train
208
+ env_set: {}
209
+ env_copy: []
210
+ config:
211
+ override_dirname:
212
+ kv_sep: '='
213
+ item_sep: ','
214
+ exclude_keys: []
215
+ runtime:
216
+ version: 1.3.2
217
+ version_base: '1.1'
218
+ cwd: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA
219
+ config_sources:
220
+ - path: hydra.conf
221
+ schema: pkg
222
+ provider: hydra
223
+ - path: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA/examples/rl_games/config
224
+ schema: file
225
+ provider: main
226
+ - path: ''
227
+ schema: structured
228
+ provider: schema
229
+ output_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints/openvla_flappy_fix_latency_2_small/hydra
230
+ choices:
231
+ cross_task_setup: null
232
+ checkpoint: default
233
+ mode: single
234
+ init: bridge
235
+ env: flappy
236
+ model: openvla
237
+ hydra/env: default
238
+ hydra/callbacks: null
239
+ hydra/job_logging: default
240
+ hydra/hydra_logging: default
241
+ hydra/hydra_help: default
242
+ hydra/help: default
243
+ hydra/sweeper: basic
244
+ hydra/launcher: basic
245
+ hydra/output: default
246
+ verbose: false
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2_small/hydra/.hydra/overrides.yaml ADDED
@@ -0,0 +1,85 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ - model=openvla
2
+ - env=flappy
3
+ - init=bridge
4
+ - mode=single
5
+ - run_id=openvla_flappy_fix_latency_2_small
6
+ - run_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints
7
+ - seed=42
8
+ - wandb_entity=dongqianyu99-zhejiang-university
9
+ - wandb_project=starVLA_rl_games
10
+ - rl_games.env_eval.enabled=true
11
+ - checkpoint.sync.enabled=false
12
+ - checkpoint.sync.keep_last_n=0
13
+ - checkpoint.local.keep_last_n=1
14
+ - checkpoint.save_best_model=true
15
+ - checkpoint.save_pt_file=false
16
+ - trainer.is_resume=false
17
+ - trainer.pretrained_checkpoint=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
18
+ - trainer.resume_step=0
19
+ - trainer.max_train_steps=1500
20
+ - trainer.num_warmup_steps=100
21
+ - trainer.save_interval=100
22
+ - trainer.eval_interval=100
23
+ - trainer.eval_num_batches=20
24
+ - trainer.logging_frequency=10
25
+ - trainer.gradient_accumulation_steps=16
26
+ - trainer.distributed_backend=none
27
+ - trainer.learning_rate.base=2e-05
28
+ - trainer.learning_rate.qwen_vl_interface=1e-05
29
+ - trainer.learning_rate.action_model=0.0001
30
+ - trainer.lr_scheduler_type=cosine_with_min_lr
31
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
32
+ - trainer.loss_scale.vla=1.0
33
+ - trainer.loss_scale.vlm=0.1
34
+ - trainer.max_grad_norm=1.0
35
+ - trainer.weight_decay=0.0
36
+ - trainer.gradient_clipping=1.0
37
+ - trainer.optimizer.name=AdamW
38
+ - trainer.optimizer.betas=[0.9, 0.95]
39
+ - trainer.optimizer.eps=1e-08
40
+ - trainer.optimizer.weight_decay=1e-08
41
+ - trainer.optimizer.fused=false
42
+ - trainer.save_format=pt
43
+ - framework.name=QwenOFT
44
+ - framework.qwenvl.attn_implementation=flash_attention_2
45
+ - framework.qwenvl.enable_gradient_checkpointing=true
46
+ - framework.action_model.action_dim=7
47
+ - framework.action_model.action_env_dim=2
48
+ - framework.action_model.state_dim=7
49
+ - framework.action_model.loss_type=discrete_ce
50
+ - framework.action_model.action_horizon=1
51
+ - framework.action_model.future_action_window_size=0
52
+ - framework.action_model.past_action_window_size=0
53
+ - datasets.vla_data.include_state=true
54
+ - datasets.vla_data.action_type=discrete
55
+ - datasets.vla_data.sequential_step_sampling=false
56
+ - datasets.vla_data.per_device_batch_size=32
57
+ - datasets.vla_data.load_all_data_for_training=true
58
+ - datasets.vla_data.video_backend=torchvision_av
59
+ - datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_2_small
60
+ - datasets.vla_data.data_mix=flappy_train__bridge
61
+ - datasets.vla_data.eval_data_mix=flappy_train__bridge__val
62
+ - framework.qwenvl.base_vlm=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct
63
+ - rl_games.task=flappy
64
+ - rl_games.model_alias=openvla
65
+ - rl_games.initialization_mode=bridge
66
+ - rl_games.action_carrier=bridge
67
+ - rl_games.env_eval.distributed_mode=none
68
+ - rl_games.env_eval.latency.mode=single
69
+ - rl_games.env_eval.frameskip=1
70
+ - rl_games.env_eval.image_size=224
71
+ - rl_games.env_eval.seed=42
72
+ - rl_games.env_eval.fixed_episode_seeds=true
73
+ - rl_games.env_eval.latency_seed_stride=0
74
+ - rl_games.env_eval.task_seed_stride=0
75
+ - rl_games.env_eval.latency.values=[0]
76
+ - rl_games.env_eval.latency.prompt_map_path=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_2_small/flappy_train__bridge/latency_prompt_map.json
77
+ - rl_games.env_eval.mid_train.enabled=true
78
+ - rl_games.env_eval.mid_train.interval_steps=100
79
+ - rl_games.env_eval.mid_train.num_episodes=5
80
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
81
+ - rl_games.env_eval.mid_train.latencies=[2]
82
+ - rl_games.env_eval.post_train.enabled=true
83
+ - rl_games.env_eval.post_train.num_episodes=5
84
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
85
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2_small/hydra/train_starvla_hydra.log ADDED
The diff for this file is too large to render. See raw diff
 
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_2_small/summary.jsonl ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"steps": 100}
2
+ {"steps": 200}
3
+ {"steps": 300}
4
+ {"steps": 400}
5
+ {"steps": 500}
6
+ {"steps": 600}
7
+ {"steps": 700}
8
+ {"steps": 800}
9
+ {"steps": 900}
10
+ {"steps": 1000}
11
+ {"steps": 1100}
12
+ {"steps": 1200}
13
+ {"steps": 1300}
14
+ {"steps": 1400}
15
+ {"steps": 1500}
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_3_small/checkpoints/best_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dfd6f77b285dff5a628b8a2e8aee12d2c8f11c2196464ed2505f7e138ec2a828
3
+ size 9138230516
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_3_small/checkpoints/steps_1500_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8b61cc767d8a885e8bf0e3fecc383c8e3233beb3f3ad2544e6f7c8c4265ae614
3
+ size 9138230516
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_3_small/config.full.yaml ADDED
@@ -0,0 +1,187 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ name: QwenOFT
3
+ qwenvl:
4
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
5
+ attn_implementation: flash_attention_2
6
+ enable_gradient_checkpointing: true
7
+ action_model:
8
+ action_model_type: MLP
9
+ action_dim: 7
10
+ action_hidden_dim: 2560
11
+ future_action_window_size: 0
12
+ past_action_window_size: 0
13
+ loss_type: discrete_ce
14
+ state_dim: 7
15
+ action_horizon: 1
16
+ action_env_dim: 2
17
+ datasets:
18
+ vla_data:
19
+ dataset_py: lerobot_datasets
20
+ include_state: true
21
+ data_root_dir: /workspace/starVLA/data/flappy_fix_latency_3_small
22
+ data_mix: flappy_train__bridge
23
+ eval_data_mix: flappy_train__bridge__val
24
+ custom_mixtures_path: null
25
+ action_type: discrete
26
+ sequential_step_sampling: false
27
+ per_device_batch_size: 32
28
+ load_all_data_for_training: true
29
+ obs_image_size: null
30
+ video_backend: torchvision_av
31
+ dataset:
32
+ source_hf: ''
33
+ converted_name: flappy_train
34
+ single_source_hf: ''
35
+ mixed_source_hf: ''
36
+ single_converted_name: flappy_train
37
+ mixed_converted_name: flappy_mixed_latency_train
38
+ single_latency_filter: null
39
+ mixed_latency_filter: null
40
+ force_download: false
41
+ setup_force: false
42
+ verify_rows: 200
43
+ max_episodes: null
44
+ episodes_per_latency: null
45
+ latency_filter: null
46
+ debug_subset:
47
+ enabled: false
48
+ max_episodes: 5
49
+ suffix: debug
50
+ base_model:
51
+ repo_id: Qwen/Qwen3-VL-4B-Instruct
52
+ initialization:
53
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
54
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
55
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
56
+ trainer:
57
+ max_train_steps: 1500
58
+ num_warmup_steps: 100
59
+ save_interval: 100
60
+ eval_interval: 100
61
+ eval_num_batches: 20
62
+ learning_rate:
63
+ base: 2.0e-05
64
+ qwen_vl_interface: 1.0e-05
65
+ action_model: 0.0001
66
+ lr_scheduler_type: cosine_with_min_lr
67
+ scheduler_specific_kwargs:
68
+ min_lr: 1.0e-06
69
+ freeze_modules: ''
70
+ loss_scale:
71
+ vla: 1.0
72
+ vlm: 0.1
73
+ max_grad_norm: 1.0
74
+ weight_decay: 0.0
75
+ logging_frequency: 10
76
+ gradient_clipping: 1.0
77
+ gradient_accumulation_steps: 16
78
+ distributed_backend: none
79
+ is_resume: false
80
+ pretrained_checkpoint: null
81
+ resume_step: 0
82
+ reload_modules: null
83
+ optimizer:
84
+ name: AdamW
85
+ betas:
86
+ - 0.9
87
+ - 0.95
88
+ eps: 1.0e-08
89
+ weight_decay: 1.0e-08
90
+ fused: false
91
+ save_format: pt
92
+ workspace_dir: WORKSPACE_DIR
93
+ run_root_dir: /workspace/starVLA/results/Checkpoints
94
+ seed: 42
95
+ wandb_entity: dongqianyu99-zhejiang-university
96
+ wandb_project: starVLA_rl_games
97
+ auth:
98
+ env_file: null
99
+ hf_token_env: HF_TOKEN
100
+ wandb_api_key_env: WANDB_API_KEY
101
+ paths:
102
+ run_root_dir: results/Checkpoints
103
+ dataset_local_dir: playground/Datasets/rl_games
104
+ dataset_cache_dir: null
105
+ base_model_dir: playground/Pretrained_models/Qwen3-VL-4B-Instruct
106
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
107
+ launch:
108
+ use_accelerate: true
109
+ gpus: null
110
+ num_processes: 1
111
+ dry_run: false
112
+ conda:
113
+ enabled: true
114
+ env_name: null
115
+ rl_games:
116
+ model_alias: openvla
117
+ env_eval:
118
+ image_size: 224
119
+ frameskip: 1
120
+ seed: 42
121
+ fixed_episode_seeds: true
122
+ latency_seed_stride: 0
123
+ task_seed_stride: 0
124
+ task_description: ''
125
+ enabled: true
126
+ distributed_mode: none
127
+ vectorized:
128
+ enabled: false
129
+ batch_size: 1
130
+ latency:
131
+ prompt_map_path: /workspace/starVLA/data/flappy_fix_latency_3_small/flappy_train__bridge/latency_prompt_map.json
132
+ mode: single
133
+ values:
134
+ - 0
135
+ mid_train:
136
+ enabled: true
137
+ interval_steps: 100
138
+ latencies:
139
+ - 3
140
+ num_episodes: 5
141
+ max_steps_per_episode: 3600
142
+ post_train:
143
+ enabled: true
144
+ latencies:
145
+ - 0
146
+ - 1
147
+ - 2
148
+ - 3
149
+ - 4
150
+ num_episodes: 5
151
+ max_steps_per_episode: 3600
152
+ task: flappy
153
+ initialization_mode: bridge
154
+ action_carrier: bridge
155
+ model: openvla
156
+ env: flappy
157
+ init: bridge
158
+ bridge_base_model:
159
+ repo_id:
160
+ openvla: Qwen/Qwen3-VL-4B-Instruct
161
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
162
+ pi05: Qwen/Qwen3-VL-4B-Instruct
163
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
164
+ local_dir:
165
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
166
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
167
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
168
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
169
+ mode: single
170
+ checkpoint:
171
+ load: auto
172
+ hf_repo_id: null
173
+ save_best_model: true
174
+ save_pt_file: false
175
+ local:
176
+ keep_last_n: 1
177
+ sync:
178
+ enabled: false
179
+ repo_id: null
180
+ keep_last_n: 0
181
+ sync_every_n_checkpoints: 1
182
+ resume_policy: local_latest
183
+ run_id: openvla_flappy_fix_latency_3_small
184
+ output_dir: /workspace/starVLA/results/Checkpoints/openvla_flappy_fix_latency_3_small
185
+ config_yaml: null
186
+ is_debug: false
187
+ version_id: '0.21'
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_3_small/config.yaml ADDED
@@ -0,0 +1,89 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ checkpoint:
2
+ local:
3
+ keep_last_n: 1
4
+ save_best_model: true
5
+ save_pt_file: false
6
+ sync:
7
+ enabled: false
8
+ keep_last_n: 0
9
+ repo_id: null
10
+ datasets:
11
+ vla_data:
12
+ data_mix: flappy_train__bridge
13
+ dataset_py: lerobot_datasets
14
+ eval_data_mix: flappy_train__bridge__val
15
+ include_state: true
16
+ obs_image_size: null
17
+ per_device_batch_size: 32
18
+ framework:
19
+ action_model:
20
+ action_dim: 7
21
+ action_env_dim: 2
22
+ action_hidden_dim: 2560
23
+ action_horizon: 1
24
+ action_model_type: MLP
25
+ loss_type: discrete_ce
26
+ state_dim: 7
27
+ name: QwenOFT
28
+ qwenvl:
29
+ attn_implementation: flash_attention_2
30
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
31
+ enable_gradient_checkpointing: true
32
+ output_dir: /workspace/starVLA/results/Checkpoints/openvla_flappy_fix_latency_3_small
33
+ rl_games:
34
+ env_eval:
35
+ distributed_mode: none
36
+ enabled: true
37
+ fixed_episode_seeds: true
38
+ frameskip: 1
39
+ image_size: 224
40
+ latency:
41
+ prompt_map_path: /workspace/starVLA/data/flappy_fix_latency_3_small/flappy_train__bridge/latency_prompt_map.json
42
+ latency_seed_stride: 0
43
+ mid_train:
44
+ enabled: true
45
+ interval_steps: 100
46
+ latencies:
47
+ - 3
48
+ max_steps_per_episode: 3600
49
+ num_episodes: 5
50
+ seed: 42
51
+ task_description: ''
52
+ task_seed_stride: 0
53
+ vectorized:
54
+ enabled: false
55
+ model_alias: openvla
56
+ task: flappy
57
+ run_id: openvla_flappy_fix_latency_3_small
58
+ run_root_dir: /workspace/starVLA/results/Checkpoints
59
+ seed: 42
60
+ trainer:
61
+ distributed_backend: none
62
+ eval_interval: 100
63
+ eval_num_batches: 20
64
+ freeze_modules: ''
65
+ gradient_accumulation_steps: 16
66
+ gradient_clipping: 1.0
67
+ is_resume: false
68
+ learning_rate:
69
+ action_model: 0.0001
70
+ base: 2.0e-05
71
+ qwen_vl_interface: 1.0e-05
72
+ logging_frequency: 10
73
+ lr_scheduler_type: cosine_with_min_lr
74
+ max_train_steps: 1500
75
+ num_warmup_steps: 100
76
+ optimizer:
77
+ betas:
78
+ - 0.9
79
+ - 0.95
80
+ eps: 1.0e-08
81
+ fused: false
82
+ weight_decay: 1.0e-08
83
+ pretrained_checkpoint: null
84
+ reload_modules: null
85
+ save_interval: 100
86
+ scheduler_specific_kwargs:
87
+ min_lr: 1.0e-06
88
+ wandb_entity: dongqianyu99-zhejiang-university
89
+ wandb_project: starVLA_rl_games
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_3_small/dataset_statistics.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.8046913743019104,
6
+ 0.1953086405992508,
7
+ 0.0,
8
+ 0.0,
9
+ 0.0,
10
+ 0.0,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.3963683247566223,
15
+ 0.3963683247566223,
16
+ 0.0,
17
+ 0.0,
18
+ 0.0,
19
+ 0.0,
20
+ 0.0
21
+ ],
22
+ "max": [
23
+ 1.0,
24
+ 1.0,
25
+ 0.0,
26
+ 0.0,
27
+ 0.0,
28
+ 0.0,
29
+ 0.0
30
+ ],
31
+ "min": [
32
+ 0.0,
33
+ 0.0,
34
+ 0.0,
35
+ 0.0,
36
+ 0.0,
37
+ 0.0,
38
+ 0.0
39
+ ],
40
+ "q01": [
41
+ 0.0,
42
+ 0.0,
43
+ 0.0,
44
+ 0.0,
45
+ 0.0,
46
+ 0.0,
47
+ 0.0
48
+ ],
49
+ "q99": [
50
+ 1.0,
51
+ 1.0,
52
+ 0.0,
53
+ 0.0,
54
+ 0.0,
55
+ 0.0,
56
+ 0.0
57
+ ],
58
+ "mask": [
59
+ true,
60
+ true,
61
+ true,
62
+ true,
63
+ true,
64
+ true,
65
+ true
66
+ ]
67
+ },
68
+ "state": {
69
+ "mean": [
70
+ 0.0,
71
+ 0.0,
72
+ 0.0,
73
+ 0.0,
74
+ 0.0,
75
+ 0.0,
76
+ 0.0
77
+ ],
78
+ "std": [
79
+ 0.0,
80
+ 0.0,
81
+ 0.0,
82
+ 0.0,
83
+ 0.0,
84
+ 0.0,
85
+ 0.0
86
+ ],
87
+ "max": [
88
+ 0.0,
89
+ 0.0,
90
+ 0.0,
91
+ 0.0,
92
+ 0.0,
93
+ 0.0,
94
+ 0.0
95
+ ],
96
+ "min": [
97
+ 0.0,
98
+ 0.0,
99
+ 0.0,
100
+ 0.0,
101
+ 0.0,
102
+ 0.0,
103
+ 0.0
104
+ ],
105
+ "q01": [
106
+ 0.0,
107
+ 0.0,
108
+ 0.0,
109
+ 0.0,
110
+ 0.0,
111
+ 0.0,
112
+ 0.0
113
+ ],
114
+ "q99": [
115
+ 0.0,
116
+ 0.0,
117
+ 0.0,
118
+ 0.0,
119
+ 0.0,
120
+ 0.0,
121
+ 0.0
122
+ ]
123
+ },
124
+ "num_transitions": 97200,
125
+ "num_trajectories": 27
126
+ }
127
+ }
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_3_small/dataset_statistics_eval.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.8032407164573669,
6
+ 0.19675925374031067,
7
+ 0.0,
8
+ 0.0,
9
+ 0.0,
10
+ 0.0,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.3975333571434021,
15
+ 0.3975333571434021,
16
+ 0.0,
17
+ 0.0,
18
+ 0.0,
19
+ 0.0,
20
+ 0.0
21
+ ],
22
+ "max": [
23
+ 1.0,
24
+ 1.0,
25
+ 0.0,
26
+ 0.0,
27
+ 0.0,
28
+ 0.0,
29
+ 0.0
30
+ ],
31
+ "min": [
32
+ 0.0,
33
+ 0.0,
34
+ 0.0,
35
+ 0.0,
36
+ 0.0,
37
+ 0.0,
38
+ 0.0
39
+ ],
40
+ "q01": [
41
+ 0.0,
42
+ 0.0,
43
+ 0.0,
44
+ 0.0,
45
+ 0.0,
46
+ 0.0,
47
+ 0.0
48
+ ],
49
+ "q99": [
50
+ 1.0,
51
+ 1.0,
52
+ 0.0,
53
+ 0.0,
54
+ 0.0,
55
+ 0.0,
56
+ 0.0
57
+ ],
58
+ "mask": [
59
+ true,
60
+ true,
61
+ true,
62
+ true,
63
+ true,
64
+ true,
65
+ true
66
+ ]
67
+ },
68
+ "state": {
69
+ "mean": [
70
+ 0.0,
71
+ 0.0,
72
+ 0.0,
73
+ 0.0,
74
+ 0.0,
75
+ 0.0,
76
+ 0.0
77
+ ],
78
+ "std": [
79
+ 0.0,
80
+ 0.0,
81
+ 0.0,
82
+ 0.0,
83
+ 0.0,
84
+ 0.0,
85
+ 0.0
86
+ ],
87
+ "max": [
88
+ 0.0,
89
+ 0.0,
90
+ 0.0,
91
+ 0.0,
92
+ 0.0,
93
+ 0.0,
94
+ 0.0
95
+ ],
96
+ "min": [
97
+ 0.0,
98
+ 0.0,
99
+ 0.0,
100
+ 0.0,
101
+ 0.0,
102
+ 0.0,
103
+ 0.0
104
+ ],
105
+ "q01": [
106
+ 0.0,
107
+ 0.0,
108
+ 0.0,
109
+ 0.0,
110
+ 0.0,
111
+ 0.0,
112
+ 0.0
113
+ ],
114
+ "q99": [
115
+ 0.0,
116
+ 0.0,
117
+ 0.0,
118
+ 0.0,
119
+ 0.0,
120
+ 0.0,
121
+ 0.0
122
+ ]
123
+ },
124
+ "num_transitions": 10800,
125
+ "num_trajectories": 3
126
+ }
127
+ }
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_3_small/hydra/.hydra/config.yaml ADDED
@@ -0,0 +1,185 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ qwenvl:
3
+ base_vlm: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct
4
+ attn_implementation: flash_attention_2
5
+ enable_gradient_checkpointing: true
6
+ action_model:
7
+ state_dim: 7
8
+ loss_type: discrete_ce
9
+ action_horizon: 1
10
+ future_action_window_size: 0
11
+ past_action_window_size: 0
12
+ action_dim: 7
13
+ action_env_dim: 2
14
+ name: QwenOFT
15
+ datasets:
16
+ vla_data:
17
+ dataset_py: lerobot_datasets
18
+ include_state: true
19
+ data_root_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_3_small
20
+ data_mix: flappy_train__bridge
21
+ eval_data_mix: flappy_train__bridge__val
22
+ custom_mixtures_path: null
23
+ action_type: discrete
24
+ sequential_step_sampling: false
25
+ per_device_batch_size: 32
26
+ load_all_data_for_training: true
27
+ obs_image_size: null
28
+ video_backend: torchvision_av
29
+ dataset:
30
+ source_hf: ${dataset.single_source_hf}
31
+ converted_name: ${dataset.single_converted_name}
32
+ single_source_hf: ''
33
+ mixed_source_hf: ''
34
+ single_converted_name: flappy_train
35
+ mixed_converted_name: flappy_mixed_latency_train
36
+ single_latency_filter: null
37
+ mixed_latency_filter: null
38
+ force_download: false
39
+ setup_force: false
40
+ verify_rows: 200
41
+ max_episodes: null
42
+ episodes_per_latency: null
43
+ latency_filter: ${dataset.single_latency_filter}
44
+ debug_subset:
45
+ enabled: false
46
+ max_episodes: 5
47
+ suffix: debug
48
+ base_model:
49
+ repo_id: ${bridge_base_model.repo_id.${model}}
50
+ initialization:
51
+ checkpoint_local_dir: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
52
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
53
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
54
+ trainer:
55
+ max_train_steps: 1500
56
+ num_warmup_steps: 100
57
+ save_interval: 100
58
+ eval_interval: 100
59
+ eval_num_batches: 20
60
+ learning_rate:
61
+ base: 2.0e-05
62
+ qwen_vl_interface: 1.0e-05
63
+ action_model: 0.0001
64
+ lr_scheduler_type: cosine_with_min_lr
65
+ scheduler_specific_kwargs:
66
+ min_lr: 1.0e-06
67
+ freeze_modules: ''
68
+ loss_scale:
69
+ vla: 1.0
70
+ vlm: 0.1
71
+ max_grad_norm: 1.0
72
+ weight_decay: 0.0
73
+ logging_frequency: 10
74
+ gradient_clipping: 1.0
75
+ gradient_accumulation_steps: 16
76
+ distributed_backend: none
77
+ is_resume: false
78
+ pretrained_checkpoint: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
79
+ resume_step: 0
80
+ reload_modules: null
81
+ optimizer:
82
+ name: AdamW
83
+ betas:
84
+ - 0.9
85
+ - 0.95
86
+ eps: 1.0e-08
87
+ weight_decay: 1.0e-08
88
+ fused: false
89
+ save_format: pt
90
+ workspace_dir: WORKSPACE_DIR
91
+ run_root_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints
92
+ seed: 42
93
+ wandb_entity: dongqianyu99-zhejiang-university
94
+ wandb_project: starVLA_rl_games
95
+ auth:
96
+ env_file: null
97
+ hf_token_env: HF_TOKEN
98
+ wandb_api_key_env: WANDB_API_KEY
99
+ paths:
100
+ run_root_dir: results/Checkpoints
101
+ dataset_local_dir: playground/Datasets/rl_games
102
+ dataset_cache_dir: null
103
+ base_model_dir: ${bridge_base_model.local_dir.${model}}
104
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
105
+ launch:
106
+ use_accelerate: true
107
+ gpus: null
108
+ num_processes: 1
109
+ dry_run: false
110
+ conda:
111
+ enabled: true
112
+ env_name: null
113
+ rl_games:
114
+ model_alias: openvla
115
+ env_eval:
116
+ image_size: 224
117
+ frameskip: 1
118
+ seed: 42
119
+ fixed_episode_seeds: true
120
+ latency_seed_stride: 0
121
+ task_seed_stride: 0
122
+ task_description: ''
123
+ enabled: true
124
+ distributed_mode: none
125
+ vectorized:
126
+ enabled: false
127
+ batch_size: 1
128
+ latency:
129
+ prompt_map_path: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_3_small/flappy_train__bridge/latency_prompt_map.json
130
+ mode: single
131
+ values:
132
+ - 0
133
+ mid_train:
134
+ enabled: true
135
+ interval_steps: 100
136
+ latencies:
137
+ - 3
138
+ num_episodes: 5
139
+ max_steps_per_episode: 3600
140
+ post_train:
141
+ enabled: true
142
+ latencies:
143
+ - 0
144
+ - 1
145
+ - 2
146
+ - 3
147
+ - 4
148
+ num_episodes: 5
149
+ max_steps_per_episode: 3600
150
+ task: flappy
151
+ initialization_mode: bridge
152
+ action_carrier: bridge
153
+ model: openvla
154
+ env: flappy
155
+ init: bridge
156
+ bridge_base_model:
157
+ repo_id:
158
+ openvla: Qwen/Qwen3-VL-4B-Instruct
159
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
160
+ pi05: Qwen/Qwen3-VL-4B-Instruct
161
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
162
+ local_dir:
163
+ openvla: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
164
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
165
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
166
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
167
+ mode: single
168
+ checkpoint:
169
+ load: auto
170
+ hf_repo_id: null
171
+ save_best_model: true
172
+ save_pt_file: false
173
+ local:
174
+ keep_last_n: 1
175
+ sync:
176
+ enabled: false
177
+ repo_id: null
178
+ keep_last_n: 0
179
+ sync_every_n_checkpoints: 1
180
+ resume_policy: local_latest
181
+ run_id: openvla_flappy_fix_latency_3_small
182
+ output_dir: null
183
+ config_yaml: null
184
+ is_debug: false
185
+ version_id: '0.21'
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_3_small/hydra/.hydra/hydra.yaml ADDED
@@ -0,0 +1,246 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ hydra:
2
+ run:
3
+ dir: ${run_root_dir}/${run_id}/hydra
4
+ sweep:
5
+ dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
6
+ subdir: ${hydra.job.num}
7
+ launcher:
8
+ _target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
9
+ sweeper:
10
+ _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
11
+ max_batch_size: null
12
+ params: null
13
+ help:
14
+ app_name: ${hydra.job.name}
15
+ header: '${hydra.help.app_name} is powered by Hydra.
16
+
17
+ '
18
+ footer: 'Powered by Hydra (https://hydra.cc)
19
+
20
+ Use --hydra-help to view Hydra specific help
21
+
22
+ '
23
+ template: '${hydra.help.header}
24
+
25
+ == Configuration groups ==
26
+
27
+ Compose your configuration from those groups (group=option)
28
+
29
+
30
+ $APP_CONFIG_GROUPS
31
+
32
+
33
+ == Config ==
34
+
35
+ Override anything in the config (foo.bar=value)
36
+
37
+
38
+ $CONFIG
39
+
40
+
41
+ ${hydra.help.footer}
42
+
43
+ '
44
+ hydra_help:
45
+ template: 'Hydra (${hydra.runtime.version})
46
+
47
+ See https://hydra.cc for more info.
48
+
49
+
50
+ == Flags ==
51
+
52
+ $FLAGS_HELP
53
+
54
+
55
+ == Configuration groups ==
56
+
57
+ Compose your configuration from those groups (For example, append hydra/job_logging=disabled
58
+ to command line)
59
+
60
+
61
+ $HYDRA_CONFIG_GROUPS
62
+
63
+
64
+ Use ''--cfg hydra'' to Show the Hydra config.
65
+
66
+ '
67
+ hydra_help: ???
68
+ hydra_logging:
69
+ version: 1
70
+ formatters:
71
+ simple:
72
+ format: '[%(asctime)s][HYDRA] %(message)s'
73
+ handlers:
74
+ console:
75
+ class: logging.StreamHandler
76
+ formatter: simple
77
+ stream: ext://sys.stdout
78
+ root:
79
+ level: INFO
80
+ handlers:
81
+ - console
82
+ loggers:
83
+ logging_example:
84
+ level: DEBUG
85
+ disable_existing_loggers: false
86
+ job_logging:
87
+ version: 1
88
+ formatters:
89
+ simple:
90
+ format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
91
+ handlers:
92
+ console:
93
+ class: logging.StreamHandler
94
+ formatter: simple
95
+ stream: ext://sys.stdout
96
+ file:
97
+ class: logging.FileHandler
98
+ formatter: simple
99
+ filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
100
+ root:
101
+ level: INFO
102
+ handlers:
103
+ - console
104
+ - file
105
+ disable_existing_loggers: false
106
+ env: {}
107
+ mode: RUN
108
+ searchpath: []
109
+ callbacks: {}
110
+ output_subdir: .hydra
111
+ overrides:
112
+ hydra:
113
+ - hydra.mode=RUN
114
+ task:
115
+ - model=openvla
116
+ - env=flappy
117
+ - init=bridge
118
+ - mode=single
119
+ - run_id=openvla_flappy_fix_latency_3_small
120
+ - run_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints
121
+ - seed=42
122
+ - wandb_entity=dongqianyu99-zhejiang-university
123
+ - wandb_project=starVLA_rl_games
124
+ - rl_games.env_eval.enabled=true
125
+ - checkpoint.sync.enabled=false
126
+ - checkpoint.sync.keep_last_n=0
127
+ - checkpoint.local.keep_last_n=1
128
+ - checkpoint.save_best_model=true
129
+ - checkpoint.save_pt_file=false
130
+ - trainer.is_resume=false
131
+ - trainer.pretrained_checkpoint=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
132
+ - trainer.resume_step=0
133
+ - trainer.max_train_steps=1500
134
+ - trainer.num_warmup_steps=100
135
+ - trainer.save_interval=100
136
+ - trainer.eval_interval=100
137
+ - trainer.eval_num_batches=20
138
+ - trainer.logging_frequency=10
139
+ - trainer.gradient_accumulation_steps=16
140
+ - trainer.distributed_backend=none
141
+ - trainer.learning_rate.base=2e-05
142
+ - trainer.learning_rate.qwen_vl_interface=1e-05
143
+ - trainer.learning_rate.action_model=0.0001
144
+ - trainer.lr_scheduler_type=cosine_with_min_lr
145
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
146
+ - trainer.loss_scale.vla=1.0
147
+ - trainer.loss_scale.vlm=0.1
148
+ - trainer.max_grad_norm=1.0
149
+ - trainer.weight_decay=0.0
150
+ - trainer.gradient_clipping=1.0
151
+ - trainer.optimizer.name=AdamW
152
+ - trainer.optimizer.betas=[0.9, 0.95]
153
+ - trainer.optimizer.eps=1e-08
154
+ - trainer.optimizer.weight_decay=1e-08
155
+ - trainer.optimizer.fused=false
156
+ - trainer.save_format=pt
157
+ - framework.name=QwenOFT
158
+ - framework.qwenvl.attn_implementation=flash_attention_2
159
+ - framework.qwenvl.enable_gradient_checkpointing=true
160
+ - framework.action_model.action_dim=7
161
+ - framework.action_model.action_env_dim=2
162
+ - framework.action_model.state_dim=7
163
+ - framework.action_model.loss_type=discrete_ce
164
+ - framework.action_model.action_horizon=1
165
+ - framework.action_model.future_action_window_size=0
166
+ - framework.action_model.past_action_window_size=0
167
+ - datasets.vla_data.include_state=true
168
+ - datasets.vla_data.action_type=discrete
169
+ - datasets.vla_data.sequential_step_sampling=false
170
+ - datasets.vla_data.per_device_batch_size=32
171
+ - datasets.vla_data.load_all_data_for_training=true
172
+ - datasets.vla_data.video_backend=torchvision_av
173
+ - datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_3_small
174
+ - datasets.vla_data.data_mix=flappy_train__bridge
175
+ - datasets.vla_data.eval_data_mix=flappy_train__bridge__val
176
+ - framework.qwenvl.base_vlm=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct
177
+ - rl_games.task=flappy
178
+ - rl_games.model_alias=openvla
179
+ - rl_games.initialization_mode=bridge
180
+ - rl_games.action_carrier=bridge
181
+ - rl_games.env_eval.distributed_mode=none
182
+ - rl_games.env_eval.latency.mode=single
183
+ - rl_games.env_eval.frameskip=1
184
+ - rl_games.env_eval.image_size=224
185
+ - rl_games.env_eval.seed=42
186
+ - rl_games.env_eval.fixed_episode_seeds=true
187
+ - rl_games.env_eval.latency_seed_stride=0
188
+ - rl_games.env_eval.task_seed_stride=0
189
+ - rl_games.env_eval.latency.values=[0]
190
+ - rl_games.env_eval.latency.prompt_map_path=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_3_small/flappy_train__bridge/latency_prompt_map.json
191
+ - rl_games.env_eval.mid_train.enabled=true
192
+ - rl_games.env_eval.mid_train.interval_steps=100
193
+ - rl_games.env_eval.mid_train.num_episodes=5
194
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
195
+ - rl_games.env_eval.mid_train.latencies=[3]
196
+ - rl_games.env_eval.post_train.enabled=true
197
+ - rl_games.env_eval.post_train.num_episodes=5
198
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
199
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
200
+ job:
201
+ name: train_starvla_hydra
202
+ chdir: false
203
+ override_dirname: checkpoint.local.keep_last_n=1,checkpoint.save_best_model=true,checkpoint.save_pt_file=false,checkpoint.sync.enabled=false,checkpoint.sync.keep_last_n=0,datasets.vla_data.action_type=discrete,datasets.vla_data.data_mix=flappy_train__bridge,datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_3_small,datasets.vla_data.eval_data_mix=flappy_train__bridge__val,datasets.vla_data.include_state=true,datasets.vla_data.load_all_data_for_training=true,datasets.vla_data.per_device_batch_size=32,datasets.vla_data.sequential_step_sampling=false,datasets.vla_data.video_backend=torchvision_av,env=flappy,framework.action_model.action_dim=7,framework.action_model.action_env_dim=2,framework.action_model.action_horizon=1,framework.action_model.future_action_window_size=0,framework.action_model.loss_type=discrete_ce,framework.action_model.past_action_window_size=0,framework.action_model.state_dim=7,framework.name=QwenOFT,framework.qwenvl.attn_implementation=flash_attention_2,framework.qwenvl.base_vlm=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct,framework.qwenvl.enable_gradient_checkpointing=true,init=bridge,mode=single,model=openvla,rl_games.action_carrier=bridge,rl_games.env_eval.distributed_mode=none,rl_games.env_eval.enabled=true,rl_games.env_eval.fixed_episode_seeds=true,rl_games.env_eval.frameskip=1,rl_games.env_eval.image_size=224,rl_games.env_eval.latency.mode=single,rl_games.env_eval.latency.prompt_map_path=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_3_small/flappy_train__bridge/latency_prompt_map.json,rl_games.env_eval.latency.values=[0],rl_games.env_eval.latency_seed_stride=0,rl_games.env_eval.mid_train.enabled=true,rl_games.env_eval.mid_train.interval_steps=100,rl_games.env_eval.mid_train.latencies=[3],rl_games.env_eval.mid_train.max_steps_per_episode=3600,rl_games.env_eval.mid_train.num_episodes=5,rl_games.env_eval.post_train.enabled=true,rl_games.env_eval.post_train.latencies=[0,1,2,3,4],rl_games.env_eval.post_train.max_steps_per_episode=3600,rl_games.env_eval.post_train.num_episodes=5,rl_games.env_eval.seed=42,rl_games.env_eval.task_seed_stride=0,rl_games.initialization_mode=bridge,rl_games.model_alias=openvla,rl_games.task=flappy,run_id=openvla_flappy_fix_latency_3_small,run_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints,seed=42,trainer.distributed_backend=none,trainer.eval_interval=100,trainer.eval_num_batches=20,trainer.gradient_accumulation_steps=16,trainer.gradient_clipping=1.0,trainer.is_resume=false,trainer.learning_rate.action_model=0.0001,trainer.learning_rate.base=2e-05,trainer.learning_rate.qwen_vl_interface=1e-05,trainer.logging_frequency=10,trainer.loss_scale.vla=1.0,trainer.loss_scale.vlm=0.1,trainer.lr_scheduler_type=cosine_with_min_lr,trainer.max_grad_norm=1.0,trainer.max_train_steps=1500,trainer.num_warmup_steps=100,trainer.optimizer.betas=[0.9,
204
+ 0.95],trainer.optimizer.eps=1e-08,trainer.optimizer.fused=false,trainer.optimizer.name=AdamW,trainer.optimizer.weight_decay=1e-08,trainer.pretrained_checkpoint=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt,trainer.resume_step=0,trainer.save_format=pt,trainer.save_interval=100,trainer.scheduler_specific_kwargs.min_lr=1e-06,trainer.weight_decay=0.0,wandb_entity=dongqianyu99-zhejiang-university,wandb_project=starVLA_rl_games
205
+ id: ???
206
+ num: ???
207
+ config_name: train
208
+ env_set: {}
209
+ env_copy: []
210
+ config:
211
+ override_dirname:
212
+ kv_sep: '='
213
+ item_sep: ','
214
+ exclude_keys: []
215
+ runtime:
216
+ version: 1.3.2
217
+ version_base: '1.1'
218
+ cwd: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA
219
+ config_sources:
220
+ - path: hydra.conf
221
+ schema: pkg
222
+ provider: hydra
223
+ - path: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA/examples/rl_games/config
224
+ schema: file
225
+ provider: main
226
+ - path: ''
227
+ schema: structured
228
+ provider: schema
229
+ output_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints/openvla_flappy_fix_latency_3_small/hydra
230
+ choices:
231
+ cross_task_setup: null
232
+ checkpoint: default
233
+ mode: single
234
+ init: bridge
235
+ env: flappy
236
+ model: openvla
237
+ hydra/env: default
238
+ hydra/callbacks: null
239
+ hydra/job_logging: default
240
+ hydra/hydra_logging: default
241
+ hydra/hydra_help: default
242
+ hydra/help: default
243
+ hydra/sweeper: basic
244
+ hydra/launcher: basic
245
+ hydra/output: default
246
+ verbose: false
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_3_small/hydra/.hydra/overrides.yaml ADDED
@@ -0,0 +1,85 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ - model=openvla
2
+ - env=flappy
3
+ - init=bridge
4
+ - mode=single
5
+ - run_id=openvla_flappy_fix_latency_3_small
6
+ - run_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints
7
+ - seed=42
8
+ - wandb_entity=dongqianyu99-zhejiang-university
9
+ - wandb_project=starVLA_rl_games
10
+ - rl_games.env_eval.enabled=true
11
+ - checkpoint.sync.enabled=false
12
+ - checkpoint.sync.keep_last_n=0
13
+ - checkpoint.local.keep_last_n=1
14
+ - checkpoint.save_best_model=true
15
+ - checkpoint.save_pt_file=false
16
+ - trainer.is_resume=false
17
+ - trainer.pretrained_checkpoint=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
18
+ - trainer.resume_step=0
19
+ - trainer.max_train_steps=1500
20
+ - trainer.num_warmup_steps=100
21
+ - trainer.save_interval=100
22
+ - trainer.eval_interval=100
23
+ - trainer.eval_num_batches=20
24
+ - trainer.logging_frequency=10
25
+ - trainer.gradient_accumulation_steps=16
26
+ - trainer.distributed_backend=none
27
+ - trainer.learning_rate.base=2e-05
28
+ - trainer.learning_rate.qwen_vl_interface=1e-05
29
+ - trainer.learning_rate.action_model=0.0001
30
+ - trainer.lr_scheduler_type=cosine_with_min_lr
31
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
32
+ - trainer.loss_scale.vla=1.0
33
+ - trainer.loss_scale.vlm=0.1
34
+ - trainer.max_grad_norm=1.0
35
+ - trainer.weight_decay=0.0
36
+ - trainer.gradient_clipping=1.0
37
+ - trainer.optimizer.name=AdamW
38
+ - trainer.optimizer.betas=[0.9, 0.95]
39
+ - trainer.optimizer.eps=1e-08
40
+ - trainer.optimizer.weight_decay=1e-08
41
+ - trainer.optimizer.fused=false
42
+ - trainer.save_format=pt
43
+ - framework.name=QwenOFT
44
+ - framework.qwenvl.attn_implementation=flash_attention_2
45
+ - framework.qwenvl.enable_gradient_checkpointing=true
46
+ - framework.action_model.action_dim=7
47
+ - framework.action_model.action_env_dim=2
48
+ - framework.action_model.state_dim=7
49
+ - framework.action_model.loss_type=discrete_ce
50
+ - framework.action_model.action_horizon=1
51
+ - framework.action_model.future_action_window_size=0
52
+ - framework.action_model.past_action_window_size=0
53
+ - datasets.vla_data.include_state=true
54
+ - datasets.vla_data.action_type=discrete
55
+ - datasets.vla_data.sequential_step_sampling=false
56
+ - datasets.vla_data.per_device_batch_size=32
57
+ - datasets.vla_data.load_all_data_for_training=true
58
+ - datasets.vla_data.video_backend=torchvision_av
59
+ - datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_3_small
60
+ - datasets.vla_data.data_mix=flappy_train__bridge
61
+ - datasets.vla_data.eval_data_mix=flappy_train__bridge__val
62
+ - framework.qwenvl.base_vlm=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct
63
+ - rl_games.task=flappy
64
+ - rl_games.model_alias=openvla
65
+ - rl_games.initialization_mode=bridge
66
+ - rl_games.action_carrier=bridge
67
+ - rl_games.env_eval.distributed_mode=none
68
+ - rl_games.env_eval.latency.mode=single
69
+ - rl_games.env_eval.frameskip=1
70
+ - rl_games.env_eval.image_size=224
71
+ - rl_games.env_eval.seed=42
72
+ - rl_games.env_eval.fixed_episode_seeds=true
73
+ - rl_games.env_eval.latency_seed_stride=0
74
+ - rl_games.env_eval.task_seed_stride=0
75
+ - rl_games.env_eval.latency.values=[0]
76
+ - rl_games.env_eval.latency.prompt_map_path=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_3_small/flappy_train__bridge/latency_prompt_map.json
77
+ - rl_games.env_eval.mid_train.enabled=true
78
+ - rl_games.env_eval.mid_train.interval_steps=100
79
+ - rl_games.env_eval.mid_train.num_episodes=5
80
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
81
+ - rl_games.env_eval.mid_train.latencies=[3]
82
+ - rl_games.env_eval.post_train.enabled=true
83
+ - rl_games.env_eval.post_train.num_episodes=5
84
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
85
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_3_small/hydra/train_starvla_hydra.log ADDED
The diff for this file is too large to render. See raw diff
 
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_3_small/summary.jsonl ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"steps": 100}
2
+ {"steps": 200}
3
+ {"steps": 300}
4
+ {"steps": 400}
5
+ {"steps": 500}
6
+ {"steps": 600}
7
+ {"steps": 700}
8
+ {"steps": 800}
9
+ {"steps": 900}
10
+ {"steps": 1000}
11
+ {"steps": 1100}
12
+ {"steps": 1200}
13
+ {"steps": 1300}
14
+ {"steps": 1400}
15
+ {"steps": 1500}
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_exp/checkpoints/steps_2500_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b4a6e3ade6d47371998034284d7a0ccbeb1f92b9713a657eb6041604cdbe9235
3
+ size 9138230516
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_exp/checkpoints/steps_2500_pytorch_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1610fb6615308c849e844acde8d98b233fe6f6c4d01657d184fd4bd9945e55f5
3
+ size 9138476048
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_exp/config.full.yaml ADDED
@@ -0,0 +1,179 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ name: QwenOFT
3
+ qwenvl:
4
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
5
+ attn_implementation: flash_attention_2
6
+ enable_gradient_checkpointing: true
7
+ action_model:
8
+ action_model_type: MLP
9
+ action_dim: 7
10
+ action_hidden_dim: 2560
11
+ future_action_window_size: 0
12
+ past_action_window_size: 0
13
+ loss_type: discrete_ce
14
+ state_dim: 7
15
+ action_horizon: 1
16
+ action_env_dim: 2
17
+ datasets:
18
+ vla_data:
19
+ dataset_py: lerobot_datasets
20
+ include_state: true
21
+ data_root_dir: /workspace/starVLA/data/flappy_fix_latency_4
22
+ data_mix: flappy_train__bridge
23
+ eval_data_mix: flappy_train__bridge__val
24
+ action_type: discrete
25
+ sequential_step_sampling: false
26
+ per_device_batch_size: 32
27
+ load_all_data_for_training: true
28
+ obs_image_size: null
29
+ video_backend: torchvision_av
30
+ dataset:
31
+ source_hf: ''
32
+ converted_name: flappy_train
33
+ single_source_hf: ''
34
+ mixed_source_hf: ''
35
+ single_converted_name: flappy_train
36
+ mixed_converted_name: flappy_mixed_latency_train
37
+ single_latency_filter: null
38
+ mixed_latency_filter: null
39
+ force_download: false
40
+ setup_force: false
41
+ verify_rows: 200
42
+ max_episodes: null
43
+ latency_filter: null
44
+ debug_subset:
45
+ enabled: false
46
+ max_episodes: 5
47
+ suffix: debug
48
+ base_model:
49
+ repo_id: Qwen/Qwen3-VL-4B-Instruct
50
+ initialization:
51
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
52
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
53
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
54
+ trainer:
55
+ max_train_steps: 2500
56
+ num_warmup_steps: 100
57
+ save_interval: 100
58
+ eval_interval: 100
59
+ eval_num_batches: 20
60
+ learning_rate:
61
+ base: 2.0e-05
62
+ qwen_vl_interface: 1.0e-05
63
+ action_model: 0.0001
64
+ lr_scheduler_type: cosine_with_min_lr
65
+ scheduler_specific_kwargs:
66
+ min_lr: 1.0e-06
67
+ freeze_modules: ''
68
+ loss_scale:
69
+ vla: 1.0
70
+ vlm: 0.1
71
+ max_grad_norm: 1.0
72
+ weight_decay: 0.0
73
+ logging_frequency: 10
74
+ gradient_clipping: 1.0
75
+ gradient_accumulation_steps: 16
76
+ distributed_backend: none
77
+ is_resume: false
78
+ pretrained_checkpoint: /workspace/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
79
+ resume_step: 0
80
+ reload_modules: null
81
+ optimizer:
82
+ name: AdamW
83
+ betas:
84
+ - 0.9
85
+ - 0.95
86
+ eps: 1.0e-08
87
+ weight_decay: 1.0e-08
88
+ fused: false
89
+ save_format: pt
90
+ workspace_dir: WORKSPACE_DIR
91
+ run_root_dir: /workspace/starVLA/results/Checkpoints
92
+ seed: 42
93
+ wandb_entity: dongqianyu99-zhejiang-university
94
+ wandb_project: starVLA_rl_games
95
+ auth:
96
+ env_file: null
97
+ hf_token_env: HF_TOKEN
98
+ wandb_api_key_env: WANDB_API_KEY
99
+ paths:
100
+ run_root_dir: results/Checkpoints
101
+ dataset_local_dir: playground/Datasets/rl_games
102
+ dataset_cache_dir: null
103
+ base_model_dir: playground/Pretrained_models/Qwen3-VL-4B-Instruct
104
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
105
+ launch:
106
+ use_accelerate: true
107
+ gpus: null
108
+ num_processes: 1
109
+ dry_run: false
110
+ conda:
111
+ enabled: true
112
+ env_name: null
113
+ rl_games:
114
+ model_alias: openvla
115
+ env_eval:
116
+ image_size: 224
117
+ frameskip: 1
118
+ seed: 42
119
+ fixed_episode_seeds: true
120
+ latency_seed_stride: 0
121
+ task_seed_stride: 100000
122
+ task_description: ''
123
+ enabled: true
124
+ latency:
125
+ prompt_map_path: /workspace/starVLA/data/flappy_fix_latency_4/flappy_train__bridge/latency_prompt_map.json
126
+ mode: single
127
+ values:
128
+ - 0
129
+ mid_train:
130
+ enabled: true
131
+ interval_steps: 100
132
+ latencies:
133
+ - 4
134
+ num_episodes: 5
135
+ max_steps_per_episode: 3600
136
+ post_train:
137
+ enabled: true
138
+ latencies:
139
+ - 0
140
+ - 1
141
+ - 2
142
+ - 3
143
+ - 4
144
+ num_episodes: 5
145
+ max_steps_per_episode: 3600
146
+ task: flappy
147
+ initialization_mode: bridge
148
+ action_carrier: bridge
149
+ model: openvla
150
+ env: flappy
151
+ init: bridge
152
+ bridge_base_model:
153
+ repo_id:
154
+ openvla: Qwen/Qwen3-VL-4B-Instruct
155
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
156
+ pi05: Qwen/Qwen3-VL-4B-Instruct
157
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
158
+ local_dir:
159
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
160
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
161
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
162
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
163
+ mode: single
164
+ checkpoint:
165
+ load: auto
166
+ hf_repo_id: null
167
+ local:
168
+ keep_last_n: 2
169
+ sync:
170
+ enabled: false
171
+ repo_id: null
172
+ keep_last_n: 0
173
+ sync_every_n_checkpoints: 1
174
+ resume_policy: local_latest
175
+ run_id: openvla_flappy_fix_latency_4_exp
176
+ output_dir: /workspace/starVLA/results/Checkpoints/openvla_flappy_fix_latency_4_exp
177
+ config_yaml: null
178
+ is_debug: false
179
+ version_id: '0.21'
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_exp/config.yaml ADDED
@@ -0,0 +1,83 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ checkpoint:
2
+ local:
3
+ keep_last_n: 2
4
+ sync:
5
+ enabled: false
6
+ keep_last_n: 0
7
+ repo_id: null
8
+ datasets:
9
+ vla_data:
10
+ data_mix: flappy_train__bridge
11
+ dataset_py: lerobot_datasets
12
+ eval_data_mix: flappy_train__bridge__val
13
+ include_state: true
14
+ obs_image_size: null
15
+ per_device_batch_size: 32
16
+ framework:
17
+ action_model:
18
+ action_dim: 7
19
+ action_env_dim: 2
20
+ action_hidden_dim: 2560
21
+ action_horizon: 1
22
+ action_model_type: MLP
23
+ loss_type: discrete_ce
24
+ state_dim: 7
25
+ name: QwenOFT
26
+ qwenvl:
27
+ attn_implementation: flash_attention_2
28
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
29
+ enable_gradient_checkpointing: true
30
+ output_dir: /workspace/starVLA/results/Checkpoints/openvla_flappy_fix_latency_4_exp
31
+ rl_games:
32
+ env_eval:
33
+ enabled: true
34
+ fixed_episode_seeds: true
35
+ frameskip: 1
36
+ image_size: 224
37
+ latency:
38
+ prompt_map_path: /workspace/starVLA/data/flappy_fix_latency_4/flappy_train__bridge/latency_prompt_map.json
39
+ latency_seed_stride: 0
40
+ mid_train:
41
+ enabled: true
42
+ interval_steps: 100
43
+ latencies:
44
+ - 4
45
+ max_steps_per_episode: 3600
46
+ num_episodes: 5
47
+ seed: 42
48
+ task_description: ''
49
+ task_seed_stride: 100000
50
+ model_alias: openvla
51
+ task: flappy
52
+ run_id: openvla_flappy_fix_latency_4_exp
53
+ run_root_dir: /workspace/starVLA/results/Checkpoints
54
+ seed: 42
55
+ trainer:
56
+ eval_interval: 100
57
+ eval_num_batches: 20
58
+ freeze_modules: ''
59
+ gradient_clipping: 1.0
60
+ is_resume: false
61
+ learning_rate:
62
+ action_model: 0.0001
63
+ base: 2.0e-05
64
+ qwen_vl_interface: 1.0e-05
65
+ logging_frequency: 10
66
+ lr_scheduler_type: cosine_with_min_lr
67
+ max_train_steps: 2500
68
+ num_warmup_steps: 100
69
+ optimizer:
70
+ betas:
71
+ - 0.9
72
+ - 0.95
73
+ eps: 1.0e-08
74
+ fused: false
75
+ weight_decay: 1.0e-08
76
+ pretrained_checkpoint: /workspace/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
77
+ reload_modules: null
78
+ save_format: pt
79
+ save_interval: 100
80
+ scheduler_specific_kwargs:
81
+ min_lr: 1.0e-06
82
+ wandb_entity: dongqianyu99-zhejiang-university
83
+ wandb_project: starVLA_rl_games
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_exp/dataset_statistics.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.8187143802642822,
6
+ 0.18128561973571777,
7
+ 0.0,
8
+ 0.0,
9
+ 0.0,
10
+ 0.0,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.3849780857563019,
15
+ 0.3849780857563019,
16
+ 0.0,
17
+ 0.0,
18
+ 0.0,
19
+ 0.0,
20
+ 0.0
21
+ ],
22
+ "max": [
23
+ 1.0,
24
+ 1.0,
25
+ 0.0,
26
+ 0.0,
27
+ 0.0,
28
+ 0.0,
29
+ 0.0
30
+ ],
31
+ "min": [
32
+ 0.0,
33
+ 0.0,
34
+ 0.0,
35
+ 0.0,
36
+ 0.0,
37
+ 0.0,
38
+ 0.0
39
+ ],
40
+ "q01": [
41
+ 0.0,
42
+ 0.0,
43
+ 0.0,
44
+ 0.0,
45
+ 0.0,
46
+ 0.0,
47
+ 0.0
48
+ ],
49
+ "q99": [
50
+ 1.0,
51
+ 1.0,
52
+ 0.0,
53
+ 0.0,
54
+ 0.0,
55
+ 0.0,
56
+ 0.0
57
+ ],
58
+ "mask": [
59
+ true,
60
+ true,
61
+ true,
62
+ true,
63
+ true,
64
+ true,
65
+ true
66
+ ]
67
+ },
68
+ "state": {
69
+ "mean": [
70
+ 0.0,
71
+ 0.0,
72
+ 0.0,
73
+ 0.0,
74
+ 0.0,
75
+ 0.0,
76
+ 0.0
77
+ ],
78
+ "std": [
79
+ 0.0,
80
+ 0.0,
81
+ 0.0,
82
+ 0.0,
83
+ 0.0,
84
+ 0.0,
85
+ 0.0
86
+ ],
87
+ "max": [
88
+ 0.0,
89
+ 0.0,
90
+ 0.0,
91
+ 0.0,
92
+ 0.0,
93
+ 0.0,
94
+ 0.0
95
+ ],
96
+ "min": [
97
+ 0.0,
98
+ 0.0,
99
+ 0.0,
100
+ 0.0,
101
+ 0.0,
102
+ 0.0,
103
+ 0.0
104
+ ],
105
+ "q01": [
106
+ 0.0,
107
+ 0.0,
108
+ 0.0,
109
+ 0.0,
110
+ 0.0,
111
+ 0.0,
112
+ 0.0
113
+ ],
114
+ "q99": [
115
+ 0.0,
116
+ 0.0,
117
+ 0.0,
118
+ 0.0,
119
+ 0.0,
120
+ 0.0,
121
+ 0.0
122
+ ]
123
+ },
124
+ "num_transitions": 318696,
125
+ "num_trajectories": 90
126
+ }
127
+ }
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_exp/dataset_statistics_eval.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.820472240447998,
6
+ 0.17952777445316315,
7
+ 0.0,
8
+ 0.0,
9
+ 0.0,
10
+ 0.0,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.38379305601119995,
15
+ 0.38379305601119995,
16
+ 0.0,
17
+ 0.0,
18
+ 0.0,
19
+ 0.0,
20
+ 0.0
21
+ ],
22
+ "max": [
23
+ 1.0,
24
+ 1.0,
25
+ 0.0,
26
+ 0.0,
27
+ 0.0,
28
+ 0.0,
29
+ 0.0
30
+ ],
31
+ "min": [
32
+ 0.0,
33
+ 0.0,
34
+ 0.0,
35
+ 0.0,
36
+ 0.0,
37
+ 0.0,
38
+ 0.0
39
+ ],
40
+ "q01": [
41
+ 0.0,
42
+ 0.0,
43
+ 0.0,
44
+ 0.0,
45
+ 0.0,
46
+ 0.0,
47
+ 0.0
48
+ ],
49
+ "q99": [
50
+ 1.0,
51
+ 1.0,
52
+ 0.0,
53
+ 0.0,
54
+ 0.0,
55
+ 0.0,
56
+ 0.0
57
+ ],
58
+ "mask": [
59
+ true,
60
+ true,
61
+ true,
62
+ true,
63
+ true,
64
+ true,
65
+ true
66
+ ]
67
+ },
68
+ "state": {
69
+ "mean": [
70
+ 0.0,
71
+ 0.0,
72
+ 0.0,
73
+ 0.0,
74
+ 0.0,
75
+ 0.0,
76
+ 0.0
77
+ ],
78
+ "std": [
79
+ 0.0,
80
+ 0.0,
81
+ 0.0,
82
+ 0.0,
83
+ 0.0,
84
+ 0.0,
85
+ 0.0
86
+ ],
87
+ "max": [
88
+ 0.0,
89
+ 0.0,
90
+ 0.0,
91
+ 0.0,
92
+ 0.0,
93
+ 0.0,
94
+ 0.0
95
+ ],
96
+ "min": [
97
+ 0.0,
98
+ 0.0,
99
+ 0.0,
100
+ 0.0,
101
+ 0.0,
102
+ 0.0,
103
+ 0.0
104
+ ],
105
+ "q01": [
106
+ 0.0,
107
+ 0.0,
108
+ 0.0,
109
+ 0.0,
110
+ 0.0,
111
+ 0.0,
112
+ 0.0
113
+ ],
114
+ "q99": [
115
+ 0.0,
116
+ 0.0,
117
+ 0.0,
118
+ 0.0,
119
+ 0.0,
120
+ 0.0,
121
+ 0.0
122
+ ]
123
+ },
124
+ "num_transitions": 36000,
125
+ "num_trajectories": 10
126
+ }
127
+ }
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_exp/hydra/.hydra/config.yaml ADDED
@@ -0,0 +1,177 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ qwenvl:
3
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
4
+ attn_implementation: flash_attention_2
5
+ enable_gradient_checkpointing: true
6
+ action_model:
7
+ state_dim: 7
8
+ loss_type: discrete_ce
9
+ action_horizon: 1
10
+ future_action_window_size: 0
11
+ past_action_window_size: 0
12
+ action_dim: 7
13
+ action_env_dim: 2
14
+ name: QwenOFT
15
+ datasets:
16
+ vla_data:
17
+ dataset_py: lerobot_datasets
18
+ include_state: true
19
+ data_root_dir: /workspace/starVLA/data/flappy_fix_latency_4
20
+ data_mix: flappy_train__bridge
21
+ eval_data_mix: flappy_train__bridge__val
22
+ action_type: discrete
23
+ sequential_step_sampling: false
24
+ per_device_batch_size: 32
25
+ load_all_data_for_training: true
26
+ obs_image_size: null
27
+ video_backend: torchvision_av
28
+ dataset:
29
+ source_hf: ${dataset.single_source_hf}
30
+ converted_name: ${dataset.single_converted_name}
31
+ single_source_hf: ''
32
+ mixed_source_hf: ''
33
+ single_converted_name: flappy_train
34
+ mixed_converted_name: flappy_mixed_latency_train
35
+ single_latency_filter: null
36
+ mixed_latency_filter: null
37
+ force_download: false
38
+ setup_force: false
39
+ verify_rows: 200
40
+ max_episodes: null
41
+ latency_filter: ${dataset.single_latency_filter}
42
+ debug_subset:
43
+ enabled: false
44
+ max_episodes: 5
45
+ suffix: debug
46
+ base_model:
47
+ repo_id: ${bridge_base_model.repo_id.${model}}
48
+ initialization:
49
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
50
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
51
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
52
+ trainer:
53
+ max_train_steps: 2500
54
+ num_warmup_steps: 100
55
+ save_interval: 100
56
+ eval_interval: 100
57
+ eval_num_batches: 20
58
+ learning_rate:
59
+ base: 2.0e-05
60
+ qwen_vl_interface: 1.0e-05
61
+ action_model: 0.0001
62
+ lr_scheduler_type: cosine_with_min_lr
63
+ scheduler_specific_kwargs:
64
+ min_lr: 1.0e-06
65
+ freeze_modules: ''
66
+ loss_scale:
67
+ vla: 1.0
68
+ vlm: 0.1
69
+ max_grad_norm: 1.0
70
+ weight_decay: 0.0
71
+ logging_frequency: 10
72
+ gradient_clipping: 1.0
73
+ gradient_accumulation_steps: 16
74
+ distributed_backend: none
75
+ is_resume: false
76
+ pretrained_checkpoint: /workspace/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
77
+ resume_step: 0
78
+ reload_modules: null
79
+ optimizer:
80
+ name: AdamW
81
+ betas:
82
+ - 0.9
83
+ - 0.95
84
+ eps: 1.0e-08
85
+ weight_decay: 1.0e-08
86
+ fused: false
87
+ save_format: pt
88
+ workspace_dir: WORKSPACE_DIR
89
+ run_root_dir: /workspace/starVLA/results/Checkpoints
90
+ seed: 42
91
+ wandb_entity: dongqianyu99-zhejiang-university
92
+ wandb_project: starVLA_rl_games
93
+ auth:
94
+ env_file: null
95
+ hf_token_env: HF_TOKEN
96
+ wandb_api_key_env: WANDB_API_KEY
97
+ paths:
98
+ run_root_dir: results/Checkpoints
99
+ dataset_local_dir: playground/Datasets/rl_games
100
+ dataset_cache_dir: null
101
+ base_model_dir: ${bridge_base_model.local_dir.${model}}
102
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
103
+ launch:
104
+ use_accelerate: true
105
+ gpus: null
106
+ num_processes: 1
107
+ dry_run: false
108
+ conda:
109
+ enabled: true
110
+ env_name: null
111
+ rl_games:
112
+ model_alias: openvla
113
+ env_eval:
114
+ image_size: 224
115
+ frameskip: 1
116
+ seed: 42
117
+ fixed_episode_seeds: true
118
+ latency_seed_stride: 0
119
+ task_seed_stride: 100000
120
+ task_description: ''
121
+ enabled: true
122
+ latency:
123
+ prompt_map_path: /workspace/starVLA/data/flappy_fix_latency_4/flappy_train__bridge/latency_prompt_map.json
124
+ mode: single
125
+ values:
126
+ - 0
127
+ mid_train:
128
+ enabled: true
129
+ interval_steps: 100
130
+ latencies:
131
+ - 4
132
+ num_episodes: 5
133
+ max_steps_per_episode: 3600
134
+ post_train:
135
+ enabled: true
136
+ latencies:
137
+ - 0
138
+ - 1
139
+ - 2
140
+ - 3
141
+ - 4
142
+ num_episodes: 5
143
+ max_steps_per_episode: 3600
144
+ task: flappy
145
+ initialization_mode: bridge
146
+ action_carrier: bridge
147
+ model: openvla
148
+ env: flappy
149
+ init: bridge
150
+ bridge_base_model:
151
+ repo_id:
152
+ openvla: Qwen/Qwen3-VL-4B-Instruct
153
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
154
+ pi05: Qwen/Qwen3-VL-4B-Instruct
155
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
156
+ local_dir:
157
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
158
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
159
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
160
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
161
+ mode: single
162
+ checkpoint:
163
+ load: auto
164
+ hf_repo_id: null
165
+ local:
166
+ keep_last_n: 2
167
+ sync:
168
+ enabled: false
169
+ repo_id: null
170
+ keep_last_n: 0
171
+ sync_every_n_checkpoints: 1
172
+ resume_policy: local_latest
173
+ run_id: openvla_flappy_fix_latency_4_exp
174
+ output_dir: null
175
+ config_yaml: null
176
+ is_debug: false
177
+ version_id: '0.21'
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_exp/hydra/.hydra/hydra.yaml ADDED
@@ -0,0 +1,242 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ hydra:
2
+ run:
3
+ dir: ${run_root_dir}/${run_id}/hydra
4
+ sweep:
5
+ dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
6
+ subdir: ${hydra.job.num}
7
+ launcher:
8
+ _target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
9
+ sweeper:
10
+ _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
11
+ max_batch_size: null
12
+ params: null
13
+ help:
14
+ app_name: ${hydra.job.name}
15
+ header: '${hydra.help.app_name} is powered by Hydra.
16
+
17
+ '
18
+ footer: 'Powered by Hydra (https://hydra.cc)
19
+
20
+ Use --hydra-help to view Hydra specific help
21
+
22
+ '
23
+ template: '${hydra.help.header}
24
+
25
+ == Configuration groups ==
26
+
27
+ Compose your configuration from those groups (group=option)
28
+
29
+
30
+ $APP_CONFIG_GROUPS
31
+
32
+
33
+ == Config ==
34
+
35
+ Override anything in the config (foo.bar=value)
36
+
37
+
38
+ $CONFIG
39
+
40
+
41
+ ${hydra.help.footer}
42
+
43
+ '
44
+ hydra_help:
45
+ template: 'Hydra (${hydra.runtime.version})
46
+
47
+ See https://hydra.cc for more info.
48
+
49
+
50
+ == Flags ==
51
+
52
+ $FLAGS_HELP
53
+
54
+
55
+ == Configuration groups ==
56
+
57
+ Compose your configuration from those groups (For example, append hydra/job_logging=disabled
58
+ to command line)
59
+
60
+
61
+ $HYDRA_CONFIG_GROUPS
62
+
63
+
64
+ Use ''--cfg hydra'' to Show the Hydra config.
65
+
66
+ '
67
+ hydra_help: ???
68
+ hydra_logging:
69
+ version: 1
70
+ formatters:
71
+ simple:
72
+ format: '[%(asctime)s][HYDRA] %(message)s'
73
+ handlers:
74
+ console:
75
+ class: logging.StreamHandler
76
+ formatter: simple
77
+ stream: ext://sys.stdout
78
+ root:
79
+ level: INFO
80
+ handlers:
81
+ - console
82
+ loggers:
83
+ logging_example:
84
+ level: DEBUG
85
+ disable_existing_loggers: false
86
+ job_logging:
87
+ version: 1
88
+ formatters:
89
+ simple:
90
+ format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
91
+ handlers:
92
+ console:
93
+ class: logging.StreamHandler
94
+ formatter: simple
95
+ stream: ext://sys.stdout
96
+ file:
97
+ class: logging.FileHandler
98
+ formatter: simple
99
+ filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
100
+ root:
101
+ level: INFO
102
+ handlers:
103
+ - console
104
+ - file
105
+ disable_existing_loggers: false
106
+ env: {}
107
+ mode: RUN
108
+ searchpath: []
109
+ callbacks: {}
110
+ output_subdir: .hydra
111
+ overrides:
112
+ hydra:
113
+ - hydra.mode=RUN
114
+ task:
115
+ - model=openvla
116
+ - env=flappy
117
+ - init=bridge
118
+ - mode=single
119
+ - run_id=openvla_flappy_fix_latency_4_exp
120
+ - run_root_dir=/workspace/starVLA/results/Checkpoints
121
+ - seed=42
122
+ - wandb_entity=dongqianyu99-zhejiang-university
123
+ - wandb_project=starVLA_rl_games
124
+ - rl_games.env_eval.enabled=true
125
+ - checkpoint.sync.enabled=false
126
+ - checkpoint.sync.keep_last_n=0
127
+ - checkpoint.local.keep_last_n=2
128
+ - trainer.is_resume=false
129
+ - trainer.pretrained_checkpoint=/workspace/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
130
+ - trainer.resume_step=0
131
+ - trainer.max_train_steps=2500
132
+ - trainer.num_warmup_steps=100
133
+ - trainer.save_interval=100
134
+ - trainer.eval_interval=100
135
+ - trainer.eval_num_batches=20
136
+ - trainer.logging_frequency=10
137
+ - trainer.gradient_accumulation_steps=16
138
+ - trainer.distributed_backend=none
139
+ - trainer.learning_rate.base=2e-05
140
+ - trainer.learning_rate.qwen_vl_interface=1e-05
141
+ - trainer.learning_rate.action_model=0.0001
142
+ - trainer.lr_scheduler_type=cosine_with_min_lr
143
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
144
+ - trainer.loss_scale.vla=1.0
145
+ - trainer.loss_scale.vlm=0.1
146
+ - trainer.max_grad_norm=1.0
147
+ - trainer.weight_decay=0.0
148
+ - trainer.gradient_clipping=1.0
149
+ - trainer.optimizer.name=AdamW
150
+ - trainer.optimizer.betas=[0.9, 0.95]
151
+ - trainer.optimizer.eps=1e-08
152
+ - trainer.optimizer.weight_decay=1e-08
153
+ - trainer.optimizer.fused=false
154
+ - trainer.save_format=pt
155
+ - framework.name=QwenOFT
156
+ - framework.qwenvl.attn_implementation=flash_attention_2
157
+ - framework.qwenvl.enable_gradient_checkpointing=true
158
+ - framework.action_model.action_dim=7
159
+ - framework.action_model.action_env_dim=2
160
+ - framework.action_model.state_dim=7
161
+ - framework.action_model.loss_type=discrete_ce
162
+ - framework.action_model.action_horizon=1
163
+ - framework.action_model.future_action_window_size=0
164
+ - framework.action_model.past_action_window_size=0
165
+ - datasets.vla_data.include_state=true
166
+ - datasets.vla_data.action_type=discrete
167
+ - datasets.vla_data.sequential_step_sampling=false
168
+ - datasets.vla_data.per_device_batch_size=32
169
+ - datasets.vla_data.load_all_data_for_training=true
170
+ - datasets.vla_data.video_backend=torchvision_av
171
+ - datasets.vla_data.data_root_dir=/workspace/starVLA/data/flappy_fix_latency_4
172
+ - datasets.vla_data.data_mix=flappy_train__bridge
173
+ - datasets.vla_data.eval_data_mix=flappy_train__bridge__val
174
+ - framework.qwenvl.base_vlm=/workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
175
+ - rl_games.task=flappy
176
+ - rl_games.model_alias=openvla
177
+ - rl_games.initialization_mode=bridge
178
+ - rl_games.action_carrier=bridge
179
+ - rl_games.env_eval.latency.mode=single
180
+ - rl_games.env_eval.frameskip=1
181
+ - rl_games.env_eval.image_size=224
182
+ - rl_games.env_eval.seed=42
183
+ - rl_games.env_eval.fixed_episode_seeds=true
184
+ - rl_games.env_eval.latency_seed_stride=0
185
+ - rl_games.env_eval.task_seed_stride=100000
186
+ - rl_games.env_eval.latency.values=[0]
187
+ - rl_games.env_eval.latency.prompt_map_path=/workspace/starVLA/data/flappy_fix_latency_4/flappy_train__bridge/latency_prompt_map.json
188
+ - rl_games.env_eval.mid_train.enabled=true
189
+ - rl_games.env_eval.mid_train.interval_steps=100
190
+ - rl_games.env_eval.mid_train.num_episodes=5
191
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
192
+ - rl_games.env_eval.mid_train.latencies=[4]
193
+ - rl_games.env_eval.post_train.enabled=true
194
+ - rl_games.env_eval.post_train.num_episodes=5
195
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
196
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
197
+ job:
198
+ name: train_starvla_hydra
199
+ chdir: false
200
+ override_dirname: checkpoint.local.keep_last_n=2,checkpoint.sync.enabled=false,checkpoint.sync.keep_last_n=0,datasets.vla_data.action_type=discrete,datasets.vla_data.data_mix=flappy_train__bridge,datasets.vla_data.data_root_dir=/workspace/starVLA/data/flappy_fix_latency_4,datasets.vla_data.eval_data_mix=flappy_train__bridge__val,datasets.vla_data.include_state=true,datasets.vla_data.load_all_data_for_training=true,datasets.vla_data.per_device_batch_size=32,datasets.vla_data.sequential_step_sampling=false,datasets.vla_data.video_backend=torchvision_av,env=flappy,framework.action_model.action_dim=7,framework.action_model.action_env_dim=2,framework.action_model.action_horizon=1,framework.action_model.future_action_window_size=0,framework.action_model.loss_type=discrete_ce,framework.action_model.past_action_window_size=0,framework.action_model.state_dim=7,framework.name=QwenOFT,framework.qwenvl.attn_implementation=flash_attention_2,framework.qwenvl.base_vlm=/workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct,framework.qwenvl.enable_gradient_checkpointing=true,init=bridge,mode=single,model=openvla,rl_games.action_carrier=bridge,rl_games.env_eval.enabled=true,rl_games.env_eval.fixed_episode_seeds=true,rl_games.env_eval.frameskip=1,rl_games.env_eval.image_size=224,rl_games.env_eval.latency.mode=single,rl_games.env_eval.latency.prompt_map_path=/workspace/starVLA/data/flappy_fix_latency_4/flappy_train__bridge/latency_prompt_map.json,rl_games.env_eval.latency.values=[0],rl_games.env_eval.latency_seed_stride=0,rl_games.env_eval.mid_train.enabled=true,rl_games.env_eval.mid_train.interval_steps=100,rl_games.env_eval.mid_train.latencies=[4],rl_games.env_eval.mid_train.max_steps_per_episode=3600,rl_games.env_eval.mid_train.num_episodes=5,rl_games.env_eval.post_train.enabled=true,rl_games.env_eval.post_train.latencies=[0,1,2,3,4],rl_games.env_eval.post_train.max_steps_per_episode=3600,rl_games.env_eval.post_train.num_episodes=5,rl_games.env_eval.seed=42,rl_games.env_eval.task_seed_stride=100000,rl_games.initialization_mode=bridge,rl_games.model_alias=openvla,rl_games.task=flappy,run_id=openvla_flappy_fix_latency_4_exp,run_root_dir=/workspace/starVLA/results/Checkpoints,seed=42,trainer.distributed_backend=none,trainer.eval_interval=100,trainer.eval_num_batches=20,trainer.gradient_accumulation_steps=16,trainer.gradient_clipping=1.0,trainer.is_resume=false,trainer.learning_rate.action_model=0.0001,trainer.learning_rate.base=2e-05,trainer.learning_rate.qwen_vl_interface=1e-05,trainer.logging_frequency=10,trainer.loss_scale.vla=1.0,trainer.loss_scale.vlm=0.1,trainer.lr_scheduler_type=cosine_with_min_lr,trainer.max_grad_norm=1.0,trainer.max_train_steps=2500,trainer.num_warmup_steps=100,trainer.optimizer.betas=[0.9,
201
+ 0.95],trainer.optimizer.eps=1e-08,trainer.optimizer.fused=false,trainer.optimizer.name=AdamW,trainer.optimizer.weight_decay=1e-08,trainer.pretrained_checkpoint=/workspace/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt,trainer.resume_step=0,trainer.save_format=pt,trainer.save_interval=100,trainer.scheduler_specific_kwargs.min_lr=1e-06,trainer.weight_decay=0.0,wandb_entity=dongqianyu99-zhejiang-university,wandb_project=starVLA_rl_games
202
+ id: ???
203
+ num: ???
204
+ config_name: train
205
+ env_set: {}
206
+ env_copy: []
207
+ config:
208
+ override_dirname:
209
+ kv_sep: '='
210
+ item_sep: ','
211
+ exclude_keys: []
212
+ runtime:
213
+ version: 1.3.2
214
+ version_base: '1.1'
215
+ cwd: /workspace/starVLA
216
+ config_sources:
217
+ - path: hydra.conf
218
+ schema: pkg
219
+ provider: hydra
220
+ - path: /workspace/starVLA/examples/rl_games/config
221
+ schema: file
222
+ provider: main
223
+ - path: ''
224
+ schema: structured
225
+ provider: schema
226
+ output_dir: /workspace/starVLA/results/Checkpoints/openvla_flappy_fix_latency_4_exp/hydra
227
+ choices:
228
+ checkpoint: default
229
+ mode: single
230
+ init: bridge
231
+ env: flappy
232
+ model: openvla
233
+ hydra/env: default
234
+ hydra/callbacks: null
235
+ hydra/job_logging: default
236
+ hydra/hydra_logging: default
237
+ hydra/hydra_help: default
238
+ hydra/help: default
239
+ hydra/sweeper: basic
240
+ hydra/launcher: basic
241
+ hydra/output: default
242
+ verbose: false
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_exp/hydra/.hydra/overrides.yaml ADDED
@@ -0,0 +1,82 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ - model=openvla
2
+ - env=flappy
3
+ - init=bridge
4
+ - mode=single
5
+ - run_id=openvla_flappy_fix_latency_4_exp
6
+ - run_root_dir=/workspace/starVLA/results/Checkpoints
7
+ - seed=42
8
+ - wandb_entity=dongqianyu99-zhejiang-university
9
+ - wandb_project=starVLA_rl_games
10
+ - rl_games.env_eval.enabled=true
11
+ - checkpoint.sync.enabled=false
12
+ - checkpoint.sync.keep_last_n=0
13
+ - checkpoint.local.keep_last_n=2
14
+ - trainer.is_resume=false
15
+ - trainer.pretrained_checkpoint=/workspace/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
16
+ - trainer.resume_step=0
17
+ - trainer.max_train_steps=2500
18
+ - trainer.num_warmup_steps=100
19
+ - trainer.save_interval=100
20
+ - trainer.eval_interval=100
21
+ - trainer.eval_num_batches=20
22
+ - trainer.logging_frequency=10
23
+ - trainer.gradient_accumulation_steps=16
24
+ - trainer.distributed_backend=none
25
+ - trainer.learning_rate.base=2e-05
26
+ - trainer.learning_rate.qwen_vl_interface=1e-05
27
+ - trainer.learning_rate.action_model=0.0001
28
+ - trainer.lr_scheduler_type=cosine_with_min_lr
29
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
30
+ - trainer.loss_scale.vla=1.0
31
+ - trainer.loss_scale.vlm=0.1
32
+ - trainer.max_grad_norm=1.0
33
+ - trainer.weight_decay=0.0
34
+ - trainer.gradient_clipping=1.0
35
+ - trainer.optimizer.name=AdamW
36
+ - trainer.optimizer.betas=[0.9, 0.95]
37
+ - trainer.optimizer.eps=1e-08
38
+ - trainer.optimizer.weight_decay=1e-08
39
+ - trainer.optimizer.fused=false
40
+ - trainer.save_format=pt
41
+ - framework.name=QwenOFT
42
+ - framework.qwenvl.attn_implementation=flash_attention_2
43
+ - framework.qwenvl.enable_gradient_checkpointing=true
44
+ - framework.action_model.action_dim=7
45
+ - framework.action_model.action_env_dim=2
46
+ - framework.action_model.state_dim=7
47
+ - framework.action_model.loss_type=discrete_ce
48
+ - framework.action_model.action_horizon=1
49
+ - framework.action_model.future_action_window_size=0
50
+ - framework.action_model.past_action_window_size=0
51
+ - datasets.vla_data.include_state=true
52
+ - datasets.vla_data.action_type=discrete
53
+ - datasets.vla_data.sequential_step_sampling=false
54
+ - datasets.vla_data.per_device_batch_size=32
55
+ - datasets.vla_data.load_all_data_for_training=true
56
+ - datasets.vla_data.video_backend=torchvision_av
57
+ - datasets.vla_data.data_root_dir=/workspace/starVLA/data/flappy_fix_latency_4
58
+ - datasets.vla_data.data_mix=flappy_train__bridge
59
+ - datasets.vla_data.eval_data_mix=flappy_train__bridge__val
60
+ - framework.qwenvl.base_vlm=/workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
61
+ - rl_games.task=flappy
62
+ - rl_games.model_alias=openvla
63
+ - rl_games.initialization_mode=bridge
64
+ - rl_games.action_carrier=bridge
65
+ - rl_games.env_eval.latency.mode=single
66
+ - rl_games.env_eval.frameskip=1
67
+ - rl_games.env_eval.image_size=224
68
+ - rl_games.env_eval.seed=42
69
+ - rl_games.env_eval.fixed_episode_seeds=true
70
+ - rl_games.env_eval.latency_seed_stride=0
71
+ - rl_games.env_eval.task_seed_stride=100000
72
+ - rl_games.env_eval.latency.values=[0]
73
+ - rl_games.env_eval.latency.prompt_map_path=/workspace/starVLA/data/flappy_fix_latency_4/flappy_train__bridge/latency_prompt_map.json
74
+ - rl_games.env_eval.mid_train.enabled=true
75
+ - rl_games.env_eval.mid_train.interval_steps=100
76
+ - rl_games.env_eval.mid_train.num_episodes=5
77
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
78
+ - rl_games.env_eval.mid_train.latencies=[4]
79
+ - rl_games.env_eval.post_train.enabled=true
80
+ - rl_games.env_eval.post_train.num_episodes=5
81
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
82
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_exp/hydra/train_starvla_hydra.log ADDED
The diff for this file is too large to render. See raw diff
 
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_exp/summary.jsonl ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"steps": 100}
2
+ {"steps": 200}
3
+ {"steps": 300}
4
+ {"steps": 400}
5
+ {"steps": 500}
6
+ {"steps": 600}
7
+ {"steps": 700}
8
+ {"steps": 800}
9
+ {"steps": 900}
10
+ {"steps": 1000}
11
+ {"steps": 1100}
12
+ {"steps": 1200}
13
+ {"steps": 1300}
14
+ {"steps": 1400}
15
+ {"steps": 1500}
16
+ {"steps": 1600}
17
+ {"steps": 1700}
18
+ {"steps": 1800}
19
+ {"steps": 1900}
20
+ {"steps": 2000}
21
+ {"steps": 2100}
22
+ {"steps": 2200}
23
+ {"steps": 2300}
24
+ {"steps": 2400}
25
+ {"steps": 2500}
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_small/checkpoints/best_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f893f7feb006ca5eee1dbc84ac222cf50c7b0dddcb7d572cf6d52135605bb0b9
3
+ size 9138230516
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_small/checkpoints/steps_1500_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e3af4a77ec1ff9bb6d5cc58129a8e8b5b060d80d978228564561e188791d8029
3
+ size 9138230516
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_small/config.full.yaml ADDED
@@ -0,0 +1,187 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ name: QwenOFT
3
+ qwenvl:
4
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
5
+ attn_implementation: flash_attention_2
6
+ enable_gradient_checkpointing: true
7
+ action_model:
8
+ action_model_type: MLP
9
+ action_dim: 7
10
+ action_hidden_dim: 2560
11
+ future_action_window_size: 0
12
+ past_action_window_size: 0
13
+ loss_type: discrete_ce
14
+ state_dim: 7
15
+ action_horizon: 1
16
+ action_env_dim: 2
17
+ datasets:
18
+ vla_data:
19
+ dataset_py: lerobot_datasets
20
+ include_state: true
21
+ data_root_dir: /workspace/starVLA/data/flappy_fix_latency_4_small
22
+ data_mix: flappy_train__bridge
23
+ eval_data_mix: flappy_train__bridge__val
24
+ custom_mixtures_path: null
25
+ action_type: discrete
26
+ sequential_step_sampling: false
27
+ per_device_batch_size: 32
28
+ load_all_data_for_training: true
29
+ obs_image_size: null
30
+ video_backend: torchvision_av
31
+ dataset:
32
+ source_hf: ''
33
+ converted_name: flappy_train
34
+ single_source_hf: ''
35
+ mixed_source_hf: ''
36
+ single_converted_name: flappy_train
37
+ mixed_converted_name: flappy_mixed_latency_train
38
+ single_latency_filter: null
39
+ mixed_latency_filter: null
40
+ force_download: false
41
+ setup_force: false
42
+ verify_rows: 200
43
+ max_episodes: null
44
+ episodes_per_latency: null
45
+ latency_filter: null
46
+ debug_subset:
47
+ enabled: false
48
+ max_episodes: 5
49
+ suffix: debug
50
+ base_model:
51
+ repo_id: Qwen/Qwen3-VL-4B-Instruct
52
+ initialization:
53
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
54
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
55
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
56
+ trainer:
57
+ max_train_steps: 1500
58
+ num_warmup_steps: 100
59
+ save_interval: 100
60
+ eval_interval: 100
61
+ eval_num_batches: 20
62
+ learning_rate:
63
+ base: 2.0e-05
64
+ qwen_vl_interface: 1.0e-05
65
+ action_model: 0.0001
66
+ lr_scheduler_type: cosine_with_min_lr
67
+ scheduler_specific_kwargs:
68
+ min_lr: 1.0e-06
69
+ freeze_modules: ''
70
+ loss_scale:
71
+ vla: 1.0
72
+ vlm: 0.1
73
+ max_grad_norm: 1.0
74
+ weight_decay: 0.0
75
+ logging_frequency: 10
76
+ gradient_clipping: 1.0
77
+ gradient_accumulation_steps: 16
78
+ distributed_backend: none
79
+ is_resume: false
80
+ pretrained_checkpoint: null
81
+ resume_step: 0
82
+ reload_modules: null
83
+ optimizer:
84
+ name: AdamW
85
+ betas:
86
+ - 0.9
87
+ - 0.95
88
+ eps: 1.0e-08
89
+ weight_decay: 1.0e-08
90
+ fused: false
91
+ save_format: pt
92
+ workspace_dir: WORKSPACE_DIR
93
+ run_root_dir: /workspace/starVLA/results/Checkpoints
94
+ seed: 42
95
+ wandb_entity: dongqianyu99-zhejiang-university
96
+ wandb_project: starVLA_rl_games
97
+ auth:
98
+ env_file: null
99
+ hf_token_env: HF_TOKEN
100
+ wandb_api_key_env: WANDB_API_KEY
101
+ paths:
102
+ run_root_dir: results/Checkpoints
103
+ dataset_local_dir: playground/Datasets/rl_games
104
+ dataset_cache_dir: null
105
+ base_model_dir: playground/Pretrained_models/Qwen3-VL-4B-Instruct
106
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
107
+ launch:
108
+ use_accelerate: true
109
+ gpus: null
110
+ num_processes: 1
111
+ dry_run: false
112
+ conda:
113
+ enabled: true
114
+ env_name: null
115
+ rl_games:
116
+ model_alias: openvla
117
+ env_eval:
118
+ image_size: 224
119
+ frameskip: 1
120
+ seed: 42
121
+ fixed_episode_seeds: true
122
+ latency_seed_stride: 0
123
+ task_seed_stride: 0
124
+ task_description: ''
125
+ enabled: true
126
+ distributed_mode: none
127
+ vectorized:
128
+ enabled: false
129
+ batch_size: 1
130
+ latency:
131
+ prompt_map_path: /workspace/starVLA/data/flappy_fix_latency_4_small/flappy_train__bridge/latency_prompt_map.json
132
+ mode: single
133
+ values:
134
+ - 0
135
+ mid_train:
136
+ enabled: true
137
+ interval_steps: 100
138
+ latencies:
139
+ - 4
140
+ num_episodes: 5
141
+ max_steps_per_episode: 3600
142
+ post_train:
143
+ enabled: true
144
+ latencies:
145
+ - 0
146
+ - 1
147
+ - 2
148
+ - 3
149
+ - 4
150
+ num_episodes: 5
151
+ max_steps_per_episode: 3600
152
+ task: flappy
153
+ initialization_mode: bridge
154
+ action_carrier: bridge
155
+ model: openvla
156
+ env: flappy
157
+ init: bridge
158
+ bridge_base_model:
159
+ repo_id:
160
+ openvla: Qwen/Qwen3-VL-4B-Instruct
161
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
162
+ pi05: Qwen/Qwen3-VL-4B-Instruct
163
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
164
+ local_dir:
165
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
166
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
167
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
168
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
169
+ mode: single
170
+ checkpoint:
171
+ load: auto
172
+ hf_repo_id: null
173
+ save_best_model: true
174
+ save_pt_file: false
175
+ local:
176
+ keep_last_n: 1
177
+ sync:
178
+ enabled: false
179
+ repo_id: null
180
+ keep_last_n: 0
181
+ sync_every_n_checkpoints: 1
182
+ resume_policy: local_latest
183
+ run_id: openvla_flappy_fix_latency_4_small
184
+ output_dir: /workspace/starVLA/results/Checkpoints/openvla_flappy_fix_latency_4_small
185
+ config_yaml: null
186
+ is_debug: false
187
+ version_id: '0.21'
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_small/config.yaml ADDED
@@ -0,0 +1,89 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ checkpoint:
2
+ local:
3
+ keep_last_n: 1
4
+ save_best_model: true
5
+ save_pt_file: false
6
+ sync:
7
+ enabled: false
8
+ keep_last_n: 0
9
+ repo_id: null
10
+ datasets:
11
+ vla_data:
12
+ data_mix: flappy_train__bridge
13
+ dataset_py: lerobot_datasets
14
+ eval_data_mix: flappy_train__bridge__val
15
+ include_state: true
16
+ obs_image_size: null
17
+ per_device_batch_size: 32
18
+ framework:
19
+ action_model:
20
+ action_dim: 7
21
+ action_env_dim: 2
22
+ action_hidden_dim: 2560
23
+ action_horizon: 1
24
+ action_model_type: MLP
25
+ loss_type: discrete_ce
26
+ state_dim: 7
27
+ name: QwenOFT
28
+ qwenvl:
29
+ attn_implementation: flash_attention_2
30
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
31
+ enable_gradient_checkpointing: true
32
+ output_dir: /workspace/starVLA/results/Checkpoints/openvla_flappy_fix_latency_4_small
33
+ rl_games:
34
+ env_eval:
35
+ distributed_mode: none
36
+ enabled: true
37
+ fixed_episode_seeds: true
38
+ frameskip: 1
39
+ image_size: 224
40
+ latency:
41
+ prompt_map_path: /workspace/starVLA/data/flappy_fix_latency_4_small/flappy_train__bridge/latency_prompt_map.json
42
+ latency_seed_stride: 0
43
+ mid_train:
44
+ enabled: true
45
+ interval_steps: 100
46
+ latencies:
47
+ - 4
48
+ max_steps_per_episode: 3600
49
+ num_episodes: 5
50
+ seed: 42
51
+ task_description: ''
52
+ task_seed_stride: 0
53
+ vectorized:
54
+ enabled: false
55
+ model_alias: openvla
56
+ task: flappy
57
+ run_id: openvla_flappy_fix_latency_4_small
58
+ run_root_dir: /workspace/starVLA/results/Checkpoints
59
+ seed: 42
60
+ trainer:
61
+ distributed_backend: none
62
+ eval_interval: 100
63
+ eval_num_batches: 20
64
+ freeze_modules: ''
65
+ gradient_accumulation_steps: 16
66
+ gradient_clipping: 1.0
67
+ is_resume: false
68
+ learning_rate:
69
+ action_model: 0.0001
70
+ base: 2.0e-05
71
+ qwen_vl_interface: 1.0e-05
72
+ logging_frequency: 10
73
+ lr_scheduler_type: cosine_with_min_lr
74
+ max_train_steps: 1500
75
+ num_warmup_steps: 100
76
+ optimizer:
77
+ betas:
78
+ - 0.9
79
+ - 0.95
80
+ eps: 1.0e-08
81
+ fused: false
82
+ weight_decay: 1.0e-08
83
+ pretrained_checkpoint: null
84
+ reload_modules: null
85
+ save_interval: 100
86
+ scheduler_specific_kwargs:
87
+ min_lr: 1.0e-06
88
+ wandb_entity: dongqianyu99-zhejiang-university
89
+ wandb_project: starVLA_rl_games
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_small/dataset_statistics.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.8208409547805786,
6
+ 0.1791590452194214,
7
+ 0.0,
8
+ 0.0,
9
+ 0.0,
10
+ 0.0,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.38357409834861755,
15
+ 0.38357409834861755,
16
+ 0.0,
17
+ 0.0,
18
+ 0.0,
19
+ 0.0,
20
+ 0.0
21
+ ],
22
+ "max": [
23
+ 1.0,
24
+ 1.0,
25
+ 0.0,
26
+ 0.0,
27
+ 0.0,
28
+ 0.0,
29
+ 0.0
30
+ ],
31
+ "min": [
32
+ 0.0,
33
+ 0.0,
34
+ 0.0,
35
+ 0.0,
36
+ 0.0,
37
+ 0.0,
38
+ 0.0
39
+ ],
40
+ "q01": [
41
+ 0.0,
42
+ 0.0,
43
+ 0.0,
44
+ 0.0,
45
+ 0.0,
46
+ 0.0,
47
+ 0.0
48
+ ],
49
+ "q99": [
50
+ 1.0,
51
+ 1.0,
52
+ 0.0,
53
+ 0.0,
54
+ 0.0,
55
+ 0.0,
56
+ 0.0
57
+ ],
58
+ "mask": [
59
+ true,
60
+ true,
61
+ true,
62
+ true,
63
+ true,
64
+ true,
65
+ true
66
+ ]
67
+ },
68
+ "state": {
69
+ "mean": [
70
+ 0.0,
71
+ 0.0,
72
+ 0.0,
73
+ 0.0,
74
+ 0.0,
75
+ 0.0,
76
+ 0.0
77
+ ],
78
+ "std": [
79
+ 0.0,
80
+ 0.0,
81
+ 0.0,
82
+ 0.0,
83
+ 0.0,
84
+ 0.0,
85
+ 0.0
86
+ ],
87
+ "max": [
88
+ 0.0,
89
+ 0.0,
90
+ 0.0,
91
+ 0.0,
92
+ 0.0,
93
+ 0.0,
94
+ 0.0
95
+ ],
96
+ "min": [
97
+ 0.0,
98
+ 0.0,
99
+ 0.0,
100
+ 0.0,
101
+ 0.0,
102
+ 0.0,
103
+ 0.0
104
+ ],
105
+ "q01": [
106
+ 0.0,
107
+ 0.0,
108
+ 0.0,
109
+ 0.0,
110
+ 0.0,
111
+ 0.0,
112
+ 0.0
113
+ ],
114
+ "q99": [
115
+ 0.0,
116
+ 0.0,
117
+ 0.0,
118
+ 0.0,
119
+ 0.0,
120
+ 0.0,
121
+ 0.0
122
+ ]
123
+ },
124
+ "num_transitions": 91896,
125
+ "num_trajectories": 27
126
+ }
127
+ }
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_small/dataset_statistics_eval.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.8154629468917847,
6
+ 0.18453703820705414,
7
+ 0.0,
8
+ 0.0,
9
+ 0.0,
10
+ 0.0,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.3879301846027374,
15
+ 0.3879301846027374,
16
+ 0.0,
17
+ 0.0,
18
+ 0.0,
19
+ 0.0,
20
+ 0.0
21
+ ],
22
+ "max": [
23
+ 1.0,
24
+ 1.0,
25
+ 0.0,
26
+ 0.0,
27
+ 0.0,
28
+ 0.0,
29
+ 0.0
30
+ ],
31
+ "min": [
32
+ 0.0,
33
+ 0.0,
34
+ 0.0,
35
+ 0.0,
36
+ 0.0,
37
+ 0.0,
38
+ 0.0
39
+ ],
40
+ "q01": [
41
+ 0.0,
42
+ 0.0,
43
+ 0.0,
44
+ 0.0,
45
+ 0.0,
46
+ 0.0,
47
+ 0.0
48
+ ],
49
+ "q99": [
50
+ 1.0,
51
+ 1.0,
52
+ 0.0,
53
+ 0.0,
54
+ 0.0,
55
+ 0.0,
56
+ 0.0
57
+ ],
58
+ "mask": [
59
+ true,
60
+ true,
61
+ true,
62
+ true,
63
+ true,
64
+ true,
65
+ true
66
+ ]
67
+ },
68
+ "state": {
69
+ "mean": [
70
+ 0.0,
71
+ 0.0,
72
+ 0.0,
73
+ 0.0,
74
+ 0.0,
75
+ 0.0,
76
+ 0.0
77
+ ],
78
+ "std": [
79
+ 0.0,
80
+ 0.0,
81
+ 0.0,
82
+ 0.0,
83
+ 0.0,
84
+ 0.0,
85
+ 0.0
86
+ ],
87
+ "max": [
88
+ 0.0,
89
+ 0.0,
90
+ 0.0,
91
+ 0.0,
92
+ 0.0,
93
+ 0.0,
94
+ 0.0
95
+ ],
96
+ "min": [
97
+ 0.0,
98
+ 0.0,
99
+ 0.0,
100
+ 0.0,
101
+ 0.0,
102
+ 0.0,
103
+ 0.0
104
+ ],
105
+ "q01": [
106
+ 0.0,
107
+ 0.0,
108
+ 0.0,
109
+ 0.0,
110
+ 0.0,
111
+ 0.0,
112
+ 0.0
113
+ ],
114
+ "q99": [
115
+ 0.0,
116
+ 0.0,
117
+ 0.0,
118
+ 0.0,
119
+ 0.0,
120
+ 0.0,
121
+ 0.0
122
+ ]
123
+ },
124
+ "num_transitions": 10800,
125
+ "num_trajectories": 3
126
+ }
127
+ }
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_small/hydra/.hydra/config.yaml ADDED
@@ -0,0 +1,185 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ qwenvl:
3
+ base_vlm: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct
4
+ attn_implementation: flash_attention_2
5
+ enable_gradient_checkpointing: true
6
+ action_model:
7
+ state_dim: 7
8
+ loss_type: discrete_ce
9
+ action_horizon: 1
10
+ future_action_window_size: 0
11
+ past_action_window_size: 0
12
+ action_dim: 7
13
+ action_env_dim: 2
14
+ name: QwenOFT
15
+ datasets:
16
+ vla_data:
17
+ dataset_py: lerobot_datasets
18
+ include_state: true
19
+ data_root_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_4_small
20
+ data_mix: flappy_train__bridge
21
+ eval_data_mix: flappy_train__bridge__val
22
+ custom_mixtures_path: null
23
+ action_type: discrete
24
+ sequential_step_sampling: false
25
+ per_device_batch_size: 32
26
+ load_all_data_for_training: true
27
+ obs_image_size: null
28
+ video_backend: torchvision_av
29
+ dataset:
30
+ source_hf: ${dataset.single_source_hf}
31
+ converted_name: ${dataset.single_converted_name}
32
+ single_source_hf: ''
33
+ mixed_source_hf: ''
34
+ single_converted_name: flappy_train
35
+ mixed_converted_name: flappy_mixed_latency_train
36
+ single_latency_filter: null
37
+ mixed_latency_filter: null
38
+ force_download: false
39
+ setup_force: false
40
+ verify_rows: 200
41
+ max_episodes: null
42
+ episodes_per_latency: null
43
+ latency_filter: ${dataset.single_latency_filter}
44
+ debug_subset:
45
+ enabled: false
46
+ max_episodes: 5
47
+ suffix: debug
48
+ base_model:
49
+ repo_id: ${bridge_base_model.repo_id.${model}}
50
+ initialization:
51
+ checkpoint_local_dir: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
52
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
53
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
54
+ trainer:
55
+ max_train_steps: 1500
56
+ num_warmup_steps: 100
57
+ save_interval: 100
58
+ eval_interval: 100
59
+ eval_num_batches: 20
60
+ learning_rate:
61
+ base: 2.0e-05
62
+ qwen_vl_interface: 1.0e-05
63
+ action_model: 0.0001
64
+ lr_scheduler_type: cosine_with_min_lr
65
+ scheduler_specific_kwargs:
66
+ min_lr: 1.0e-06
67
+ freeze_modules: ''
68
+ loss_scale:
69
+ vla: 1.0
70
+ vlm: 0.1
71
+ max_grad_norm: 1.0
72
+ weight_decay: 0.0
73
+ logging_frequency: 10
74
+ gradient_clipping: 1.0
75
+ gradient_accumulation_steps: 16
76
+ distributed_backend: none
77
+ is_resume: false
78
+ pretrained_checkpoint: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
79
+ resume_step: 0
80
+ reload_modules: null
81
+ optimizer:
82
+ name: AdamW
83
+ betas:
84
+ - 0.9
85
+ - 0.95
86
+ eps: 1.0e-08
87
+ weight_decay: 1.0e-08
88
+ fused: false
89
+ save_format: pt
90
+ workspace_dir: WORKSPACE_DIR
91
+ run_root_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints
92
+ seed: 42
93
+ wandb_entity: dongqianyu99-zhejiang-university
94
+ wandb_project: starVLA_rl_games
95
+ auth:
96
+ env_file: null
97
+ hf_token_env: HF_TOKEN
98
+ wandb_api_key_env: WANDB_API_KEY
99
+ paths:
100
+ run_root_dir: results/Checkpoints
101
+ dataset_local_dir: playground/Datasets/rl_games
102
+ dataset_cache_dir: null
103
+ base_model_dir: ${bridge_base_model.local_dir.${model}}
104
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
105
+ launch:
106
+ use_accelerate: true
107
+ gpus: null
108
+ num_processes: 1
109
+ dry_run: false
110
+ conda:
111
+ enabled: true
112
+ env_name: null
113
+ rl_games:
114
+ model_alias: openvla
115
+ env_eval:
116
+ image_size: 224
117
+ frameskip: 1
118
+ seed: 42
119
+ fixed_episode_seeds: true
120
+ latency_seed_stride: 0
121
+ task_seed_stride: 0
122
+ task_description: ''
123
+ enabled: true
124
+ distributed_mode: none
125
+ vectorized:
126
+ enabled: false
127
+ batch_size: 1
128
+ latency:
129
+ prompt_map_path: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_4_small/flappy_train__bridge/latency_prompt_map.json
130
+ mode: single
131
+ values:
132
+ - 0
133
+ mid_train:
134
+ enabled: true
135
+ interval_steps: 100
136
+ latencies:
137
+ - 4
138
+ num_episodes: 5
139
+ max_steps_per_episode: 3600
140
+ post_train:
141
+ enabled: true
142
+ latencies:
143
+ - 0
144
+ - 1
145
+ - 2
146
+ - 3
147
+ - 4
148
+ num_episodes: 5
149
+ max_steps_per_episode: 3600
150
+ task: flappy
151
+ initialization_mode: bridge
152
+ action_carrier: bridge
153
+ model: openvla
154
+ env: flappy
155
+ init: bridge
156
+ bridge_base_model:
157
+ repo_id:
158
+ openvla: Qwen/Qwen3-VL-4B-Instruct
159
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
160
+ pi05: Qwen/Qwen3-VL-4B-Instruct
161
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
162
+ local_dir:
163
+ openvla: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
164
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
165
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
166
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
167
+ mode: single
168
+ checkpoint:
169
+ load: auto
170
+ hf_repo_id: null
171
+ save_best_model: true
172
+ save_pt_file: false
173
+ local:
174
+ keep_last_n: 1
175
+ sync:
176
+ enabled: false
177
+ repo_id: null
178
+ keep_last_n: 0
179
+ sync_every_n_checkpoints: 1
180
+ resume_policy: local_latest
181
+ run_id: openvla_flappy_fix_latency_4_small
182
+ output_dir: null
183
+ config_yaml: null
184
+ is_debug: false
185
+ version_id: '0.21'
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_small/hydra/.hydra/hydra.yaml ADDED
@@ -0,0 +1,246 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ hydra:
2
+ run:
3
+ dir: ${run_root_dir}/${run_id}/hydra
4
+ sweep:
5
+ dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
6
+ subdir: ${hydra.job.num}
7
+ launcher:
8
+ _target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
9
+ sweeper:
10
+ _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
11
+ max_batch_size: null
12
+ params: null
13
+ help:
14
+ app_name: ${hydra.job.name}
15
+ header: '${hydra.help.app_name} is powered by Hydra.
16
+
17
+ '
18
+ footer: 'Powered by Hydra (https://hydra.cc)
19
+
20
+ Use --hydra-help to view Hydra specific help
21
+
22
+ '
23
+ template: '${hydra.help.header}
24
+
25
+ == Configuration groups ==
26
+
27
+ Compose your configuration from those groups (group=option)
28
+
29
+
30
+ $APP_CONFIG_GROUPS
31
+
32
+
33
+ == Config ==
34
+
35
+ Override anything in the config (foo.bar=value)
36
+
37
+
38
+ $CONFIG
39
+
40
+
41
+ ${hydra.help.footer}
42
+
43
+ '
44
+ hydra_help:
45
+ template: 'Hydra (${hydra.runtime.version})
46
+
47
+ See https://hydra.cc for more info.
48
+
49
+
50
+ == Flags ==
51
+
52
+ $FLAGS_HELP
53
+
54
+
55
+ == Configuration groups ==
56
+
57
+ Compose your configuration from those groups (For example, append hydra/job_logging=disabled
58
+ to command line)
59
+
60
+
61
+ $HYDRA_CONFIG_GROUPS
62
+
63
+
64
+ Use ''--cfg hydra'' to Show the Hydra config.
65
+
66
+ '
67
+ hydra_help: ???
68
+ hydra_logging:
69
+ version: 1
70
+ formatters:
71
+ simple:
72
+ format: '[%(asctime)s][HYDRA] %(message)s'
73
+ handlers:
74
+ console:
75
+ class: logging.StreamHandler
76
+ formatter: simple
77
+ stream: ext://sys.stdout
78
+ root:
79
+ level: INFO
80
+ handlers:
81
+ - console
82
+ loggers:
83
+ logging_example:
84
+ level: DEBUG
85
+ disable_existing_loggers: false
86
+ job_logging:
87
+ version: 1
88
+ formatters:
89
+ simple:
90
+ format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
91
+ handlers:
92
+ console:
93
+ class: logging.StreamHandler
94
+ formatter: simple
95
+ stream: ext://sys.stdout
96
+ file:
97
+ class: logging.FileHandler
98
+ formatter: simple
99
+ filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
100
+ root:
101
+ level: INFO
102
+ handlers:
103
+ - console
104
+ - file
105
+ disable_existing_loggers: false
106
+ env: {}
107
+ mode: RUN
108
+ searchpath: []
109
+ callbacks: {}
110
+ output_subdir: .hydra
111
+ overrides:
112
+ hydra:
113
+ - hydra.mode=RUN
114
+ task:
115
+ - model=openvla
116
+ - env=flappy
117
+ - init=bridge
118
+ - mode=single
119
+ - run_id=openvla_flappy_fix_latency_4_small
120
+ - run_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints
121
+ - seed=42
122
+ - wandb_entity=dongqianyu99-zhejiang-university
123
+ - wandb_project=starVLA_rl_games
124
+ - rl_games.env_eval.enabled=true
125
+ - checkpoint.sync.enabled=false
126
+ - checkpoint.sync.keep_last_n=0
127
+ - checkpoint.local.keep_last_n=1
128
+ - checkpoint.save_best_model=true
129
+ - checkpoint.save_pt_file=false
130
+ - trainer.is_resume=false
131
+ - trainer.pretrained_checkpoint=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
132
+ - trainer.resume_step=0
133
+ - trainer.max_train_steps=1500
134
+ - trainer.num_warmup_steps=100
135
+ - trainer.save_interval=100
136
+ - trainer.eval_interval=100
137
+ - trainer.eval_num_batches=20
138
+ - trainer.logging_frequency=10
139
+ - trainer.gradient_accumulation_steps=16
140
+ - trainer.distributed_backend=none
141
+ - trainer.learning_rate.base=2e-05
142
+ - trainer.learning_rate.qwen_vl_interface=1e-05
143
+ - trainer.learning_rate.action_model=0.0001
144
+ - trainer.lr_scheduler_type=cosine_with_min_lr
145
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
146
+ - trainer.loss_scale.vla=1.0
147
+ - trainer.loss_scale.vlm=0.1
148
+ - trainer.max_grad_norm=1.0
149
+ - trainer.weight_decay=0.0
150
+ - trainer.gradient_clipping=1.0
151
+ - trainer.optimizer.name=AdamW
152
+ - trainer.optimizer.betas=[0.9, 0.95]
153
+ - trainer.optimizer.eps=1e-08
154
+ - trainer.optimizer.weight_decay=1e-08
155
+ - trainer.optimizer.fused=false
156
+ - trainer.save_format=pt
157
+ - framework.name=QwenOFT
158
+ - framework.qwenvl.attn_implementation=flash_attention_2
159
+ - framework.qwenvl.enable_gradient_checkpointing=true
160
+ - framework.action_model.action_dim=7
161
+ - framework.action_model.action_env_dim=2
162
+ - framework.action_model.state_dim=7
163
+ - framework.action_model.loss_type=discrete_ce
164
+ - framework.action_model.action_horizon=1
165
+ - framework.action_model.future_action_window_size=0
166
+ - framework.action_model.past_action_window_size=0
167
+ - datasets.vla_data.include_state=true
168
+ - datasets.vla_data.action_type=discrete
169
+ - datasets.vla_data.sequential_step_sampling=false
170
+ - datasets.vla_data.per_device_batch_size=32
171
+ - datasets.vla_data.load_all_data_for_training=true
172
+ - datasets.vla_data.video_backend=torchvision_av
173
+ - datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_4_small
174
+ - datasets.vla_data.data_mix=flappy_train__bridge
175
+ - datasets.vla_data.eval_data_mix=flappy_train__bridge__val
176
+ - framework.qwenvl.base_vlm=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct
177
+ - rl_games.task=flappy
178
+ - rl_games.model_alias=openvla
179
+ - rl_games.initialization_mode=bridge
180
+ - rl_games.action_carrier=bridge
181
+ - rl_games.env_eval.distributed_mode=none
182
+ - rl_games.env_eval.latency.mode=single
183
+ - rl_games.env_eval.frameskip=1
184
+ - rl_games.env_eval.image_size=224
185
+ - rl_games.env_eval.seed=42
186
+ - rl_games.env_eval.fixed_episode_seeds=true
187
+ - rl_games.env_eval.latency_seed_stride=0
188
+ - rl_games.env_eval.task_seed_stride=0
189
+ - rl_games.env_eval.latency.values=[0]
190
+ - rl_games.env_eval.latency.prompt_map_path=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_4_small/flappy_train__bridge/latency_prompt_map.json
191
+ - rl_games.env_eval.mid_train.enabled=true
192
+ - rl_games.env_eval.mid_train.interval_steps=100
193
+ - rl_games.env_eval.mid_train.num_episodes=5
194
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
195
+ - rl_games.env_eval.mid_train.latencies=[4]
196
+ - rl_games.env_eval.post_train.enabled=true
197
+ - rl_games.env_eval.post_train.num_episodes=5
198
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
199
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
200
+ job:
201
+ name: train_starvla_hydra
202
+ chdir: false
203
+ override_dirname: checkpoint.local.keep_last_n=1,checkpoint.save_best_model=true,checkpoint.save_pt_file=false,checkpoint.sync.enabled=false,checkpoint.sync.keep_last_n=0,datasets.vla_data.action_type=discrete,datasets.vla_data.data_mix=flappy_train__bridge,datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_4_small,datasets.vla_data.eval_data_mix=flappy_train__bridge__val,datasets.vla_data.include_state=true,datasets.vla_data.load_all_data_for_training=true,datasets.vla_data.per_device_batch_size=32,datasets.vla_data.sequential_step_sampling=false,datasets.vla_data.video_backend=torchvision_av,env=flappy,framework.action_model.action_dim=7,framework.action_model.action_env_dim=2,framework.action_model.action_horizon=1,framework.action_model.future_action_window_size=0,framework.action_model.loss_type=discrete_ce,framework.action_model.past_action_window_size=0,framework.action_model.state_dim=7,framework.name=QwenOFT,framework.qwenvl.attn_implementation=flash_attention_2,framework.qwenvl.base_vlm=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct,framework.qwenvl.enable_gradient_checkpointing=true,init=bridge,mode=single,model=openvla,rl_games.action_carrier=bridge,rl_games.env_eval.distributed_mode=none,rl_games.env_eval.enabled=true,rl_games.env_eval.fixed_episode_seeds=true,rl_games.env_eval.frameskip=1,rl_games.env_eval.image_size=224,rl_games.env_eval.latency.mode=single,rl_games.env_eval.latency.prompt_map_path=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_4_small/flappy_train__bridge/latency_prompt_map.json,rl_games.env_eval.latency.values=[0],rl_games.env_eval.latency_seed_stride=0,rl_games.env_eval.mid_train.enabled=true,rl_games.env_eval.mid_train.interval_steps=100,rl_games.env_eval.mid_train.latencies=[4],rl_games.env_eval.mid_train.max_steps_per_episode=3600,rl_games.env_eval.mid_train.num_episodes=5,rl_games.env_eval.post_train.enabled=true,rl_games.env_eval.post_train.latencies=[0,1,2,3,4],rl_games.env_eval.post_train.max_steps_per_episode=3600,rl_games.env_eval.post_train.num_episodes=5,rl_games.env_eval.seed=42,rl_games.env_eval.task_seed_stride=0,rl_games.initialization_mode=bridge,rl_games.model_alias=openvla,rl_games.task=flappy,run_id=openvla_flappy_fix_latency_4_small,run_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints,seed=42,trainer.distributed_backend=none,trainer.eval_interval=100,trainer.eval_num_batches=20,trainer.gradient_accumulation_steps=16,trainer.gradient_clipping=1.0,trainer.is_resume=false,trainer.learning_rate.action_model=0.0001,trainer.learning_rate.base=2e-05,trainer.learning_rate.qwen_vl_interface=1e-05,trainer.logging_frequency=10,trainer.loss_scale.vla=1.0,trainer.loss_scale.vlm=0.1,trainer.lr_scheduler_type=cosine_with_min_lr,trainer.max_grad_norm=1.0,trainer.max_train_steps=1500,trainer.num_warmup_steps=100,trainer.optimizer.betas=[0.9,
204
+ 0.95],trainer.optimizer.eps=1e-08,trainer.optimizer.fused=false,trainer.optimizer.name=AdamW,trainer.optimizer.weight_decay=1e-08,trainer.pretrained_checkpoint=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt,trainer.resume_step=0,trainer.save_format=pt,trainer.save_interval=100,trainer.scheduler_specific_kwargs.min_lr=1e-06,trainer.weight_decay=0.0,wandb_entity=dongqianyu99-zhejiang-university,wandb_project=starVLA_rl_games
205
+ id: ???
206
+ num: ???
207
+ config_name: train
208
+ env_set: {}
209
+ env_copy: []
210
+ config:
211
+ override_dirname:
212
+ kv_sep: '='
213
+ item_sep: ','
214
+ exclude_keys: []
215
+ runtime:
216
+ version: 1.3.2
217
+ version_base: '1.1'
218
+ cwd: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA
219
+ config_sources:
220
+ - path: hydra.conf
221
+ schema: pkg
222
+ provider: hydra
223
+ - path: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA/examples/rl_games/config
224
+ schema: file
225
+ provider: main
226
+ - path: ''
227
+ schema: structured
228
+ provider: schema
229
+ output_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints/openvla_flappy_fix_latency_4_small/hydra
230
+ choices:
231
+ cross_task_setup: null
232
+ checkpoint: default
233
+ mode: single
234
+ init: bridge
235
+ env: flappy
236
+ model: openvla
237
+ hydra/env: default
238
+ hydra/callbacks: null
239
+ hydra/job_logging: default
240
+ hydra/hydra_logging: default
241
+ hydra/hydra_help: default
242
+ hydra/help: default
243
+ hydra/sweeper: basic
244
+ hydra/launcher: basic
245
+ hydra/output: default
246
+ verbose: false
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_small/hydra/.hydra/overrides.yaml ADDED
@@ -0,0 +1,85 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ - model=openvla
2
+ - env=flappy
3
+ - init=bridge
4
+ - mode=single
5
+ - run_id=openvla_flappy_fix_latency_4_small
6
+ - run_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints
7
+ - seed=42
8
+ - wandb_entity=dongqianyu99-zhejiang-university
9
+ - wandb_project=starVLA_rl_games
10
+ - rl_games.env_eval.enabled=true
11
+ - checkpoint.sync.enabled=false
12
+ - checkpoint.sync.keep_last_n=0
13
+ - checkpoint.local.keep_last_n=1
14
+ - checkpoint.save_best_model=true
15
+ - checkpoint.save_pt_file=false
16
+ - trainer.is_resume=false
17
+ - trainer.pretrained_checkpoint=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
18
+ - trainer.resume_step=0
19
+ - trainer.max_train_steps=1500
20
+ - trainer.num_warmup_steps=100
21
+ - trainer.save_interval=100
22
+ - trainer.eval_interval=100
23
+ - trainer.eval_num_batches=20
24
+ - trainer.logging_frequency=10
25
+ - trainer.gradient_accumulation_steps=16
26
+ - trainer.distributed_backend=none
27
+ - trainer.learning_rate.base=2e-05
28
+ - trainer.learning_rate.qwen_vl_interface=1e-05
29
+ - trainer.learning_rate.action_model=0.0001
30
+ - trainer.lr_scheduler_type=cosine_with_min_lr
31
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
32
+ - trainer.loss_scale.vla=1.0
33
+ - trainer.loss_scale.vlm=0.1
34
+ - trainer.max_grad_norm=1.0
35
+ - trainer.weight_decay=0.0
36
+ - trainer.gradient_clipping=1.0
37
+ - trainer.optimizer.name=AdamW
38
+ - trainer.optimizer.betas=[0.9, 0.95]
39
+ - trainer.optimizer.eps=1e-08
40
+ - trainer.optimizer.weight_decay=1e-08
41
+ - trainer.optimizer.fused=false
42
+ - trainer.save_format=pt
43
+ - framework.name=QwenOFT
44
+ - framework.qwenvl.attn_implementation=flash_attention_2
45
+ - framework.qwenvl.enable_gradient_checkpointing=true
46
+ - framework.action_model.action_dim=7
47
+ - framework.action_model.action_env_dim=2
48
+ - framework.action_model.state_dim=7
49
+ - framework.action_model.loss_type=discrete_ce
50
+ - framework.action_model.action_horizon=1
51
+ - framework.action_model.future_action_window_size=0
52
+ - framework.action_model.past_action_window_size=0
53
+ - datasets.vla_data.include_state=true
54
+ - datasets.vla_data.action_type=discrete
55
+ - datasets.vla_data.sequential_step_sampling=false
56
+ - datasets.vla_data.per_device_batch_size=32
57
+ - datasets.vla_data.load_all_data_for_training=true
58
+ - datasets.vla_data.video_backend=torchvision_av
59
+ - datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_4_small
60
+ - datasets.vla_data.data_mix=flappy_train__bridge
61
+ - datasets.vla_data.eval_data_mix=flappy_train__bridge__val
62
+ - framework.qwenvl.base_vlm=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct
63
+ - rl_games.task=flappy
64
+ - rl_games.model_alias=openvla
65
+ - rl_games.initialization_mode=bridge
66
+ - rl_games.action_carrier=bridge
67
+ - rl_games.env_eval.distributed_mode=none
68
+ - rl_games.env_eval.latency.mode=single
69
+ - rl_games.env_eval.frameskip=1
70
+ - rl_games.env_eval.image_size=224
71
+ - rl_games.env_eval.seed=42
72
+ - rl_games.env_eval.fixed_episode_seeds=true
73
+ - rl_games.env_eval.latency_seed_stride=0
74
+ - rl_games.env_eval.task_seed_stride=0
75
+ - rl_games.env_eval.latency.values=[0]
76
+ - rl_games.env_eval.latency.prompt_map_path=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/flappy_fix_latency_4_small/flappy_train__bridge/latency_prompt_map.json
77
+ - rl_games.env_eval.mid_train.enabled=true
78
+ - rl_games.env_eval.mid_train.interval_steps=100
79
+ - rl_games.env_eval.mid_train.num_episodes=5
80
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
81
+ - rl_games.env_eval.mid_train.latencies=[4]
82
+ - rl_games.env_eval.post_train.enabled=true
83
+ - rl_games.env_eval.post_train.num_episodes=5
84
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
85
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_small/hydra/train_starvla_hydra.log ADDED
The diff for this file is too large to render. See raw diff
 
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_4_small/summary.jsonl ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"steps": 100}
2
+ {"steps": 200}
3
+ {"steps": 300}
4
+ {"steps": 400}
5
+ {"steps": 500}
6
+ {"steps": 600}
7
+ {"steps": 700}
8
+ {"steps": 800}
9
+ {"steps": 900}
10
+ {"steps": 1000}
11
+ {"steps": 1100}
12
+ {"steps": 1200}
13
+ {"steps": 1300}
14
+ {"steps": 1400}
15
+ {"steps": 1500}
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_6/checkpoints/steps_5000_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a24aac79d3d3fd72e0a8e782c0fd1f5b3294ee74eca4bbce1cbff73128b72122
3
+ size 9138230516
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_6/checkpoints/steps_5000_pytorch_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:26943b407d9280f89a461ca674615037f1e2ddda896fe00b88015ff5ce22a9f0
3
+ size 9138476048
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_6/config.full.yaml ADDED
@@ -0,0 +1,178 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ name: QwenOFT
3
+ qwenvl:
4
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
5
+ attn_implementation: flash_attention_2
6
+ enable_gradient_checkpointing: true
7
+ action_model:
8
+ action_model_type: MLP
9
+ action_dim: 7
10
+ action_hidden_dim: 2560
11
+ future_action_window_size: 0
12
+ past_action_window_size: 0
13
+ loss_type: discrete_ce
14
+ state_dim: 7
15
+ action_horizon: 1
16
+ action_env_dim: 2
17
+ datasets:
18
+ vla_data:
19
+ dataset_py: lerobot_datasets
20
+ include_state: true
21
+ data_root_dir: /workspace/starVLA/data/flappy_fix_latency_6
22
+ data_mix: flappy_train__bridge
23
+ eval_data_mix: flappy_train__bridge__val
24
+ action_type: discrete
25
+ sequential_step_sampling: false
26
+ per_device_batch_size: 16
27
+ load_all_data_for_training: true
28
+ obs_image_size: null
29
+ video_backend: torchvision_av
30
+ dataset:
31
+ source_hf: ''
32
+ converted_name: flappy_train
33
+ single_source_hf: ''
34
+ mixed_source_hf: ''
35
+ single_converted_name: flappy_train
36
+ mixed_converted_name: flappy_mixed_latency_train
37
+ single_latency_filter: null
38
+ mixed_latency_filter: null
39
+ force_download: false
40
+ setup_force: false
41
+ verify_rows: 200
42
+ max_episodes: null
43
+ latency_filter: null
44
+ debug_subset:
45
+ enabled: false
46
+ max_episodes: 5
47
+ suffix: debug
48
+ base_model:
49
+ repo_id: Qwen/Qwen3-VL-4B-Instruct
50
+ initialization:
51
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
52
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
53
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
54
+ trainer:
55
+ max_train_steps: 5000
56
+ num_warmup_steps: 100
57
+ save_interval: 100
58
+ eval_interval: 100
59
+ eval_num_batches: 20
60
+ learning_rate:
61
+ base: 2.0e-05
62
+ qwen_vl_interface: 1.0e-05
63
+ action_model: 0.0001
64
+ lr_scheduler_type: cosine_with_min_lr
65
+ scheduler_specific_kwargs:
66
+ min_lr: 1.0e-06
67
+ freeze_modules: ''
68
+ loss_scale:
69
+ vla: 1.0
70
+ vlm: 0.1
71
+ max_grad_norm: 1.0
72
+ weight_decay: 0.0
73
+ logging_frequency: 10
74
+ gradient_clipping: 1.0
75
+ gradient_accumulation_steps: 16
76
+ distributed_backend: none
77
+ is_resume: false
78
+ pretrained_checkpoint: /workspace/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
79
+ resume_step: 0
80
+ reload_modules: null
81
+ optimizer:
82
+ name: AdamW
83
+ betas:
84
+ - 0.9
85
+ - 0.95
86
+ eps: 1.0e-08
87
+ weight_decay: 1.0e-08
88
+ fused: false
89
+ save_format: pt
90
+ workspace_dir: WORKSPACE_DIR
91
+ run_root_dir: /workspace/starVLA/results/Checkpoints
92
+ seed: 42
93
+ wandb_entity: dongqianyu99-zhejiang-university
94
+ wandb_project: starVLA_rl_games
95
+ auth:
96
+ env_file: null
97
+ hf_token_env: HF_TOKEN
98
+ wandb_api_key_env: WANDB_API_KEY
99
+ paths:
100
+ run_root_dir: results/Checkpoints
101
+ dataset_local_dir: playground/Datasets/rl_games
102
+ dataset_cache_dir: null
103
+ base_model_dir: playground/Pretrained_models/Qwen3-VL-4B-Instruct
104
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
105
+ launch:
106
+ use_accelerate: true
107
+ gpus: null
108
+ num_processes: 1
109
+ dry_run: false
110
+ conda:
111
+ enabled: true
112
+ env_name: null
113
+ rl_games:
114
+ model_alias: openvla
115
+ env_eval:
116
+ image_size: 224
117
+ frameskip: 1
118
+ task_description: ''
119
+ enabled: true
120
+ latency:
121
+ prompt_map_path: /workspace/starVLA/data/flappy_fix_latency_6/flappy_train__bridge/latency_prompt_map.json
122
+ mode: single
123
+ values:
124
+ - 0
125
+ mid_train:
126
+ enabled: true
127
+ interval_steps: 100
128
+ latencies:
129
+ - 6
130
+ num_episodes: 1
131
+ max_steps_per_episode: 3600
132
+ post_train:
133
+ enabled: true
134
+ latencies:
135
+ - 0
136
+ - 1
137
+ - 2
138
+ - 3
139
+ - 4
140
+ - 5
141
+ - 6
142
+ - 7
143
+ num_episodes: 5
144
+ max_steps_per_episode: 3600
145
+ task: flappy
146
+ initialization_mode: bridge
147
+ action_carrier: bridge
148
+ model: openvla
149
+ env: flappy
150
+ init: bridge
151
+ bridge_base_model:
152
+ repo_id:
153
+ openvla: Qwen/Qwen3-VL-4B-Instruct
154
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
155
+ pi05: Qwen/Qwen3-VL-4B-Instruct
156
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
157
+ local_dir:
158
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
159
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
160
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
161
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
162
+ mode: single
163
+ checkpoint:
164
+ load: auto
165
+ hf_repo_id: null
166
+ local:
167
+ keep_last_n: 1
168
+ sync:
169
+ enabled: false
170
+ repo_id: null
171
+ keep_last_n: 0
172
+ sync_every_n_checkpoints: 1
173
+ resume_policy: local_latest
174
+ run_id: openvla_flappy_fix_latency_6
175
+ output_dir: /workspace/starVLA/results/Checkpoints/openvla_flappy_fix_latency_6
176
+ config_yaml: null
177
+ is_debug: false
178
+ version_id: '0.21'
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_6/config.yaml ADDED
@@ -0,0 +1,79 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ checkpoint:
2
+ local:
3
+ keep_last_n: 1
4
+ sync:
5
+ enabled: false
6
+ keep_last_n: 0
7
+ repo_id: null
8
+ datasets:
9
+ vla_data:
10
+ data_mix: flappy_train__bridge
11
+ dataset_py: lerobot_datasets
12
+ eval_data_mix: flappy_train__bridge__val
13
+ include_state: true
14
+ obs_image_size: null
15
+ per_device_batch_size: 16
16
+ framework:
17
+ action_model:
18
+ action_dim: 7
19
+ action_env_dim: 2
20
+ action_hidden_dim: 2560
21
+ action_horizon: 1
22
+ action_model_type: MLP
23
+ loss_type: discrete_ce
24
+ state_dim: 7
25
+ name: QwenOFT
26
+ qwenvl:
27
+ attn_implementation: flash_attention_2
28
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
29
+ enable_gradient_checkpointing: true
30
+ output_dir: /workspace/starVLA/results/Checkpoints/openvla_flappy_fix_latency_6
31
+ rl_games:
32
+ env_eval:
33
+ enabled: true
34
+ frameskip: 1
35
+ image_size: 224
36
+ latency:
37
+ prompt_map_path: /workspace/starVLA/data/flappy_fix_latency_6/flappy_train__bridge/latency_prompt_map.json
38
+ mid_train:
39
+ enabled: true
40
+ interval_steps: 100
41
+ latencies:
42
+ - 6
43
+ max_steps_per_episode: 3600
44
+ num_episodes: 1
45
+ task_description: ''
46
+ model_alias: openvla
47
+ task: flappy
48
+ run_id: openvla_flappy_fix_latency_6
49
+ run_root_dir: /workspace/starVLA/results/Checkpoints
50
+ seed: 42
51
+ trainer:
52
+ eval_interval: 100
53
+ eval_num_batches: 20
54
+ freeze_modules: ''
55
+ gradient_clipping: 1.0
56
+ is_resume: false
57
+ learning_rate:
58
+ action_model: 0.0001
59
+ base: 2.0e-05
60
+ qwen_vl_interface: 1.0e-05
61
+ logging_frequency: 10
62
+ lr_scheduler_type: cosine_with_min_lr
63
+ max_train_steps: 5000
64
+ num_warmup_steps: 100
65
+ optimizer:
66
+ betas:
67
+ - 0.9
68
+ - 0.95
69
+ eps: 1.0e-08
70
+ fused: false
71
+ weight_decay: 1.0e-08
72
+ pretrained_checkpoint: /workspace/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
73
+ reload_modules: null
74
+ save_format: pt
75
+ save_interval: 100
76
+ scheduler_specific_kwargs:
77
+ min_lr: 1.0e-06
78
+ wandb_entity: dongqianyu99-zhejiang-university
79
+ wandb_project: starVLA_rl_games
latency-aware/flappy/vla/starvla-qwenoft-h1/openvla_flappy_fix_latency_6/dataset_statistics.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.8444219827651978,
6
+ 0.15557800233364105,
7
+ 0.0,
8
+ 0.0,
9
+ 0.0,
10
+ 0.0,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.3620394766330719,
15
+ 0.3620394766330719,
16
+ 0.0,
17
+ 0.0,
18
+ 0.0,
19
+ 0.0,
20
+ 0.0
21
+ ],
22
+ "max": [
23
+ 1.0,
24
+ 1.0,
25
+ 0.0,
26
+ 0.0,
27
+ 0.0,
28
+ 0.0,
29
+ 0.0
30
+ ],
31
+ "min": [
32
+ 0.0,
33
+ 0.0,
34
+ 0.0,
35
+ 0.0,
36
+ 0.0,
37
+ 0.0,
38
+ 0.0
39
+ ],
40
+ "q01": [
41
+ 0.0,
42
+ 0.0,
43
+ 0.0,
44
+ 0.0,
45
+ 0.0,
46
+ 0.0,
47
+ 0.0
48
+ ],
49
+ "q99": [
50
+ 1.0,
51
+ 1.0,
52
+ 0.0,
53
+ 0.0,
54
+ 0.0,
55
+ 0.0,
56
+ 0.0
57
+ ],
58
+ "mask": [
59
+ true,
60
+ true,
61
+ true,
62
+ true,
63
+ true,
64
+ true,
65
+ true
66
+ ]
67
+ },
68
+ "state": {
69
+ "mean": [
70
+ 0.0,
71
+ 0.0,
72
+ 0.0,
73
+ 0.0,
74
+ 0.0,
75
+ 0.0,
76
+ 0.0
77
+ ],
78
+ "std": [
79
+ 0.0,
80
+ 0.0,
81
+ 0.0,
82
+ 0.0,
83
+ 0.0,
84
+ 0.0,
85
+ 0.0
86
+ ],
87
+ "max": [
88
+ 0.0,
89
+ 0.0,
90
+ 0.0,
91
+ 0.0,
92
+ 0.0,
93
+ 0.0,
94
+ 0.0
95
+ ],
96
+ "min": [
97
+ 0.0,
98
+ 0.0,
99
+ 0.0,
100
+ 0.0,
101
+ 0.0,
102
+ 0.0,
103
+ 0.0
104
+ ],
105
+ "q01": [
106
+ 0.0,
107
+ 0.0,
108
+ 0.0,
109
+ 0.0,
110
+ 0.0,
111
+ 0.0,
112
+ 0.0
113
+ ],
114
+ "q99": [
115
+ 0.0,
116
+ 0.0,
117
+ 0.0,
118
+ 0.0,
119
+ 0.0,
120
+ 0.0,
121
+ 0.0
122
+ ]
123
+ },
124
+ "num_transitions": 313894,
125
+ "num_trajectories": 90
126
+ }
127
+ }