Caesarrr commited on
Commit
f99bd51
·
verified ·
1 Parent(s): 2aeb8e3

Complete fine-tuning-scope asset ownership transfer before removing old repositories

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/flappy_fix_latency_2_200ep_full_tuning/checkpoints/steps_5000_model.safetensors +3 -0
  2. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning/config.full.yaml +219 -0
  3. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning/config.yaml +99 -0
  4. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning/dataset_statistics.json +127 -0
  5. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning/dataset_statistics_eval.json +127 -0
  6. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning/hydra/.hydra/config.yaml +217 -0
  7. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning/hydra/.hydra/hydra.yaml +261 -0
  8. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning/hydra/.hydra/overrides.yaml +101 -0
  9. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning/hydra/train_starvla_hydra.log +0 -0
  10. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning/summary.jsonl +10 -0
  11. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected/checkpoints/steps_5000_model.safetensors +3 -0
  12. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected/config.full.yaml +219 -0
  13. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected/config.yaml +99 -0
  14. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected/dataset_statistics.json +127 -0
  15. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected/dataset_statistics_eval.json +127 -0
  16. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected/hydra/.hydra/config.yaml +217 -0
  17. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected/hydra/.hydra/hydra.yaml +266 -0
  18. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected/hydra/.hydra/overrides.yaml +106 -0
  19. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected/hydra/train_starvla_hydra.log +0 -0
  20. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected/summary.jsonl +10 -0
  21. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/checkpoints/steps_3000_model.safetensors +3 -0
  22. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/config.full.yaml +224 -0
  23. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/config.yaml +103 -0
  24. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/dataset_statistics.json +127 -0
  25. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/dataset_statistics_eval.json +127 -0
  26. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/hydra/.hydra/config.yaml +222 -0
  27. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/hydra/.hydra/hydra.yaml +341 -0
  28. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/hydra/.hydra/overrides.yaml +181 -0
  29. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/hydra/train_starvla_hydra.log +0 -0
  30. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/summary.jsonl +6 -0
  31. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/checkpoints/steps_4000_model.safetensors +3 -0
  32. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/config.full.yaml +224 -0
  33. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/config.yaml +103 -0
  34. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/dataset_statistics.json +127 -0
  35. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/dataset_statistics_eval.json +127 -0
  36. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/hydra/.hydra/config.yaml +222 -0
  37. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/hydra/.hydra/hydra.yaml +341 -0
  38. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/hydra/.hydra/overrides.yaml +181 -0
  39. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/hydra/train_starvla_hydra.log +0 -0
  40. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/summary.jsonl +8 -0
  41. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_last_8_layers/checkpoints/steps_5000_model.safetensors +3 -0
  42. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_last_8_layers/config.full.yaml +217 -0
  43. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_last_8_layers/config.yaml +97 -0
  44. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_last_8_layers/dataset_statistics.json +127 -0
  45. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_last_8_layers/dataset_statistics_eval.json +127 -0
  46. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_last_8_layers/hydra/.hydra/config.yaml +215 -0
  47. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_last_8_layers/hydra/.hydra/hydra.yaml +259 -0
  48. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_last_8_layers/hydra/.hydra/overrides.yaml +99 -0
  49. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_last_8_layers/hydra/train_starvla_hydra.log +0 -0
  50. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_last_8_layers/summary.jsonl +10 -0
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning/checkpoints/steps_5000_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2ad30759d93084b8a8c6e03129e3239fac9a644b4c17c638deb76d82aa8610d5
3
+ size 9138230516
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning/config.full.yaml ADDED
@@ -0,0 +1,219 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ name: QwenOFT
3
+ qwenvl:
4
+ base_vlm: /workspace/latency-sensitive-bench/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/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep
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
+ eval_sequential_step_sampling: null
28
+ num_workers: 8
29
+ eval_num_workers: 8
30
+ prefetch_factor: 4
31
+ persistent_workers: true
32
+ pin_memory: true
33
+ shuffle: true
34
+ action_balance:
35
+ enabled: false
36
+ strategy: balanced_epoch
37
+ action_key: action_id
38
+ target_flap_fraction: 0.3
39
+ noop_id: 0
40
+ flap_id: 1
41
+ latency_curriculum:
42
+ enabled: false
43
+ strategy: exclusive
44
+ latencies: null
45
+ phase_steps: null
46
+ per_device_batch_size: 32
47
+ load_all_data_for_training: true
48
+ num_obs_frames: 1
49
+ image_mode: single
50
+ stitch_grid:
51
+ - 2
52
+ - 2
53
+ obs_image_size: null
54
+ video_backend: torchvision_av
55
+ dataset:
56
+ source_hf: ''
57
+ config_name: null
58
+ source_subdir: null
59
+ converted_name: flappy_train
60
+ single_source_hf: ''
61
+ mixed_source_hf: ''
62
+ single_converted_name: flappy_train
63
+ mixed_converted_name: flappy_mixed_latency_train
64
+ single_latency_filter: null
65
+ mixed_latency_filter: null
66
+ force_download: false
67
+ setup_force: false
68
+ skip_verification: false
69
+ verify_rows: 200
70
+ max_episodes: null
71
+ episodes_per_latency: null
72
+ latency_filter: null
73
+ debug_subset:
74
+ enabled: false
75
+ max_episodes: 5
76
+ suffix: debug
77
+ base_model:
78
+ repo_id: Qwen/Qwen3-VL-4B-Instruct
79
+ initialization:
80
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
81
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
82
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
83
+ trainer:
84
+ max_train_steps: 5000
85
+ num_warmup_steps: 100
86
+ save_interval: 500
87
+ eval_interval: 100
88
+ eval_num_batches: 100
89
+ per_latency_eval_num_batches: null
90
+ eval_action_classification: true
91
+ eval_action_classification_interval: null
92
+ cc_f1_tolerance: 1
93
+ learning_rate:
94
+ base: 2.0e-05
95
+ qwen_vl_interface: 1.0e-05
96
+ action_model: 0.0001
97
+ lr_scheduler_type: cosine_with_min_lr
98
+ scheduler_specific_kwargs:
99
+ min_lr: 1.0e-06
100
+ freeze_modules: ''
101
+ freeze_llm_bottom_ratio: 0.0
102
+ loss_scale:
103
+ vla: 1.0
104
+ vlm: 0.1
105
+ max_grad_norm: 1.0
106
+ weight_decay: 0.0
107
+ logging_frequency: 1
108
+ gradient_clipping: 1.0
109
+ gradient_accumulation_steps: 4
110
+ distributed_backend: none
111
+ is_resume: false
112
+ pretrained_checkpoint: null
113
+ resume_step: 0
114
+ reload_modules: null
115
+ optimizer:
116
+ name: AdamW
117
+ betas:
118
+ - 0.9
119
+ - 0.95
120
+ eps: 1.0e-08
121
+ weight_decay: 1.0e-08
122
+ fused: true
123
+ save_format: pt
124
+ workspace_dir: WORKSPACE_DIR
125
+ run_root_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints
126
+ seed: 42
127
+ wandb_entity: saberrr-zju
128
+ wandb_project: starVLA_rl_games
129
+ auth:
130
+ env_file: null
131
+ hf_token_env: HF_TOKEN
132
+ wandb_api_key_env: WANDB_API_KEY
133
+ paths:
134
+ run_root_dir: results/Checkpoints
135
+ dataset_local_dir: playground/Datasets/rl_games
136
+ dataset_cache_dir: null
137
+ base_model_dir: playground/Pretrained_models/Qwen3-VL-4B-Instruct
138
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
139
+ launch:
140
+ use_accelerate: true
141
+ gpus: null
142
+ num_processes: 1
143
+ dry_run: false
144
+ conda:
145
+ enabled: true
146
+ env_name: null
147
+ rl_games:
148
+ model_alias: openvla
149
+ env_eval:
150
+ image_size: 224
151
+ frameskip: 1
152
+ seed: 42
153
+ fixed_episode_seeds: true
154
+ latency_seed_stride: 0
155
+ task_seed_stride: 0
156
+ task_description: ''
157
+ enabled: true
158
+ distributed_mode: none
159
+ vectorized:
160
+ enabled: false
161
+ batch_size: 1
162
+ latency:
163
+ prompt_map_path: /workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
164
+ mode: single
165
+ values:
166
+ - 0
167
+ mid_train:
168
+ enabled: true
169
+ interval_steps: 250
170
+ latencies:
171
+ - 2
172
+ num_episodes: 5
173
+ max_steps_per_episode: 3600
174
+ post_train:
175
+ enabled: false
176
+ latencies:
177
+ - 0
178
+ - 1
179
+ - 2
180
+ - 3
181
+ - 4
182
+ num_episodes: 5
183
+ max_steps_per_episode: 3600
184
+ task: flappy
185
+ initialization_mode: bridge
186
+ action_carrier: bridge
187
+ model: openvla
188
+ env: flappy
189
+ init: bridge
190
+ bridge_base_model:
191
+ repo_id:
192
+ openvla: Qwen/Qwen3-VL-4B-Instruct
193
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
194
+ pi05: Qwen/Qwen3-VL-4B-Instruct
195
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
196
+ local_dir:
197
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
198
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
199
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
200
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
201
+ mode: single
202
+ checkpoint:
203
+ load: auto
204
+ hf_repo_id: null
205
+ save_best_model: false
206
+ save_pt_file: false
207
+ local:
208
+ keep_last_n: 1
209
+ sync:
210
+ enabled: false
211
+ repo_id: null
212
+ keep_last_n: 0
213
+ sync_every_n_checkpoints: 1
214
+ resume_policy: local_latest
215
+ run_id: flappy_fix_latency_2_200ep_full_tuning
216
+ output_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning
217
+ config_yaml: null
218
+ is_debug: false
219
+ version_id: '0.21'
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning/config.yaml ADDED
@@ -0,0 +1,99 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ checkpoint:
2
+ local:
3
+ keep_last_n: 1
4
+ save_best_model: false
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
+ eval_num_workers: 8
16
+ include_state: true
17
+ latency_curriculum:
18
+ enabled: false
19
+ obs_image_size: null
20
+ per_device_batch_size: 32
21
+ persistent_workers: true
22
+ pin_memory: true
23
+ prefetch_factor: 4
24
+ framework:
25
+ action_model:
26
+ action_dim: 7
27
+ action_env_dim: 2
28
+ action_hidden_dim: 2560
29
+ action_horizon: 1
30
+ action_model_type: MLP
31
+ loss_type: discrete_ce
32
+ state_dim: 7
33
+ name: QwenOFT
34
+ qwenvl:
35
+ attn_implementation: flash_attention_2
36
+ base_vlm: /workspace/latency-sensitive-bench/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
37
+ enable_gradient_checkpointing: true
38
+ output_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning
39
+ rl_games:
40
+ env_eval:
41
+ distributed_mode: none
42
+ enabled: true
43
+ fixed_episode_seeds: true
44
+ frameskip: 1
45
+ image_size: 224
46
+ latency:
47
+ prompt_map_path: /workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
48
+ latency_seed_stride: 0
49
+ mid_train:
50
+ enabled: true
51
+ interval_steps: 250
52
+ latencies:
53
+ - 2
54
+ max_steps_per_episode: 3600
55
+ num_episodes: 5
56
+ seed: 42
57
+ task_description: ''
58
+ task_seed_stride: 0
59
+ vectorized:
60
+ enabled: false
61
+ model_alias: openvla
62
+ task: flappy
63
+ run_id: flappy_fix_latency_2_200ep_full_tuning
64
+ run_root_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints
65
+ seed: 42
66
+ trainer:
67
+ distributed_backend: none
68
+ eval_action_classification: true
69
+ eval_action_classification_interval: null
70
+ eval_interval: 100
71
+ eval_num_batches: 100
72
+ freeze_llm_bottom_ratio: 0.0
73
+ freeze_modules: ''
74
+ gradient_accumulation_steps: 4
75
+ gradient_clipping: 1.0
76
+ is_resume: false
77
+ learning_rate:
78
+ action_model: 0.0001
79
+ base: 2.0e-05
80
+ qwen_vl_interface: 1.0e-05
81
+ logging_frequency: 1
82
+ lr_scheduler_type: cosine_with_min_lr
83
+ max_train_steps: 5000
84
+ num_warmup_steps: 100
85
+ optimizer:
86
+ betas:
87
+ - 0.9
88
+ - 0.95
89
+ eps: 1.0e-08
90
+ fused: true
91
+ weight_decay: 1.0e-08
92
+ per_latency_eval_num_batches: null
93
+ pretrained_checkpoint: null
94
+ reload_modules: null
95
+ save_interval: 500
96
+ scheduler_specific_kwargs:
97
+ min_lr: 1.0e-06
98
+ wandb_entity: saberrr-zju
99
+ wandb_project: starVLA_rl_games
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning/dataset_statistics.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.6028500199317932,
6
+ 0.3971499800682068,
7
+ 0.0,
8
+ 0.0,
9
+ 0.0,
10
+ 0.0,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.4890792667865753,
15
+ 0.4890792667865753,
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": 330734,
125
+ "num_trajectories": 180
126
+ }
127
+ }
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning/dataset_statistics_eval.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.7959861159324646,
6
+ 0.2040138840675354,
7
+ 0.0,
8
+ 0.0,
9
+ 0.0,
10
+ 0.0,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.4030573070049286,
15
+ 0.4030573070049286,
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": 72000,
125
+ "num_trajectories": 20
126
+ }
127
+ }
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning/hydra/.hydra/config.yaml ADDED
@@ -0,0 +1,217 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ qwenvl:
3
+ base_vlm: /workspace/latency-sensitive-bench/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/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep
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
+ eval_sequential_step_sampling: null
26
+ num_workers: 8
27
+ eval_num_workers: 8
28
+ prefetch_factor: 4
29
+ persistent_workers: true
30
+ pin_memory: true
31
+ shuffle: true
32
+ action_balance:
33
+ enabled: false
34
+ strategy: balanced_epoch
35
+ action_key: action_id
36
+ target_flap_fraction: 0.3
37
+ noop_id: 0
38
+ flap_id: 1
39
+ latency_curriculum:
40
+ enabled: false
41
+ strategy: exclusive
42
+ latencies: null
43
+ phase_steps: null
44
+ per_device_batch_size: 32
45
+ load_all_data_for_training: true
46
+ num_obs_frames: 1
47
+ image_mode: single
48
+ stitch_grid:
49
+ - 2
50
+ - 2
51
+ obs_image_size: null
52
+ video_backend: torchvision_av
53
+ dataset:
54
+ source_hf: ${dataset.single_source_hf}
55
+ config_name: null
56
+ source_subdir: null
57
+ converted_name: ${dataset.single_converted_name}
58
+ single_source_hf: ''
59
+ mixed_source_hf: ''
60
+ single_converted_name: flappy_train
61
+ mixed_converted_name: flappy_mixed_latency_train
62
+ single_latency_filter: null
63
+ mixed_latency_filter: null
64
+ force_download: false
65
+ setup_force: false
66
+ skip_verification: false
67
+ verify_rows: 200
68
+ max_episodes: null
69
+ episodes_per_latency: null
70
+ latency_filter: ${dataset.single_latency_filter}
71
+ debug_subset:
72
+ enabled: false
73
+ max_episodes: 5
74
+ suffix: debug
75
+ base_model:
76
+ repo_id: ${bridge_base_model.repo_id.${model}}
77
+ initialization:
78
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
79
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
80
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
81
+ trainer:
82
+ max_train_steps: 5000
83
+ num_warmup_steps: 100
84
+ save_interval: 500
85
+ eval_interval: 100
86
+ eval_num_batches: 100
87
+ per_latency_eval_num_batches: null
88
+ eval_action_classification: true
89
+ eval_action_classification_interval: null
90
+ cc_f1_tolerance: 1
91
+ learning_rate:
92
+ base: 2.0e-05
93
+ qwen_vl_interface: 1.0e-05
94
+ action_model: 0.0001
95
+ lr_scheduler_type: cosine_with_min_lr
96
+ scheduler_specific_kwargs:
97
+ min_lr: 1.0e-06
98
+ freeze_modules: ''
99
+ freeze_llm_bottom_ratio: 0.0
100
+ loss_scale:
101
+ vla: 1.0
102
+ vlm: 0.1
103
+ max_grad_norm: 1.0
104
+ weight_decay: 0.0
105
+ logging_frequency: 1
106
+ gradient_clipping: 1.0
107
+ gradient_accumulation_steps: 4
108
+ distributed_backend: none
109
+ is_resume: false
110
+ pretrained_checkpoint: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning/checkpoints/_initialization/StarVLA_Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
111
+ resume_step: 0
112
+ reload_modules: null
113
+ optimizer:
114
+ name: AdamW
115
+ betas:
116
+ - 0.9
117
+ - 0.95
118
+ eps: 1.0e-08
119
+ weight_decay: 1.0e-08
120
+ fused: true
121
+ save_format: pt
122
+ workspace_dir: WORKSPACE_DIR
123
+ run_root_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints
124
+ seed: 42
125
+ wandb_entity: saberrr-zju
126
+ wandb_project: starVLA_rl_games
127
+ auth:
128
+ env_file: null
129
+ hf_token_env: HF_TOKEN
130
+ wandb_api_key_env: WANDB_API_KEY
131
+ paths:
132
+ run_root_dir: results/Checkpoints
133
+ dataset_local_dir: playground/Datasets/rl_games
134
+ dataset_cache_dir: null
135
+ base_model_dir: ${bridge_base_model.local_dir.${model}}
136
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
137
+ launch:
138
+ use_accelerate: true
139
+ gpus: null
140
+ num_processes: 1
141
+ dry_run: false
142
+ conda:
143
+ enabled: true
144
+ env_name: null
145
+ rl_games:
146
+ model_alias: openvla
147
+ env_eval:
148
+ image_size: 224
149
+ frameskip: 1
150
+ seed: 42
151
+ fixed_episode_seeds: true
152
+ latency_seed_stride: 0
153
+ task_seed_stride: 0
154
+ task_description: ''
155
+ enabled: true
156
+ distributed_mode: none
157
+ vectorized:
158
+ enabled: false
159
+ batch_size: 1
160
+ latency:
161
+ prompt_map_path: /workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
162
+ mode: single
163
+ values:
164
+ - 0
165
+ mid_train:
166
+ enabled: true
167
+ interval_steps: 250
168
+ latencies:
169
+ - 2
170
+ num_episodes: 5
171
+ max_steps_per_episode: 3600
172
+ post_train:
173
+ enabled: false
174
+ latencies:
175
+ - 0
176
+ - 1
177
+ - 2
178
+ - 3
179
+ - 4
180
+ num_episodes: 5
181
+ max_steps_per_episode: 3600
182
+ task: flappy
183
+ initialization_mode: bridge
184
+ action_carrier: bridge
185
+ model: openvla
186
+ env: flappy
187
+ init: bridge
188
+ bridge_base_model:
189
+ repo_id:
190
+ openvla: Qwen/Qwen3-VL-4B-Instruct
191
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
192
+ pi05: Qwen/Qwen3-VL-4B-Instruct
193
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
194
+ local_dir:
195
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
196
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
197
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
198
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
199
+ mode: single
200
+ checkpoint:
201
+ load: auto
202
+ hf_repo_id: null
203
+ save_best_model: false
204
+ save_pt_file: false
205
+ local:
206
+ keep_last_n: 1
207
+ sync:
208
+ enabled: false
209
+ repo_id: null
210
+ keep_last_n: 0
211
+ sync_every_n_checkpoints: 1
212
+ resume_policy: local_latest
213
+ run_id: flappy_fix_latency_2_200ep_full_tuning
214
+ output_dir: null
215
+ config_yaml: null
216
+ is_debug: false
217
+ version_id: '0.21'
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning/hydra/.hydra/hydra.yaml ADDED
@@ -0,0 +1,261 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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=flappy_fix_latency_2_200ep_full_tuning
120
+ - run_root_dir=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints
121
+ - seed=42
122
+ - wandb_entity=saberrr-zju
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=false
129
+ - checkpoint.save_pt_file=false
130
+ - trainer.is_resume=false
131
+ - trainer.pretrained_checkpoint=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning/checkpoints/_initialization/StarVLA_Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
132
+ - trainer.resume_step=0
133
+ - trainer.max_train_steps=5000
134
+ - trainer.num_warmup_steps=100
135
+ - trainer.save_interval=500
136
+ - trainer.eval_interval=100
137
+ - trainer.eval_num_batches=100
138
+ - trainer.eval_action_classification=true
139
+ - trainer.logging_frequency=1
140
+ - trainer.gradient_accumulation_steps=4
141
+ - trainer.distributed_backend=none
142
+ - trainer.learning_rate.base=2e-05
143
+ - trainer.learning_rate.qwen_vl_interface=1e-05
144
+ - trainer.learning_rate.action_model=0.0001
145
+ - trainer.lr_scheduler_type=cosine_with_min_lr
146
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
147
+ - trainer.freeze_llm_bottom_ratio=0.0
148
+ - trainer.loss_scale.vla=1.0
149
+ - trainer.loss_scale.vlm=0.1
150
+ - trainer.max_grad_norm=1.0
151
+ - trainer.weight_decay=0.0
152
+ - trainer.gradient_clipping=1.0
153
+ - trainer.optimizer.name=AdamW
154
+ - trainer.optimizer.betas=[0.9,0.95]
155
+ - trainer.optimizer.eps=1e-08
156
+ - trainer.optimizer.weight_decay=1e-08
157
+ - trainer.optimizer.fused=true
158
+ - trainer.save_format=pt
159
+ - framework.name=QwenOFT
160
+ - framework.qwenvl.attn_implementation=flash_attention_2
161
+ - framework.qwenvl.enable_gradient_checkpointing=true
162
+ - framework.action_model.action_dim=7
163
+ - framework.action_model.action_env_dim=2
164
+ - framework.action_model.state_dim=7
165
+ - framework.action_model.loss_type=discrete_ce
166
+ - framework.action_model.action_horizon=1
167
+ - framework.action_model.future_action_window_size=0
168
+ - framework.action_model.past_action_window_size=0
169
+ - datasets.vla_data.include_state=true
170
+ - datasets.vla_data.action_type=discrete
171
+ - datasets.vla_data.sequential_step_sampling=false
172
+ - datasets.vla_data.shuffle=true
173
+ - datasets.vla_data.action_balance.enabled=false
174
+ - datasets.vla_data.action_balance.strategy=balanced_epoch
175
+ - datasets.vla_data.action_balance.action_key=action_id
176
+ - datasets.vla_data.action_balance.target_flap_fraction=0.3
177
+ - datasets.vla_data.action_balance.noop_id=0
178
+ - datasets.vla_data.action_balance.flap_id=1
179
+ - datasets.vla_data.latency_curriculum.enabled=false
180
+ - datasets.vla_data.latency_curriculum.strategy=exclusive
181
+ - datasets.vla_data.per_device_batch_size=32
182
+ - datasets.vla_data.num_workers=8
183
+ - datasets.vla_data.eval_num_workers=8
184
+ - datasets.vla_data.prefetch_factor=4
185
+ - datasets.vla_data.persistent_workers=true
186
+ - datasets.vla_data.pin_memory=true
187
+ - datasets.vla_data.load_all_data_for_training=true
188
+ - datasets.vla_data.video_backend=torchvision_av
189
+ - datasets.vla_data.data_root_dir=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep
190
+ - datasets.vla_data.data_mix=flappy_train__bridge
191
+ - datasets.vla_data.eval_data_mix=flappy_train__bridge__val
192
+ - framework.qwenvl.base_vlm=/workspace/latency-sensitive-bench/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
193
+ - rl_games.task=flappy
194
+ - rl_games.model_alias=openvla
195
+ - rl_games.initialization_mode=bridge
196
+ - rl_games.action_carrier=bridge
197
+ - rl_games.env_eval.distributed_mode=none
198
+ - rl_games.env_eval.latency.mode=single
199
+ - rl_games.env_eval.frameskip=1
200
+ - rl_games.env_eval.image_size=224
201
+ - rl_games.env_eval.seed=42
202
+ - rl_games.env_eval.fixed_episode_seeds=true
203
+ - rl_games.env_eval.latency_seed_stride=0
204
+ - rl_games.env_eval.task_seed_stride=0
205
+ - rl_games.env_eval.latency.values=[0]
206
+ - rl_games.env_eval.latency.prompt_map_path=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
207
+ - rl_games.env_eval.mid_train.enabled=true
208
+ - rl_games.env_eval.mid_train.interval_steps=250
209
+ - rl_games.env_eval.mid_train.num_episodes=5
210
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
211
+ - rl_games.env_eval.mid_train.latencies=[2]
212
+ - rl_games.env_eval.post_train.enabled=false
213
+ - rl_games.env_eval.post_train.num_episodes=5
214
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
215
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
216
+ job:
217
+ name: train_starvla_hydra
218
+ chdir: false
219
+ override_dirname: checkpoint.local.keep_last_n=1,checkpoint.save_best_model=false,checkpoint.save_pt_file=false,checkpoint.sync.enabled=false,checkpoint.sync.keep_last_n=0,datasets.vla_data.action_balance.action_key=action_id,datasets.vla_data.action_balance.enabled=false,datasets.vla_data.action_balance.flap_id=1,datasets.vla_data.action_balance.noop_id=0,datasets.vla_data.action_balance.strategy=balanced_epoch,datasets.vla_data.action_balance.target_flap_fraction=0.3,datasets.vla_data.action_type=discrete,datasets.vla_data.data_mix=flappy_train__bridge,datasets.vla_data.data_root_dir=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep,datasets.vla_data.eval_data_mix=flappy_train__bridge__val,datasets.vla_data.eval_num_workers=8,datasets.vla_data.include_state=true,datasets.vla_data.latency_curriculum.enabled=false,datasets.vla_data.latency_curriculum.strategy=exclusive,datasets.vla_data.load_all_data_for_training=true,datasets.vla_data.num_workers=8,datasets.vla_data.per_device_batch_size=32,datasets.vla_data.persistent_workers=true,datasets.vla_data.pin_memory=true,datasets.vla_data.prefetch_factor=4,datasets.vla_data.sequential_step_sampling=false,datasets.vla_data.shuffle=true,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/latency-sensitive-bench/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=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/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=250,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=false,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=flappy_fix_latency_2_200ep_full_tuning,run_root_dir=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints,seed=42,trainer.distributed_backend=none,trainer.eval_action_classification=true,trainer.eval_interval=100,trainer.eval_num_batches=100,trainer.freeze_llm_bottom_ratio=0.0,trainer.gradient_accumulation_steps=4,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=1,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=5000,trainer.num_warmup_steps=100,trainer.optimizer.betas=[0.9,0.95],trainer.optimizer.eps=1e-08,trainer.optimizer.fused=true,trainer.optimizer.name=AdamW,trainer.optimizer.weight_decay=1e-08,trainer.pretrained_checkpoint=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning/checkpoints/_initialization/StarVLA_Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt,trainer.resume_step=0,trainer.save_format=pt,trainer.save_interval=500,trainer.scheduler_specific_kwargs.min_lr=1e-06,trainer.weight_decay=0.0,wandb_entity=saberrr-zju,wandb_project=starVLA_rl_games
220
+ id: ???
221
+ num: ???
222
+ config_name: train
223
+ env_set: {}
224
+ env_copy: []
225
+ config:
226
+ override_dirname:
227
+ kv_sep: '='
228
+ item_sep: ','
229
+ exclude_keys: []
230
+ runtime:
231
+ version: 1.3.3
232
+ version_base: '1.1'
233
+ cwd: /workspace/latency-sensitive-bench/starVLA
234
+ config_sources:
235
+ - path: hydra.conf
236
+ schema: pkg
237
+ provider: hydra
238
+ - path: /workspace/latency-sensitive-bench/starVLA/examples/rl_games/config
239
+ schema: file
240
+ provider: main
241
+ - path: ''
242
+ schema: structured
243
+ provider: schema
244
+ output_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning/hydra
245
+ choices:
246
+ cross_task_setup: null
247
+ checkpoint: default
248
+ mode: single
249
+ init: bridge
250
+ env: flappy
251
+ model: openvla
252
+ hydra/env: default
253
+ hydra/callbacks: null
254
+ hydra/job_logging: default
255
+ hydra/hydra_logging: default
256
+ hydra/hydra_help: default
257
+ hydra/help: default
258
+ hydra/sweeper: basic
259
+ hydra/launcher: basic
260
+ hydra/output: default
261
+ verbose: false
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning/hydra/.hydra/overrides.yaml ADDED
@@ -0,0 +1,101 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ - model=openvla
2
+ - env=flappy
3
+ - init=bridge
4
+ - mode=single
5
+ - run_id=flappy_fix_latency_2_200ep_full_tuning
6
+ - run_root_dir=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints
7
+ - seed=42
8
+ - wandb_entity=saberrr-zju
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=false
15
+ - checkpoint.save_pt_file=false
16
+ - trainer.is_resume=false
17
+ - trainer.pretrained_checkpoint=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning/checkpoints/_initialization/StarVLA_Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
18
+ - trainer.resume_step=0
19
+ - trainer.max_train_steps=5000
20
+ - trainer.num_warmup_steps=100
21
+ - trainer.save_interval=500
22
+ - trainer.eval_interval=100
23
+ - trainer.eval_num_batches=100
24
+ - trainer.eval_action_classification=true
25
+ - trainer.logging_frequency=1
26
+ - trainer.gradient_accumulation_steps=4
27
+ - trainer.distributed_backend=none
28
+ - trainer.learning_rate.base=2e-05
29
+ - trainer.learning_rate.qwen_vl_interface=1e-05
30
+ - trainer.learning_rate.action_model=0.0001
31
+ - trainer.lr_scheduler_type=cosine_with_min_lr
32
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
33
+ - trainer.freeze_llm_bottom_ratio=0.0
34
+ - trainer.loss_scale.vla=1.0
35
+ - trainer.loss_scale.vlm=0.1
36
+ - trainer.max_grad_norm=1.0
37
+ - trainer.weight_decay=0.0
38
+ - trainer.gradient_clipping=1.0
39
+ - trainer.optimizer.name=AdamW
40
+ - trainer.optimizer.betas=[0.9,0.95]
41
+ - trainer.optimizer.eps=1e-08
42
+ - trainer.optimizer.weight_decay=1e-08
43
+ - trainer.optimizer.fused=true
44
+ - trainer.save_format=pt
45
+ - framework.name=QwenOFT
46
+ - framework.qwenvl.attn_implementation=flash_attention_2
47
+ - framework.qwenvl.enable_gradient_checkpointing=true
48
+ - framework.action_model.action_dim=7
49
+ - framework.action_model.action_env_dim=2
50
+ - framework.action_model.state_dim=7
51
+ - framework.action_model.loss_type=discrete_ce
52
+ - framework.action_model.action_horizon=1
53
+ - framework.action_model.future_action_window_size=0
54
+ - framework.action_model.past_action_window_size=0
55
+ - datasets.vla_data.include_state=true
56
+ - datasets.vla_data.action_type=discrete
57
+ - datasets.vla_data.sequential_step_sampling=false
58
+ - datasets.vla_data.shuffle=true
59
+ - datasets.vla_data.action_balance.enabled=false
60
+ - datasets.vla_data.action_balance.strategy=balanced_epoch
61
+ - datasets.vla_data.action_balance.action_key=action_id
62
+ - datasets.vla_data.action_balance.target_flap_fraction=0.3
63
+ - datasets.vla_data.action_balance.noop_id=0
64
+ - datasets.vla_data.action_balance.flap_id=1
65
+ - datasets.vla_data.latency_curriculum.enabled=false
66
+ - datasets.vla_data.latency_curriculum.strategy=exclusive
67
+ - datasets.vla_data.per_device_batch_size=32
68
+ - datasets.vla_data.num_workers=8
69
+ - datasets.vla_data.eval_num_workers=8
70
+ - datasets.vla_data.prefetch_factor=4
71
+ - datasets.vla_data.persistent_workers=true
72
+ - datasets.vla_data.pin_memory=true
73
+ - datasets.vla_data.load_all_data_for_training=true
74
+ - datasets.vla_data.video_backend=torchvision_av
75
+ - datasets.vla_data.data_root_dir=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep
76
+ - datasets.vla_data.data_mix=flappy_train__bridge
77
+ - datasets.vla_data.eval_data_mix=flappy_train__bridge__val
78
+ - framework.qwenvl.base_vlm=/workspace/latency-sensitive-bench/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
79
+ - rl_games.task=flappy
80
+ - rl_games.model_alias=openvla
81
+ - rl_games.initialization_mode=bridge
82
+ - rl_games.action_carrier=bridge
83
+ - rl_games.env_eval.distributed_mode=none
84
+ - rl_games.env_eval.latency.mode=single
85
+ - rl_games.env_eval.frameskip=1
86
+ - rl_games.env_eval.image_size=224
87
+ - rl_games.env_eval.seed=42
88
+ - rl_games.env_eval.fixed_episode_seeds=true
89
+ - rl_games.env_eval.latency_seed_stride=0
90
+ - rl_games.env_eval.task_seed_stride=0
91
+ - rl_games.env_eval.latency.values=[0]
92
+ - rl_games.env_eval.latency.prompt_map_path=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
93
+ - rl_games.env_eval.mid_train.enabled=true
94
+ - rl_games.env_eval.mid_train.interval_steps=250
95
+ - rl_games.env_eval.mid_train.num_episodes=5
96
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
97
+ - rl_games.env_eval.mid_train.latencies=[2]
98
+ - rl_games.env_eval.post_train.enabled=false
99
+ - rl_games.env_eval.post_train.num_episodes=5
100
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
101
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning/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/flappy_fix_latency_2_200ep_full_tuning/summary.jsonl ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {"steps": 500}
2
+ {"steps": 1000}
3
+ {"steps": 1500}
4
+ {"steps": 2000}
5
+ {"steps": 2500}
6
+ {"steps": 3000}
7
+ {"steps": 3500}
8
+ {"steps": 4000}
9
+ {"steps": 4500}
10
+ {"steps": 5000}
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected/checkpoints/steps_5000_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ff343b81e73667493a90c8c0696872ab7f3ed1bf55fae59cf8b57f9785b040eb
3
+ size 9138230516
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected/config.full.yaml ADDED
@@ -0,0 +1,219 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ name: QwenOFT
3
+ qwenvl:
4
+ base_vlm: /workspace/latency-sensitive-bench/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/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep
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
+ eval_sequential_step_sampling: null
28
+ num_workers: 8
29
+ eval_num_workers: 8
30
+ prefetch_factor: 4
31
+ persistent_workers: true
32
+ pin_memory: true
33
+ shuffle: true
34
+ action_balance:
35
+ enabled: false
36
+ strategy: balanced_epoch
37
+ action_key: action_id
38
+ target_flap_fraction: 0.3
39
+ noop_id: 0
40
+ flap_id: 1
41
+ latency_curriculum:
42
+ enabled: false
43
+ strategy: exclusive
44
+ latencies: null
45
+ phase_steps: null
46
+ per_device_batch_size: 32
47
+ load_all_data_for_training: true
48
+ num_obs_frames: 1
49
+ image_mode: single
50
+ stitch_grid:
51
+ - 2
52
+ - 2
53
+ obs_image_size: null
54
+ video_backend: torchvision_av
55
+ dataset:
56
+ source_hf: ''
57
+ config_name: null
58
+ source_subdir: null
59
+ converted_name: flappy_train
60
+ single_source_hf: ''
61
+ mixed_source_hf: ''
62
+ single_converted_name: flappy_train
63
+ mixed_converted_name: flappy_mixed_latency_train
64
+ single_latency_filter: null
65
+ mixed_latency_filter: null
66
+ force_download: false
67
+ setup_force: false
68
+ skip_verification: false
69
+ verify_rows: 200
70
+ max_episodes: null
71
+ episodes_per_latency: null
72
+ latency_filter: null
73
+ debug_subset:
74
+ enabled: false
75
+ max_episodes: 5
76
+ suffix: debug
77
+ base_model:
78
+ repo_id: Qwen/Qwen3-VL-4B-Instruct
79
+ initialization:
80
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
81
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
82
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
83
+ trainer:
84
+ max_train_steps: 5000
85
+ num_warmup_steps: 100
86
+ save_interval: 500
87
+ eval_interval: 100
88
+ eval_num_batches: 100
89
+ per_latency_eval_num_batches: null
90
+ eval_action_classification: true
91
+ eval_action_classification_interval: null
92
+ cc_f1_tolerance: 1
93
+ learning_rate:
94
+ base: 2.0e-05
95
+ qwen_vl_interface: 1.0e-05
96
+ action_model: 0.0001
97
+ lr_scheduler_type: cosine_with_min_lr
98
+ scheduler_specific_kwargs:
99
+ min_lr: 1.0e-06
100
+ freeze_modules: ''
101
+ freeze_llm_bottom_ratio: 0.0
102
+ loss_scale:
103
+ vla: 1.0
104
+ vlm: 0.1
105
+ max_grad_norm: 1.0
106
+ weight_decay: 0.0
107
+ logging_frequency: 1
108
+ gradient_clipping: 1.0
109
+ gradient_accumulation_steps: 4
110
+ distributed_backend: none
111
+ is_resume: false
112
+ pretrained_checkpoint: null
113
+ resume_step: 0
114
+ reload_modules: null
115
+ optimizer:
116
+ name: AdamW
117
+ betas:
118
+ - 0.9
119
+ - 0.95
120
+ eps: 1.0e-08
121
+ weight_decay: 1.0e-08
122
+ fused: true
123
+ save_format: pt
124
+ workspace_dir: WORKSPACE_DIR
125
+ run_root_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints
126
+ seed: 42
127
+ wandb_entity: saberrr-zju
128
+ wandb_project: starVLA_rl_games
129
+ auth:
130
+ env_file: null
131
+ hf_token_env: HF_TOKEN
132
+ wandb_api_key_env: WANDB_API_KEY
133
+ paths:
134
+ run_root_dir: results/Checkpoints
135
+ dataset_local_dir: playground/Datasets/rl_games
136
+ dataset_cache_dir: null
137
+ base_model_dir: playground/Pretrained_models/Qwen3-VL-4B-Instruct
138
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
139
+ launch:
140
+ use_accelerate: true
141
+ gpus: null
142
+ num_processes: 1
143
+ dry_run: false
144
+ conda:
145
+ enabled: true
146
+ env_name: null
147
+ rl_games:
148
+ model_alias: openvla
149
+ env_eval:
150
+ image_size: 224
151
+ frameskip: 1
152
+ seed: 42
153
+ fixed_episode_seeds: true
154
+ latency_seed_stride: 0
155
+ task_seed_stride: 0
156
+ task_description: ''
157
+ enabled: true
158
+ distributed_mode: none
159
+ vectorized:
160
+ enabled: false
161
+ batch_size: 1
162
+ latency:
163
+ prompt_map_path: /workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
164
+ mode: single
165
+ values:
166
+ - 0
167
+ mid_train:
168
+ enabled: true
169
+ interval_steps: 250
170
+ latencies:
171
+ - 2
172
+ num_episodes: 5
173
+ max_steps_per_episode: 3600
174
+ post_train:
175
+ enabled: false
176
+ latencies:
177
+ - 0
178
+ - 1
179
+ - 2
180
+ - 3
181
+ - 4
182
+ num_episodes: 5
183
+ max_steps_per_episode: 3600
184
+ task: flappy
185
+ initialization_mode: bridge
186
+ action_carrier: bridge
187
+ model: openvla
188
+ env: flappy
189
+ init: bridge
190
+ bridge_base_model:
191
+ repo_id:
192
+ openvla: Qwen/Qwen3-VL-4B-Instruct
193
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
194
+ pi05: Qwen/Qwen3-VL-4B-Instruct
195
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
196
+ local_dir:
197
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
198
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
199
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
200
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
201
+ mode: single
202
+ checkpoint:
203
+ load: auto
204
+ hf_repo_id: null
205
+ save_best_model: false
206
+ save_pt_file: false
207
+ local:
208
+ keep_last_n: 1
209
+ sync:
210
+ enabled: false
211
+ repo_id: null
212
+ keep_last_n: 0
213
+ sync_every_n_checkpoints: 1
214
+ resume_policy: local_latest
215
+ run_id: flappy_fix_latency_2_200ep_full_tuning_corrected
216
+ output_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning_corrected
217
+ config_yaml: null
218
+ is_debug: false
219
+ version_id: '0.21'
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected/config.yaml ADDED
@@ -0,0 +1,99 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ checkpoint:
2
+ local:
3
+ keep_last_n: 1
4
+ save_best_model: false
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
+ eval_num_workers: 8
16
+ include_state: true
17
+ latency_curriculum:
18
+ enabled: false
19
+ obs_image_size: null
20
+ per_device_batch_size: 32
21
+ persistent_workers: true
22
+ pin_memory: true
23
+ prefetch_factor: 4
24
+ framework:
25
+ action_model:
26
+ action_dim: 7
27
+ action_env_dim: 2
28
+ action_hidden_dim: 2560
29
+ action_horizon: 1
30
+ action_model_type: MLP
31
+ loss_type: discrete_ce
32
+ state_dim: 7
33
+ name: QwenOFT
34
+ qwenvl:
35
+ attn_implementation: flash_attention_2
36
+ base_vlm: /workspace/latency-sensitive-bench/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
37
+ enable_gradient_checkpointing: true
38
+ output_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning_corrected
39
+ rl_games:
40
+ env_eval:
41
+ distributed_mode: none
42
+ enabled: true
43
+ fixed_episode_seeds: true
44
+ frameskip: 1
45
+ image_size: 224
46
+ latency:
47
+ prompt_map_path: /workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
48
+ latency_seed_stride: 0
49
+ mid_train:
50
+ enabled: true
51
+ interval_steps: 250
52
+ latencies:
53
+ - 2
54
+ max_steps_per_episode: 3600
55
+ num_episodes: 5
56
+ seed: 42
57
+ task_description: ''
58
+ task_seed_stride: 0
59
+ vectorized:
60
+ enabled: false
61
+ model_alias: openvla
62
+ task: flappy
63
+ run_id: flappy_fix_latency_2_200ep_full_tuning_corrected
64
+ run_root_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints
65
+ seed: 42
66
+ trainer:
67
+ distributed_backend: none
68
+ eval_action_classification: true
69
+ eval_action_classification_interval: null
70
+ eval_interval: 100
71
+ eval_num_batches: 100
72
+ freeze_llm_bottom_ratio: 0.0
73
+ freeze_modules: ''
74
+ gradient_accumulation_steps: 4
75
+ gradient_clipping: 1.0
76
+ is_resume: false
77
+ learning_rate:
78
+ action_model: 0.0001
79
+ base: 2.0e-05
80
+ qwen_vl_interface: 1.0e-05
81
+ logging_frequency: 1
82
+ lr_scheduler_type: cosine_with_min_lr
83
+ max_train_steps: 5000
84
+ num_warmup_steps: 100
85
+ optimizer:
86
+ betas:
87
+ - 0.9
88
+ - 0.95
89
+ eps: 1.0e-08
90
+ fused: true
91
+ weight_decay: 1.0e-08
92
+ per_latency_eval_num_batches: null
93
+ pretrained_checkpoint: null
94
+ reload_modules: null
95
+ save_interval: 500
96
+ scheduler_specific_kwargs:
97
+ min_lr: 1.0e-06
98
+ wandb_entity: saberrr-zju
99
+ wandb_project: starVLA_rl_games
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected/dataset_statistics.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.6028500199317932,
6
+ 0.3971499800682068,
7
+ 0.0,
8
+ 0.0,
9
+ 0.0,
10
+ 0.0,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.4890792667865753,
15
+ 0.4890792667865753,
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": 330734,
125
+ "num_trajectories": 180
126
+ }
127
+ }
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected/dataset_statistics_eval.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.7959861159324646,
6
+ 0.2040138840675354,
7
+ 0.0,
8
+ 0.0,
9
+ 0.0,
10
+ 0.0,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.4030573070049286,
15
+ 0.4030573070049286,
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": 72000,
125
+ "num_trajectories": 20
126
+ }
127
+ }
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected/hydra/.hydra/config.yaml ADDED
@@ -0,0 +1,217 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ qwenvl:
3
+ base_vlm: /workspace/latency-sensitive-bench/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/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep
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
+ eval_sequential_step_sampling: null
26
+ num_workers: 8
27
+ eval_num_workers: 8
28
+ prefetch_factor: 4
29
+ persistent_workers: true
30
+ pin_memory: true
31
+ shuffle: true
32
+ action_balance:
33
+ enabled: false
34
+ strategy: balanced_epoch
35
+ action_key: action_id
36
+ target_flap_fraction: 0.3
37
+ noop_id: 0
38
+ flap_id: 1
39
+ latency_curriculum:
40
+ enabled: false
41
+ strategy: exclusive
42
+ latencies: null
43
+ phase_steps: null
44
+ per_device_batch_size: 32
45
+ load_all_data_for_training: true
46
+ num_obs_frames: 1
47
+ image_mode: single
48
+ stitch_grid:
49
+ - 2
50
+ - 2
51
+ obs_image_size: null
52
+ video_backend: torchvision_av
53
+ dataset:
54
+ source_hf: ${dataset.single_source_hf}
55
+ config_name: null
56
+ source_subdir: null
57
+ converted_name: ${dataset.single_converted_name}
58
+ single_source_hf: ''
59
+ mixed_source_hf: ''
60
+ single_converted_name: flappy_train
61
+ mixed_converted_name: flappy_mixed_latency_train
62
+ single_latency_filter: null
63
+ mixed_latency_filter: null
64
+ force_download: false
65
+ setup_force: false
66
+ skip_verification: false
67
+ verify_rows: 200
68
+ max_episodes: null
69
+ episodes_per_latency: null
70
+ latency_filter: ${dataset.single_latency_filter}
71
+ debug_subset:
72
+ enabled: false
73
+ max_episodes: 5
74
+ suffix: debug
75
+ base_model:
76
+ repo_id: ${bridge_base_model.repo_id.${model}}
77
+ initialization:
78
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
79
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
80
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
81
+ trainer:
82
+ max_train_steps: 5000
83
+ num_warmup_steps: 100
84
+ save_interval: 500
85
+ eval_interval: 100
86
+ eval_num_batches: 100
87
+ per_latency_eval_num_batches: null
88
+ eval_action_classification: true
89
+ eval_action_classification_interval: null
90
+ cc_f1_tolerance: 1
91
+ learning_rate:
92
+ base: 2.0e-05
93
+ qwen_vl_interface: 1.0e-05
94
+ action_model: 0.0001
95
+ lr_scheduler_type: cosine_with_min_lr
96
+ scheduler_specific_kwargs:
97
+ min_lr: 1.0e-06
98
+ freeze_modules: ''
99
+ freeze_llm_bottom_ratio: 0.0
100
+ loss_scale:
101
+ vla: 1.0
102
+ vlm: 0.1
103
+ max_grad_norm: 1.0
104
+ weight_decay: 0.0
105
+ logging_frequency: 1
106
+ gradient_clipping: 1.0
107
+ gradient_accumulation_steps: 4
108
+ distributed_backend: none
109
+ is_resume: false
110
+ pretrained_checkpoint: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning_corrected/checkpoints/_initialization/StarVLA_Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
111
+ resume_step: 0
112
+ reload_modules: null
113
+ optimizer:
114
+ name: AdamW
115
+ betas:
116
+ - 0.9
117
+ - 0.95
118
+ eps: 1.0e-08
119
+ weight_decay: 1.0e-08
120
+ fused: true
121
+ save_format: pt
122
+ workspace_dir: WORKSPACE_DIR
123
+ run_root_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints
124
+ seed: 42
125
+ wandb_entity: saberrr-zju
126
+ wandb_project: starVLA_rl_games
127
+ auth:
128
+ env_file: null
129
+ hf_token_env: HF_TOKEN
130
+ wandb_api_key_env: WANDB_API_KEY
131
+ paths:
132
+ run_root_dir: results/Checkpoints
133
+ dataset_local_dir: playground/Datasets/rl_games
134
+ dataset_cache_dir: null
135
+ base_model_dir: ${bridge_base_model.local_dir.${model}}
136
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
137
+ launch:
138
+ use_accelerate: true
139
+ gpus: null
140
+ num_processes: 1
141
+ dry_run: false
142
+ conda:
143
+ enabled: true
144
+ env_name: null
145
+ rl_games:
146
+ model_alias: openvla
147
+ env_eval:
148
+ image_size: 224
149
+ frameskip: 1
150
+ seed: 42
151
+ fixed_episode_seeds: true
152
+ latency_seed_stride: 0
153
+ task_seed_stride: 0
154
+ task_description: ''
155
+ enabled: true
156
+ distributed_mode: none
157
+ vectorized:
158
+ enabled: false
159
+ batch_size: 1
160
+ latency:
161
+ prompt_map_path: /workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
162
+ mode: single
163
+ values:
164
+ - 0
165
+ mid_train:
166
+ enabled: true
167
+ interval_steps: 250
168
+ latencies:
169
+ - 2
170
+ num_episodes: 5
171
+ max_steps_per_episode: 3600
172
+ post_train:
173
+ enabled: false
174
+ latencies:
175
+ - 0
176
+ - 1
177
+ - 2
178
+ - 3
179
+ - 4
180
+ num_episodes: 5
181
+ max_steps_per_episode: 3600
182
+ task: flappy
183
+ initialization_mode: bridge
184
+ action_carrier: bridge
185
+ model: openvla
186
+ env: flappy
187
+ init: bridge
188
+ bridge_base_model:
189
+ repo_id:
190
+ openvla: Qwen/Qwen3-VL-4B-Instruct
191
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
192
+ pi05: Qwen/Qwen3-VL-4B-Instruct
193
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
194
+ local_dir:
195
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
196
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
197
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
198
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
199
+ mode: single
200
+ checkpoint:
201
+ load: auto
202
+ hf_repo_id: null
203
+ save_best_model: false
204
+ save_pt_file: false
205
+ local:
206
+ keep_last_n: 1
207
+ sync:
208
+ enabled: false
209
+ repo_id: null
210
+ keep_last_n: 0
211
+ sync_every_n_checkpoints: 1
212
+ resume_policy: local_latest
213
+ run_id: flappy_fix_latency_2_200ep_full_tuning_corrected
214
+ output_dir: null
215
+ config_yaml: null
216
+ is_debug: false
217
+ version_id: '0.21'
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected/hydra/.hydra/hydra.yaml ADDED
@@ -0,0 +1,266 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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=flappy_fix_latency_2_200ep_full_tuning_corrected
120
+ - run_root_dir=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints
121
+ - seed=42
122
+ - wandb_entity=saberrr-zju
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=false
129
+ - checkpoint.save_pt_file=false
130
+ - trainer.is_resume=false
131
+ - trainer.pretrained_checkpoint=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning_corrected/checkpoints/_initialization/StarVLA_Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
132
+ - trainer.resume_step=0
133
+ - trainer.max_train_steps=5000
134
+ - trainer.num_warmup_steps=100
135
+ - trainer.save_interval=500
136
+ - trainer.eval_interval=100
137
+ - trainer.eval_num_batches=100
138
+ - trainer.eval_action_classification=true
139
+ - trainer.logging_frequency=1
140
+ - trainer.gradient_accumulation_steps=4
141
+ - trainer.distributed_backend=none
142
+ - trainer.learning_rate.base=2e-05
143
+ - trainer.learning_rate.qwen_vl_interface=1e-05
144
+ - trainer.learning_rate.action_model=0.0001
145
+ - trainer.lr_scheduler_type=cosine_with_min_lr
146
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
147
+ - trainer.freeze_llm_bottom_ratio=0.0
148
+ - trainer.loss_scale.vla=1.0
149
+ - trainer.loss_scale.vlm=0.1
150
+ - trainer.max_grad_norm=1.0
151
+ - trainer.weight_decay=0.0
152
+ - trainer.gradient_clipping=1.0
153
+ - trainer.optimizer.name=AdamW
154
+ - trainer.optimizer.betas=[0.9,0.95]
155
+ - trainer.optimizer.eps=1e-08
156
+ - trainer.optimizer.weight_decay=1e-08
157
+ - trainer.optimizer.fused=true
158
+ - trainer.save_format=pt
159
+ - framework.name=QwenOFT
160
+ - framework.qwenvl.attn_implementation=flash_attention_2
161
+ - framework.qwenvl.enable_gradient_checkpointing=true
162
+ - framework.action_model.action_dim=7
163
+ - framework.action_model.action_env_dim=2
164
+ - framework.action_model.state_dim=7
165
+ - framework.action_model.loss_type=discrete_ce
166
+ - framework.action_model.action_horizon=1
167
+ - framework.action_model.future_action_window_size=0
168
+ - framework.action_model.past_action_window_size=0
169
+ - datasets.vla_data.include_state=true
170
+ - datasets.vla_data.action_type=discrete
171
+ - datasets.vla_data.sequential_step_sampling=false
172
+ - datasets.vla_data.shuffle=true
173
+ - datasets.vla_data.num_workers=8
174
+ - datasets.vla_data.eval_num_workers=8
175
+ - datasets.vla_data.prefetch_factor=4
176
+ - datasets.vla_data.persistent_workers=true
177
+ - datasets.vla_data.pin_memory=true
178
+ - datasets.vla_data.action_balance.enabled=false
179
+ - datasets.vla_data.action_balance.strategy=balanced_epoch
180
+ - datasets.vla_data.action_balance.action_key=action_id
181
+ - datasets.vla_data.action_balance.target_flap_fraction=0.3
182
+ - datasets.vla_data.action_balance.noop_id=0
183
+ - datasets.vla_data.action_balance.flap_id=1
184
+ - datasets.vla_data.latency_curriculum.enabled=false
185
+ - datasets.vla_data.latency_curriculum.strategy=exclusive
186
+ - datasets.vla_data.per_device_batch_size=32
187
+ - datasets.vla_data.num_workers=8
188
+ - datasets.vla_data.eval_num_workers=8
189
+ - datasets.vla_data.prefetch_factor=4
190
+ - datasets.vla_data.persistent_workers=true
191
+ - datasets.vla_data.pin_memory=true
192
+ - datasets.vla_data.load_all_data_for_training=true
193
+ - datasets.vla_data.video_backend=torchvision_av
194
+ - datasets.vla_data.data_root_dir=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep
195
+ - datasets.vla_data.data_mix=flappy_train__bridge
196
+ - datasets.vla_data.eval_data_mix=flappy_train__bridge__val
197
+ - framework.qwenvl.base_vlm=/workspace/latency-sensitive-bench/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
198
+ - rl_games.task=flappy
199
+ - rl_games.model_alias=openvla
200
+ - rl_games.initialization_mode=bridge
201
+ - rl_games.action_carrier=bridge
202
+ - rl_games.env_eval.distributed_mode=none
203
+ - rl_games.env_eval.latency.mode=single
204
+ - rl_games.env_eval.frameskip=1
205
+ - rl_games.env_eval.image_size=224
206
+ - rl_games.env_eval.seed=42
207
+ - rl_games.env_eval.fixed_episode_seeds=true
208
+ - rl_games.env_eval.latency_seed_stride=0
209
+ - rl_games.env_eval.task_seed_stride=0
210
+ - rl_games.env_eval.latency.values=[0]
211
+ - rl_games.env_eval.latency.prompt_map_path=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
212
+ - rl_games.env_eval.mid_train.enabled=true
213
+ - rl_games.env_eval.mid_train.interval_steps=250
214
+ - rl_games.env_eval.mid_train.num_episodes=5
215
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
216
+ - rl_games.env_eval.mid_train.latencies=[2]
217
+ - rl_games.env_eval.post_train.enabled=false
218
+ - rl_games.env_eval.post_train.num_episodes=5
219
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
220
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
221
+ job:
222
+ name: train_starvla_hydra
223
+ chdir: false
224
+ override_dirname: checkpoint.local.keep_last_n=1,checkpoint.save_best_model=false,checkpoint.save_pt_file=false,checkpoint.sync.enabled=false,checkpoint.sync.keep_last_n=0,datasets.vla_data.action_balance.action_key=action_id,datasets.vla_data.action_balance.enabled=false,datasets.vla_data.action_balance.flap_id=1,datasets.vla_data.action_balance.noop_id=0,datasets.vla_data.action_balance.strategy=balanced_epoch,datasets.vla_data.action_balance.target_flap_fraction=0.3,datasets.vla_data.action_type=discrete,datasets.vla_data.data_mix=flappy_train__bridge,datasets.vla_data.data_root_dir=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep,datasets.vla_data.eval_data_mix=flappy_train__bridge__val,datasets.vla_data.eval_num_workers=8,datasets.vla_data.eval_num_workers=8,datasets.vla_data.include_state=true,datasets.vla_data.latency_curriculum.enabled=false,datasets.vla_data.latency_curriculum.strategy=exclusive,datasets.vla_data.load_all_data_for_training=true,datasets.vla_data.num_workers=8,datasets.vla_data.num_workers=8,datasets.vla_data.per_device_batch_size=32,datasets.vla_data.persistent_workers=true,datasets.vla_data.persistent_workers=true,datasets.vla_data.pin_memory=true,datasets.vla_data.pin_memory=true,datasets.vla_data.prefetch_factor=4,datasets.vla_data.prefetch_factor=4,datasets.vla_data.sequential_step_sampling=false,datasets.vla_data.shuffle=true,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/latency-sensitive-bench/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=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/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=250,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=false,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=flappy_fix_latency_2_200ep_full_tuning_corrected,run_root_dir=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints,seed=42,trainer.distributed_backend=none,trainer.eval_action_classification=true,trainer.eval_interval=100,trainer.eval_num_batches=100,trainer.freeze_llm_bottom_ratio=0.0,trainer.gradient_accumulation_steps=4,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=1,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=5000,trainer.num_warmup_steps=100,trainer.optimizer.betas=[0.9,0.95],trainer.optimizer.eps=1e-08,trainer.optimizer.fused=true,trainer.optimizer.name=AdamW,trainer.optimizer.weight_decay=1e-08,trainer.pretrained_checkpoint=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning_corrected/checkpoints/_initialization/StarVLA_Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt,trainer.resume_step=0,trainer.save_format=pt,trainer.save_interval=500,trainer.scheduler_specific_kwargs.min_lr=1e-06,trainer.weight_decay=0.0,wandb_entity=saberrr-zju,wandb_project=starVLA_rl_games
225
+ id: ???
226
+ num: ???
227
+ config_name: train
228
+ env_set: {}
229
+ env_copy: []
230
+ config:
231
+ override_dirname:
232
+ kv_sep: '='
233
+ item_sep: ','
234
+ exclude_keys: []
235
+ runtime:
236
+ version: 1.3.3
237
+ version_base: '1.1'
238
+ cwd: /workspace/latency-sensitive-bench/starVLA
239
+ config_sources:
240
+ - path: hydra.conf
241
+ schema: pkg
242
+ provider: hydra
243
+ - path: /workspace/latency-sensitive-bench/starVLA/examples/rl_games/config
244
+ schema: file
245
+ provider: main
246
+ - path: ''
247
+ schema: structured
248
+ provider: schema
249
+ output_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning_corrected/hydra
250
+ choices:
251
+ cross_task_setup: null
252
+ checkpoint: default
253
+ mode: single
254
+ init: bridge
255
+ env: flappy
256
+ model: openvla
257
+ hydra/env: default
258
+ hydra/callbacks: null
259
+ hydra/job_logging: default
260
+ hydra/hydra_logging: default
261
+ hydra/hydra_help: default
262
+ hydra/help: default
263
+ hydra/sweeper: basic
264
+ hydra/launcher: basic
265
+ hydra/output: default
266
+ verbose: false
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected/hydra/.hydra/overrides.yaml ADDED
@@ -0,0 +1,106 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ - model=openvla
2
+ - env=flappy
3
+ - init=bridge
4
+ - mode=single
5
+ - run_id=flappy_fix_latency_2_200ep_full_tuning_corrected
6
+ - run_root_dir=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints
7
+ - seed=42
8
+ - wandb_entity=saberrr-zju
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=false
15
+ - checkpoint.save_pt_file=false
16
+ - trainer.is_resume=false
17
+ - trainer.pretrained_checkpoint=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning_corrected/checkpoints/_initialization/StarVLA_Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
18
+ - trainer.resume_step=0
19
+ - trainer.max_train_steps=5000
20
+ - trainer.num_warmup_steps=100
21
+ - trainer.save_interval=500
22
+ - trainer.eval_interval=100
23
+ - trainer.eval_num_batches=100
24
+ - trainer.eval_action_classification=true
25
+ - trainer.logging_frequency=1
26
+ - trainer.gradient_accumulation_steps=4
27
+ - trainer.distributed_backend=none
28
+ - trainer.learning_rate.base=2e-05
29
+ - trainer.learning_rate.qwen_vl_interface=1e-05
30
+ - trainer.learning_rate.action_model=0.0001
31
+ - trainer.lr_scheduler_type=cosine_with_min_lr
32
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
33
+ - trainer.freeze_llm_bottom_ratio=0.0
34
+ - trainer.loss_scale.vla=1.0
35
+ - trainer.loss_scale.vlm=0.1
36
+ - trainer.max_grad_norm=1.0
37
+ - trainer.weight_decay=0.0
38
+ - trainer.gradient_clipping=1.0
39
+ - trainer.optimizer.name=AdamW
40
+ - trainer.optimizer.betas=[0.9,0.95]
41
+ - trainer.optimizer.eps=1e-08
42
+ - trainer.optimizer.weight_decay=1e-08
43
+ - trainer.optimizer.fused=true
44
+ - trainer.save_format=pt
45
+ - framework.name=QwenOFT
46
+ - framework.qwenvl.attn_implementation=flash_attention_2
47
+ - framework.qwenvl.enable_gradient_checkpointing=true
48
+ - framework.action_model.action_dim=7
49
+ - framework.action_model.action_env_dim=2
50
+ - framework.action_model.state_dim=7
51
+ - framework.action_model.loss_type=discrete_ce
52
+ - framework.action_model.action_horizon=1
53
+ - framework.action_model.future_action_window_size=0
54
+ - framework.action_model.past_action_window_size=0
55
+ - datasets.vla_data.include_state=true
56
+ - datasets.vla_data.action_type=discrete
57
+ - datasets.vla_data.sequential_step_sampling=false
58
+ - datasets.vla_data.shuffle=true
59
+ - datasets.vla_data.num_workers=8
60
+ - datasets.vla_data.eval_num_workers=8
61
+ - datasets.vla_data.prefetch_factor=4
62
+ - datasets.vla_data.persistent_workers=true
63
+ - datasets.vla_data.pin_memory=true
64
+ - datasets.vla_data.action_balance.enabled=false
65
+ - datasets.vla_data.action_balance.strategy=balanced_epoch
66
+ - datasets.vla_data.action_balance.action_key=action_id
67
+ - datasets.vla_data.action_balance.target_flap_fraction=0.3
68
+ - datasets.vla_data.action_balance.noop_id=0
69
+ - datasets.vla_data.action_balance.flap_id=1
70
+ - datasets.vla_data.latency_curriculum.enabled=false
71
+ - datasets.vla_data.latency_curriculum.strategy=exclusive
72
+ - datasets.vla_data.per_device_batch_size=32
73
+ - datasets.vla_data.num_workers=8
74
+ - datasets.vla_data.eval_num_workers=8
75
+ - datasets.vla_data.prefetch_factor=4
76
+ - datasets.vla_data.persistent_workers=true
77
+ - datasets.vla_data.pin_memory=true
78
+ - datasets.vla_data.load_all_data_for_training=true
79
+ - datasets.vla_data.video_backend=torchvision_av
80
+ - datasets.vla_data.data_root_dir=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep
81
+ - datasets.vla_data.data_mix=flappy_train__bridge
82
+ - datasets.vla_data.eval_data_mix=flappy_train__bridge__val
83
+ - framework.qwenvl.base_vlm=/workspace/latency-sensitive-bench/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
84
+ - rl_games.task=flappy
85
+ - rl_games.model_alias=openvla
86
+ - rl_games.initialization_mode=bridge
87
+ - rl_games.action_carrier=bridge
88
+ - rl_games.env_eval.distributed_mode=none
89
+ - rl_games.env_eval.latency.mode=single
90
+ - rl_games.env_eval.frameskip=1
91
+ - rl_games.env_eval.image_size=224
92
+ - rl_games.env_eval.seed=42
93
+ - rl_games.env_eval.fixed_episode_seeds=true
94
+ - rl_games.env_eval.latency_seed_stride=0
95
+ - rl_games.env_eval.task_seed_stride=0
96
+ - rl_games.env_eval.latency.values=[0]
97
+ - rl_games.env_eval.latency.prompt_map_path=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
98
+ - rl_games.env_eval.mid_train.enabled=true
99
+ - rl_games.env_eval.mid_train.interval_steps=250
100
+ - rl_games.env_eval.mid_train.num_episodes=5
101
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
102
+ - rl_games.env_eval.mid_train.latencies=[2]
103
+ - rl_games.env_eval.post_train.enabled=false
104
+ - rl_games.env_eval.post_train.num_episodes=5
105
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
106
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected/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/flappy_fix_latency_2_200ep_full_tuning_corrected/summary.jsonl ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {"steps": 500}
2
+ {"steps": 1000}
3
+ {"steps": 1500}
4
+ {"steps": 2000}
5
+ {"steps": 2500}
6
+ {"steps": 3000}
7
+ {"steps": 3500}
8
+ {"steps": 4000}
9
+ {"steps": 4500}
10
+ {"steps": 5000}
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/checkpoints/steps_3000_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dee60fc99cf35e2e6d5908129093c8c66c2cafb62d3b87a236d45acd86522d59
3
+ size 9138230516
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/config.full.yaml ADDED
@@ -0,0 +1,224 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ name: QwenOFT
3
+ qwenvl:
4
+ base_vlm: /workspace/latency-sensitive-bench/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/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep
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
+ eval_sequential_step_sampling: null
28
+ num_workers: 8
29
+ eval_num_workers: 8
30
+ prefetch_factor: 4
31
+ persistent_workers: true
32
+ pin_memory: true
33
+ shuffle: true
34
+ action_balance:
35
+ enabled: false
36
+ strategy: balanced_epoch
37
+ action_key: action_id
38
+ target_flap_fraction: 0.3
39
+ noop_id: 0
40
+ flap_id: 1
41
+ latency_curriculum:
42
+ enabled: false
43
+ strategy: exclusive
44
+ latencies: null
45
+ phase_steps: null
46
+ per_device_batch_size: 32
47
+ load_all_data_for_training: true
48
+ num_obs_frames: 1
49
+ image_mode: single
50
+ stitch_grid:
51
+ - 2
52
+ - 2
53
+ obs_image_size: null
54
+ video_backend: torchvision_av
55
+ dataset:
56
+ source_hf: ''
57
+ config_name: null
58
+ source_subdir: null
59
+ converted_name: flappy_train
60
+ single_source_hf: ''
61
+ mixed_source_hf: ''
62
+ single_converted_name: flappy_train
63
+ mixed_converted_name: flappy_mixed_latency_train
64
+ single_latency_filter: null
65
+ mixed_latency_filter: null
66
+ force_download: false
67
+ setup_force: false
68
+ skip_verification: false
69
+ verify_rows: 200
70
+ max_episodes: null
71
+ episodes_per_latency: null
72
+ latency_filter: null
73
+ debug_subset:
74
+ enabled: false
75
+ max_episodes: 5
76
+ suffix: debug
77
+ base_model:
78
+ repo_id: Qwen/Qwen3-VL-4B-Instruct
79
+ initialization:
80
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
81
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
82
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
83
+ trainer:
84
+ max_train_steps: 3000
85
+ num_warmup_steps: 100
86
+ save_interval: 500
87
+ eval_interval: 250
88
+ eval_num_batches: 50
89
+ per_latency_eval_num_batches: null
90
+ eval_action_classification: true
91
+ eval_action_classification_interval: null
92
+ cc_f1_tolerance: 1
93
+ learning_rate:
94
+ base: 2.0e-05
95
+ qwen_vl_interface: 1.0e-05
96
+ action_model: 0.0001
97
+ lr_scheduler_type: cosine_with_min_lr
98
+ scheduler_specific_kwargs:
99
+ min_lr: 1.0e-06
100
+ freeze_modules: ''
101
+ freeze_vit: false
102
+ freeze_tied_embedding: false
103
+ freeze_llm_layers: []
104
+ loss_scale:
105
+ vla: 1.0
106
+ vlm: 0.1
107
+ max_grad_norm: 1.0
108
+ weight_decay: 0.0
109
+ logging_frequency: 1
110
+ profile_timing:
111
+ enabled: false
112
+ log_interval: 10
113
+ gradient_clipping: 1.0
114
+ gradient_accumulation_steps: 1
115
+ distributed_backend: none
116
+ is_resume: false
117
+ pretrained_checkpoint: null
118
+ resume_step: 0
119
+ reload_modules: null
120
+ optimizer:
121
+ name: AdamW
122
+ betas:
123
+ - 0.9
124
+ - 0.95
125
+ eps: 1.0e-08
126
+ weight_decay: 1.0e-08
127
+ fused: true
128
+ save_format: pt
129
+ workspace_dir: WORKSPACE_DIR
130
+ run_root_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints
131
+ seed: 42
132
+ wandb_entity: saberrr-zju
133
+ wandb_project: starVLA_rl_games
134
+ auth:
135
+ env_file: null
136
+ hf_token_env: HF_TOKEN
137
+ wandb_api_key_env: WANDB_API_KEY
138
+ paths:
139
+ run_root_dir: results/Checkpoints
140
+ dataset_local_dir: data/flappy_fix_latency_2_200ep
141
+ dataset_cache_dir: null
142
+ base_model_dir: playground/Pretrained_models/Qwen3-VL-4B-Instruct
143
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
144
+ launch:
145
+ use_accelerate: true
146
+ gpus: null
147
+ num_processes: 1
148
+ dry_run: false
149
+ conda:
150
+ enabled: true
151
+ env_name: null
152
+ rl_games:
153
+ model_alias: openvla
154
+ env_eval:
155
+ image_size: 224
156
+ frameskip: 1
157
+ seed: 42
158
+ fixed_episode_seeds: true
159
+ latency_seed_stride: 0
160
+ task_seed_stride: 0
161
+ task_description: ''
162
+ enabled: true
163
+ distributed_mode: none
164
+ vectorized:
165
+ enabled: false
166
+ batch_size: 1
167
+ latency:
168
+ prompt_map_path: /workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
169
+ mode: single
170
+ values:
171
+ - 0
172
+ mid_train:
173
+ enabled: true
174
+ interval_steps: 250
175
+ latencies:
176
+ - 2
177
+ num_episodes: 5
178
+ max_steps_per_episode: 3600
179
+ post_train:
180
+ enabled: false
181
+ latencies:
182
+ - 0
183
+ - 1
184
+ - 2
185
+ - 3
186
+ - 4
187
+ num_episodes: 5
188
+ max_steps_per_episode: 3600
189
+ task: flappy
190
+ initialization_mode: bridge
191
+ action_carrier: bridge
192
+ model: openvla
193
+ env: flappy
194
+ init: bridge
195
+ bridge_base_model:
196
+ repo_id:
197
+ openvla: Qwen/Qwen3-VL-4B-Instruct
198
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
199
+ pi05: Qwen/Qwen3-VL-4B-Instruct
200
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
201
+ local_dir:
202
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
203
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
204
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
205
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
206
+ mode: single
207
+ checkpoint:
208
+ load: none
209
+ hf_repo_id: null
210
+ save_best_model: false
211
+ save_pt_file: false
212
+ local:
213
+ keep_last_n: 1
214
+ sync:
215
+ enabled: false
216
+ repo_id: null
217
+ keep_last_n: 0
218
+ sync_every_n_checkpoints: 1
219
+ resume_policy: local_latest
220
+ run_id: flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps
221
+ output_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps
222
+ config_yaml: null
223
+ is_debug: false
224
+ version_id: '0.21'
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/config.yaml ADDED
@@ -0,0 +1,103 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ checkpoint:
2
+ local:
3
+ keep_last_n: 1
4
+ save_best_model: false
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
+ eval_num_workers: 8
16
+ include_state: true
17
+ latency_curriculum:
18
+ enabled: false
19
+ obs_image_size: null
20
+ per_device_batch_size: 32
21
+ persistent_workers: true
22
+ pin_memory: true
23
+ prefetch_factor: 4
24
+ framework:
25
+ action_model:
26
+ action_dim: 7
27
+ action_env_dim: 2
28
+ action_hidden_dim: 2560
29
+ action_horizon: 1
30
+ action_model_type: MLP
31
+ loss_type: discrete_ce
32
+ state_dim: 7
33
+ name: QwenOFT
34
+ qwenvl:
35
+ attn_implementation: flash_attention_2
36
+ base_vlm: /workspace/latency-sensitive-bench/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
37
+ enable_gradient_checkpointing: true
38
+ output_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps
39
+ rl_games:
40
+ env_eval:
41
+ distributed_mode: none
42
+ enabled: true
43
+ fixed_episode_seeds: true
44
+ frameskip: 1
45
+ image_size: 224
46
+ latency:
47
+ prompt_map_path: /workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
48
+ latency_seed_stride: 0
49
+ mid_train:
50
+ enabled: true
51
+ interval_steps: 250
52
+ latencies:
53
+ - 2
54
+ max_steps_per_episode: 3600
55
+ num_episodes: 5
56
+ seed: 42
57
+ task_description: ''
58
+ task_seed_stride: 0
59
+ vectorized:
60
+ enabled: false
61
+ model_alias: openvla
62
+ task: flappy
63
+ run_id: flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps
64
+ run_root_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints
65
+ seed: 42
66
+ trainer:
67
+ distributed_backend: none
68
+ eval_action_classification: true
69
+ eval_action_classification_interval: null
70
+ eval_interval: 250
71
+ eval_num_batches: 50
72
+ freeze_llm_layers: []
73
+ freeze_modules: ''
74
+ freeze_tied_embedding: false
75
+ freeze_vit: false
76
+ gradient_accumulation_steps: 1
77
+ gradient_clipping: 1.0
78
+ is_resume: false
79
+ learning_rate:
80
+ action_model: 0.0001
81
+ base: 2.0e-05
82
+ qwen_vl_interface: 1.0e-05
83
+ logging_frequency: 1
84
+ lr_scheduler_type: cosine_with_min_lr
85
+ max_train_steps: 3000
86
+ num_warmup_steps: 100
87
+ optimizer:
88
+ betas:
89
+ - 0.9
90
+ - 0.95
91
+ eps: 1.0e-08
92
+ fused: true
93
+ weight_decay: 1.0e-08
94
+ per_latency_eval_num_batches: null
95
+ pretrained_checkpoint: null
96
+ profile_timing:
97
+ enabled: false
98
+ reload_modules: null
99
+ save_interval: 500
100
+ scheduler_specific_kwargs:
101
+ min_lr: 1.0e-06
102
+ wandb_entity: saberrr-zju
103
+ wandb_project: starVLA_rl_games
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/dataset_statistics.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.6028500199317932,
6
+ 0.3971499800682068,
7
+ 0.0,
8
+ 0.0,
9
+ 0.0,
10
+ 0.0,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.4890792667865753,
15
+ 0.4890792667865753,
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": 330734,
125
+ "num_trajectories": 180
126
+ }
127
+ }
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/dataset_statistics_eval.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.7959861159324646,
6
+ 0.2040138840675354,
7
+ 0.0,
8
+ 0.0,
9
+ 0.0,
10
+ 0.0,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.4030573070049286,
15
+ 0.4030573070049286,
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": 72000,
125
+ "num_trajectories": 20
126
+ }
127
+ }
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/hydra/.hydra/config.yaml ADDED
@@ -0,0 +1,222 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ qwenvl:
3
+ base_vlm: /workspace/latency-sensitive-bench/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/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep
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
+ eval_sequential_step_sampling: null
26
+ num_workers: 8
27
+ eval_num_workers: 8
28
+ prefetch_factor: 4
29
+ persistent_workers: true
30
+ pin_memory: true
31
+ shuffle: true
32
+ action_balance:
33
+ enabled: false
34
+ strategy: balanced_epoch
35
+ action_key: action_id
36
+ target_flap_fraction: 0.3
37
+ noop_id: 0
38
+ flap_id: 1
39
+ latency_curriculum:
40
+ enabled: false
41
+ strategy: exclusive
42
+ latencies: null
43
+ phase_steps: null
44
+ per_device_batch_size: 32
45
+ load_all_data_for_training: true
46
+ num_obs_frames: 1
47
+ image_mode: single
48
+ stitch_grid:
49
+ - 2
50
+ - 2
51
+ obs_image_size: null
52
+ video_backend: torchvision_av
53
+ dataset:
54
+ source_hf: ''
55
+ config_name: null
56
+ source_subdir: null
57
+ converted_name: flappy_train
58
+ single_source_hf: ''
59
+ mixed_source_hf: ''
60
+ single_converted_name: flappy_train
61
+ mixed_converted_name: flappy_mixed_latency_train
62
+ single_latency_filter: null
63
+ mixed_latency_filter: null
64
+ force_download: false
65
+ setup_force: false
66
+ skip_verification: false
67
+ verify_rows: 200
68
+ max_episodes: null
69
+ episodes_per_latency: null
70
+ latency_filter: null
71
+ debug_subset:
72
+ enabled: false
73
+ max_episodes: 5
74
+ suffix: debug
75
+ base_model:
76
+ repo_id: Qwen/Qwen3-VL-4B-Instruct
77
+ initialization:
78
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
79
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
80
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
81
+ trainer:
82
+ max_train_steps: 3000
83
+ num_warmup_steps: 100
84
+ save_interval: 500
85
+ eval_interval: 250
86
+ eval_num_batches: 50
87
+ per_latency_eval_num_batches: null
88
+ eval_action_classification: true
89
+ eval_action_classification_interval: null
90
+ cc_f1_tolerance: 1
91
+ learning_rate:
92
+ base: 2.0e-05
93
+ qwen_vl_interface: 1.0e-05
94
+ action_model: 0.0001
95
+ lr_scheduler_type: cosine_with_min_lr
96
+ scheduler_specific_kwargs:
97
+ min_lr: 1.0e-06
98
+ freeze_modules: ''
99
+ freeze_vit: false
100
+ freeze_tied_embedding: false
101
+ freeze_llm_layers: []
102
+ loss_scale:
103
+ vla: 1.0
104
+ vlm: 0.1
105
+ max_grad_norm: 1.0
106
+ weight_decay: 0.0
107
+ logging_frequency: 1
108
+ profile_timing:
109
+ enabled: false
110
+ log_interval: 10
111
+ gradient_clipping: 1.0
112
+ gradient_accumulation_steps: 1
113
+ distributed_backend: none
114
+ is_resume: false
115
+ pretrained_checkpoint: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/checkpoints/_initialization/StarVLA_Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
116
+ resume_step: 0
117
+ reload_modules: null
118
+ optimizer:
119
+ name: AdamW
120
+ betas:
121
+ - 0.9
122
+ - 0.95
123
+ eps: 1.0e-08
124
+ weight_decay: 1.0e-08
125
+ fused: true
126
+ save_format: pt
127
+ workspace_dir: WORKSPACE_DIR
128
+ run_root_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints
129
+ seed: 42
130
+ wandb_entity: saberrr-zju
131
+ wandb_project: starVLA_rl_games
132
+ auth:
133
+ env_file: null
134
+ hf_token_env: HF_TOKEN
135
+ wandb_api_key_env: WANDB_API_KEY
136
+ paths:
137
+ run_root_dir: results/Checkpoints
138
+ dataset_local_dir: data/flappy_fix_latency_2_200ep
139
+ dataset_cache_dir: null
140
+ base_model_dir: playground/Pretrained_models/Qwen3-VL-4B-Instruct
141
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
142
+ launch:
143
+ use_accelerate: true
144
+ gpus: null
145
+ num_processes: 1
146
+ dry_run: false
147
+ conda:
148
+ enabled: true
149
+ env_name: null
150
+ rl_games:
151
+ model_alias: openvla
152
+ env_eval:
153
+ image_size: 224
154
+ frameskip: 1
155
+ seed: 42
156
+ fixed_episode_seeds: true
157
+ latency_seed_stride: 0
158
+ task_seed_stride: 0
159
+ task_description: ''
160
+ enabled: true
161
+ distributed_mode: none
162
+ vectorized:
163
+ enabled: false
164
+ batch_size: 1
165
+ latency:
166
+ prompt_map_path: /workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
167
+ mode: single
168
+ values:
169
+ - 0
170
+ mid_train:
171
+ enabled: true
172
+ interval_steps: 250
173
+ latencies:
174
+ - 2
175
+ num_episodes: 5
176
+ max_steps_per_episode: 3600
177
+ post_train:
178
+ enabled: false
179
+ latencies:
180
+ - 0
181
+ - 1
182
+ - 2
183
+ - 3
184
+ - 4
185
+ num_episodes: 5
186
+ max_steps_per_episode: 3600
187
+ task: flappy
188
+ initialization_mode: bridge
189
+ action_carrier: bridge
190
+ model: openvla
191
+ env: flappy
192
+ init: bridge
193
+ bridge_base_model:
194
+ repo_id:
195
+ openvla: Qwen/Qwen3-VL-4B-Instruct
196
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
197
+ pi05: Qwen/Qwen3-VL-4B-Instruct
198
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
199
+ local_dir:
200
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
201
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
202
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
203
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
204
+ mode: single
205
+ checkpoint:
206
+ load: none
207
+ hf_repo_id: null
208
+ save_best_model: false
209
+ save_pt_file: false
210
+ local:
211
+ keep_last_n: 1
212
+ sync:
213
+ enabled: false
214
+ repo_id: null
215
+ keep_last_n: 0
216
+ sync_every_n_checkpoints: 1
217
+ resume_policy: local_latest
218
+ run_id: flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps
219
+ output_dir: null
220
+ config_yaml: null
221
+ is_debug: false
222
+ version_id: 0.21
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/hydra/.hydra/hydra.yaml ADDED
@@ -0,0 +1,341 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ - ++framework.qwenvl.base_vlm=playground/Pretrained_models/Qwen3-VL-4B-Instruct
120
+ - ++framework.qwenvl.attn_implementation=flash_attention_2
121
+ - ++framework.qwenvl.enable_gradient_checkpointing=true
122
+ - ++framework.action_model.state_dim=7
123
+ - ++framework.action_model.loss_type=discrete_ce
124
+ - ++framework.action_model.action_horizon=1
125
+ - ++framework.action_model.future_action_window_size=0
126
+ - ++framework.action_model.past_action_window_size=0
127
+ - ++framework.action_model.action_dim=7
128
+ - ++framework.action_model.action_env_dim=2
129
+ - ++framework.name=QwenOFT
130
+ - ++datasets.vla_data.dataset_py=lerobot_datasets
131
+ - ++datasets.vla_data.include_state=true
132
+ - ++datasets.vla_data.data_root_dir=playground/Datasets/rl_games
133
+ - ++datasets.vla_data.data_mix=flappy_train
134
+ - ++datasets.vla_data.eval_data_mix=null
135
+ - ++datasets.vla_data.custom_mixtures_path=null
136
+ - ++datasets.vla_data.action_type=discrete
137
+ - ++datasets.vla_data.sequential_step_sampling=false
138
+ - ++datasets.vla_data.eval_sequential_step_sampling=null
139
+ - ++datasets.vla_data.num_workers=8
140
+ - ++datasets.vla_data.eval_num_workers=8
141
+ - ++datasets.vla_data.prefetch_factor=4
142
+ - ++datasets.vla_data.persistent_workers=true
143
+ - ++datasets.vla_data.pin_memory=true
144
+ - ++datasets.vla_data.shuffle=true
145
+ - ++datasets.vla_data.action_balance.enabled=false
146
+ - ++datasets.vla_data.action_balance.strategy=balanced_epoch
147
+ - ++datasets.vla_data.action_balance.action_key=action_id
148
+ - ++datasets.vla_data.action_balance.target_flap_fraction=0.3
149
+ - ++datasets.vla_data.action_balance.noop_id=0
150
+ - ++datasets.vla_data.action_balance.flap_id=1
151
+ - ++datasets.vla_data.latency_curriculum.enabled=false
152
+ - ++datasets.vla_data.latency_curriculum.strategy=exclusive
153
+ - ++datasets.vla_data.latency_curriculum.latencies=null
154
+ - ++datasets.vla_data.latency_curriculum.phase_steps=null
155
+ - ++datasets.vla_data.per_device_batch_size=32
156
+ - ++datasets.vla_data.load_all_data_for_training=true
157
+ - ++datasets.vla_data.num_obs_frames=1
158
+ - ++datasets.vla_data.image_mode=single
159
+ - ++datasets.vla_data.stitch_grid=[2,2]
160
+ - ++datasets.vla_data.obs_image_size=null
161
+ - ++datasets.vla_data.video_backend=torchvision_av
162
+ - ++dataset.source_hf=
163
+ - ++dataset.config_name=null
164
+ - ++dataset.source_subdir=null
165
+ - ++dataset.converted_name=flappy_train
166
+ - ++dataset.single_source_hf=
167
+ - ++dataset.mixed_source_hf=
168
+ - ++dataset.single_converted_name=flappy_train
169
+ - ++dataset.mixed_converted_name=flappy_mixed_latency_train
170
+ - ++dataset.single_latency_filter=null
171
+ - ++dataset.mixed_latency_filter=null
172
+ - ++dataset.force_download=false
173
+ - ++dataset.setup_force=false
174
+ - ++dataset.skip_verification=false
175
+ - ++dataset.verify_rows=200
176
+ - ++dataset.max_episodes=null
177
+ - ++dataset.episodes_per_latency=null
178
+ - ++dataset.latency_filter=null
179
+ - ++dataset.debug_subset.enabled=false
180
+ - ++dataset.debug_subset.max_episodes=5
181
+ - ++dataset.debug_subset.suffix=debug
182
+ - ++base_model.repo_id=Qwen/Qwen3-VL-4B-Instruct
183
+ - ++initialization.checkpoint_local_dir=playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
184
+ - ++initialization.checkpoint_hf_repo_id=StarVLA/Qwen3VL-OFT-Bridge-RT-1
185
+ - ++initialization.checkpoint_filename=checkpoints/steps_5000_pytorch_model.pt
186
+ - ++trainer.max_train_steps=3000
187
+ - ++trainer.num_warmup_steps=100
188
+ - ++trainer.save_interval=500
189
+ - ++trainer.eval_interval=250
190
+ - ++trainer.eval_num_batches=50
191
+ - ++trainer.per_latency_eval_num_batches=null
192
+ - ++trainer.eval_action_classification=true
193
+ - ++trainer.eval_action_classification_interval=null
194
+ - ++trainer.cc_f1_tolerance=1
195
+ - ++trainer.learning_rate.base=2e-05
196
+ - ++trainer.learning_rate.qwen_vl_interface=1e-05
197
+ - ++trainer.learning_rate.action_model=0.0001
198
+ - ++trainer.lr_scheduler_type=cosine_with_min_lr
199
+ - ++trainer.scheduler_specific_kwargs.min_lr=1e-06
200
+ - ++trainer.freeze_modules=
201
+ - ++trainer.freeze_vit=false
202
+ - ++trainer.freeze_tied_embedding=false
203
+ - ++trainer.freeze_llm_layers=[]
204
+ - ++trainer.loss_scale.vla=1.0
205
+ - ++trainer.loss_scale.vlm=0.1
206
+ - ++trainer.max_grad_norm=1.0
207
+ - ++trainer.weight_decay=0.0
208
+ - ++trainer.logging_frequency=1
209
+ - ++trainer.profile_timing.enabled=false
210
+ - ++trainer.profile_timing.log_interval=10
211
+ - ++trainer.gradient_clipping=1.0
212
+ - ++trainer.gradient_accumulation_steps=1
213
+ - ++trainer.distributed_backend=none
214
+ - ++trainer.is_resume=false
215
+ - ++trainer.pretrained_checkpoint=null
216
+ - ++trainer.resume_step=0
217
+ - ++trainer.reload_modules=null
218
+ - ++trainer.optimizer.name=AdamW
219
+ - ++trainer.optimizer.betas=[0.9,0.95]
220
+ - ++trainer.optimizer.eps=1e-08
221
+ - ++trainer.optimizer.weight_decay=1e-08
222
+ - ++trainer.optimizer.fused=true
223
+ - ++trainer.save_format=pt
224
+ - ++workspace_dir=WORKSPACE_DIR
225
+ - ++run_root_dir=results/Checkpoints
226
+ - ++seed=42
227
+ - ++wandb_entity=saberrr-zju
228
+ - ++wandb_project=starVLA_rl_games
229
+ - ++auth.env_file=null
230
+ - ++auth.hf_token_env=HF_TOKEN
231
+ - ++auth.wandb_api_key_env=WANDB_API_KEY
232
+ - ++paths.run_root_dir=results/Checkpoints
233
+ - ++paths.dataset_local_dir=data/flappy_fix_latency_2_200ep
234
+ - ++paths.dataset_cache_dir=null
235
+ - ++paths.base_model_dir=playground/Pretrained_models/Qwen3-VL-4B-Instruct
236
+ - ++paths.accelerate_config=starVLA/config/deepseeds/deepspeed_zero2.yaml
237
+ - ++rl_games.model_alias=openvla
238
+ - ++rl_games.env_eval.image_size=224
239
+ - ++rl_games.env_eval.frameskip=1
240
+ - ++rl_games.env_eval.seed=42
241
+ - ++rl_games.env_eval.fixed_episode_seeds=true
242
+ - ++rl_games.env_eval.latency_seed_stride=0
243
+ - ++rl_games.env_eval.task_seed_stride=0
244
+ - ++rl_games.env_eval.task_description=
245
+ - ++rl_games.env_eval.enabled=true
246
+ - ++rl_games.env_eval.distributed_mode=none
247
+ - ++rl_games.env_eval.vectorized.enabled=false
248
+ - ++rl_games.env_eval.vectorized.batch_size=1
249
+ - ++rl_games.env_eval.latency.prompt_map_path=null
250
+ - ++rl_games.env_eval.latency.mode=single
251
+ - ++rl_games.env_eval.latency.values=[0]
252
+ - ++rl_games.env_eval.mid_train.enabled=true
253
+ - ++rl_games.env_eval.mid_train.interval_steps=250
254
+ - ++rl_games.env_eval.mid_train.latencies=[2]
255
+ - ++rl_games.env_eval.mid_train.num_episodes=5
256
+ - ++rl_games.env_eval.mid_train.max_steps_per_episode=3600
257
+ - ++rl_games.env_eval.post_train.enabled=false
258
+ - ++rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
259
+ - ++rl_games.env_eval.post_train.num_episodes=5
260
+ - ++rl_games.env_eval.post_train.max_steps_per_episode=3600
261
+ - ++rl_games.task=flappy
262
+ - ++rl_games.initialization_mode=bridge
263
+ - ++rl_games.action_carrier=bridge
264
+ - ++bridge_base_model.repo_id.openvla=Qwen/Qwen3-VL-4B-Instruct
265
+ - ++bridge_base_model.repo_id.pi0=StarVLA/Qwen2.5-VL-3B-Instruct-Action
266
+ - ++bridge_base_model.repo_id.pi05=Qwen/Qwen3-VL-4B-Instruct
267
+ - ++bridge_base_model.repo_id.gr00t=Qwen/Qwen3-VL-4B-Instruct
268
+ - ++bridge_base_model.local_dir.openvla=playground/Pretrained_models/Qwen3-VL-4B-Instruct
269
+ - ++bridge_base_model.local_dir.pi0=playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
270
+ - ++bridge_base_model.local_dir.pi05=playground/Pretrained_models/Qwen3-VL-4B-Instruct
271
+ - ++bridge_base_model.local_dir.gr00t=playground/Pretrained_models/Qwen3-VL-4B-Instruct
272
+ - ++checkpoint.load=none
273
+ - ++checkpoint.hf_repo_id=null
274
+ - ++checkpoint.save_best_model=false
275
+ - ++checkpoint.save_pt_file=false
276
+ - ++checkpoint.local.keep_last_n=1
277
+ - ++checkpoint.sync.enabled=false
278
+ - ++checkpoint.sync.repo_id=null
279
+ - ++checkpoint.sync.keep_last_n=0
280
+ - ++checkpoint.sync.sync_every_n_checkpoints=1
281
+ - ++checkpoint.sync.resume_policy=local_latest
282
+ - ++run_id=flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps
283
+ - ++output_dir=null
284
+ - ++config_yaml=null
285
+ - ++is_debug=false
286
+ - ++version_id=0.21
287
+ - ++run_root_dir=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints
288
+ - ++trainer.is_resume=false
289
+ - ++trainer.pretrained_checkpoint=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/checkpoints/_initialization/StarVLA_Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
290
+ - ++trainer.resume_step=0
291
+ - ++datasets.vla_data.data_root_dir=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep
292
+ - ++datasets.vla_data.data_mix=flappy_train__bridge
293
+ - ++datasets.vla_data.eval_data_mix=flappy_train__bridge__val
294
+ - ++framework.qwenvl.base_vlm=/workspace/latency-sensitive-bench/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
295
+ - ++rl_games.env_eval.latency.prompt_map_path=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
296
+ job:
297
+ name: train_starvla_hydra
298
+ chdir: false
299
+ override_dirname: ++auth.env_file=null,++auth.hf_token_env=HF_TOKEN,++auth.wandb_api_key_env=WANDB_API_KEY,++base_model.repo_id=Qwen/Qwen3-VL-4B-Instruct,++bridge_base_model.local_dir.gr00t=playground/Pretrained_models/Qwen3-VL-4B-Instruct,++bridge_base_model.local_dir.openvla=playground/Pretrained_models/Qwen3-VL-4B-Instruct,++bridge_base_model.local_dir.pi05=playground/Pretrained_models/Qwen3-VL-4B-Instruct,++bridge_base_model.local_dir.pi0=playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action,++bridge_base_model.repo_id.gr00t=Qwen/Qwen3-VL-4B-Instruct,++bridge_base_model.repo_id.openvla=Qwen/Qwen3-VL-4B-Instruct,++bridge_base_model.repo_id.pi05=Qwen/Qwen3-VL-4B-Instruct,++bridge_base_model.repo_id.pi0=StarVLA/Qwen2.5-VL-3B-Instruct-Action,++checkpoint.hf_repo_id=null,++checkpoint.load=none,++checkpoint.local.keep_last_n=1,++checkpoint.save_best_model=false,++checkpoint.save_pt_file=false,++checkpoint.sync.enabled=false,++checkpoint.sync.keep_last_n=0,++checkpoint.sync.repo_id=null,++checkpoint.sync.resume_policy=local_latest,++checkpoint.sync.sync_every_n_checkpoints=1,++config_yaml=null,++dataset.config_name=null,++dataset.converted_name=flappy_train,++dataset.debug_subset.enabled=false,++dataset.debug_subset.max_episodes=5,++dataset.debug_subset.suffix=debug,++dataset.episodes_per_latency=null,++dataset.force_download=false,++dataset.latency_filter=null,++dataset.max_episodes=null,++dataset.mixed_converted_name=flappy_mixed_latency_train,++dataset.mixed_latency_filter=null,++dataset.mixed_source_hf=,++dataset.setup_force=false,++dataset.single_converted_name=flappy_train,++dataset.single_latency_filter=null,++dataset.single_source_hf=,++dataset.skip_verification=false,++dataset.source_hf=,++dataset.source_subdir=null,++dataset.verify_rows=200,++datasets.vla_data.action_balance.action_key=action_id,++datasets.vla_data.action_balance.enabled=false,++datasets.vla_data.action_balance.flap_id=1,++datasets.vla_data.action_balance.noop_id=0,++datasets.vla_data.action_balance.strategy=balanced_epoch,++datasets.vla_data.action_balance.target_flap_fraction=0.3,++datasets.vla_data.action_type=discrete,++datasets.vla_data.custom_mixtures_path=null,++datasets.vla_data.data_mix=flappy_train,++datasets.vla_data.data_mix=flappy_train__bridge,++datasets.vla_data.data_root_dir=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep,++datasets.vla_data.data_root_dir=playground/Datasets/rl_games,++datasets.vla_data.dataset_py=lerobot_datasets,++datasets.vla_data.eval_data_mix=flappy_train__bridge__val,++datasets.vla_data.eval_data_mix=null,++datasets.vla_data.eval_num_workers=8,++datasets.vla_data.eval_sequential_step_sampling=null,++datasets.vla_data.image_mode=single,++datasets.vla_data.include_state=true,++datasets.vla_data.latency_curriculum.enabled=false,++datasets.vla_data.latency_curriculum.latencies=null,++datasets.vla_data.latency_curriculum.phase_steps=null,++datasets.vla_data.latency_curriculum.strategy=exclusive,++datasets.vla_data.load_all_data_for_training=true,++datasets.vla_data.num_obs_frames=1,++datasets.vla_data.num_workers=8,++datasets.vla_data.obs_image_size=null,++datasets.vla_data.per_device_batch_size=32,++datasets.vla_data.persistent_workers=true,++datasets.vla_data.pin_memory=true,++datasets.vla_data.prefetch_factor=4,++datasets.vla_data.sequential_step_sampling=false,++datasets.vla_data.shuffle=true,++datasets.vla_data.stitch_grid=[2,2],++datasets.vla_data.video_backend=torchvision_av,++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/latency-sensitive-bench/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct,++framework.qwenvl.base_vlm=playground/Pretrained_models/Qwen3-VL-4B-Instruct,++framework.qwenvl.enable_gradient_checkpointing=true,++initialization.checkpoint_filename=checkpoints/steps_5000_pytorch_model.pt,++initialization.checkpoint_hf_repo_id=StarVLA/Qwen3VL-OFT-Bridge-RT-1,++initialization.checkpoint_local_dir=playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1,++is_debug=false,++output_dir=null,++paths.accelerate_config=starVLA/config/deepseeds/deepspeed_zero2.yaml,++paths.base_model_dir=playground/Pretrained_models/Qwen3-VL-4B-Instruct,++paths.dataset_cache_dir=null,++paths.dataset_local_dir=data/flappy_fix_latency_2_200ep,++paths.run_root_dir=results/Checkpoints,++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=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json,++rl_games.env_eval.latency.prompt_map_path=null,++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=250,++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=false,++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_description=,++rl_games.env_eval.task_seed_stride=0,++rl_games.env_eval.vectorized.batch_size=1,++rl_games.env_eval.vectorized.enabled=false,++rl_games.initialization_mode=bridge,++rl_games.model_alias=openvla,++rl_games.task=flappy,++run_id=flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps,++run_root_dir=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints,++run_root_dir=results/Checkpoints,++seed=42,++trainer.cc_f1_tolerance=1,++trainer.distributed_backend=none,++trainer.eval_action_classification=true,++trainer.eval_action_classification_interval=null,++trainer.eval_interval=250,++trainer.eval_num_batches=50,++trainer.freeze_llm_layers=[],++trainer.freeze_modules=,++trainer.freeze_tied_embedding=false,++trainer.freeze_vit=false,++trainer.gradient_accumulation_steps=1,++trainer.gradient_clipping=1.0,++trainer.is_resume=false,++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=1,++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=3000,++trainer.num_warmup_steps=100,++trainer.optimizer.betas=[0.9,0.95],++trainer.optimizer.eps=1e-08,++trainer.optimizer.fused=true,++trainer.optimizer.name=AdamW,++trainer.optimizer.weight_decay=1e-08,++trainer.per_latency_eval_num_batches=null,++trainer.pretrained_checkpoint=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/checkpoints/_initialization/StarVLA_Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt,++trainer.pretrained_checkpoint=null,++trainer.profile_timing.enabled=false,++trainer.profile_timing.log_interval=10,++trainer.reload_modules=null,++trainer.resume_step=0,++trainer.resume_step=0,++trainer.save_format=pt,++trainer.save_interval=500,++trainer.scheduler_specific_kwargs.min_lr=1e-06,++trainer.weight_decay=0.0,++version_id=0.21,++wandb_entity=saberrr-zju,++wandb_project=starVLA_rl_games,++workspace_dir=WORKSPACE_DIR,env=flappy,init=bridge,mode=single,model=openvla
300
+ id: ???
301
+ num: ???
302
+ config_name: train
303
+ env_set: {}
304
+ env_copy: []
305
+ config:
306
+ override_dirname:
307
+ kv_sep: '='
308
+ item_sep: ','
309
+ exclude_keys: []
310
+ runtime:
311
+ version: 1.3.3
312
+ version_base: '1.1'
313
+ cwd: /workspace/latency-sensitive-bench/starVLA
314
+ config_sources:
315
+ - path: hydra.conf
316
+ schema: pkg
317
+ provider: hydra
318
+ - path: /workspace/latency-sensitive-bench/starVLA/examples/rl_games/config
319
+ schema: file
320
+ provider: main
321
+ - path: ''
322
+ schema: structured
323
+ provider: schema
324
+ output_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/hydra
325
+ choices:
326
+ cross_task_setup: null
327
+ checkpoint: default
328
+ mode: single
329
+ init: bridge
330
+ env: flappy
331
+ model: openvla
332
+ hydra/env: default
333
+ hydra/callbacks: null
334
+ hydra/job_logging: default
335
+ hydra/hydra_logging: default
336
+ hydra/hydra_help: default
337
+ hydra/help: default
338
+ hydra/sweeper: basic
339
+ hydra/launcher: basic
340
+ hydra/output: default
341
+ verbose: false
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/hydra/.hydra/overrides.yaml ADDED
@@ -0,0 +1,181 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ - model=openvla
2
+ - env=flappy
3
+ - init=bridge
4
+ - mode=single
5
+ - ++framework.qwenvl.base_vlm=playground/Pretrained_models/Qwen3-VL-4B-Instruct
6
+ - ++framework.qwenvl.attn_implementation=flash_attention_2
7
+ - ++framework.qwenvl.enable_gradient_checkpointing=true
8
+ - ++framework.action_model.state_dim=7
9
+ - ++framework.action_model.loss_type=discrete_ce
10
+ - ++framework.action_model.action_horizon=1
11
+ - ++framework.action_model.future_action_window_size=0
12
+ - ++framework.action_model.past_action_window_size=0
13
+ - ++framework.action_model.action_dim=7
14
+ - ++framework.action_model.action_env_dim=2
15
+ - ++framework.name=QwenOFT
16
+ - ++datasets.vla_data.dataset_py=lerobot_datasets
17
+ - ++datasets.vla_data.include_state=true
18
+ - ++datasets.vla_data.data_root_dir=playground/Datasets/rl_games
19
+ - ++datasets.vla_data.data_mix=flappy_train
20
+ - ++datasets.vla_data.eval_data_mix=null
21
+ - ++datasets.vla_data.custom_mixtures_path=null
22
+ - ++datasets.vla_data.action_type=discrete
23
+ - ++datasets.vla_data.sequential_step_sampling=false
24
+ - ++datasets.vla_data.eval_sequential_step_sampling=null
25
+ - ++datasets.vla_data.num_workers=8
26
+ - ++datasets.vla_data.eval_num_workers=8
27
+ - ++datasets.vla_data.prefetch_factor=4
28
+ - ++datasets.vla_data.persistent_workers=true
29
+ - ++datasets.vla_data.pin_memory=true
30
+ - ++datasets.vla_data.shuffle=true
31
+ - ++datasets.vla_data.action_balance.enabled=false
32
+ - ++datasets.vla_data.action_balance.strategy=balanced_epoch
33
+ - ++datasets.vla_data.action_balance.action_key=action_id
34
+ - ++datasets.vla_data.action_balance.target_flap_fraction=0.3
35
+ - ++datasets.vla_data.action_balance.noop_id=0
36
+ - ++datasets.vla_data.action_balance.flap_id=1
37
+ - ++datasets.vla_data.latency_curriculum.enabled=false
38
+ - ++datasets.vla_data.latency_curriculum.strategy=exclusive
39
+ - ++datasets.vla_data.latency_curriculum.latencies=null
40
+ - ++datasets.vla_data.latency_curriculum.phase_steps=null
41
+ - ++datasets.vla_data.per_device_batch_size=32
42
+ - ++datasets.vla_data.load_all_data_for_training=true
43
+ - ++datasets.vla_data.num_obs_frames=1
44
+ - ++datasets.vla_data.image_mode=single
45
+ - ++datasets.vla_data.stitch_grid=[2,2]
46
+ - ++datasets.vla_data.obs_image_size=null
47
+ - ++datasets.vla_data.video_backend=torchvision_av
48
+ - ++dataset.source_hf=
49
+ - ++dataset.config_name=null
50
+ - ++dataset.source_subdir=null
51
+ - ++dataset.converted_name=flappy_train
52
+ - ++dataset.single_source_hf=
53
+ - ++dataset.mixed_source_hf=
54
+ - ++dataset.single_converted_name=flappy_train
55
+ - ++dataset.mixed_converted_name=flappy_mixed_latency_train
56
+ - ++dataset.single_latency_filter=null
57
+ - ++dataset.mixed_latency_filter=null
58
+ - ++dataset.force_download=false
59
+ - ++dataset.setup_force=false
60
+ - ++dataset.skip_verification=false
61
+ - ++dataset.verify_rows=200
62
+ - ++dataset.max_episodes=null
63
+ - ++dataset.episodes_per_latency=null
64
+ - ++dataset.latency_filter=null
65
+ - ++dataset.debug_subset.enabled=false
66
+ - ++dataset.debug_subset.max_episodes=5
67
+ - ++dataset.debug_subset.suffix=debug
68
+ - ++base_model.repo_id=Qwen/Qwen3-VL-4B-Instruct
69
+ - ++initialization.checkpoint_local_dir=playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
70
+ - ++initialization.checkpoint_hf_repo_id=StarVLA/Qwen3VL-OFT-Bridge-RT-1
71
+ - ++initialization.checkpoint_filename=checkpoints/steps_5000_pytorch_model.pt
72
+ - ++trainer.max_train_steps=3000
73
+ - ++trainer.num_warmup_steps=100
74
+ - ++trainer.save_interval=500
75
+ - ++trainer.eval_interval=250
76
+ - ++trainer.eval_num_batches=50
77
+ - ++trainer.per_latency_eval_num_batches=null
78
+ - ++trainer.eval_action_classification=true
79
+ - ++trainer.eval_action_classification_interval=null
80
+ - ++trainer.cc_f1_tolerance=1
81
+ - ++trainer.learning_rate.base=2e-05
82
+ - ++trainer.learning_rate.qwen_vl_interface=1e-05
83
+ - ++trainer.learning_rate.action_model=0.0001
84
+ - ++trainer.lr_scheduler_type=cosine_with_min_lr
85
+ - ++trainer.scheduler_specific_kwargs.min_lr=1e-06
86
+ - ++trainer.freeze_modules=
87
+ - ++trainer.freeze_vit=false
88
+ - ++trainer.freeze_tied_embedding=false
89
+ - ++trainer.freeze_llm_layers=[]
90
+ - ++trainer.loss_scale.vla=1.0
91
+ - ++trainer.loss_scale.vlm=0.1
92
+ - ++trainer.max_grad_norm=1.0
93
+ - ++trainer.weight_decay=0.0
94
+ - ++trainer.logging_frequency=1
95
+ - ++trainer.profile_timing.enabled=false
96
+ - ++trainer.profile_timing.log_interval=10
97
+ - ++trainer.gradient_clipping=1.0
98
+ - ++trainer.gradient_accumulation_steps=1
99
+ - ++trainer.distributed_backend=none
100
+ - ++trainer.is_resume=false
101
+ - ++trainer.pretrained_checkpoint=null
102
+ - ++trainer.resume_step=0
103
+ - ++trainer.reload_modules=null
104
+ - ++trainer.optimizer.name=AdamW
105
+ - ++trainer.optimizer.betas=[0.9,0.95]
106
+ - ++trainer.optimizer.eps=1e-08
107
+ - ++trainer.optimizer.weight_decay=1e-08
108
+ - ++trainer.optimizer.fused=true
109
+ - ++trainer.save_format=pt
110
+ - ++workspace_dir=WORKSPACE_DIR
111
+ - ++run_root_dir=results/Checkpoints
112
+ - ++seed=42
113
+ - ++wandb_entity=saberrr-zju
114
+ - ++wandb_project=starVLA_rl_games
115
+ - ++auth.env_file=null
116
+ - ++auth.hf_token_env=HF_TOKEN
117
+ - ++auth.wandb_api_key_env=WANDB_API_KEY
118
+ - ++paths.run_root_dir=results/Checkpoints
119
+ - ++paths.dataset_local_dir=data/flappy_fix_latency_2_200ep
120
+ - ++paths.dataset_cache_dir=null
121
+ - ++paths.base_model_dir=playground/Pretrained_models/Qwen3-VL-4B-Instruct
122
+ - ++paths.accelerate_config=starVLA/config/deepseeds/deepspeed_zero2.yaml
123
+ - ++rl_games.model_alias=openvla
124
+ - ++rl_games.env_eval.image_size=224
125
+ - ++rl_games.env_eval.frameskip=1
126
+ - ++rl_games.env_eval.seed=42
127
+ - ++rl_games.env_eval.fixed_episode_seeds=true
128
+ - ++rl_games.env_eval.latency_seed_stride=0
129
+ - ++rl_games.env_eval.task_seed_stride=0
130
+ - ++rl_games.env_eval.task_description=
131
+ - ++rl_games.env_eval.enabled=true
132
+ - ++rl_games.env_eval.distributed_mode=none
133
+ - ++rl_games.env_eval.vectorized.enabled=false
134
+ - ++rl_games.env_eval.vectorized.batch_size=1
135
+ - ++rl_games.env_eval.latency.prompt_map_path=null
136
+ - ++rl_games.env_eval.latency.mode=single
137
+ - ++rl_games.env_eval.latency.values=[0]
138
+ - ++rl_games.env_eval.mid_train.enabled=true
139
+ - ++rl_games.env_eval.mid_train.interval_steps=250
140
+ - ++rl_games.env_eval.mid_train.latencies=[2]
141
+ - ++rl_games.env_eval.mid_train.num_episodes=5
142
+ - ++rl_games.env_eval.mid_train.max_steps_per_episode=3600
143
+ - ++rl_games.env_eval.post_train.enabled=false
144
+ - ++rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
145
+ - ++rl_games.env_eval.post_train.num_episodes=5
146
+ - ++rl_games.env_eval.post_train.max_steps_per_episode=3600
147
+ - ++rl_games.task=flappy
148
+ - ++rl_games.initialization_mode=bridge
149
+ - ++rl_games.action_carrier=bridge
150
+ - ++bridge_base_model.repo_id.openvla=Qwen/Qwen3-VL-4B-Instruct
151
+ - ++bridge_base_model.repo_id.pi0=StarVLA/Qwen2.5-VL-3B-Instruct-Action
152
+ - ++bridge_base_model.repo_id.pi05=Qwen/Qwen3-VL-4B-Instruct
153
+ - ++bridge_base_model.repo_id.gr00t=Qwen/Qwen3-VL-4B-Instruct
154
+ - ++bridge_base_model.local_dir.openvla=playground/Pretrained_models/Qwen3-VL-4B-Instruct
155
+ - ++bridge_base_model.local_dir.pi0=playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
156
+ - ++bridge_base_model.local_dir.pi05=playground/Pretrained_models/Qwen3-VL-4B-Instruct
157
+ - ++bridge_base_model.local_dir.gr00t=playground/Pretrained_models/Qwen3-VL-4B-Instruct
158
+ - ++checkpoint.load=none
159
+ - ++checkpoint.hf_repo_id=null
160
+ - ++checkpoint.save_best_model=false
161
+ - ++checkpoint.save_pt_file=false
162
+ - ++checkpoint.local.keep_last_n=1
163
+ - ++checkpoint.sync.enabled=false
164
+ - ++checkpoint.sync.repo_id=null
165
+ - ++checkpoint.sync.keep_last_n=0
166
+ - ++checkpoint.sync.sync_every_n_checkpoints=1
167
+ - ++checkpoint.sync.resume_policy=local_latest
168
+ - ++run_id=flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps
169
+ - ++output_dir=null
170
+ - ++config_yaml=null
171
+ - ++is_debug=false
172
+ - ++version_id=0.21
173
+ - ++run_root_dir=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints
174
+ - ++trainer.is_resume=false
175
+ - ++trainer.pretrained_checkpoint=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/checkpoints/_initialization/StarVLA_Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
176
+ - ++trainer.resume_step=0
177
+ - ++datasets.vla_data.data_root_dir=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep
178
+ - ++datasets.vla_data.data_mix=flappy_train__bridge
179
+ - ++datasets.vla_data.eval_data_mix=flappy_train__bridge__val
180
+ - ++framework.qwenvl.base_vlm=/workspace/latency-sensitive-bench/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
181
+ - ++rl_games.env_eval.latency.prompt_map_path=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/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/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_3k_steps/summary.jsonl ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {"steps": 500}
2
+ {"steps": 1000}
3
+ {"steps": 1500}
4
+ {"steps": 2000}
5
+ {"steps": 2500}
6
+ {"steps": 3000}
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/checkpoints/steps_4000_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:074a513f33ae5b81499478f5b7cd258757d271ce29077b30e299b49f5ba81bdb
3
+ size 9138230516
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/config.full.yaml ADDED
@@ -0,0 +1,224 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ name: QwenOFT
3
+ qwenvl:
4
+ base_vlm: /workspace/latency-sensitive-bench/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/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep
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
+ eval_sequential_step_sampling: null
28
+ num_workers: 8
29
+ eval_num_workers: 8
30
+ prefetch_factor: 4
31
+ persistent_workers: true
32
+ pin_memory: true
33
+ shuffle: true
34
+ action_balance:
35
+ enabled: false
36
+ strategy: balanced_epoch
37
+ action_key: action_id
38
+ target_flap_fraction: 0.3
39
+ noop_id: 0
40
+ flap_id: 1
41
+ latency_curriculum:
42
+ enabled: false
43
+ strategy: exclusive
44
+ latencies: null
45
+ phase_steps: null
46
+ per_device_batch_size: 32
47
+ load_all_data_for_training: true
48
+ num_obs_frames: 1
49
+ image_mode: single
50
+ stitch_grid:
51
+ - 2
52
+ - 2
53
+ obs_image_size: null
54
+ video_backend: torchvision_av
55
+ dataset:
56
+ source_hf: ''
57
+ config_name: null
58
+ source_subdir: null
59
+ converted_name: flappy_train
60
+ single_source_hf: ''
61
+ mixed_source_hf: ''
62
+ single_converted_name: flappy_train
63
+ mixed_converted_name: flappy_mixed_latency_train
64
+ single_latency_filter: null
65
+ mixed_latency_filter: null
66
+ force_download: false
67
+ setup_force: false
68
+ skip_verification: false
69
+ verify_rows: 200
70
+ max_episodes: null
71
+ episodes_per_latency: null
72
+ latency_filter: null
73
+ debug_subset:
74
+ enabled: false
75
+ max_episodes: 5
76
+ suffix: debug
77
+ base_model:
78
+ repo_id: Qwen/Qwen3-VL-4B-Instruct
79
+ initialization:
80
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
81
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
82
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
83
+ trainer:
84
+ max_train_steps: 4000
85
+ num_warmup_steps: 100
86
+ save_interval: 500
87
+ eval_interval: 250
88
+ eval_num_batches: 50
89
+ per_latency_eval_num_batches: null
90
+ eval_action_classification: true
91
+ eval_action_classification_interval: null
92
+ cc_f1_tolerance: 1
93
+ learning_rate:
94
+ base: 2.0e-05
95
+ qwen_vl_interface: 1.0e-05
96
+ action_model: 0.0001
97
+ lr_scheduler_type: cosine_with_min_lr
98
+ scheduler_specific_kwargs:
99
+ min_lr: 1.0e-06
100
+ freeze_modules: ''
101
+ freeze_vit: false
102
+ freeze_tied_embedding: false
103
+ freeze_llm_layers: []
104
+ loss_scale:
105
+ vla: 1.0
106
+ vlm: 0.1
107
+ max_grad_norm: 1.0
108
+ weight_decay: 0.0
109
+ logging_frequency: 1
110
+ profile_timing:
111
+ enabled: false
112
+ log_interval: 10
113
+ gradient_clipping: 1.0
114
+ gradient_accumulation_steps: 1
115
+ distributed_backend: none
116
+ is_resume: false
117
+ pretrained_checkpoint: null
118
+ resume_step: 0
119
+ reload_modules: null
120
+ optimizer:
121
+ name: AdamW
122
+ betas:
123
+ - 0.9
124
+ - 0.95
125
+ eps: 1.0e-08
126
+ weight_decay: 1.0e-08
127
+ fused: true
128
+ save_format: pt
129
+ workspace_dir: WORKSPACE_DIR
130
+ run_root_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints
131
+ seed: 42
132
+ wandb_entity: saberrr-zju
133
+ wandb_project: starVLA_rl_games
134
+ auth:
135
+ env_file: null
136
+ hf_token_env: HF_TOKEN
137
+ wandb_api_key_env: WANDB_API_KEY
138
+ paths:
139
+ run_root_dir: results/Checkpoints
140
+ dataset_local_dir: data/flappy_fix_latency_2_200ep
141
+ dataset_cache_dir: null
142
+ base_model_dir: playground/Pretrained_models/Qwen3-VL-4B-Instruct
143
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
144
+ launch:
145
+ use_accelerate: true
146
+ gpus: null
147
+ num_processes: 1
148
+ dry_run: false
149
+ conda:
150
+ enabled: true
151
+ env_name: null
152
+ rl_games:
153
+ model_alias: openvla
154
+ env_eval:
155
+ image_size: 224
156
+ frameskip: 1
157
+ seed: 42
158
+ fixed_episode_seeds: true
159
+ latency_seed_stride: 0
160
+ task_seed_stride: 0
161
+ task_description: ''
162
+ enabled: true
163
+ distributed_mode: none
164
+ vectorized:
165
+ enabled: false
166
+ batch_size: 1
167
+ latency:
168
+ prompt_map_path: /workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
169
+ mode: single
170
+ values:
171
+ - 0
172
+ mid_train:
173
+ enabled: true
174
+ interval_steps: 250
175
+ latencies:
176
+ - 2
177
+ num_episodes: 5
178
+ max_steps_per_episode: 3600
179
+ post_train:
180
+ enabled: false
181
+ latencies:
182
+ - 0
183
+ - 1
184
+ - 2
185
+ - 3
186
+ - 4
187
+ num_episodes: 5
188
+ max_steps_per_episode: 3600
189
+ task: flappy
190
+ initialization_mode: bridge
191
+ action_carrier: bridge
192
+ model: openvla
193
+ env: flappy
194
+ init: bridge
195
+ bridge_base_model:
196
+ repo_id:
197
+ openvla: Qwen/Qwen3-VL-4B-Instruct
198
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
199
+ pi05: Qwen/Qwen3-VL-4B-Instruct
200
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
201
+ local_dir:
202
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
203
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
204
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
205
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
206
+ mode: single
207
+ checkpoint:
208
+ load: none
209
+ hf_repo_id: null
210
+ save_best_model: false
211
+ save_pt_file: false
212
+ local:
213
+ keep_last_n: 1
214
+ sync:
215
+ enabled: false
216
+ repo_id: null
217
+ keep_last_n: 0
218
+ sync_every_n_checkpoints: 1
219
+ resume_policy: local_latest
220
+ run_id: flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps
221
+ output_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps
222
+ config_yaml: null
223
+ is_debug: false
224
+ version_id: '0.21'
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/config.yaml ADDED
@@ -0,0 +1,103 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ checkpoint:
2
+ local:
3
+ keep_last_n: 1
4
+ save_best_model: false
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
+ eval_num_workers: 8
16
+ include_state: true
17
+ latency_curriculum:
18
+ enabled: false
19
+ obs_image_size: null
20
+ per_device_batch_size: 32
21
+ persistent_workers: true
22
+ pin_memory: true
23
+ prefetch_factor: 4
24
+ framework:
25
+ action_model:
26
+ action_dim: 7
27
+ action_env_dim: 2
28
+ action_hidden_dim: 2560
29
+ action_horizon: 1
30
+ action_model_type: MLP
31
+ loss_type: discrete_ce
32
+ state_dim: 7
33
+ name: QwenOFT
34
+ qwenvl:
35
+ attn_implementation: flash_attention_2
36
+ base_vlm: /workspace/latency-sensitive-bench/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
37
+ enable_gradient_checkpointing: true
38
+ output_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps
39
+ rl_games:
40
+ env_eval:
41
+ distributed_mode: none
42
+ enabled: true
43
+ fixed_episode_seeds: true
44
+ frameskip: 1
45
+ image_size: 224
46
+ latency:
47
+ prompt_map_path: /workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
48
+ latency_seed_stride: 0
49
+ mid_train:
50
+ enabled: true
51
+ interval_steps: 250
52
+ latencies:
53
+ - 2
54
+ max_steps_per_episode: 3600
55
+ num_episodes: 5
56
+ seed: 42
57
+ task_description: ''
58
+ task_seed_stride: 0
59
+ vectorized:
60
+ enabled: false
61
+ model_alias: openvla
62
+ task: flappy
63
+ run_id: flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps
64
+ run_root_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints
65
+ seed: 42
66
+ trainer:
67
+ distributed_backend: none
68
+ eval_action_classification: true
69
+ eval_action_classification_interval: null
70
+ eval_interval: 250
71
+ eval_num_batches: 50
72
+ freeze_llm_layers: []
73
+ freeze_modules: ''
74
+ freeze_tied_embedding: false
75
+ freeze_vit: false
76
+ gradient_accumulation_steps: 1
77
+ gradient_clipping: 1.0
78
+ is_resume: false
79
+ learning_rate:
80
+ action_model: 0.0001
81
+ base: 2.0e-05
82
+ qwen_vl_interface: 1.0e-05
83
+ logging_frequency: 1
84
+ lr_scheduler_type: cosine_with_min_lr
85
+ max_train_steps: 4000
86
+ num_warmup_steps: 100
87
+ optimizer:
88
+ betas:
89
+ - 0.9
90
+ - 0.95
91
+ eps: 1.0e-08
92
+ fused: true
93
+ weight_decay: 1.0e-08
94
+ per_latency_eval_num_batches: null
95
+ pretrained_checkpoint: null
96
+ profile_timing:
97
+ enabled: false
98
+ reload_modules: null
99
+ save_interval: 500
100
+ scheduler_specific_kwargs:
101
+ min_lr: 1.0e-06
102
+ wandb_entity: saberrr-zju
103
+ wandb_project: starVLA_rl_games
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/dataset_statistics.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.6028500199317932,
6
+ 0.3971499800682068,
7
+ 0.0,
8
+ 0.0,
9
+ 0.0,
10
+ 0.0,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.4890792667865753,
15
+ 0.4890792667865753,
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": 330734,
125
+ "num_trajectories": 180
126
+ }
127
+ }
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/dataset_statistics_eval.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.7959861159324646,
6
+ 0.2040138840675354,
7
+ 0.0,
8
+ 0.0,
9
+ 0.0,
10
+ 0.0,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.4030573070049286,
15
+ 0.4030573070049286,
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": 72000,
125
+ "num_trajectories": 20
126
+ }
127
+ }
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/hydra/.hydra/config.yaml ADDED
@@ -0,0 +1,222 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ qwenvl:
3
+ base_vlm: /workspace/latency-sensitive-bench/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/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep
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
+ eval_sequential_step_sampling: null
26
+ num_workers: 8
27
+ eval_num_workers: 8
28
+ prefetch_factor: 4
29
+ persistent_workers: true
30
+ pin_memory: true
31
+ shuffle: true
32
+ action_balance:
33
+ enabled: false
34
+ strategy: balanced_epoch
35
+ action_key: action_id
36
+ target_flap_fraction: 0.3
37
+ noop_id: 0
38
+ flap_id: 1
39
+ latency_curriculum:
40
+ enabled: false
41
+ strategy: exclusive
42
+ latencies: null
43
+ phase_steps: null
44
+ per_device_batch_size: 32
45
+ load_all_data_for_training: true
46
+ num_obs_frames: 1
47
+ image_mode: single
48
+ stitch_grid:
49
+ - 2
50
+ - 2
51
+ obs_image_size: null
52
+ video_backend: torchvision_av
53
+ dataset:
54
+ source_hf: ''
55
+ config_name: null
56
+ source_subdir: null
57
+ converted_name: flappy_train
58
+ single_source_hf: ''
59
+ mixed_source_hf: ''
60
+ single_converted_name: flappy_train
61
+ mixed_converted_name: flappy_mixed_latency_train
62
+ single_latency_filter: null
63
+ mixed_latency_filter: null
64
+ force_download: false
65
+ setup_force: false
66
+ skip_verification: false
67
+ verify_rows: 200
68
+ max_episodes: null
69
+ episodes_per_latency: null
70
+ latency_filter: null
71
+ debug_subset:
72
+ enabled: false
73
+ max_episodes: 5
74
+ suffix: debug
75
+ base_model:
76
+ repo_id: Qwen/Qwen3-VL-4B-Instruct
77
+ initialization:
78
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
79
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
80
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
81
+ trainer:
82
+ max_train_steps: 4000
83
+ num_warmup_steps: 100
84
+ save_interval: 500
85
+ eval_interval: 250
86
+ eval_num_batches: 50
87
+ per_latency_eval_num_batches: null
88
+ eval_action_classification: true
89
+ eval_action_classification_interval: null
90
+ cc_f1_tolerance: 1
91
+ learning_rate:
92
+ base: 2.0e-05
93
+ qwen_vl_interface: 1.0e-05
94
+ action_model: 0.0001
95
+ lr_scheduler_type: cosine_with_min_lr
96
+ scheduler_specific_kwargs:
97
+ min_lr: 1.0e-06
98
+ freeze_modules: ''
99
+ freeze_vit: false
100
+ freeze_tied_embedding: false
101
+ freeze_llm_layers: []
102
+ loss_scale:
103
+ vla: 1.0
104
+ vlm: 0.1
105
+ max_grad_norm: 1.0
106
+ weight_decay: 0.0
107
+ logging_frequency: 1
108
+ profile_timing:
109
+ enabled: false
110
+ log_interval: 10
111
+ gradient_clipping: 1.0
112
+ gradient_accumulation_steps: 1
113
+ distributed_backend: none
114
+ is_resume: false
115
+ pretrained_checkpoint: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/checkpoints/_initialization/StarVLA_Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
116
+ resume_step: 0
117
+ reload_modules: null
118
+ optimizer:
119
+ name: AdamW
120
+ betas:
121
+ - 0.9
122
+ - 0.95
123
+ eps: 1.0e-08
124
+ weight_decay: 1.0e-08
125
+ fused: true
126
+ save_format: pt
127
+ workspace_dir: WORKSPACE_DIR
128
+ run_root_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints
129
+ seed: 42
130
+ wandb_entity: saberrr-zju
131
+ wandb_project: starVLA_rl_games
132
+ auth:
133
+ env_file: null
134
+ hf_token_env: HF_TOKEN
135
+ wandb_api_key_env: WANDB_API_KEY
136
+ paths:
137
+ run_root_dir: results/Checkpoints
138
+ dataset_local_dir: data/flappy_fix_latency_2_200ep
139
+ dataset_cache_dir: null
140
+ base_model_dir: playground/Pretrained_models/Qwen3-VL-4B-Instruct
141
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
142
+ launch:
143
+ use_accelerate: true
144
+ gpus: null
145
+ num_processes: 1
146
+ dry_run: false
147
+ conda:
148
+ enabled: true
149
+ env_name: null
150
+ rl_games:
151
+ model_alias: openvla
152
+ env_eval:
153
+ image_size: 224
154
+ frameskip: 1
155
+ seed: 42
156
+ fixed_episode_seeds: true
157
+ latency_seed_stride: 0
158
+ task_seed_stride: 0
159
+ task_description: ''
160
+ enabled: true
161
+ distributed_mode: none
162
+ vectorized:
163
+ enabled: false
164
+ batch_size: 1
165
+ latency:
166
+ prompt_map_path: /workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
167
+ mode: single
168
+ values:
169
+ - 0
170
+ mid_train:
171
+ enabled: true
172
+ interval_steps: 250
173
+ latencies:
174
+ - 2
175
+ num_episodes: 5
176
+ max_steps_per_episode: 3600
177
+ post_train:
178
+ enabled: false
179
+ latencies:
180
+ - 0
181
+ - 1
182
+ - 2
183
+ - 3
184
+ - 4
185
+ num_episodes: 5
186
+ max_steps_per_episode: 3600
187
+ task: flappy
188
+ initialization_mode: bridge
189
+ action_carrier: bridge
190
+ model: openvla
191
+ env: flappy
192
+ init: bridge
193
+ bridge_base_model:
194
+ repo_id:
195
+ openvla: Qwen/Qwen3-VL-4B-Instruct
196
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
197
+ pi05: Qwen/Qwen3-VL-4B-Instruct
198
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
199
+ local_dir:
200
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
201
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
202
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
203
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
204
+ mode: single
205
+ checkpoint:
206
+ load: none
207
+ hf_repo_id: null
208
+ save_best_model: false
209
+ save_pt_file: false
210
+ local:
211
+ keep_last_n: 1
212
+ sync:
213
+ enabled: false
214
+ repo_id: null
215
+ keep_last_n: 0
216
+ sync_every_n_checkpoints: 1
217
+ resume_policy: local_latest
218
+ run_id: flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps
219
+ output_dir: null
220
+ config_yaml: null
221
+ is_debug: false
222
+ version_id: 0.21
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/hydra/.hydra/hydra.yaml ADDED
@@ -0,0 +1,341 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ - ++framework.qwenvl.base_vlm=playground/Pretrained_models/Qwen3-VL-4B-Instruct
120
+ - ++framework.qwenvl.attn_implementation=flash_attention_2
121
+ - ++framework.qwenvl.enable_gradient_checkpointing=true
122
+ - ++framework.action_model.state_dim=7
123
+ - ++framework.action_model.loss_type=discrete_ce
124
+ - ++framework.action_model.action_horizon=1
125
+ - ++framework.action_model.future_action_window_size=0
126
+ - ++framework.action_model.past_action_window_size=0
127
+ - ++framework.action_model.action_dim=7
128
+ - ++framework.action_model.action_env_dim=2
129
+ - ++framework.name=QwenOFT
130
+ - ++datasets.vla_data.dataset_py=lerobot_datasets
131
+ - ++datasets.vla_data.include_state=true
132
+ - ++datasets.vla_data.data_root_dir=playground/Datasets/rl_games
133
+ - ++datasets.vla_data.data_mix=flappy_train
134
+ - ++datasets.vla_data.eval_data_mix=null
135
+ - ++datasets.vla_data.custom_mixtures_path=null
136
+ - ++datasets.vla_data.action_type=discrete
137
+ - ++datasets.vla_data.sequential_step_sampling=false
138
+ - ++datasets.vla_data.eval_sequential_step_sampling=null
139
+ - ++datasets.vla_data.num_workers=8
140
+ - ++datasets.vla_data.eval_num_workers=8
141
+ - ++datasets.vla_data.prefetch_factor=4
142
+ - ++datasets.vla_data.persistent_workers=true
143
+ - ++datasets.vla_data.pin_memory=true
144
+ - ++datasets.vla_data.shuffle=true
145
+ - ++datasets.vla_data.action_balance.enabled=false
146
+ - ++datasets.vla_data.action_balance.strategy=balanced_epoch
147
+ - ++datasets.vla_data.action_balance.action_key=action_id
148
+ - ++datasets.vla_data.action_balance.target_flap_fraction=0.3
149
+ - ++datasets.vla_data.action_balance.noop_id=0
150
+ - ++datasets.vla_data.action_balance.flap_id=1
151
+ - ++datasets.vla_data.latency_curriculum.enabled=false
152
+ - ++datasets.vla_data.latency_curriculum.strategy=exclusive
153
+ - ++datasets.vla_data.latency_curriculum.latencies=null
154
+ - ++datasets.vla_data.latency_curriculum.phase_steps=null
155
+ - ++datasets.vla_data.per_device_batch_size=32
156
+ - ++datasets.vla_data.load_all_data_for_training=true
157
+ - ++datasets.vla_data.num_obs_frames=1
158
+ - ++datasets.vla_data.image_mode=single
159
+ - ++datasets.vla_data.stitch_grid=[2,2]
160
+ - ++datasets.vla_data.obs_image_size=null
161
+ - ++datasets.vla_data.video_backend=torchvision_av
162
+ - ++dataset.source_hf=
163
+ - ++dataset.config_name=null
164
+ - ++dataset.source_subdir=null
165
+ - ++dataset.converted_name=flappy_train
166
+ - ++dataset.single_source_hf=
167
+ - ++dataset.mixed_source_hf=
168
+ - ++dataset.single_converted_name=flappy_train
169
+ - ++dataset.mixed_converted_name=flappy_mixed_latency_train
170
+ - ++dataset.single_latency_filter=null
171
+ - ++dataset.mixed_latency_filter=null
172
+ - ++dataset.force_download=false
173
+ - ++dataset.setup_force=false
174
+ - ++dataset.skip_verification=false
175
+ - ++dataset.verify_rows=200
176
+ - ++dataset.max_episodes=null
177
+ - ++dataset.episodes_per_latency=null
178
+ - ++dataset.latency_filter=null
179
+ - ++dataset.debug_subset.enabled=false
180
+ - ++dataset.debug_subset.max_episodes=5
181
+ - ++dataset.debug_subset.suffix=debug
182
+ - ++base_model.repo_id=Qwen/Qwen3-VL-4B-Instruct
183
+ - ++initialization.checkpoint_local_dir=playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
184
+ - ++initialization.checkpoint_hf_repo_id=StarVLA/Qwen3VL-OFT-Bridge-RT-1
185
+ - ++initialization.checkpoint_filename=checkpoints/steps_5000_pytorch_model.pt
186
+ - ++trainer.max_train_steps=4000
187
+ - ++trainer.num_warmup_steps=100
188
+ - ++trainer.save_interval=500
189
+ - ++trainer.eval_interval=250
190
+ - ++trainer.eval_num_batches=50
191
+ - ++trainer.per_latency_eval_num_batches=null
192
+ - ++trainer.eval_action_classification=true
193
+ - ++trainer.eval_action_classification_interval=null
194
+ - ++trainer.cc_f1_tolerance=1
195
+ - ++trainer.learning_rate.base=2e-05
196
+ - ++trainer.learning_rate.qwen_vl_interface=1e-05
197
+ - ++trainer.learning_rate.action_model=0.0001
198
+ - ++trainer.lr_scheduler_type=cosine_with_min_lr
199
+ - ++trainer.scheduler_specific_kwargs.min_lr=1e-06
200
+ - ++trainer.freeze_modules=
201
+ - ++trainer.freeze_vit=false
202
+ - ++trainer.freeze_tied_embedding=false
203
+ - ++trainer.freeze_llm_layers=[]
204
+ - ++trainer.loss_scale.vla=1.0
205
+ - ++trainer.loss_scale.vlm=0.1
206
+ - ++trainer.max_grad_norm=1.0
207
+ - ++trainer.weight_decay=0.0
208
+ - ++trainer.logging_frequency=1
209
+ - ++trainer.profile_timing.enabled=false
210
+ - ++trainer.profile_timing.log_interval=10
211
+ - ++trainer.gradient_clipping=1.0
212
+ - ++trainer.gradient_accumulation_steps=1
213
+ - ++trainer.distributed_backend=none
214
+ - ++trainer.is_resume=false
215
+ - ++trainer.pretrained_checkpoint=null
216
+ - ++trainer.resume_step=0
217
+ - ++trainer.reload_modules=null
218
+ - ++trainer.optimizer.name=AdamW
219
+ - ++trainer.optimizer.betas=[0.9,0.95]
220
+ - ++trainer.optimizer.eps=1e-08
221
+ - ++trainer.optimizer.weight_decay=1e-08
222
+ - ++trainer.optimizer.fused=true
223
+ - ++trainer.save_format=pt
224
+ - ++workspace_dir=WORKSPACE_DIR
225
+ - ++run_root_dir=results/Checkpoints
226
+ - ++seed=42
227
+ - ++wandb_entity=saberrr-zju
228
+ - ++wandb_project=starVLA_rl_games
229
+ - ++auth.env_file=null
230
+ - ++auth.hf_token_env=HF_TOKEN
231
+ - ++auth.wandb_api_key_env=WANDB_API_KEY
232
+ - ++paths.run_root_dir=results/Checkpoints
233
+ - ++paths.dataset_local_dir=data/flappy_fix_latency_2_200ep
234
+ - ++paths.dataset_cache_dir=null
235
+ - ++paths.base_model_dir=playground/Pretrained_models/Qwen3-VL-4B-Instruct
236
+ - ++paths.accelerate_config=starVLA/config/deepseeds/deepspeed_zero2.yaml
237
+ - ++rl_games.model_alias=openvla
238
+ - ++rl_games.env_eval.image_size=224
239
+ - ++rl_games.env_eval.frameskip=1
240
+ - ++rl_games.env_eval.seed=42
241
+ - ++rl_games.env_eval.fixed_episode_seeds=true
242
+ - ++rl_games.env_eval.latency_seed_stride=0
243
+ - ++rl_games.env_eval.task_seed_stride=0
244
+ - ++rl_games.env_eval.task_description=
245
+ - ++rl_games.env_eval.enabled=true
246
+ - ++rl_games.env_eval.distributed_mode=none
247
+ - ++rl_games.env_eval.vectorized.enabled=false
248
+ - ++rl_games.env_eval.vectorized.batch_size=1
249
+ - ++rl_games.env_eval.latency.prompt_map_path=null
250
+ - ++rl_games.env_eval.latency.mode=single
251
+ - ++rl_games.env_eval.latency.values=[0]
252
+ - ++rl_games.env_eval.mid_train.enabled=true
253
+ - ++rl_games.env_eval.mid_train.interval_steps=250
254
+ - ++rl_games.env_eval.mid_train.latencies=[2]
255
+ - ++rl_games.env_eval.mid_train.num_episodes=5
256
+ - ++rl_games.env_eval.mid_train.max_steps_per_episode=3600
257
+ - ++rl_games.env_eval.post_train.enabled=false
258
+ - ++rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
259
+ - ++rl_games.env_eval.post_train.num_episodes=5
260
+ - ++rl_games.env_eval.post_train.max_steps_per_episode=3600
261
+ - ++rl_games.task=flappy
262
+ - ++rl_games.initialization_mode=bridge
263
+ - ++rl_games.action_carrier=bridge
264
+ - ++bridge_base_model.repo_id.openvla=Qwen/Qwen3-VL-4B-Instruct
265
+ - ++bridge_base_model.repo_id.pi0=StarVLA/Qwen2.5-VL-3B-Instruct-Action
266
+ - ++bridge_base_model.repo_id.pi05=Qwen/Qwen3-VL-4B-Instruct
267
+ - ++bridge_base_model.repo_id.gr00t=Qwen/Qwen3-VL-4B-Instruct
268
+ - ++bridge_base_model.local_dir.openvla=playground/Pretrained_models/Qwen3-VL-4B-Instruct
269
+ - ++bridge_base_model.local_dir.pi0=playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
270
+ - ++bridge_base_model.local_dir.pi05=playground/Pretrained_models/Qwen3-VL-4B-Instruct
271
+ - ++bridge_base_model.local_dir.gr00t=playground/Pretrained_models/Qwen3-VL-4B-Instruct
272
+ - ++checkpoint.load=none
273
+ - ++checkpoint.hf_repo_id=null
274
+ - ++checkpoint.save_best_model=false
275
+ - ++checkpoint.save_pt_file=false
276
+ - ++checkpoint.local.keep_last_n=1
277
+ - ++checkpoint.sync.enabled=false
278
+ - ++checkpoint.sync.repo_id=null
279
+ - ++checkpoint.sync.keep_last_n=0
280
+ - ++checkpoint.sync.sync_every_n_checkpoints=1
281
+ - ++checkpoint.sync.resume_policy=local_latest
282
+ - ++run_id=flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps
283
+ - ++output_dir=null
284
+ - ++config_yaml=null
285
+ - ++is_debug=false
286
+ - ++version_id=0.21
287
+ - ++run_root_dir=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints
288
+ - ++trainer.is_resume=false
289
+ - ++trainer.pretrained_checkpoint=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/checkpoints/_initialization/StarVLA_Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
290
+ - ++trainer.resume_step=0
291
+ - ++datasets.vla_data.data_root_dir=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep
292
+ - ++datasets.vla_data.data_mix=flappy_train__bridge
293
+ - ++datasets.vla_data.eval_data_mix=flappy_train__bridge__val
294
+ - ++framework.qwenvl.base_vlm=/workspace/latency-sensitive-bench/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
295
+ - ++rl_games.env_eval.latency.prompt_map_path=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
296
+ job:
297
+ name: train_starvla_hydra
298
+ chdir: false
299
+ override_dirname: ++auth.env_file=null,++auth.hf_token_env=HF_TOKEN,++auth.wandb_api_key_env=WANDB_API_KEY,++base_model.repo_id=Qwen/Qwen3-VL-4B-Instruct,++bridge_base_model.local_dir.gr00t=playground/Pretrained_models/Qwen3-VL-4B-Instruct,++bridge_base_model.local_dir.openvla=playground/Pretrained_models/Qwen3-VL-4B-Instruct,++bridge_base_model.local_dir.pi05=playground/Pretrained_models/Qwen3-VL-4B-Instruct,++bridge_base_model.local_dir.pi0=playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action,++bridge_base_model.repo_id.gr00t=Qwen/Qwen3-VL-4B-Instruct,++bridge_base_model.repo_id.openvla=Qwen/Qwen3-VL-4B-Instruct,++bridge_base_model.repo_id.pi05=Qwen/Qwen3-VL-4B-Instruct,++bridge_base_model.repo_id.pi0=StarVLA/Qwen2.5-VL-3B-Instruct-Action,++checkpoint.hf_repo_id=null,++checkpoint.load=none,++checkpoint.local.keep_last_n=1,++checkpoint.save_best_model=false,++checkpoint.save_pt_file=false,++checkpoint.sync.enabled=false,++checkpoint.sync.keep_last_n=0,++checkpoint.sync.repo_id=null,++checkpoint.sync.resume_policy=local_latest,++checkpoint.sync.sync_every_n_checkpoints=1,++config_yaml=null,++dataset.config_name=null,++dataset.converted_name=flappy_train,++dataset.debug_subset.enabled=false,++dataset.debug_subset.max_episodes=5,++dataset.debug_subset.suffix=debug,++dataset.episodes_per_latency=null,++dataset.force_download=false,++dataset.latency_filter=null,++dataset.max_episodes=null,++dataset.mixed_converted_name=flappy_mixed_latency_train,++dataset.mixed_latency_filter=null,++dataset.mixed_source_hf=,++dataset.setup_force=false,++dataset.single_converted_name=flappy_train,++dataset.single_latency_filter=null,++dataset.single_source_hf=,++dataset.skip_verification=false,++dataset.source_hf=,++dataset.source_subdir=null,++dataset.verify_rows=200,++datasets.vla_data.action_balance.action_key=action_id,++datasets.vla_data.action_balance.enabled=false,++datasets.vla_data.action_balance.flap_id=1,++datasets.vla_data.action_balance.noop_id=0,++datasets.vla_data.action_balance.strategy=balanced_epoch,++datasets.vla_data.action_balance.target_flap_fraction=0.3,++datasets.vla_data.action_type=discrete,++datasets.vla_data.custom_mixtures_path=null,++datasets.vla_data.data_mix=flappy_train,++datasets.vla_data.data_mix=flappy_train__bridge,++datasets.vla_data.data_root_dir=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep,++datasets.vla_data.data_root_dir=playground/Datasets/rl_games,++datasets.vla_data.dataset_py=lerobot_datasets,++datasets.vla_data.eval_data_mix=flappy_train__bridge__val,++datasets.vla_data.eval_data_mix=null,++datasets.vla_data.eval_num_workers=8,++datasets.vla_data.eval_sequential_step_sampling=null,++datasets.vla_data.image_mode=single,++datasets.vla_data.include_state=true,++datasets.vla_data.latency_curriculum.enabled=false,++datasets.vla_data.latency_curriculum.latencies=null,++datasets.vla_data.latency_curriculum.phase_steps=null,++datasets.vla_data.latency_curriculum.strategy=exclusive,++datasets.vla_data.load_all_data_for_training=true,++datasets.vla_data.num_obs_frames=1,++datasets.vla_data.num_workers=8,++datasets.vla_data.obs_image_size=null,++datasets.vla_data.per_device_batch_size=32,++datasets.vla_data.persistent_workers=true,++datasets.vla_data.pin_memory=true,++datasets.vla_data.prefetch_factor=4,++datasets.vla_data.sequential_step_sampling=false,++datasets.vla_data.shuffle=true,++datasets.vla_data.stitch_grid=[2,2],++datasets.vla_data.video_backend=torchvision_av,++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/latency-sensitive-bench/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct,++framework.qwenvl.base_vlm=playground/Pretrained_models/Qwen3-VL-4B-Instruct,++framework.qwenvl.enable_gradient_checkpointing=true,++initialization.checkpoint_filename=checkpoints/steps_5000_pytorch_model.pt,++initialization.checkpoint_hf_repo_id=StarVLA/Qwen3VL-OFT-Bridge-RT-1,++initialization.checkpoint_local_dir=playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1,++is_debug=false,++output_dir=null,++paths.accelerate_config=starVLA/config/deepseeds/deepspeed_zero2.yaml,++paths.base_model_dir=playground/Pretrained_models/Qwen3-VL-4B-Instruct,++paths.dataset_cache_dir=null,++paths.dataset_local_dir=data/flappy_fix_latency_2_200ep,++paths.run_root_dir=results/Checkpoints,++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=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json,++rl_games.env_eval.latency.prompt_map_path=null,++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=250,++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=false,++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_description=,++rl_games.env_eval.task_seed_stride=0,++rl_games.env_eval.vectorized.batch_size=1,++rl_games.env_eval.vectorized.enabled=false,++rl_games.initialization_mode=bridge,++rl_games.model_alias=openvla,++rl_games.task=flappy,++run_id=flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps,++run_root_dir=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints,++run_root_dir=results/Checkpoints,++seed=42,++trainer.cc_f1_tolerance=1,++trainer.distributed_backend=none,++trainer.eval_action_classification=true,++trainer.eval_action_classification_interval=null,++trainer.eval_interval=250,++trainer.eval_num_batches=50,++trainer.freeze_llm_layers=[],++trainer.freeze_modules=,++trainer.freeze_tied_embedding=false,++trainer.freeze_vit=false,++trainer.gradient_accumulation_steps=1,++trainer.gradient_clipping=1.0,++trainer.is_resume=false,++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=1,++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=4000,++trainer.num_warmup_steps=100,++trainer.optimizer.betas=[0.9,0.95],++trainer.optimizer.eps=1e-08,++trainer.optimizer.fused=true,++trainer.optimizer.name=AdamW,++trainer.optimizer.weight_decay=1e-08,++trainer.per_latency_eval_num_batches=null,++trainer.pretrained_checkpoint=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/checkpoints/_initialization/StarVLA_Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt,++trainer.pretrained_checkpoint=null,++trainer.profile_timing.enabled=false,++trainer.profile_timing.log_interval=10,++trainer.reload_modules=null,++trainer.resume_step=0,++trainer.resume_step=0,++trainer.save_format=pt,++trainer.save_interval=500,++trainer.scheduler_specific_kwargs.min_lr=1e-06,++trainer.weight_decay=0.0,++version_id=0.21,++wandb_entity=saberrr-zju,++wandb_project=starVLA_rl_games,++workspace_dir=WORKSPACE_DIR,env=flappy,init=bridge,mode=single,model=openvla
300
+ id: ???
301
+ num: ???
302
+ config_name: train
303
+ env_set: {}
304
+ env_copy: []
305
+ config:
306
+ override_dirname:
307
+ kv_sep: '='
308
+ item_sep: ','
309
+ exclude_keys: []
310
+ runtime:
311
+ version: 1.3.3
312
+ version_base: '1.1'
313
+ cwd: /workspace/latency-sensitive-bench/starVLA
314
+ config_sources:
315
+ - path: hydra.conf
316
+ schema: pkg
317
+ provider: hydra
318
+ - path: /workspace/latency-sensitive-bench/starVLA/examples/rl_games/config
319
+ schema: file
320
+ provider: main
321
+ - path: ''
322
+ schema: structured
323
+ provider: schema
324
+ output_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/hydra
325
+ choices:
326
+ cross_task_setup: null
327
+ checkpoint: default
328
+ mode: single
329
+ init: bridge
330
+ env: flappy
331
+ model: openvla
332
+ hydra/env: default
333
+ hydra/callbacks: null
334
+ hydra/job_logging: default
335
+ hydra/hydra_logging: default
336
+ hydra/hydra_help: default
337
+ hydra/help: default
338
+ hydra/sweeper: basic
339
+ hydra/launcher: basic
340
+ hydra/output: default
341
+ verbose: false
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/hydra/.hydra/overrides.yaml ADDED
@@ -0,0 +1,181 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ - model=openvla
2
+ - env=flappy
3
+ - init=bridge
4
+ - mode=single
5
+ - ++framework.qwenvl.base_vlm=playground/Pretrained_models/Qwen3-VL-4B-Instruct
6
+ - ++framework.qwenvl.attn_implementation=flash_attention_2
7
+ - ++framework.qwenvl.enable_gradient_checkpointing=true
8
+ - ++framework.action_model.state_dim=7
9
+ - ++framework.action_model.loss_type=discrete_ce
10
+ - ++framework.action_model.action_horizon=1
11
+ - ++framework.action_model.future_action_window_size=0
12
+ - ++framework.action_model.past_action_window_size=0
13
+ - ++framework.action_model.action_dim=7
14
+ - ++framework.action_model.action_env_dim=2
15
+ - ++framework.name=QwenOFT
16
+ - ++datasets.vla_data.dataset_py=lerobot_datasets
17
+ - ++datasets.vla_data.include_state=true
18
+ - ++datasets.vla_data.data_root_dir=playground/Datasets/rl_games
19
+ - ++datasets.vla_data.data_mix=flappy_train
20
+ - ++datasets.vla_data.eval_data_mix=null
21
+ - ++datasets.vla_data.custom_mixtures_path=null
22
+ - ++datasets.vla_data.action_type=discrete
23
+ - ++datasets.vla_data.sequential_step_sampling=false
24
+ - ++datasets.vla_data.eval_sequential_step_sampling=null
25
+ - ++datasets.vla_data.num_workers=8
26
+ - ++datasets.vla_data.eval_num_workers=8
27
+ - ++datasets.vla_data.prefetch_factor=4
28
+ - ++datasets.vla_data.persistent_workers=true
29
+ - ++datasets.vla_data.pin_memory=true
30
+ - ++datasets.vla_data.shuffle=true
31
+ - ++datasets.vla_data.action_balance.enabled=false
32
+ - ++datasets.vla_data.action_balance.strategy=balanced_epoch
33
+ - ++datasets.vla_data.action_balance.action_key=action_id
34
+ - ++datasets.vla_data.action_balance.target_flap_fraction=0.3
35
+ - ++datasets.vla_data.action_balance.noop_id=0
36
+ - ++datasets.vla_data.action_balance.flap_id=1
37
+ - ++datasets.vla_data.latency_curriculum.enabled=false
38
+ - ++datasets.vla_data.latency_curriculum.strategy=exclusive
39
+ - ++datasets.vla_data.latency_curriculum.latencies=null
40
+ - ++datasets.vla_data.latency_curriculum.phase_steps=null
41
+ - ++datasets.vla_data.per_device_batch_size=32
42
+ - ++datasets.vla_data.load_all_data_for_training=true
43
+ - ++datasets.vla_data.num_obs_frames=1
44
+ - ++datasets.vla_data.image_mode=single
45
+ - ++datasets.vla_data.stitch_grid=[2,2]
46
+ - ++datasets.vla_data.obs_image_size=null
47
+ - ++datasets.vla_data.video_backend=torchvision_av
48
+ - ++dataset.source_hf=
49
+ - ++dataset.config_name=null
50
+ - ++dataset.source_subdir=null
51
+ - ++dataset.converted_name=flappy_train
52
+ - ++dataset.single_source_hf=
53
+ - ++dataset.mixed_source_hf=
54
+ - ++dataset.single_converted_name=flappy_train
55
+ - ++dataset.mixed_converted_name=flappy_mixed_latency_train
56
+ - ++dataset.single_latency_filter=null
57
+ - ++dataset.mixed_latency_filter=null
58
+ - ++dataset.force_download=false
59
+ - ++dataset.setup_force=false
60
+ - ++dataset.skip_verification=false
61
+ - ++dataset.verify_rows=200
62
+ - ++dataset.max_episodes=null
63
+ - ++dataset.episodes_per_latency=null
64
+ - ++dataset.latency_filter=null
65
+ - ++dataset.debug_subset.enabled=false
66
+ - ++dataset.debug_subset.max_episodes=5
67
+ - ++dataset.debug_subset.suffix=debug
68
+ - ++base_model.repo_id=Qwen/Qwen3-VL-4B-Instruct
69
+ - ++initialization.checkpoint_local_dir=playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
70
+ - ++initialization.checkpoint_hf_repo_id=StarVLA/Qwen3VL-OFT-Bridge-RT-1
71
+ - ++initialization.checkpoint_filename=checkpoints/steps_5000_pytorch_model.pt
72
+ - ++trainer.max_train_steps=4000
73
+ - ++trainer.num_warmup_steps=100
74
+ - ++trainer.save_interval=500
75
+ - ++trainer.eval_interval=250
76
+ - ++trainer.eval_num_batches=50
77
+ - ++trainer.per_latency_eval_num_batches=null
78
+ - ++trainer.eval_action_classification=true
79
+ - ++trainer.eval_action_classification_interval=null
80
+ - ++trainer.cc_f1_tolerance=1
81
+ - ++trainer.learning_rate.base=2e-05
82
+ - ++trainer.learning_rate.qwen_vl_interface=1e-05
83
+ - ++trainer.learning_rate.action_model=0.0001
84
+ - ++trainer.lr_scheduler_type=cosine_with_min_lr
85
+ - ++trainer.scheduler_specific_kwargs.min_lr=1e-06
86
+ - ++trainer.freeze_modules=
87
+ - ++trainer.freeze_vit=false
88
+ - ++trainer.freeze_tied_embedding=false
89
+ - ++trainer.freeze_llm_layers=[]
90
+ - ++trainer.loss_scale.vla=1.0
91
+ - ++trainer.loss_scale.vlm=0.1
92
+ - ++trainer.max_grad_norm=1.0
93
+ - ++trainer.weight_decay=0.0
94
+ - ++trainer.logging_frequency=1
95
+ - ++trainer.profile_timing.enabled=false
96
+ - ++trainer.profile_timing.log_interval=10
97
+ - ++trainer.gradient_clipping=1.0
98
+ - ++trainer.gradient_accumulation_steps=1
99
+ - ++trainer.distributed_backend=none
100
+ - ++trainer.is_resume=false
101
+ - ++trainer.pretrained_checkpoint=null
102
+ - ++trainer.resume_step=0
103
+ - ++trainer.reload_modules=null
104
+ - ++trainer.optimizer.name=AdamW
105
+ - ++trainer.optimizer.betas=[0.9,0.95]
106
+ - ++trainer.optimizer.eps=1e-08
107
+ - ++trainer.optimizer.weight_decay=1e-08
108
+ - ++trainer.optimizer.fused=true
109
+ - ++trainer.save_format=pt
110
+ - ++workspace_dir=WORKSPACE_DIR
111
+ - ++run_root_dir=results/Checkpoints
112
+ - ++seed=42
113
+ - ++wandb_entity=saberrr-zju
114
+ - ++wandb_project=starVLA_rl_games
115
+ - ++auth.env_file=null
116
+ - ++auth.hf_token_env=HF_TOKEN
117
+ - ++auth.wandb_api_key_env=WANDB_API_KEY
118
+ - ++paths.run_root_dir=results/Checkpoints
119
+ - ++paths.dataset_local_dir=data/flappy_fix_latency_2_200ep
120
+ - ++paths.dataset_cache_dir=null
121
+ - ++paths.base_model_dir=playground/Pretrained_models/Qwen3-VL-4B-Instruct
122
+ - ++paths.accelerate_config=starVLA/config/deepseeds/deepspeed_zero2.yaml
123
+ - ++rl_games.model_alias=openvla
124
+ - ++rl_games.env_eval.image_size=224
125
+ - ++rl_games.env_eval.frameskip=1
126
+ - ++rl_games.env_eval.seed=42
127
+ - ++rl_games.env_eval.fixed_episode_seeds=true
128
+ - ++rl_games.env_eval.latency_seed_stride=0
129
+ - ++rl_games.env_eval.task_seed_stride=0
130
+ - ++rl_games.env_eval.task_description=
131
+ - ++rl_games.env_eval.enabled=true
132
+ - ++rl_games.env_eval.distributed_mode=none
133
+ - ++rl_games.env_eval.vectorized.enabled=false
134
+ - ++rl_games.env_eval.vectorized.batch_size=1
135
+ - ++rl_games.env_eval.latency.prompt_map_path=null
136
+ - ++rl_games.env_eval.latency.mode=single
137
+ - ++rl_games.env_eval.latency.values=[0]
138
+ - ++rl_games.env_eval.mid_train.enabled=true
139
+ - ++rl_games.env_eval.mid_train.interval_steps=250
140
+ - ++rl_games.env_eval.mid_train.latencies=[2]
141
+ - ++rl_games.env_eval.mid_train.num_episodes=5
142
+ - ++rl_games.env_eval.mid_train.max_steps_per_episode=3600
143
+ - ++rl_games.env_eval.post_train.enabled=false
144
+ - ++rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
145
+ - ++rl_games.env_eval.post_train.num_episodes=5
146
+ - ++rl_games.env_eval.post_train.max_steps_per_episode=3600
147
+ - ++rl_games.task=flappy
148
+ - ++rl_games.initialization_mode=bridge
149
+ - ++rl_games.action_carrier=bridge
150
+ - ++bridge_base_model.repo_id.openvla=Qwen/Qwen3-VL-4B-Instruct
151
+ - ++bridge_base_model.repo_id.pi0=StarVLA/Qwen2.5-VL-3B-Instruct-Action
152
+ - ++bridge_base_model.repo_id.pi05=Qwen/Qwen3-VL-4B-Instruct
153
+ - ++bridge_base_model.repo_id.gr00t=Qwen/Qwen3-VL-4B-Instruct
154
+ - ++bridge_base_model.local_dir.openvla=playground/Pretrained_models/Qwen3-VL-4B-Instruct
155
+ - ++bridge_base_model.local_dir.pi0=playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
156
+ - ++bridge_base_model.local_dir.pi05=playground/Pretrained_models/Qwen3-VL-4B-Instruct
157
+ - ++bridge_base_model.local_dir.gr00t=playground/Pretrained_models/Qwen3-VL-4B-Instruct
158
+ - ++checkpoint.load=none
159
+ - ++checkpoint.hf_repo_id=null
160
+ - ++checkpoint.save_best_model=false
161
+ - ++checkpoint.save_pt_file=false
162
+ - ++checkpoint.local.keep_last_n=1
163
+ - ++checkpoint.sync.enabled=false
164
+ - ++checkpoint.sync.repo_id=null
165
+ - ++checkpoint.sync.keep_last_n=0
166
+ - ++checkpoint.sync.sync_every_n_checkpoints=1
167
+ - ++checkpoint.sync.resume_policy=local_latest
168
+ - ++run_id=flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps
169
+ - ++output_dir=null
170
+ - ++config_yaml=null
171
+ - ++is_debug=false
172
+ - ++version_id=0.21
173
+ - ++run_root_dir=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints
174
+ - ++trainer.is_resume=false
175
+ - ++trainer.pretrained_checkpoint=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/checkpoints/_initialization/StarVLA_Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
176
+ - ++trainer.resume_step=0
177
+ - ++datasets.vla_data.data_root_dir=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep
178
+ - ++datasets.vla_data.data_mix=flappy_train__bridge
179
+ - ++datasets.vla_data.eval_data_mix=flappy_train__bridge__val
180
+ - ++framework.qwenvl.base_vlm=/workspace/latency-sensitive-bench/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
181
+ - ++rl_games.env_eval.latency.prompt_map_path=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/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/flappy_fix_latency_2_200ep_full_tuning_corrected_bs32_4k_steps/summary.jsonl ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {"steps": 500}
2
+ {"steps": 1000}
3
+ {"steps": 1500}
4
+ {"steps": 2000}
5
+ {"steps": 2500}
6
+ {"steps": 3000}
7
+ {"steps": 3500}
8
+ {"steps": 4000}
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_last_8_layers/checkpoints/steps_5000_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8afaa32fca8bd02a62fd251a0457e809370c64803c90166b0962241a2089519a
3
+ size 9138230516
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_last_8_layers/config.full.yaml ADDED
@@ -0,0 +1,217 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ name: QwenOFT
3
+ qwenvl:
4
+ base_vlm: /workspace/latency-sensitive-bench/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/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep
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
+ eval_sequential_step_sampling: null
28
+ num_workers: 8
29
+ eval_num_workers: 8
30
+ prefetch_factor: 4
31
+ shuffle: true
32
+ action_balance:
33
+ enabled: false
34
+ strategy: balanced_epoch
35
+ action_key: action_id
36
+ target_flap_fraction: 0.3
37
+ noop_id: 0
38
+ flap_id: 1
39
+ latency_curriculum:
40
+ enabled: false
41
+ strategy: exclusive
42
+ latencies: null
43
+ phase_steps: null
44
+ per_device_batch_size: 64
45
+ load_all_data_for_training: true
46
+ num_obs_frames: 1
47
+ image_mode: single
48
+ stitch_grid:
49
+ - 2
50
+ - 2
51
+ obs_image_size: null
52
+ video_backend: torchvision_av
53
+ dataset:
54
+ source_hf: ''
55
+ config_name: null
56
+ source_subdir: null
57
+ converted_name: flappy_train
58
+ single_source_hf: ''
59
+ mixed_source_hf: ''
60
+ single_converted_name: flappy_train
61
+ mixed_converted_name: flappy_mixed_latency_train
62
+ single_latency_filter: null
63
+ mixed_latency_filter: null
64
+ force_download: false
65
+ setup_force: false
66
+ skip_verification: false
67
+ verify_rows: 200
68
+ max_episodes: null
69
+ episodes_per_latency: null
70
+ latency_filter: null
71
+ debug_subset:
72
+ enabled: false
73
+ max_episodes: 5
74
+ suffix: debug
75
+ base_model:
76
+ repo_id: Qwen/Qwen3-VL-4B-Instruct
77
+ initialization:
78
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
79
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
80
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
81
+ trainer:
82
+ max_train_steps: 5000
83
+ num_warmup_steps: 100
84
+ save_interval: 500
85
+ eval_interval: 100
86
+ eval_num_batches: 100
87
+ per_latency_eval_num_batches: null
88
+ eval_action_classification: true
89
+ eval_action_classification_interval: null
90
+ cc_f1_tolerance: 1
91
+ learning_rate:
92
+ base: 2.0e-05
93
+ qwen_vl_interface: 1.0e-05
94
+ action_model: 0.0001
95
+ lr_scheduler_type: cosine_with_min_lr
96
+ scheduler_specific_kwargs:
97
+ min_lr: 1.0e-06
98
+ freeze_modules: ''
99
+ freeze_llm_bottom_ratio: 0.7778
100
+ loss_scale:
101
+ vla: 1.0
102
+ vlm: 0.1
103
+ max_grad_norm: 1.0
104
+ weight_decay: 0.0
105
+ logging_frequency: 1
106
+ gradient_clipping: 1.0
107
+ gradient_accumulation_steps: 2
108
+ distributed_backend: none
109
+ is_resume: false
110
+ pretrained_checkpoint: null
111
+ resume_step: 0
112
+ reload_modules: null
113
+ optimizer:
114
+ name: AdamW
115
+ betas:
116
+ - 0.9
117
+ - 0.95
118
+ eps: 1.0e-08
119
+ weight_decay: 1.0e-08
120
+ fused: false
121
+ save_format: pt
122
+ workspace_dir: WORKSPACE_DIR
123
+ run_root_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints
124
+ seed: 42
125
+ wandb_entity: saberrr-zju
126
+ wandb_project: starVLA_rl_games
127
+ auth:
128
+ env_file: null
129
+ hf_token_env: HF_TOKEN
130
+ wandb_api_key_env: WANDB_API_KEY
131
+ paths:
132
+ run_root_dir: results/Checkpoints
133
+ dataset_local_dir: playground/Datasets/rl_games
134
+ dataset_cache_dir: null
135
+ base_model_dir: playground/Pretrained_models/Qwen3-VL-4B-Instruct
136
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
137
+ launch:
138
+ use_accelerate: true
139
+ gpus: null
140
+ num_processes: 1
141
+ dry_run: false
142
+ conda:
143
+ enabled: true
144
+ env_name: null
145
+ rl_games:
146
+ model_alias: openvla
147
+ env_eval:
148
+ image_size: 224
149
+ frameskip: 1
150
+ seed: 42
151
+ fixed_episode_seeds: true
152
+ latency_seed_stride: 0
153
+ task_seed_stride: 0
154
+ task_description: ''
155
+ enabled: true
156
+ distributed_mode: none
157
+ vectorized:
158
+ enabled: false
159
+ batch_size: 1
160
+ latency:
161
+ prompt_map_path: /workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
162
+ mode: single
163
+ values:
164
+ - 0
165
+ mid_train:
166
+ enabled: true
167
+ interval_steps: 250
168
+ latencies:
169
+ - 2
170
+ num_episodes: 5
171
+ max_steps_per_episode: 3600
172
+ post_train:
173
+ enabled: false
174
+ latencies:
175
+ - 0
176
+ - 1
177
+ - 2
178
+ - 3
179
+ - 4
180
+ num_episodes: 5
181
+ max_steps_per_episode: 3600
182
+ task: flappy
183
+ initialization_mode: bridge
184
+ action_carrier: bridge
185
+ model: openvla
186
+ env: flappy
187
+ init: bridge
188
+ bridge_base_model:
189
+ repo_id:
190
+ openvla: Qwen/Qwen3-VL-4B-Instruct
191
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
192
+ pi05: Qwen/Qwen3-VL-4B-Instruct
193
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
194
+ local_dir:
195
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
196
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
197
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
198
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
199
+ mode: single
200
+ checkpoint:
201
+ load: auto
202
+ hf_repo_id: null
203
+ save_best_model: false
204
+ save_pt_file: false
205
+ local:
206
+ keep_last_n: 1
207
+ sync:
208
+ enabled: false
209
+ repo_id: null
210
+ keep_last_n: 0
211
+ sync_every_n_checkpoints: 1
212
+ resume_policy: local_latest
213
+ run_id: flappy_fix_latency_2_200ep_last_8_layers
214
+ output_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_last_8_layers
215
+ config_yaml: null
216
+ is_debug: false
217
+ version_id: '0.21'
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_last_8_layers/config.yaml ADDED
@@ -0,0 +1,97 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ checkpoint:
2
+ local:
3
+ keep_last_n: 1
4
+ save_best_model: false
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
+ eval_num_workers: 8
16
+ include_state: true
17
+ latency_curriculum:
18
+ enabled: false
19
+ obs_image_size: null
20
+ per_device_batch_size: 64
21
+ prefetch_factor: 4
22
+ framework:
23
+ action_model:
24
+ action_dim: 7
25
+ action_env_dim: 2
26
+ action_hidden_dim: 2560
27
+ action_horizon: 1
28
+ action_model_type: MLP
29
+ loss_type: discrete_ce
30
+ state_dim: 7
31
+ name: QwenOFT
32
+ qwenvl:
33
+ attn_implementation: flash_attention_2
34
+ base_vlm: /workspace/latency-sensitive-bench/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
35
+ enable_gradient_checkpointing: true
36
+ output_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_last_8_layers
37
+ rl_games:
38
+ env_eval:
39
+ distributed_mode: none
40
+ enabled: true
41
+ fixed_episode_seeds: true
42
+ frameskip: 1
43
+ image_size: 224
44
+ latency:
45
+ prompt_map_path: /workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
46
+ latency_seed_stride: 0
47
+ mid_train:
48
+ enabled: true
49
+ interval_steps: 250
50
+ latencies:
51
+ - 2
52
+ max_steps_per_episode: 3600
53
+ num_episodes: 5
54
+ seed: 42
55
+ task_description: ''
56
+ task_seed_stride: 0
57
+ vectorized:
58
+ enabled: false
59
+ model_alias: openvla
60
+ task: flappy
61
+ run_id: flappy_fix_latency_2_200ep_last_8_layers
62
+ run_root_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints
63
+ seed: 42
64
+ trainer:
65
+ distributed_backend: none
66
+ eval_action_classification: true
67
+ eval_action_classification_interval: null
68
+ eval_interval: 100
69
+ eval_num_batches: 100
70
+ freeze_llm_bottom_ratio: 0.7778
71
+ freeze_modules: ''
72
+ gradient_accumulation_steps: 2
73
+ gradient_clipping: 1.0
74
+ is_resume: false
75
+ learning_rate:
76
+ action_model: 0.0001
77
+ base: 2.0e-05
78
+ qwen_vl_interface: 1.0e-05
79
+ logging_frequency: 1
80
+ lr_scheduler_type: cosine_with_min_lr
81
+ max_train_steps: 5000
82
+ num_warmup_steps: 100
83
+ optimizer:
84
+ betas:
85
+ - 0.9
86
+ - 0.95
87
+ eps: 1.0e-08
88
+ fused: false
89
+ weight_decay: 1.0e-08
90
+ per_latency_eval_num_batches: null
91
+ pretrained_checkpoint: null
92
+ reload_modules: null
93
+ save_interval: 500
94
+ scheduler_specific_kwargs:
95
+ min_lr: 1.0e-06
96
+ wandb_entity: saberrr-zju
97
+ wandb_project: starVLA_rl_games
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_last_8_layers/dataset_statistics.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.6028500199317932,
6
+ 0.3971499800682068,
7
+ 0.0,
8
+ 0.0,
9
+ 0.0,
10
+ 0.0,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.4890792667865753,
15
+ 0.4890792667865753,
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": 330734,
125
+ "num_trajectories": 180
126
+ }
127
+ }
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_last_8_layers/dataset_statistics_eval.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.7959861159324646,
6
+ 0.2040138840675354,
7
+ 0.0,
8
+ 0.0,
9
+ 0.0,
10
+ 0.0,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.4030573070049286,
15
+ 0.4030573070049286,
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": 72000,
125
+ "num_trajectories": 20
126
+ }
127
+ }
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_last_8_layers/hydra/.hydra/config.yaml ADDED
@@ -0,0 +1,215 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ qwenvl:
3
+ base_vlm: /workspace/latency-sensitive-bench/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/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep
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
+ eval_sequential_step_sampling: null
26
+ num_workers: 8
27
+ eval_num_workers: 8
28
+ prefetch_factor: 4
29
+ shuffle: true
30
+ action_balance:
31
+ enabled: false
32
+ strategy: balanced_epoch
33
+ action_key: action_id
34
+ target_flap_fraction: 0.3
35
+ noop_id: 0
36
+ flap_id: 1
37
+ latency_curriculum:
38
+ enabled: false
39
+ strategy: exclusive
40
+ latencies: null
41
+ phase_steps: null
42
+ per_device_batch_size: 64
43
+ load_all_data_for_training: true
44
+ num_obs_frames: 1
45
+ image_mode: single
46
+ stitch_grid:
47
+ - 2
48
+ - 2
49
+ obs_image_size: null
50
+ video_backend: torchvision_av
51
+ dataset:
52
+ source_hf: ${dataset.single_source_hf}
53
+ config_name: null
54
+ source_subdir: null
55
+ converted_name: ${dataset.single_converted_name}
56
+ single_source_hf: ''
57
+ mixed_source_hf: ''
58
+ single_converted_name: flappy_train
59
+ mixed_converted_name: flappy_mixed_latency_train
60
+ single_latency_filter: null
61
+ mixed_latency_filter: null
62
+ force_download: false
63
+ setup_force: false
64
+ skip_verification: false
65
+ verify_rows: 200
66
+ max_episodes: null
67
+ episodes_per_latency: null
68
+ latency_filter: ${dataset.single_latency_filter}
69
+ debug_subset:
70
+ enabled: false
71
+ max_episodes: 5
72
+ suffix: debug
73
+ base_model:
74
+ repo_id: ${bridge_base_model.repo_id.${model}}
75
+ initialization:
76
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
77
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
78
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
79
+ trainer:
80
+ max_train_steps: 5000
81
+ num_warmup_steps: 100
82
+ save_interval: 500
83
+ eval_interval: 100
84
+ eval_num_batches: 100
85
+ per_latency_eval_num_batches: null
86
+ eval_action_classification: true
87
+ eval_action_classification_interval: null
88
+ cc_f1_tolerance: 1
89
+ learning_rate:
90
+ base: 2.0e-05
91
+ qwen_vl_interface: 1.0e-05
92
+ action_model: 0.0001
93
+ lr_scheduler_type: cosine_with_min_lr
94
+ scheduler_specific_kwargs:
95
+ min_lr: 1.0e-06
96
+ freeze_modules: ''
97
+ freeze_llm_bottom_ratio: 0.7778
98
+ loss_scale:
99
+ vla: 1.0
100
+ vlm: 0.1
101
+ max_grad_norm: 1.0
102
+ weight_decay: 0.0
103
+ logging_frequency: 1
104
+ gradient_clipping: 1.0
105
+ gradient_accumulation_steps: 2
106
+ distributed_backend: none
107
+ is_resume: false
108
+ pretrained_checkpoint: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_last_8_layers/checkpoints/_initialization/StarVLA_Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
109
+ resume_step: 0
110
+ reload_modules: null
111
+ optimizer:
112
+ name: AdamW
113
+ betas:
114
+ - 0.9
115
+ - 0.95
116
+ eps: 1.0e-08
117
+ weight_decay: 1.0e-08
118
+ fused: false
119
+ save_format: pt
120
+ workspace_dir: WORKSPACE_DIR
121
+ run_root_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints
122
+ seed: 42
123
+ wandb_entity: saberrr-zju
124
+ wandb_project: starVLA_rl_games
125
+ auth:
126
+ env_file: null
127
+ hf_token_env: HF_TOKEN
128
+ wandb_api_key_env: WANDB_API_KEY
129
+ paths:
130
+ run_root_dir: results/Checkpoints
131
+ dataset_local_dir: playground/Datasets/rl_games
132
+ dataset_cache_dir: null
133
+ base_model_dir: ${bridge_base_model.local_dir.${model}}
134
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
135
+ launch:
136
+ use_accelerate: true
137
+ gpus: null
138
+ num_processes: 1
139
+ dry_run: false
140
+ conda:
141
+ enabled: true
142
+ env_name: null
143
+ rl_games:
144
+ model_alias: openvla
145
+ env_eval:
146
+ image_size: 224
147
+ frameskip: 1
148
+ seed: 42
149
+ fixed_episode_seeds: true
150
+ latency_seed_stride: 0
151
+ task_seed_stride: 0
152
+ task_description: ''
153
+ enabled: true
154
+ distributed_mode: none
155
+ vectorized:
156
+ enabled: false
157
+ batch_size: 1
158
+ latency:
159
+ prompt_map_path: /workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
160
+ mode: single
161
+ values:
162
+ - 0
163
+ mid_train:
164
+ enabled: true
165
+ interval_steps: 250
166
+ latencies:
167
+ - 2
168
+ num_episodes: 5
169
+ max_steps_per_episode: 3600
170
+ post_train:
171
+ enabled: false
172
+ latencies:
173
+ - 0
174
+ - 1
175
+ - 2
176
+ - 3
177
+ - 4
178
+ num_episodes: 5
179
+ max_steps_per_episode: 3600
180
+ task: flappy
181
+ initialization_mode: bridge
182
+ action_carrier: bridge
183
+ model: openvla
184
+ env: flappy
185
+ init: bridge
186
+ bridge_base_model:
187
+ repo_id:
188
+ openvla: Qwen/Qwen3-VL-4B-Instruct
189
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
190
+ pi05: Qwen/Qwen3-VL-4B-Instruct
191
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
192
+ local_dir:
193
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
194
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
195
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
196
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
197
+ mode: single
198
+ checkpoint:
199
+ load: auto
200
+ hf_repo_id: null
201
+ save_best_model: false
202
+ save_pt_file: false
203
+ local:
204
+ keep_last_n: 1
205
+ sync:
206
+ enabled: false
207
+ repo_id: null
208
+ keep_last_n: 0
209
+ sync_every_n_checkpoints: 1
210
+ resume_policy: local_latest
211
+ run_id: flappy_fix_latency_2_200ep_last_8_layers
212
+ output_dir: null
213
+ config_yaml: null
214
+ is_debug: false
215
+ version_id: '0.21'
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_last_8_layers/hydra/.hydra/hydra.yaml ADDED
@@ -0,0 +1,259 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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=flappy_fix_latency_2_200ep_last_8_layers
120
+ - run_root_dir=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints
121
+ - seed=42
122
+ - wandb_entity=saberrr-zju
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=false
129
+ - checkpoint.save_pt_file=false
130
+ - trainer.is_resume=false
131
+ - trainer.pretrained_checkpoint=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_last_8_layers/checkpoints/_initialization/StarVLA_Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
132
+ - trainer.resume_step=0
133
+ - trainer.max_train_steps=5000
134
+ - trainer.num_warmup_steps=100
135
+ - trainer.save_interval=500
136
+ - trainer.eval_interval=100
137
+ - trainer.eval_num_batches=100
138
+ - trainer.eval_action_classification=true
139
+ - trainer.logging_frequency=1
140
+ - trainer.gradient_accumulation_steps=2
141
+ - trainer.distributed_backend=none
142
+ - trainer.learning_rate.base=2e-05
143
+ - trainer.learning_rate.qwen_vl_interface=1e-05
144
+ - trainer.learning_rate.action_model=0.0001
145
+ - trainer.lr_scheduler_type=cosine_with_min_lr
146
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
147
+ - trainer.freeze_llm_bottom_ratio=0.7778
148
+ - trainer.loss_scale.vla=1.0
149
+ - trainer.loss_scale.vlm=0.1
150
+ - trainer.max_grad_norm=1.0
151
+ - trainer.weight_decay=0.0
152
+ - trainer.gradient_clipping=1.0
153
+ - trainer.optimizer.name=AdamW
154
+ - trainer.optimizer.betas=[0.9,0.95]
155
+ - trainer.optimizer.eps=1e-08
156
+ - trainer.optimizer.weight_decay=1e-08
157
+ - trainer.optimizer.fused=false
158
+ - trainer.save_format=pt
159
+ - framework.name=QwenOFT
160
+ - framework.qwenvl.attn_implementation=flash_attention_2
161
+ - framework.qwenvl.enable_gradient_checkpointing=true
162
+ - framework.action_model.action_dim=7
163
+ - framework.action_model.action_env_dim=2
164
+ - framework.action_model.state_dim=7
165
+ - framework.action_model.loss_type=discrete_ce
166
+ - framework.action_model.action_horizon=1
167
+ - framework.action_model.future_action_window_size=0
168
+ - framework.action_model.past_action_window_size=0
169
+ - datasets.vla_data.include_state=true
170
+ - datasets.vla_data.action_type=discrete
171
+ - datasets.vla_data.sequential_step_sampling=false
172
+ - datasets.vla_data.shuffle=true
173
+ - datasets.vla_data.action_balance.enabled=false
174
+ - datasets.vla_data.action_balance.strategy=balanced_epoch
175
+ - datasets.vla_data.action_balance.action_key=action_id
176
+ - datasets.vla_data.action_balance.target_flap_fraction=0.3
177
+ - datasets.vla_data.action_balance.noop_id=0
178
+ - datasets.vla_data.action_balance.flap_id=1
179
+ - datasets.vla_data.latency_curriculum.enabled=false
180
+ - datasets.vla_data.latency_curriculum.strategy=exclusive
181
+ - datasets.vla_data.per_device_batch_size=64
182
+ - datasets.vla_data.num_workers=8
183
+ - datasets.vla_data.eval_num_workers=8
184
+ - datasets.vla_data.prefetch_factor=4
185
+ - datasets.vla_data.load_all_data_for_training=true
186
+ - datasets.vla_data.video_backend=torchvision_av
187
+ - datasets.vla_data.data_root_dir=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep
188
+ - datasets.vla_data.data_mix=flappy_train__bridge
189
+ - datasets.vla_data.eval_data_mix=flappy_train__bridge__val
190
+ - framework.qwenvl.base_vlm=/workspace/latency-sensitive-bench/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
191
+ - rl_games.task=flappy
192
+ - rl_games.model_alias=openvla
193
+ - rl_games.initialization_mode=bridge
194
+ - rl_games.action_carrier=bridge
195
+ - rl_games.env_eval.distributed_mode=none
196
+ - rl_games.env_eval.latency.mode=single
197
+ - rl_games.env_eval.frameskip=1
198
+ - rl_games.env_eval.image_size=224
199
+ - rl_games.env_eval.seed=42
200
+ - rl_games.env_eval.fixed_episode_seeds=true
201
+ - rl_games.env_eval.latency_seed_stride=0
202
+ - rl_games.env_eval.task_seed_stride=0
203
+ - rl_games.env_eval.latency.values=[0]
204
+ - rl_games.env_eval.latency.prompt_map_path=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
205
+ - rl_games.env_eval.mid_train.enabled=true
206
+ - rl_games.env_eval.mid_train.interval_steps=250
207
+ - rl_games.env_eval.mid_train.num_episodes=5
208
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
209
+ - rl_games.env_eval.mid_train.latencies=[2]
210
+ - rl_games.env_eval.post_train.enabled=false
211
+ - rl_games.env_eval.post_train.num_episodes=5
212
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
213
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
214
+ job:
215
+ name: train_starvla_hydra
216
+ chdir: false
217
+ override_dirname: checkpoint.local.keep_last_n=1,checkpoint.save_best_model=false,checkpoint.save_pt_file=false,checkpoint.sync.enabled=false,checkpoint.sync.keep_last_n=0,datasets.vla_data.action_balance.action_key=action_id,datasets.vla_data.action_balance.enabled=false,datasets.vla_data.action_balance.flap_id=1,datasets.vla_data.action_balance.noop_id=0,datasets.vla_data.action_balance.strategy=balanced_epoch,datasets.vla_data.action_balance.target_flap_fraction=0.3,datasets.vla_data.action_type=discrete,datasets.vla_data.data_mix=flappy_train__bridge,datasets.vla_data.data_root_dir=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep,datasets.vla_data.eval_data_mix=flappy_train__bridge__val,datasets.vla_data.eval_num_workers=8,datasets.vla_data.include_state=true,datasets.vla_data.latency_curriculum.enabled=false,datasets.vla_data.latency_curriculum.strategy=exclusive,datasets.vla_data.load_all_data_for_training=true,datasets.vla_data.num_workers=8,datasets.vla_data.per_device_batch_size=64,datasets.vla_data.prefetch_factor=4,datasets.vla_data.sequential_step_sampling=false,datasets.vla_data.shuffle=true,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/latency-sensitive-bench/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=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/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=250,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=false,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=flappy_fix_latency_2_200ep_last_8_layers,run_root_dir=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints,seed=42,trainer.distributed_backend=none,trainer.eval_action_classification=true,trainer.eval_interval=100,trainer.eval_num_batches=100,trainer.freeze_llm_bottom_ratio=0.7778,trainer.gradient_accumulation_steps=2,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=1,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=5000,trainer.num_warmup_steps=100,trainer.optimizer.betas=[0.9,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/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_last_8_layers/checkpoints/_initialization/StarVLA_Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt,trainer.resume_step=0,trainer.save_format=pt,trainer.save_interval=500,trainer.scheduler_specific_kwargs.min_lr=1e-06,trainer.weight_decay=0.0,wandb_entity=saberrr-zju,wandb_project=starVLA_rl_games
218
+ id: ???
219
+ num: ???
220
+ config_name: train
221
+ env_set: {}
222
+ env_copy: []
223
+ config:
224
+ override_dirname:
225
+ kv_sep: '='
226
+ item_sep: ','
227
+ exclude_keys: []
228
+ runtime:
229
+ version: 1.3.3
230
+ version_base: '1.1'
231
+ cwd: /workspace/latency-sensitive-bench/starVLA
232
+ config_sources:
233
+ - path: hydra.conf
234
+ schema: pkg
235
+ provider: hydra
236
+ - path: /workspace/latency-sensitive-bench/starVLA/examples/rl_games/config
237
+ schema: file
238
+ provider: main
239
+ - path: ''
240
+ schema: structured
241
+ provider: schema
242
+ output_dir: /workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_last_8_layers/hydra
243
+ choices:
244
+ cross_task_setup: null
245
+ checkpoint: default
246
+ mode: single
247
+ init: bridge
248
+ env: flappy
249
+ model: openvla
250
+ hydra/env: default
251
+ hydra/callbacks: null
252
+ hydra/job_logging: default
253
+ hydra/hydra_logging: default
254
+ hydra/hydra_help: default
255
+ hydra/help: default
256
+ hydra/sweeper: basic
257
+ hydra/launcher: basic
258
+ hydra/output: default
259
+ verbose: false
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_last_8_layers/hydra/.hydra/overrides.yaml ADDED
@@ -0,0 +1,99 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ - model=openvla
2
+ - env=flappy
3
+ - init=bridge
4
+ - mode=single
5
+ - run_id=flappy_fix_latency_2_200ep_last_8_layers
6
+ - run_root_dir=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints
7
+ - seed=42
8
+ - wandb_entity=saberrr-zju
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=false
15
+ - checkpoint.save_pt_file=false
16
+ - trainer.is_resume=false
17
+ - trainer.pretrained_checkpoint=/workspace/latency-sensitive-bench/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep_last_8_layers/checkpoints/_initialization/StarVLA_Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
18
+ - trainer.resume_step=0
19
+ - trainer.max_train_steps=5000
20
+ - trainer.num_warmup_steps=100
21
+ - trainer.save_interval=500
22
+ - trainer.eval_interval=100
23
+ - trainer.eval_num_batches=100
24
+ - trainer.eval_action_classification=true
25
+ - trainer.logging_frequency=1
26
+ - trainer.gradient_accumulation_steps=2
27
+ - trainer.distributed_backend=none
28
+ - trainer.learning_rate.base=2e-05
29
+ - trainer.learning_rate.qwen_vl_interface=1e-05
30
+ - trainer.learning_rate.action_model=0.0001
31
+ - trainer.lr_scheduler_type=cosine_with_min_lr
32
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
33
+ - trainer.freeze_llm_bottom_ratio=0.7778
34
+ - trainer.loss_scale.vla=1.0
35
+ - trainer.loss_scale.vlm=0.1
36
+ - trainer.max_grad_norm=1.0
37
+ - trainer.weight_decay=0.0
38
+ - trainer.gradient_clipping=1.0
39
+ - trainer.optimizer.name=AdamW
40
+ - trainer.optimizer.betas=[0.9,0.95]
41
+ - trainer.optimizer.eps=1e-08
42
+ - trainer.optimizer.weight_decay=1e-08
43
+ - trainer.optimizer.fused=false
44
+ - trainer.save_format=pt
45
+ - framework.name=QwenOFT
46
+ - framework.qwenvl.attn_implementation=flash_attention_2
47
+ - framework.qwenvl.enable_gradient_checkpointing=true
48
+ - framework.action_model.action_dim=7
49
+ - framework.action_model.action_env_dim=2
50
+ - framework.action_model.state_dim=7
51
+ - framework.action_model.loss_type=discrete_ce
52
+ - framework.action_model.action_horizon=1
53
+ - framework.action_model.future_action_window_size=0
54
+ - framework.action_model.past_action_window_size=0
55
+ - datasets.vla_data.include_state=true
56
+ - datasets.vla_data.action_type=discrete
57
+ - datasets.vla_data.sequential_step_sampling=false
58
+ - datasets.vla_data.shuffle=true
59
+ - datasets.vla_data.action_balance.enabled=false
60
+ - datasets.vla_data.action_balance.strategy=balanced_epoch
61
+ - datasets.vla_data.action_balance.action_key=action_id
62
+ - datasets.vla_data.action_balance.target_flap_fraction=0.3
63
+ - datasets.vla_data.action_balance.noop_id=0
64
+ - datasets.vla_data.action_balance.flap_id=1
65
+ - datasets.vla_data.latency_curriculum.enabled=false
66
+ - datasets.vla_data.latency_curriculum.strategy=exclusive
67
+ - datasets.vla_data.per_device_batch_size=64
68
+ - datasets.vla_data.num_workers=8
69
+ - datasets.vla_data.eval_num_workers=8
70
+ - datasets.vla_data.prefetch_factor=4
71
+ - datasets.vla_data.load_all_data_for_training=true
72
+ - datasets.vla_data.video_backend=torchvision_av
73
+ - datasets.vla_data.data_root_dir=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep
74
+ - datasets.vla_data.data_mix=flappy_train__bridge
75
+ - datasets.vla_data.eval_data_mix=flappy_train__bridge__val
76
+ - framework.qwenvl.base_vlm=/workspace/latency-sensitive-bench/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
77
+ - rl_games.task=flappy
78
+ - rl_games.model_alias=openvla
79
+ - rl_games.initialization_mode=bridge
80
+ - rl_games.action_carrier=bridge
81
+ - rl_games.env_eval.distributed_mode=none
82
+ - rl_games.env_eval.latency.mode=single
83
+ - rl_games.env_eval.frameskip=1
84
+ - rl_games.env_eval.image_size=224
85
+ - rl_games.env_eval.seed=42
86
+ - rl_games.env_eval.fixed_episode_seeds=true
87
+ - rl_games.env_eval.latency_seed_stride=0
88
+ - rl_games.env_eval.task_seed_stride=0
89
+ - rl_games.env_eval.latency.values=[0]
90
+ - rl_games.env_eval.latency.prompt_map_path=/workspace/latency-sensitive-bench/starVLA/data/flappy_fix_latency_2_200ep/flappy_train__bridge/latency_prompt_map.json
91
+ - rl_games.env_eval.mid_train.enabled=true
92
+ - rl_games.env_eval.mid_train.interval_steps=250
93
+ - rl_games.env_eval.mid_train.num_episodes=5
94
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
95
+ - rl_games.env_eval.mid_train.latencies=[2]
96
+ - rl_games.env_eval.post_train.enabled=false
97
+ - rl_games.env_eval.post_train.num_episodes=5
98
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
99
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep_last_8_layers/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/flappy_fix_latency_2_200ep_last_8_layers/summary.jsonl ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {"steps": 500}
2
+ {"steps": 1000}
3
+ {"steps": 1500}
4
+ {"steps": 2000}
5
+ {"steps": 2500}
6
+ {"steps": 3000}
7
+ {"steps": 3500}
8
+ {"steps": 4000}
9
+ {"steps": 4500}
10
+ {"steps": 5000}