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

Consolidate paper experiment assets by task and run (301-450)

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/deadly-corridor/vla/starvla-qwenoft-h1/openvla_deadly_corridor_fix_latency_2/checkpoints/best_model.safetensors +3 -0
  2. latency-aware/deadly-corridor/vla/starvla-qwenoft-h1/openvla_deadly_corridor_fix_latency_2/checkpoints/steps_500_model.safetensors +3 -0
  3. latency-aware/deadly-corridor/vla/starvla-qwenoft-h1/openvla_deadly_corridor_fix_latency_2/config.full.yaml +191 -0
  4. latency-aware/deadly-corridor/vla/starvla-qwenoft-h1/openvla_deadly_corridor_fix_latency_2/config.yaml +92 -0
  5. latency-aware/deadly-corridor/vla/starvla-qwenoft-h1/openvla_deadly_corridor_fix_latency_2/dataset_statistics.json +127 -0
  6. latency-aware/deadly-corridor/vla/starvla-qwenoft-h1/openvla_deadly_corridor_fix_latency_2/dataset_statistics_eval.json +127 -0
  7. latency-aware/deadly-corridor/vla/starvla-qwenoft-h1/openvla_deadly_corridor_fix_latency_2/hydra/.hydra/config.yaml +189 -0
  8. latency-aware/deadly-corridor/vla/starvla-qwenoft-h1/openvla_deadly_corridor_fix_latency_2/hydra/.hydra/hydra.yaml +252 -0
  9. latency-aware/deadly-corridor/vla/starvla-qwenoft-h1/openvla_deadly_corridor_fix_latency_2/hydra/.hydra/overrides.yaml +89 -0
  10. latency-aware/deadly-corridor/vla/starvla-qwenoft-h1/openvla_deadly_corridor_fix_latency_2/hydra/train_starvla_hydra.log +0 -0
  11. latency-aware/deadly-corridor/vla/starvla-qwenoft-h1/openvla_deadly_corridor_fix_latency_2/summary.jsonl +10 -0
  12. latency-aware/demon-attack/vla/starvla-qwenoft-h1/openvla_demon_attack_fix_latency_2_small/checkpoints/best_model.safetensors +3 -0
  13. latency-aware/demon-attack/vla/starvla-qwenoft-h1/openvla_demon_attack_fix_latency_2_small/checkpoints/steps_2000_model.safetensors +3 -0
  14. latency-aware/demon-attack/vla/starvla-qwenoft-h1/openvla_demon_attack_fix_latency_2_small/config.full.yaml +190 -0
  15. latency-aware/demon-attack/vla/starvla-qwenoft-h1/openvla_demon_attack_fix_latency_2_small/config.yaml +92 -0
  16. latency-aware/demon-attack/vla/starvla-qwenoft-h1/openvla_demon_attack_fix_latency_2_small/dataset_statistics.json +127 -0
  17. latency-aware/demon-attack/vla/starvla-qwenoft-h1/openvla_demon_attack_fix_latency_2_small/dataset_statistics_eval.json +127 -0
  18. latency-aware/demon-attack/vla/starvla-qwenoft-h1/openvla_demon_attack_fix_latency_2_small/hydra/.hydra/config.yaml +188 -0
  19. latency-aware/demon-attack/vla/starvla-qwenoft-h1/openvla_demon_attack_fix_latency_2_small/hydra/.hydra/hydra.yaml +251 -0
  20. latency-aware/demon-attack/vla/starvla-qwenoft-h1/openvla_demon_attack_fix_latency_2_small/hydra/.hydra/overrides.yaml +88 -0
  21. latency-aware/demon-attack/vla/starvla-qwenoft-h1/openvla_demon_attack_fix_latency_2_small/hydra/train_starvla_hydra.log +0 -0
  22. latency-aware/demon-attack/vla/starvla-qwenoft-h1/openvla_demon_attack_fix_latency_2_small/summary.jsonl +20 -0
  23. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_1_200ep/checkpoints/steps_5000_model.safetensors +3 -0
  24. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_1_200ep/config.full.yaml +198 -0
  25. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_1_200ep/config.yaml +71 -0
  26. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_1_200ep/dataset_statistics.json +127 -0
  27. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_1_200ep/dataset_statistics_eval.json +127 -0
  28. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_1_200ep/hydra/.hydra/config.yaml +196 -0
  29. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_1_200ep/hydra/.hydra/hydra.yaml +252 -0
  30. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_1_200ep/hydra/.hydra/overrides.yaml +91 -0
  31. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_1_200ep/hydra/train_starvla_hydra.log +0 -0
  32. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_1_200ep/summary.jsonl +50 -0
  33. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep/checkpoints/steps_5000_model.safetensors +3 -0
  34. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep/config.full.yaml +199 -0
  35. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep/config.yaml +71 -0
  36. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep/dataset_statistics.json +127 -0
  37. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep/dataset_statistics_eval.json +127 -0
  38. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep/hydra/.hydra/config.yaml +197 -0
  39. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep/hydra/.hydra/hydra.yaml +252 -0
  40. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep/hydra/.hydra/overrides.yaml +91 -0
  41. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep/hydra/train_starvla_hydra.log +0 -0
  42. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep/summary.jsonl +50 -0
  43. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_3_200ep/checkpoints/steps_5000_model.safetensors +3 -0
  44. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_3_200ep/config.full.yaml +199 -0
  45. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_3_200ep/config.yaml +71 -0
  46. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_3_200ep/dataset_statistics.json +127 -0
  47. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_3_200ep/dataset_statistics_eval.json +127 -0
  48. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_3_200ep/hydra/.hydra/config.yaml +197 -0
  49. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_3_200ep/hydra/.hydra/hydra.yaml +252 -0
  50. latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_3_200ep/hydra/.hydra/overrides.yaml +91 -0
latency-aware/deadly-corridor/vla/starvla-qwenoft-h1/openvla_deadly_corridor_fix_latency_2/checkpoints/best_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:630d55bc258d986e0de4b04d9223f418855bb94c2d310c13ae808912553d6c7f
3
+ size 9138230516
latency-aware/deadly-corridor/vla/starvla-qwenoft-h1/openvla_deadly_corridor_fix_latency_2/checkpoints/steps_500_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ad5f732f19bbfda68d6686e39e875f07957d08980bf6325dea7f85eccb4abc70
3
+ size 9138230516
latency-aware/deadly-corridor/vla/starvla-qwenoft-h1/openvla_deadly_corridor_fix_latency_2/config.full.yaml ADDED
@@ -0,0 +1,191 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ name: QwenOFT
3
+ qwenvl:
4
+ base_vlm: /inspire/hdd/global_user/liumingyu-253208120284/lzj/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: 7
17
+ datasets:
18
+ vla_data:
19
+ dataset_py: lerobot_datasets
20
+ include_state: true
21
+ data_root_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/deadly_corridor_fix_latency_2
22
+ data_mix: deadly_corridor_train__bridge
23
+ eval_data_mix: deadly_corridor_train__bridge__val
24
+ custom_mixtures_path: null
25
+ action_type: discrete
26
+ sequential_step_sampling: false
27
+ per_device_batch_size: 32
28
+ load_all_data_for_training: true
29
+ obs_image_size: null
30
+ video_backend: torchvision_av
31
+ dataset:
32
+ source_hf: ''
33
+ converted_name: deadly_corridor_train
34
+ single_source_hf: ''
35
+ mixed_source_hf: ''
36
+ single_converted_name: deadly_corridor_train
37
+ mixed_converted_name: deadly_corridor_mixed_latency_train
38
+ single_latency_filter: null
39
+ mixed_latency_filter: null
40
+ force_download: false
41
+ setup_force: false
42
+ verify_rows: 200
43
+ max_episodes: null
44
+ episodes_per_latency: null
45
+ latency_filter: null
46
+ debug_subset:
47
+ enabled: false
48
+ max_episodes: 5
49
+ suffix: debug
50
+ base_model:
51
+ repo_id: Qwen/Qwen3-VL-4B-Instruct
52
+ initialization:
53
+ checkpoint_local_dir: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
54
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
55
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
56
+ trainer:
57
+ max_train_steps: 500
58
+ num_warmup_steps: 100
59
+ save_interval: 50
60
+ eval_interval: 100
61
+ eval_num_batches: 20
62
+ learning_rate:
63
+ base: 2.0e-05
64
+ qwen_vl_interface: 1.0e-05
65
+ action_model: 0.0001
66
+ lr_scheduler_type: cosine_with_min_lr
67
+ scheduler_specific_kwargs:
68
+ min_lr: 1.0e-06
69
+ freeze_modules: ''
70
+ loss_scale:
71
+ vla: 1.0
72
+ vlm: 0.1
73
+ max_grad_norm: 1.0
74
+ weight_decay: 0.0
75
+ logging_frequency: 1
76
+ gradient_clipping: 1.0
77
+ gradient_accumulation_steps: 16
78
+ distributed_backend: none
79
+ is_resume: false
80
+ pretrained_checkpoint: null
81
+ resume_step: 0
82
+ reload_modules: null
83
+ optimizer:
84
+ name: AdamW
85
+ betas:
86
+ - 0.9
87
+ - 0.95
88
+ eps: 1.0e-08
89
+ weight_decay: 1.0e-08
90
+ fused: false
91
+ save_format: pt
92
+ workspace_dir: WORKSPACE_DIR
93
+ run_root_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints
94
+ seed: 42
95
+ wandb_entity: dongqianyu99-zhejiang-university
96
+ wandb_project: starVLA_rl_games
97
+ auth:
98
+ env_file: null
99
+ hf_token_env: HF_TOKEN
100
+ wandb_api_key_env: WANDB_API_KEY
101
+ paths:
102
+ run_root_dir: results/Checkpoints
103
+ dataset_local_dir: playground/Datasets/rl_games
104
+ dataset_cache_dir: null
105
+ base_model_dir: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
106
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
107
+ launch:
108
+ use_accelerate: true
109
+ gpus: null
110
+ num_processes: 1
111
+ dry_run: false
112
+ conda:
113
+ enabled: true
114
+ env_name: null
115
+ rl_games:
116
+ model_alias: openvla
117
+ env_eval:
118
+ image_size: 224
119
+ frameskip: 4
120
+ seed: 42
121
+ fixed_episode_seeds: true
122
+ latency_seed_stride: 0
123
+ task_seed_stride: 0
124
+ task_description: You are playing Deadly Corridor in VizDoom. Choose actions from
125
+ MOVE_FORWARD, MOVE_BACKWARD, MOVE_LEFT, MOVE_RIGHT, TURN_LEFT, TURN_RIGHT, ATTACK.
126
+ deadly:
127
+ action_layout: multibinary_7
128
+ enabled: true
129
+ distributed_mode: none
130
+ vectorized:
131
+ enabled: false
132
+ batch_size: 1
133
+ latency:
134
+ prompt_map_path: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/deadly_corridor_fix_latency_2/deadly_corridor_train__bridge/latency_prompt_map.json
135
+ mode: single
136
+ values:
137
+ - 0
138
+ mid_train:
139
+ enabled: true
140
+ interval_steps: 50
141
+ latencies:
142
+ - 2
143
+ num_episodes: 5
144
+ max_steps_per_episode: 3600
145
+ post_train:
146
+ enabled: true
147
+ latencies:
148
+ - 0
149
+ - 2
150
+ - 4
151
+ - 6
152
+ - 8
153
+ num_episodes: 5
154
+ max_steps_per_episode: 3600
155
+ task: deadly_corridor
156
+ deadly_corridor_loss_type: null
157
+ initialization_mode: bridge
158
+ action_carrier: bridge
159
+ model: openvla
160
+ env: deadly_corridor
161
+ init: bridge
162
+ bridge_base_model:
163
+ repo_id:
164
+ openvla: Qwen/Qwen3-VL-4B-Instruct
165
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
166
+ pi05: Qwen/Qwen3-VL-4B-Instruct
167
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
168
+ local_dir:
169
+ openvla: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
170
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
171
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
172
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
173
+ mode: single
174
+ checkpoint:
175
+ load: auto
176
+ hf_repo_id: null
177
+ save_best_model: true
178
+ save_pt_file: false
179
+ local:
180
+ keep_last_n: 1
181
+ sync:
182
+ enabled: false
183
+ repo_id: null
184
+ keep_last_n: 0
185
+ sync_every_n_checkpoints: 1
186
+ resume_policy: local_latest
187
+ run_id: openvla_deadly_corridor_fix_latency_2
188
+ output_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints/openvla_deadly_corridor_fix_latency_2
189
+ config_yaml: null
190
+ is_debug: false
191
+ version_id: '0.21'
latency-aware/deadly-corridor/vla/starvla-qwenoft-h1/openvla_deadly_corridor_fix_latency_2/config.yaml ADDED
@@ -0,0 +1,92 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ checkpoint:
2
+ local:
3
+ keep_last_n: 1
4
+ save_best_model: true
5
+ save_pt_file: false
6
+ sync:
7
+ enabled: false
8
+ keep_last_n: 0
9
+ repo_id: null
10
+ datasets:
11
+ vla_data:
12
+ data_mix: deadly_corridor_train__bridge
13
+ dataset_py: lerobot_datasets
14
+ eval_data_mix: deadly_corridor_train__bridge__val
15
+ include_state: true
16
+ obs_image_size: null
17
+ per_device_batch_size: 32
18
+ framework:
19
+ action_model:
20
+ action_dim: 7
21
+ action_env_dim: 7
22
+ action_hidden_dim: 2560
23
+ action_horizon: 1
24
+ action_model_type: MLP
25
+ loss_type: discrete_ce
26
+ state_dim: 7
27
+ name: QwenOFT
28
+ qwenvl:
29
+ attn_implementation: flash_attention_2
30
+ base_vlm: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct
31
+ enable_gradient_checkpointing: true
32
+ output_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints/openvla_deadly_corridor_fix_latency_2
33
+ rl_games:
34
+ env_eval:
35
+ deadly:
36
+ action_layout: multibinary_7
37
+ distributed_mode: none
38
+ enabled: true
39
+ fixed_episode_seeds: true
40
+ frameskip: 4
41
+ image_size: 224
42
+ latency:
43
+ prompt_map_path: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/deadly_corridor_fix_latency_2/deadly_corridor_train__bridge/latency_prompt_map.json
44
+ latency_seed_stride: 0
45
+ mid_train:
46
+ enabled: true
47
+ interval_steps: 50
48
+ latencies:
49
+ - 2
50
+ max_steps_per_episode: 3600
51
+ num_episodes: 5
52
+ seed: 42
53
+ task_description: You are playing Deadly Corridor in VizDoom. Choose actions from
54
+ MOVE_FORWARD, MOVE_BACKWARD, MOVE_LEFT, MOVE_RIGHT, TURN_LEFT, TURN_RIGHT, ATTACK.
55
+ task_seed_stride: 0
56
+ vectorized:
57
+ enabled: false
58
+ model_alias: openvla
59
+ task: deadly_corridor
60
+ run_id: openvla_deadly_corridor_fix_latency_2
61
+ run_root_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints
62
+ seed: 42
63
+ trainer:
64
+ distributed_backend: none
65
+ eval_interval: 100
66
+ eval_num_batches: 20
67
+ freeze_modules: ''
68
+ gradient_accumulation_steps: 16
69
+ gradient_clipping: 1.0
70
+ is_resume: false
71
+ learning_rate:
72
+ action_model: 0.0001
73
+ base: 2.0e-05
74
+ qwen_vl_interface: 1.0e-05
75
+ logging_frequency: 1
76
+ lr_scheduler_type: cosine_with_min_lr
77
+ max_train_steps: 500
78
+ num_warmup_steps: 100
79
+ optimizer:
80
+ betas:
81
+ - 0.9
82
+ - 0.95
83
+ eps: 1.0e-08
84
+ fused: false
85
+ weight_decay: 1.0e-08
86
+ pretrained_checkpoint: null
87
+ reload_modules: null
88
+ save_interval: 50
89
+ scheduler_specific_kwargs:
90
+ min_lr: 1.0e-06
91
+ wandb_entity: dongqianyu99-zhejiang-university
92
+ wandb_project: starVLA_rl_games
latency-aware/deadly-corridor/vla/starvla-qwenoft-h1/openvla_deadly_corridor_fix_latency_2/dataset_statistics.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.9800952076911926,
6
+ 0.00043271310278214514,
7
+ 0.04197317361831665,
8
+ 0.853310227394104,
9
+ 0.33102554082870483,
10
+ 0.3102552890777588,
11
+ 0.8277801871299744
12
+ ],
13
+ "std": [
14
+ 0.13967378437519073,
15
+ 0.02079719677567482,
16
+ 0.2005293071269989,
17
+ 0.3537939190864563,
18
+ 0.47058752179145813,
19
+ 0.4625938832759857,
20
+ 0.3775743842124939
21
+ ],
22
+ "max": [
23
+ 1.0,
24
+ 1.0,
25
+ 1.0,
26
+ 1.0,
27
+ 1.0,
28
+ 1.0,
29
+ 1.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
+ 0.0,
52
+ 1.0,
53
+ 1.0,
54
+ 1.0,
55
+ 1.0,
56
+ 1.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": 2311,
125
+ "num_trajectories": 90
126
+ }
127
+ }
latency-aware/deadly-corridor/vla/starvla-qwenoft-h1/openvla_deadly_corridor_fix_latency_2/dataset_statistics_eval.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.9674796462059021,
6
+ 0.0,
7
+ 0.05284553021192551,
8
+ 0.8211382031440735,
9
+ 0.3211382031440735,
10
+ 0.32926830649375916,
11
+ 0.8373983502388
12
+ ],
13
+ "std": [
14
+ 0.17737756669521332,
15
+ 0.0,
16
+ 0.22372478246688843,
17
+ 0.38323649764060974,
18
+ 0.4669141173362732,
19
+ 0.46994760632514954,
20
+ 0.36900219321250916
21
+ ],
22
+ "max": [
23
+ 1.0,
24
+ 0.0,
25
+ 1.0,
26
+ 1.0,
27
+ 1.0,
28
+ 1.0,
29
+ 1.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
+ 0.0,
52
+ 1.0,
53
+ 1.0,
54
+ 1.0,
55
+ 1.0,
56
+ 1.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": 246,
125
+ "num_trajectories": 10
126
+ }
127
+ }
latency-aware/deadly-corridor/vla/starvla-qwenoft-h1/openvla_deadly_corridor_fix_latency_2/hydra/.hydra/config.yaml ADDED
@@ -0,0 +1,189 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ qwenvl:
3
+ base_vlm: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct
4
+ attn_implementation: flash_attention_2
5
+ enable_gradient_checkpointing: true
6
+ action_model:
7
+ state_dim: 7
8
+ loss_type: discrete_ce
9
+ action_horizon: 1
10
+ future_action_window_size: 0
11
+ past_action_window_size: 0
12
+ action_dim: 7
13
+ action_env_dim: 7
14
+ name: QwenOFT
15
+ datasets:
16
+ vla_data:
17
+ dataset_py: lerobot_datasets
18
+ include_state: true
19
+ data_root_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/deadly_corridor_fix_latency_2
20
+ data_mix: deadly_corridor_train__bridge
21
+ eval_data_mix: deadly_corridor_train__bridge__val
22
+ custom_mixtures_path: null
23
+ action_type: discrete
24
+ sequential_step_sampling: false
25
+ per_device_batch_size: 32
26
+ load_all_data_for_training: true
27
+ obs_image_size: null
28
+ video_backend: torchvision_av
29
+ dataset:
30
+ source_hf: ${dataset.single_source_hf}
31
+ converted_name: ${dataset.single_converted_name}
32
+ single_source_hf: ''
33
+ mixed_source_hf: ''
34
+ single_converted_name: deadly_corridor_train
35
+ mixed_converted_name: deadly_corridor_mixed_latency_train
36
+ single_latency_filter: null
37
+ mixed_latency_filter: null
38
+ force_download: false
39
+ setup_force: false
40
+ verify_rows: 200
41
+ max_episodes: null
42
+ episodes_per_latency: null
43
+ latency_filter: ${dataset.single_latency_filter}
44
+ debug_subset:
45
+ enabled: false
46
+ max_episodes: 5
47
+ suffix: debug
48
+ base_model:
49
+ repo_id: ${bridge_base_model.repo_id.${model}}
50
+ initialization:
51
+ checkpoint_local_dir: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
52
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
53
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
54
+ trainer:
55
+ max_train_steps: 500
56
+ num_warmup_steps: 100
57
+ save_interval: 50
58
+ eval_interval: 100
59
+ eval_num_batches: 20
60
+ learning_rate:
61
+ base: 2.0e-05
62
+ qwen_vl_interface: 1.0e-05
63
+ action_model: 0.0001
64
+ lr_scheduler_type: cosine_with_min_lr
65
+ scheduler_specific_kwargs:
66
+ min_lr: 1.0e-06
67
+ freeze_modules: ''
68
+ loss_scale:
69
+ vla: 1.0
70
+ vlm: 0.1
71
+ max_grad_norm: 1.0
72
+ weight_decay: 0.0
73
+ logging_frequency: 1
74
+ gradient_clipping: 1.0
75
+ gradient_accumulation_steps: 16
76
+ distributed_backend: none
77
+ is_resume: false
78
+ pretrained_checkpoint: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
79
+ resume_step: 0
80
+ reload_modules: null
81
+ optimizer:
82
+ name: AdamW
83
+ betas:
84
+ - 0.9
85
+ - 0.95
86
+ eps: 1.0e-08
87
+ weight_decay: 1.0e-08
88
+ fused: false
89
+ save_format: pt
90
+ workspace_dir: WORKSPACE_DIR
91
+ run_root_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints
92
+ seed: 42
93
+ wandb_entity: dongqianyu99-zhejiang-university
94
+ wandb_project: starVLA_rl_games
95
+ auth:
96
+ env_file: null
97
+ hf_token_env: HF_TOKEN
98
+ wandb_api_key_env: WANDB_API_KEY
99
+ paths:
100
+ run_root_dir: results/Checkpoints
101
+ dataset_local_dir: playground/Datasets/rl_games
102
+ dataset_cache_dir: null
103
+ base_model_dir: ${bridge_base_model.local_dir.${model}}
104
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
105
+ launch:
106
+ use_accelerate: true
107
+ gpus: null
108
+ num_processes: 1
109
+ dry_run: false
110
+ conda:
111
+ enabled: true
112
+ env_name: null
113
+ rl_games:
114
+ model_alias: openvla
115
+ env_eval:
116
+ image_size: 224
117
+ frameskip: 4
118
+ seed: 42
119
+ fixed_episode_seeds: true
120
+ latency_seed_stride: 0
121
+ task_seed_stride: 0
122
+ task_description: You are playing Deadly Corridor in VizDoom. Choose actions from
123
+ MOVE_FORWARD, MOVE_BACKWARD, MOVE_LEFT, MOVE_RIGHT, TURN_LEFT, TURN_RIGHT, ATTACK.
124
+ deadly:
125
+ action_layout: multibinary_7
126
+ enabled: true
127
+ distributed_mode: none
128
+ vectorized:
129
+ enabled: false
130
+ batch_size: 1
131
+ latency:
132
+ prompt_map_path: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/deadly_corridor_fix_latency_2/deadly_corridor_train__bridge/latency_prompt_map.json
133
+ mode: single
134
+ values:
135
+ - 0
136
+ mid_train:
137
+ enabled: true
138
+ interval_steps: 50
139
+ latencies:
140
+ - 2
141
+ num_episodes: 5
142
+ max_steps_per_episode: 3600
143
+ post_train:
144
+ enabled: true
145
+ latencies:
146
+ - 0
147
+ - 2
148
+ - 4
149
+ - 6
150
+ - 8
151
+ num_episodes: 5
152
+ max_steps_per_episode: 3600
153
+ task: deadly_corridor
154
+ deadly_corridor_loss_type: null
155
+ initialization_mode: bridge
156
+ action_carrier: bridge
157
+ model: openvla
158
+ env: deadly_corridor
159
+ init: bridge
160
+ bridge_base_model:
161
+ repo_id:
162
+ openvla: Qwen/Qwen3-VL-4B-Instruct
163
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
164
+ pi05: Qwen/Qwen3-VL-4B-Instruct
165
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
166
+ local_dir:
167
+ openvla: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
168
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
169
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
170
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
171
+ mode: single
172
+ checkpoint:
173
+ load: auto
174
+ hf_repo_id: null
175
+ save_best_model: true
176
+ save_pt_file: false
177
+ local:
178
+ keep_last_n: 1
179
+ sync:
180
+ enabled: false
181
+ repo_id: null
182
+ keep_last_n: 0
183
+ sync_every_n_checkpoints: 1
184
+ resume_policy: local_latest
185
+ run_id: openvla_deadly_corridor_fix_latency_2
186
+ output_dir: null
187
+ config_yaml: null
188
+ is_debug: false
189
+ version_id: '0.21'
latency-aware/deadly-corridor/vla/starvla-qwenoft-h1/openvla_deadly_corridor_fix_latency_2/hydra/.hydra/hydra.yaml ADDED
@@ -0,0 +1,252 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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=deadly_corridor
117
+ - init=bridge
118
+ - mode=single
119
+ - run_id=openvla_deadly_corridor_fix_latency_2
120
+ - run_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints
121
+ - seed=42
122
+ - wandb_entity=dongqianyu99-zhejiang-university
123
+ - wandb_project=starVLA_rl_games
124
+ - rl_games.env_eval.enabled=true
125
+ - checkpoint.sync.enabled=false
126
+ - checkpoint.sync.keep_last_n=0
127
+ - checkpoint.local.keep_last_n=1
128
+ - checkpoint.save_best_model=true
129
+ - checkpoint.save_pt_file=false
130
+ - trainer.is_resume=false
131
+ - trainer.pretrained_checkpoint=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
132
+ - trainer.resume_step=0
133
+ - trainer.max_train_steps=500
134
+ - trainer.num_warmup_steps=100
135
+ - trainer.save_interval=50
136
+ - trainer.eval_interval=100
137
+ - trainer.eval_num_batches=20
138
+ - trainer.logging_frequency=1
139
+ - trainer.gradient_accumulation_steps=16
140
+ - trainer.distributed_backend=none
141
+ - trainer.learning_rate.base=2e-05
142
+ - trainer.learning_rate.qwen_vl_interface=1e-05
143
+ - trainer.learning_rate.action_model=0.0001
144
+ - trainer.lr_scheduler_type=cosine_with_min_lr
145
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
146
+ - trainer.loss_scale.vla=1.0
147
+ - trainer.loss_scale.vlm=0.1
148
+ - trainer.max_grad_norm=1.0
149
+ - trainer.weight_decay=0.0
150
+ - trainer.gradient_clipping=1.0
151
+ - trainer.optimizer.name=AdamW
152
+ - trainer.optimizer.betas=[0.9, 0.95]
153
+ - trainer.optimizer.eps=1e-08
154
+ - trainer.optimizer.weight_decay=1e-08
155
+ - trainer.optimizer.fused=false
156
+ - trainer.save_format=pt
157
+ - framework.name=QwenOFT
158
+ - framework.qwenvl.attn_implementation=flash_attention_2
159
+ - framework.qwenvl.enable_gradient_checkpointing=true
160
+ - framework.action_model.action_dim=7
161
+ - framework.action_model.action_env_dim=7
162
+ - framework.action_model.state_dim=7
163
+ - framework.action_model.loss_type=discrete_ce
164
+ - framework.action_model.action_horizon=1
165
+ - framework.action_model.future_action_window_size=0
166
+ - framework.action_model.past_action_window_size=0
167
+ - datasets.vla_data.include_state=true
168
+ - datasets.vla_data.action_type=discrete
169
+ - datasets.vla_data.sequential_step_sampling=false
170
+ - datasets.vla_data.per_device_batch_size=32
171
+ - datasets.vla_data.load_all_data_for_training=true
172
+ - datasets.vla_data.video_backend=torchvision_av
173
+ - datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/deadly_corridor_fix_latency_2
174
+ - datasets.vla_data.data_mix=deadly_corridor_train__bridge
175
+ - datasets.vla_data.eval_data_mix=deadly_corridor_train__bridge__val
176
+ - framework.qwenvl.base_vlm=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct
177
+ - rl_games.task=deadly_corridor
178
+ - rl_games.model_alias=openvla
179
+ - rl_games.initialization_mode=bridge
180
+ - rl_games.action_carrier=bridge
181
+ - rl_games.env_eval.distributed_mode=none
182
+ - rl_games.env_eval.latency.mode=single
183
+ - rl_games.env_eval.frameskip=4
184
+ - rl_games.env_eval.image_size=224
185
+ - rl_games.env_eval.seed=42
186
+ - rl_games.env_eval.fixed_episode_seeds=true
187
+ - rl_games.env_eval.latency_seed_stride=0
188
+ - rl_games.env_eval.task_seed_stride=0
189
+ - rl_games.env_eval.task_description='You are playing Deadly Corridor in VizDoom.
190
+ Choose actions from MOVE_FORWARD, MOVE_BACKWARD, MOVE_LEFT, MOVE_RIGHT, TURN_LEFT,
191
+ TURN_RIGHT, ATTACK.'
192
+ - rl_games.env_eval.latency.values=[0]
193
+ - rl_games.env_eval.latency.prompt_map_path=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/deadly_corridor_fix_latency_2/deadly_corridor_train__bridge/latency_prompt_map.json
194
+ - rl_games.env_eval.mid_train.enabled=true
195
+ - rl_games.env_eval.mid_train.interval_steps=50
196
+ - rl_games.env_eval.mid_train.num_episodes=5
197
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
198
+ - rl_games.env_eval.mid_train.latencies=[2]
199
+ - rl_games.env_eval.post_train.enabled=true
200
+ - rl_games.env_eval.post_train.num_episodes=5
201
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
202
+ - rl_games.env_eval.post_train.latencies=[0,2,4,6,8]
203
+ - rl_games.env_eval.deadly.action_layout=multibinary_7
204
+ job:
205
+ name: train_starvla_hydra
206
+ chdir: false
207
+ override_dirname: checkpoint.local.keep_last_n=1,checkpoint.save_best_model=true,checkpoint.save_pt_file=false,checkpoint.sync.enabled=false,checkpoint.sync.keep_last_n=0,datasets.vla_data.action_type=discrete,datasets.vla_data.data_mix=deadly_corridor_train__bridge,datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/deadly_corridor_fix_latency_2,datasets.vla_data.eval_data_mix=deadly_corridor_train__bridge__val,datasets.vla_data.include_state=true,datasets.vla_data.load_all_data_for_training=true,datasets.vla_data.per_device_batch_size=32,datasets.vla_data.sequential_step_sampling=false,datasets.vla_data.video_backend=torchvision_av,env=deadly_corridor,framework.action_model.action_dim=7,framework.action_model.action_env_dim=7,framework.action_model.action_horizon=1,framework.action_model.future_action_window_size=0,framework.action_model.loss_type=discrete_ce,framework.action_model.past_action_window_size=0,framework.action_model.state_dim=7,framework.name=QwenOFT,framework.qwenvl.attn_implementation=flash_attention_2,framework.qwenvl.base_vlm=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct,framework.qwenvl.enable_gradient_checkpointing=true,init=bridge,mode=single,model=openvla,rl_games.action_carrier=bridge,rl_games.env_eval.deadly.action_layout=multibinary_7,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=4,rl_games.env_eval.image_size=224,rl_games.env_eval.latency.mode=single,rl_games.env_eval.latency.prompt_map_path=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/deadly_corridor_fix_latency_2/deadly_corridor_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=50,rl_games.env_eval.mid_train.latencies=[2],rl_games.env_eval.mid_train.max_steps_per_episode=3600,rl_games.env_eval.mid_train.num_episodes=5,rl_games.env_eval.post_train.enabled=true,rl_games.env_eval.post_train.latencies=[0,2,4,6,8],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='You
208
+ are playing Deadly Corridor in VizDoom. Choose actions from MOVE_FORWARD, MOVE_BACKWARD,
209
+ MOVE_LEFT, MOVE_RIGHT, TURN_LEFT, TURN_RIGHT, ATTACK.',rl_games.env_eval.task_seed_stride=0,rl_games.initialization_mode=bridge,rl_games.model_alias=openvla,rl_games.task=deadly_corridor,run_id=openvla_deadly_corridor_fix_latency_2,run_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints,seed=42,trainer.distributed_backend=none,trainer.eval_interval=100,trainer.eval_num_batches=20,trainer.gradient_accumulation_steps=16,trainer.gradient_clipping=1.0,trainer.is_resume=false,trainer.learning_rate.action_model=0.0001,trainer.learning_rate.base=2e-05,trainer.learning_rate.qwen_vl_interface=1e-05,trainer.logging_frequency=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=500,trainer.num_warmup_steps=100,trainer.optimizer.betas=[0.9,
210
+ 0.95],trainer.optimizer.eps=1e-08,trainer.optimizer.fused=false,trainer.optimizer.name=AdamW,trainer.optimizer.weight_decay=1e-08,trainer.pretrained_checkpoint=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt,trainer.resume_step=0,trainer.save_format=pt,trainer.save_interval=50,trainer.scheduler_specific_kwargs.min_lr=1e-06,trainer.weight_decay=0.0,wandb_entity=dongqianyu99-zhejiang-university,wandb_project=starVLA_rl_games
211
+ id: ???
212
+ num: ???
213
+ config_name: train
214
+ env_set: {}
215
+ env_copy: []
216
+ config:
217
+ override_dirname:
218
+ kv_sep: '='
219
+ item_sep: ','
220
+ exclude_keys: []
221
+ runtime:
222
+ version: 1.3.2
223
+ version_base: '1.1'
224
+ cwd: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA
225
+ config_sources:
226
+ - path: hydra.conf
227
+ schema: pkg
228
+ provider: hydra
229
+ - path: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA/examples/rl_games/config
230
+ schema: file
231
+ provider: main
232
+ - path: ''
233
+ schema: structured
234
+ provider: schema
235
+ output_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints/openvla_deadly_corridor_fix_latency_2/hydra
236
+ choices:
237
+ cross_task_setup: null
238
+ checkpoint: default
239
+ mode: single
240
+ init: bridge
241
+ env: deadly_corridor
242
+ model: openvla
243
+ hydra/env: default
244
+ hydra/callbacks: null
245
+ hydra/job_logging: default
246
+ hydra/hydra_logging: default
247
+ hydra/hydra_help: default
248
+ hydra/help: default
249
+ hydra/sweeper: basic
250
+ hydra/launcher: basic
251
+ hydra/output: default
252
+ verbose: false
latency-aware/deadly-corridor/vla/starvla-qwenoft-h1/openvla_deadly_corridor_fix_latency_2/hydra/.hydra/overrides.yaml ADDED
@@ -0,0 +1,89 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ - model=openvla
2
+ - env=deadly_corridor
3
+ - init=bridge
4
+ - mode=single
5
+ - run_id=openvla_deadly_corridor_fix_latency_2
6
+ - run_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints
7
+ - seed=42
8
+ - wandb_entity=dongqianyu99-zhejiang-university
9
+ - wandb_project=starVLA_rl_games
10
+ - rl_games.env_eval.enabled=true
11
+ - checkpoint.sync.enabled=false
12
+ - checkpoint.sync.keep_last_n=0
13
+ - checkpoint.local.keep_last_n=1
14
+ - checkpoint.save_best_model=true
15
+ - checkpoint.save_pt_file=false
16
+ - trainer.is_resume=false
17
+ - trainer.pretrained_checkpoint=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
18
+ - trainer.resume_step=0
19
+ - trainer.max_train_steps=500
20
+ - trainer.num_warmup_steps=100
21
+ - trainer.save_interval=50
22
+ - trainer.eval_interval=100
23
+ - trainer.eval_num_batches=20
24
+ - trainer.logging_frequency=1
25
+ - trainer.gradient_accumulation_steps=16
26
+ - trainer.distributed_backend=none
27
+ - trainer.learning_rate.base=2e-05
28
+ - trainer.learning_rate.qwen_vl_interface=1e-05
29
+ - trainer.learning_rate.action_model=0.0001
30
+ - trainer.lr_scheduler_type=cosine_with_min_lr
31
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
32
+ - trainer.loss_scale.vla=1.0
33
+ - trainer.loss_scale.vlm=0.1
34
+ - trainer.max_grad_norm=1.0
35
+ - trainer.weight_decay=0.0
36
+ - trainer.gradient_clipping=1.0
37
+ - trainer.optimizer.name=AdamW
38
+ - trainer.optimizer.betas=[0.9, 0.95]
39
+ - trainer.optimizer.eps=1e-08
40
+ - trainer.optimizer.weight_decay=1e-08
41
+ - trainer.optimizer.fused=false
42
+ - trainer.save_format=pt
43
+ - framework.name=QwenOFT
44
+ - framework.qwenvl.attn_implementation=flash_attention_2
45
+ - framework.qwenvl.enable_gradient_checkpointing=true
46
+ - framework.action_model.action_dim=7
47
+ - framework.action_model.action_env_dim=7
48
+ - framework.action_model.state_dim=7
49
+ - framework.action_model.loss_type=discrete_ce
50
+ - framework.action_model.action_horizon=1
51
+ - framework.action_model.future_action_window_size=0
52
+ - framework.action_model.past_action_window_size=0
53
+ - datasets.vla_data.include_state=true
54
+ - datasets.vla_data.action_type=discrete
55
+ - datasets.vla_data.sequential_step_sampling=false
56
+ - datasets.vla_data.per_device_batch_size=32
57
+ - datasets.vla_data.load_all_data_for_training=true
58
+ - datasets.vla_data.video_backend=torchvision_av
59
+ - datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/deadly_corridor_fix_latency_2
60
+ - datasets.vla_data.data_mix=deadly_corridor_train__bridge
61
+ - datasets.vla_data.eval_data_mix=deadly_corridor_train__bridge__val
62
+ - framework.qwenvl.base_vlm=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct
63
+ - rl_games.task=deadly_corridor
64
+ - rl_games.model_alias=openvla
65
+ - rl_games.initialization_mode=bridge
66
+ - rl_games.action_carrier=bridge
67
+ - rl_games.env_eval.distributed_mode=none
68
+ - rl_games.env_eval.latency.mode=single
69
+ - rl_games.env_eval.frameskip=4
70
+ - rl_games.env_eval.image_size=224
71
+ - rl_games.env_eval.seed=42
72
+ - rl_games.env_eval.fixed_episode_seeds=true
73
+ - rl_games.env_eval.latency_seed_stride=0
74
+ - rl_games.env_eval.task_seed_stride=0
75
+ - rl_games.env_eval.task_description='You are playing Deadly Corridor in VizDoom.
76
+ Choose actions from MOVE_FORWARD, MOVE_BACKWARD, MOVE_LEFT, MOVE_RIGHT, TURN_LEFT,
77
+ TURN_RIGHT, ATTACK.'
78
+ - rl_games.env_eval.latency.values=[0]
79
+ - rl_games.env_eval.latency.prompt_map_path=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/deadly_corridor_fix_latency_2/deadly_corridor_train__bridge/latency_prompt_map.json
80
+ - rl_games.env_eval.mid_train.enabled=true
81
+ - rl_games.env_eval.mid_train.interval_steps=50
82
+ - rl_games.env_eval.mid_train.num_episodes=5
83
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
84
+ - rl_games.env_eval.mid_train.latencies=[2]
85
+ - rl_games.env_eval.post_train.enabled=true
86
+ - rl_games.env_eval.post_train.num_episodes=5
87
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
88
+ - rl_games.env_eval.post_train.latencies=[0,2,4,6,8]
89
+ - rl_games.env_eval.deadly.action_layout=multibinary_7
latency-aware/deadly-corridor/vla/starvla-qwenoft-h1/openvla_deadly_corridor_fix_latency_2/hydra/train_starvla_hydra.log ADDED
The diff for this file is too large to render. See raw diff
 
latency-aware/deadly-corridor/vla/starvla-qwenoft-h1/openvla_deadly_corridor_fix_latency_2/summary.jsonl ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {"steps": 50}
2
+ {"steps": 100}
3
+ {"steps": 150}
4
+ {"steps": 200}
5
+ {"steps": 250}
6
+ {"steps": 300}
7
+ {"steps": 350}
8
+ {"steps": 400}
9
+ {"steps": 450}
10
+ {"steps": 500}
latency-aware/demon-attack/vla/starvla-qwenoft-h1/openvla_demon_attack_fix_latency_2_small/checkpoints/best_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:adb853d16890002501675633ddefc2126fd3f102a63cf6ece7a5e56d0c8c662c
3
+ size 9138230516
latency-aware/demon-attack/vla/starvla-qwenoft-h1/openvla_demon_attack_fix_latency_2_small/checkpoints/steps_2000_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:adb853d16890002501675633ddefc2126fd3f102a63cf6ece7a5e56d0c8c662c
3
+ size 9138230516
latency-aware/demon-attack/vla/starvla-qwenoft-h1/openvla_demon_attack_fix_latency_2_small/config.full.yaml ADDED
@@ -0,0 +1,190 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ name: QwenOFT
3
+ qwenvl:
4
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
5
+ attn_implementation: flash_attention_2
6
+ enable_gradient_checkpointing: true
7
+ action_model:
8
+ action_model_type: MLP
9
+ action_dim: 7
10
+ action_hidden_dim: 2560
11
+ future_action_window_size: 0
12
+ past_action_window_size: 0
13
+ loss_type: discrete_ce
14
+ state_dim: 7
15
+ action_horizon: 1
16
+ action_env_dim: 6
17
+ datasets:
18
+ vla_data:
19
+ dataset_py: lerobot_datasets
20
+ include_state: true
21
+ data_root_dir: /workspace/starVLA/data/demon_attack_fix_latency_2_small
22
+ data_mix: demon_attack_train__bridge
23
+ eval_data_mix: demon_attack_train__bridge__val
24
+ custom_mixtures_path: null
25
+ action_type: discrete
26
+ sequential_step_sampling: false
27
+ per_device_batch_size: 32
28
+ load_all_data_for_training: true
29
+ obs_image_size: null
30
+ video_backend: torchvision_av
31
+ dataset:
32
+ source_hf: ''
33
+ converted_name: demon_attack_train
34
+ single_source_hf: ''
35
+ mixed_source_hf: ''
36
+ single_converted_name: demon_attack_train
37
+ mixed_converted_name: demon_attack_mixed_latency_train
38
+ single_latency_filter: null
39
+ mixed_latency_filter: null
40
+ force_download: false
41
+ setup_force: false
42
+ verify_rows: 200
43
+ max_episodes: null
44
+ episodes_per_latency: null
45
+ latency_filter: null
46
+ debug_subset:
47
+ enabled: false
48
+ max_episodes: 5
49
+ suffix: debug
50
+ base_model:
51
+ repo_id: Qwen/Qwen3-VL-4B-Instruct
52
+ initialization:
53
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
54
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
55
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
56
+ trainer:
57
+ max_train_steps: 2000
58
+ num_warmup_steps: 100
59
+ save_interval: 100
60
+ eval_interval: 100
61
+ eval_num_batches: 20
62
+ learning_rate:
63
+ base: 2.0e-05
64
+ qwen_vl_interface: 1.0e-05
65
+ action_model: 0.0001
66
+ lr_scheduler_type: cosine_with_min_lr
67
+ scheduler_specific_kwargs:
68
+ min_lr: 1.0e-06
69
+ freeze_modules: ''
70
+ loss_scale:
71
+ vla: 1.0
72
+ vlm: 0.1
73
+ max_grad_norm: 1.0
74
+ weight_decay: 0.0
75
+ logging_frequency: 10
76
+ gradient_clipping: 1.0
77
+ gradient_accumulation_steps: 16
78
+ distributed_backend: none
79
+ is_resume: true
80
+ pretrained_checkpoint: /workspace/starVLA/results/Checkpoints/openvla_demon_attack_fix_latency_2_small/checkpoints/steps_600_state
81
+ resume_step: 600
82
+ reload_modules: null
83
+ optimizer:
84
+ name: AdamW
85
+ betas:
86
+ - 0.9
87
+ - 0.95
88
+ eps: 1.0e-08
89
+ weight_decay: 1.0e-08
90
+ fused: false
91
+ save_format: pt
92
+ workspace_dir: WORKSPACE_DIR
93
+ run_root_dir: /workspace/starVLA/results/Checkpoints
94
+ seed: 42
95
+ wandb_entity: dongqianyu99-zhejiang-university
96
+ wandb_project: starVLA_rl_games
97
+ auth:
98
+ env_file: null
99
+ hf_token_env: HF_TOKEN
100
+ wandb_api_key_env: WANDB_API_KEY
101
+ paths:
102
+ run_root_dir: results/Checkpoints
103
+ dataset_local_dir: playground/Datasets/rl_games
104
+ dataset_cache_dir: null
105
+ base_model_dir: playground/Pretrained_models/Qwen3-VL-4B-Instruct
106
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
107
+ launch:
108
+ use_accelerate: true
109
+ gpus: null
110
+ num_processes: 1
111
+ dry_run: false
112
+ conda:
113
+ enabled: true
114
+ env_name: null
115
+ rl_games:
116
+ model_alias: openvla
117
+ env_eval:
118
+ image_size: 224
119
+ frameskip: 4
120
+ seed: 42
121
+ fixed_episode_seeds: true
122
+ latency_seed_stride: 0
123
+ task_seed_stride: 0
124
+ task_description: 'You are playing Demon Attack from a single game image. Choose
125
+ exactly one action from: NOOP, FIRE, RIGHT, LEFT, RIGHTFIRE, LEFTFIRE.'
126
+ demon_attack:
127
+ noop_max: 30
128
+ enabled: true
129
+ distributed_mode: none
130
+ vectorized:
131
+ enabled: false
132
+ batch_size: 1
133
+ latency:
134
+ prompt_map_path: /workspace/starVLA/data/demon_attack_fix_latency_2_small/demon_attack_train__bridge/latency_prompt_map.json
135
+ mode: single
136
+ values:
137
+ - 0
138
+ mid_train:
139
+ enabled: true
140
+ interval_steps: 100
141
+ latencies:
142
+ - 2
143
+ num_episodes: 5
144
+ max_steps_per_episode: 3600
145
+ post_train:
146
+ enabled: true
147
+ latencies:
148
+ - 0
149
+ - 1
150
+ - 2
151
+ - 3
152
+ - 4
153
+ num_episodes: 5
154
+ max_steps_per_episode: 3600
155
+ task: demon_attack
156
+ initialization_mode: bridge
157
+ action_carrier: bridge
158
+ model: openvla
159
+ env: demon_attack
160
+ init: bridge
161
+ bridge_base_model:
162
+ repo_id:
163
+ openvla: Qwen/Qwen3-VL-4B-Instruct
164
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
165
+ pi05: Qwen/Qwen3-VL-4B-Instruct
166
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
167
+ local_dir:
168
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
169
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
170
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
171
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
172
+ mode: single
173
+ checkpoint:
174
+ load: auto
175
+ hf_repo_id: null
176
+ save_best_model: true
177
+ save_pt_file: false
178
+ local:
179
+ keep_last_n: 1
180
+ sync:
181
+ enabled: false
182
+ repo_id: null
183
+ keep_last_n: 0
184
+ sync_every_n_checkpoints: 1
185
+ resume_policy: local_latest
186
+ run_id: openvla_demon_attack_fix_latency_2_small
187
+ output_dir: /workspace/starVLA/results/Checkpoints/openvla_demon_attack_fix_latency_2_small
188
+ config_yaml: null
189
+ is_debug: false
190
+ version_id: '0.21'
latency-aware/demon-attack/vla/starvla-qwenoft-h1/openvla_demon_attack_fix_latency_2_small/config.yaml ADDED
@@ -0,0 +1,92 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ checkpoint:
2
+ local:
3
+ keep_last_n: 1
4
+ save_best_model: true
5
+ save_pt_file: false
6
+ sync:
7
+ enabled: false
8
+ keep_last_n: 0
9
+ repo_id: null
10
+ datasets:
11
+ vla_data:
12
+ data_mix: demon_attack_train__bridge
13
+ dataset_py: lerobot_datasets
14
+ eval_data_mix: demon_attack_train__bridge__val
15
+ include_state: true
16
+ obs_image_size: null
17
+ per_device_batch_size: 32
18
+ framework:
19
+ action_model:
20
+ action_dim: 7
21
+ action_env_dim: 6
22
+ action_hidden_dim: 2560
23
+ action_horizon: 1
24
+ action_model_type: MLP
25
+ loss_type: discrete_ce
26
+ state_dim: 7
27
+ name: QwenOFT
28
+ qwenvl:
29
+ attn_implementation: flash_attention_2
30
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
31
+ enable_gradient_checkpointing: true
32
+ output_dir: /workspace/starVLA/results/Checkpoints/openvla_demon_attack_fix_latency_2_small
33
+ rl_games:
34
+ env_eval:
35
+ demon_attack:
36
+ noop_max: 30
37
+ distributed_mode: none
38
+ enabled: true
39
+ fixed_episode_seeds: true
40
+ frameskip: 4
41
+ image_size: 224
42
+ latency:
43
+ prompt_map_path: /workspace/starVLA/data/demon_attack_fix_latency_2_small/demon_attack_train__bridge/latency_prompt_map.json
44
+ latency_seed_stride: 0
45
+ mid_train:
46
+ enabled: true
47
+ interval_steps: 100
48
+ latencies:
49
+ - 2
50
+ max_steps_per_episode: 3600
51
+ num_episodes: 5
52
+ seed: 42
53
+ task_description: 'You are playing Demon Attack from a single game image. Choose
54
+ exactly one action from: NOOP, FIRE, RIGHT, LEFT, RIGHTFIRE, LEFTFIRE.'
55
+ task_seed_stride: 0
56
+ vectorized:
57
+ enabled: false
58
+ model_alias: openvla
59
+ task: demon_attack
60
+ run_id: openvla_demon_attack_fix_latency_2_small
61
+ run_root_dir: /workspace/starVLA/results/Checkpoints
62
+ seed: 42
63
+ trainer:
64
+ distributed_backend: none
65
+ eval_interval: 100
66
+ eval_num_batches: 20
67
+ freeze_modules: ''
68
+ gradient_accumulation_steps: 16
69
+ gradient_clipping: 1.0
70
+ is_resume: true
71
+ learning_rate:
72
+ action_model: 0.0001
73
+ base: 2.0e-05
74
+ qwen_vl_interface: 1.0e-05
75
+ logging_frequency: 10
76
+ lr_scheduler_type: cosine_with_min_lr
77
+ max_train_steps: 2000
78
+ num_warmup_steps: 100
79
+ optimizer:
80
+ betas:
81
+ - 0.9
82
+ - 0.95
83
+ eps: 1.0e-08
84
+ fused: false
85
+ weight_decay: 1.0e-08
86
+ pretrained_checkpoint: /workspace/starVLA/results/Checkpoints/openvla_demon_attack_fix_latency_2_small/checkpoints/steps_600_state
87
+ resume_step: 600
88
+ save_interval: 100
89
+ scheduler_specific_kwargs:
90
+ min_lr: 1.0e-06
91
+ wandb_entity: dongqianyu99-zhejiang-university
92
+ wandb_project: starVLA_rl_games
latency-aware/demon-attack/vla/starvla-qwenoft-h1/openvla_demon_attack_fix_latency_2_small/dataset_statistics.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.12531481683254242,
6
+ 0.3168888986110687,
7
+ 0.1617870330810547,
8
+ 0.12757407128810883,
9
+ 0.1149444431066513,
10
+ 0.15349073708057404,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.33097583055496216,
15
+ 0.46518969535827637,
16
+ 0.36836159229278564,
17
+ 0.33368170261383057,
18
+ 0.31906843185424805,
19
+ 0.36036959290504456,
20
+ 0.0
21
+ ],
22
+ "max": [
23
+ 1.0,
24
+ 1.0,
25
+ 1.0,
26
+ 1.0,
27
+ 1.0,
28
+ 1.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
+ 1.0,
53
+ 1.0,
54
+ 1.0,
55
+ 1.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": 108000,
125
+ "num_trajectories": 30
126
+ }
127
+ }
latency-aware/demon-attack/vla/starvla-qwenoft-h1/openvla_demon_attack_fix_latency_2_small/dataset_statistics_eval.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.125083327293396,
6
+ 0.3141111135482788,
7
+ 0.16361111402511597,
8
+ 0.12720833718776703,
9
+ 0.11493055522441864,
10
+ 0.15505555272102356,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.3307972252368927,
15
+ 0.4641607403755188,
16
+ 0.36990460753440857,
17
+ 0.33313921093940735,
18
+ 0.31889820098876953,
19
+ 0.3619128167629242,
20
+ 0.0
21
+ ],
22
+ "max": [
23
+ 1.0,
24
+ 1.0,
25
+ 1.0,
26
+ 1.0,
27
+ 1.0,
28
+ 1.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
+ 1.0,
53
+ 1.0,
54
+ 1.0,
55
+ 1.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/demon-attack/vla/starvla-qwenoft-h1/openvla_demon_attack_fix_latency_2_small/hydra/.hydra/config.yaml ADDED
@@ -0,0 +1,188 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ qwenvl:
3
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
4
+ attn_implementation: flash_attention_2
5
+ enable_gradient_checkpointing: true
6
+ action_model:
7
+ state_dim: 7
8
+ loss_type: discrete_ce
9
+ action_horizon: 1
10
+ future_action_window_size: 0
11
+ past_action_window_size: 0
12
+ action_dim: 7
13
+ action_env_dim: 6
14
+ name: QwenOFT
15
+ datasets:
16
+ vla_data:
17
+ dataset_py: lerobot_datasets
18
+ include_state: true
19
+ data_root_dir: /workspace/starVLA/data/demon_attack_fix_latency_2
20
+ data_mix: demon_attack_train__bridge
21
+ eval_data_mix: demon_attack_train__bridge__val
22
+ custom_mixtures_path: null
23
+ action_type: discrete
24
+ sequential_step_sampling: false
25
+ per_device_batch_size: 32
26
+ load_all_data_for_training: true
27
+ obs_image_size: null
28
+ video_backend: torchvision_av
29
+ dataset:
30
+ source_hf: ${dataset.single_source_hf}
31
+ converted_name: ${dataset.single_converted_name}
32
+ single_source_hf: ''
33
+ mixed_source_hf: ''
34
+ single_converted_name: demon_attack_train
35
+ mixed_converted_name: demon_attack_mixed_latency_train
36
+ single_latency_filter: null
37
+ mixed_latency_filter: null
38
+ force_download: false
39
+ setup_force: false
40
+ verify_rows: 200
41
+ max_episodes: null
42
+ episodes_per_latency: null
43
+ latency_filter: ${dataset.single_latency_filter}
44
+ debug_subset:
45
+ enabled: false
46
+ max_episodes: 5
47
+ suffix: debug
48
+ base_model:
49
+ repo_id: ${bridge_base_model.repo_id.${model}}
50
+ initialization:
51
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
52
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
53
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
54
+ trainer:
55
+ max_train_steps: 2000
56
+ num_warmup_steps: 100
57
+ save_interval: 100
58
+ eval_interval: 100
59
+ eval_num_batches: 20
60
+ learning_rate:
61
+ base: 2.0e-05
62
+ qwen_vl_interface: 1.0e-05
63
+ action_model: 0.0001
64
+ lr_scheduler_type: cosine_with_min_lr
65
+ scheduler_specific_kwargs:
66
+ min_lr: 1.0e-06
67
+ freeze_modules: ''
68
+ loss_scale:
69
+ vla: 1.0
70
+ vlm: 0.1
71
+ max_grad_norm: 1.0
72
+ weight_decay: 0.0
73
+ logging_frequency: 10
74
+ gradient_clipping: 1.0
75
+ gradient_accumulation_steps: 16
76
+ distributed_backend: none
77
+ is_resume: true
78
+ pretrained_checkpoint: /workspace/starVLA/results/Checkpoints/openvla_demon_attack_fix_latency_2/checkpoints/steps_600_state
79
+ resume_step: 600
80
+ reload_modules: null
81
+ optimizer:
82
+ name: AdamW
83
+ betas:
84
+ - 0.9
85
+ - 0.95
86
+ eps: 1.0e-08
87
+ weight_decay: 1.0e-08
88
+ fused: false
89
+ save_format: pt
90
+ workspace_dir: WORKSPACE_DIR
91
+ run_root_dir: /workspace/starVLA/results/Checkpoints
92
+ seed: 42
93
+ wandb_entity: dongqianyu99-zhejiang-university
94
+ wandb_project: starVLA_rl_games
95
+ auth:
96
+ env_file: null
97
+ hf_token_env: HF_TOKEN
98
+ wandb_api_key_env: WANDB_API_KEY
99
+ paths:
100
+ run_root_dir: results/Checkpoints
101
+ dataset_local_dir: playground/Datasets/rl_games
102
+ dataset_cache_dir: null
103
+ base_model_dir: ${bridge_base_model.local_dir.${model}}
104
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
105
+ launch:
106
+ use_accelerate: true
107
+ gpus: null
108
+ num_processes: 1
109
+ dry_run: false
110
+ conda:
111
+ enabled: true
112
+ env_name: null
113
+ rl_games:
114
+ model_alias: openvla
115
+ env_eval:
116
+ image_size: 224
117
+ frameskip: 4
118
+ seed: 42
119
+ fixed_episode_seeds: true
120
+ latency_seed_stride: 0
121
+ task_seed_stride: 0
122
+ task_description: 'You are playing Demon Attack from a single game image. Choose
123
+ exactly one action from: NOOP, FIRE, RIGHT, LEFT, RIGHTFIRE, LEFTFIRE.'
124
+ demon_attack:
125
+ noop_max: 30
126
+ enabled: true
127
+ distributed_mode: none
128
+ vectorized:
129
+ enabled: false
130
+ batch_size: 1
131
+ latency:
132
+ prompt_map_path: /workspace/starVLA/data/demon_attack_fix_latency_2/demon_attack_train__bridge/latency_prompt_map.json
133
+ mode: single
134
+ values:
135
+ - 0
136
+ mid_train:
137
+ enabled: true
138
+ interval_steps: 100
139
+ latencies:
140
+ - 2
141
+ num_episodes: 5
142
+ max_steps_per_episode: 3600
143
+ post_train:
144
+ enabled: true
145
+ latencies:
146
+ - 0
147
+ - 1
148
+ - 2
149
+ - 3
150
+ - 4
151
+ num_episodes: 5
152
+ max_steps_per_episode: 3600
153
+ task: demon_attack
154
+ initialization_mode: bridge
155
+ action_carrier: bridge
156
+ model: openvla
157
+ env: demon_attack
158
+ init: bridge
159
+ bridge_base_model:
160
+ repo_id:
161
+ openvla: Qwen/Qwen3-VL-4B-Instruct
162
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
163
+ pi05: Qwen/Qwen3-VL-4B-Instruct
164
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
165
+ local_dir:
166
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
167
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
168
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
169
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
170
+ mode: single
171
+ checkpoint:
172
+ load: auto
173
+ hf_repo_id: null
174
+ save_best_model: true
175
+ save_pt_file: false
176
+ local:
177
+ keep_last_n: 1
178
+ sync:
179
+ enabled: false
180
+ repo_id: null
181
+ keep_last_n: 0
182
+ sync_every_n_checkpoints: 1
183
+ resume_policy: local_latest
184
+ run_id: openvla_demon_attack_fix_latency_2
185
+ output_dir: null
186
+ config_yaml: null
187
+ is_debug: false
188
+ version_id: '0.21'
latency-aware/demon-attack/vla/starvla-qwenoft-h1/openvla_demon_attack_fix_latency_2_small/hydra/.hydra/hydra.yaml ADDED
@@ -0,0 +1,251 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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=demon_attack
117
+ - init=bridge
118
+ - mode=single
119
+ - run_id=openvla_demon_attack_fix_latency_2
120
+ - run_root_dir=/workspace/starVLA/results/Checkpoints
121
+ - seed=42
122
+ - wandb_entity=dongqianyu99-zhejiang-university
123
+ - wandb_project=starVLA_rl_games
124
+ - rl_games.env_eval.enabled=true
125
+ - checkpoint.sync.enabled=false
126
+ - checkpoint.sync.keep_last_n=0
127
+ - checkpoint.local.keep_last_n=1
128
+ - checkpoint.save_best_model=true
129
+ - checkpoint.save_pt_file=false
130
+ - trainer.is_resume=true
131
+ - trainer.pretrained_checkpoint=/workspace/starVLA/results/Checkpoints/openvla_demon_attack_fix_latency_2/checkpoints/steps_600_state
132
+ - trainer.resume_step=600
133
+ - trainer.max_train_steps=2000
134
+ - trainer.num_warmup_steps=100
135
+ - trainer.save_interval=100
136
+ - trainer.eval_interval=100
137
+ - trainer.eval_num_batches=20
138
+ - trainer.logging_frequency=10
139
+ - trainer.gradient_accumulation_steps=16
140
+ - trainer.distributed_backend=none
141
+ - trainer.learning_rate.base=2e-05
142
+ - trainer.learning_rate.qwen_vl_interface=1e-05
143
+ - trainer.learning_rate.action_model=0.0001
144
+ - trainer.lr_scheduler_type=cosine_with_min_lr
145
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
146
+ - trainer.loss_scale.vla=1.0
147
+ - trainer.loss_scale.vlm=0.1
148
+ - trainer.max_grad_norm=1.0
149
+ - trainer.weight_decay=0.0
150
+ - trainer.gradient_clipping=1.0
151
+ - trainer.optimizer.name=AdamW
152
+ - trainer.optimizer.betas=[0.9, 0.95]
153
+ - trainer.optimizer.eps=1e-08
154
+ - trainer.optimizer.weight_decay=1e-08
155
+ - trainer.optimizer.fused=false
156
+ - trainer.save_format=pt
157
+ - framework.name=QwenOFT
158
+ - framework.qwenvl.attn_implementation=flash_attention_2
159
+ - framework.qwenvl.enable_gradient_checkpointing=true
160
+ - framework.action_model.action_dim=7
161
+ - framework.action_model.action_env_dim=6
162
+ - framework.action_model.state_dim=7
163
+ - framework.action_model.loss_type=discrete_ce
164
+ - framework.action_model.action_horizon=1
165
+ - framework.action_model.future_action_window_size=0
166
+ - framework.action_model.past_action_window_size=0
167
+ - datasets.vla_data.include_state=true
168
+ - datasets.vla_data.action_type=discrete
169
+ - datasets.vla_data.sequential_step_sampling=false
170
+ - datasets.vla_data.per_device_batch_size=32
171
+ - datasets.vla_data.load_all_data_for_training=true
172
+ - datasets.vla_data.video_backend=torchvision_av
173
+ - datasets.vla_data.data_root_dir=/workspace/starVLA/data/demon_attack_fix_latency_2
174
+ - datasets.vla_data.data_mix=demon_attack_train__bridge
175
+ - datasets.vla_data.eval_data_mix=demon_attack_train__bridge__val
176
+ - framework.qwenvl.base_vlm=/workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
177
+ - rl_games.task=demon_attack
178
+ - rl_games.model_alias=openvla
179
+ - rl_games.initialization_mode=bridge
180
+ - rl_games.action_carrier=bridge
181
+ - rl_games.env_eval.distributed_mode=none
182
+ - rl_games.env_eval.latency.mode=single
183
+ - rl_games.env_eval.frameskip=4
184
+ - rl_games.env_eval.image_size=224
185
+ - rl_games.env_eval.seed=42
186
+ - rl_games.env_eval.fixed_episode_seeds=true
187
+ - rl_games.env_eval.latency_seed_stride=0
188
+ - rl_games.env_eval.task_seed_stride=0
189
+ - 'rl_games.env_eval.task_description=''You are playing Demon Attack from a single
190
+ game image. Choose exactly one action from: NOOP, FIRE, RIGHT, LEFT, RIGHTFIRE,
191
+ LEFTFIRE.'''
192
+ - rl_games.env_eval.latency.values=[0]
193
+ - rl_games.env_eval.latency.prompt_map_path=/workspace/starVLA/data/demon_attack_fix_latency_2/demon_attack_train__bridge/latency_prompt_map.json
194
+ - rl_games.env_eval.mid_train.enabled=true
195
+ - rl_games.env_eval.mid_train.interval_steps=100
196
+ - rl_games.env_eval.mid_train.num_episodes=5
197
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
198
+ - rl_games.env_eval.mid_train.latencies=[2]
199
+ - rl_games.env_eval.post_train.enabled=true
200
+ - rl_games.env_eval.post_train.num_episodes=5
201
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
202
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
203
+ job:
204
+ name: train_starvla_hydra
205
+ chdir: false
206
+ override_dirname: 'checkpoint.local.keep_last_n=1,checkpoint.save_best_model=true,checkpoint.save_pt_file=false,checkpoint.sync.enabled=false,checkpoint.sync.keep_last_n=0,datasets.vla_data.action_type=discrete,datasets.vla_data.data_mix=demon_attack_train__bridge,datasets.vla_data.data_root_dir=/workspace/starVLA/data/demon_attack_fix_latency_2,datasets.vla_data.eval_data_mix=demon_attack_train__bridge__val,datasets.vla_data.include_state=true,datasets.vla_data.load_all_data_for_training=true,datasets.vla_data.per_device_batch_size=32,datasets.vla_data.sequential_step_sampling=false,datasets.vla_data.video_backend=torchvision_av,env=demon_attack,framework.action_model.action_dim=7,framework.action_model.action_env_dim=6,framework.action_model.action_horizon=1,framework.action_model.future_action_window_size=0,framework.action_model.loss_type=discrete_ce,framework.action_model.past_action_window_size=0,framework.action_model.state_dim=7,framework.name=QwenOFT,framework.qwenvl.attn_implementation=flash_attention_2,framework.qwenvl.base_vlm=/workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct,framework.qwenvl.enable_gradient_checkpointing=true,init=bridge,mode=single,model=openvla,rl_games.action_carrier=bridge,rl_games.env_eval.distributed_mode=none,rl_games.env_eval.enabled=true,rl_games.env_eval.fixed_episode_seeds=true,rl_games.env_eval.frameskip=4,rl_games.env_eval.image_size=224,rl_games.env_eval.latency.mode=single,rl_games.env_eval.latency.prompt_map_path=/workspace/starVLA/data/demon_attack_fix_latency_2/demon_attack_train__bridge/latency_prompt_map.json,rl_games.env_eval.latency.values=[0],rl_games.env_eval.latency_seed_stride=0,rl_games.env_eval.mid_train.enabled=true,rl_games.env_eval.mid_train.interval_steps=100,rl_games.env_eval.mid_train.latencies=[2],rl_games.env_eval.mid_train.max_steps_per_episode=3600,rl_games.env_eval.mid_train.num_episodes=5,rl_games.env_eval.post_train.enabled=true,rl_games.env_eval.post_train.latencies=[0,1,2,3,4],rl_games.env_eval.post_train.max_steps_per_episode=3600,rl_games.env_eval.post_train.num_episodes=5,rl_games.env_eval.seed=42,rl_games.env_eval.task_description=''You
207
+ are playing Demon Attack from a single game image. Choose exactly one action
208
+ from: NOOP, FIRE, RIGHT, LEFT, RIGHTFIRE, LEFTFIRE.'',rl_games.env_eval.task_seed_stride=0,rl_games.initialization_mode=bridge,rl_games.model_alias=openvla,rl_games.task=demon_attack,run_id=openvla_demon_attack_fix_latency_2,run_root_dir=/workspace/starVLA/results/Checkpoints,seed=42,trainer.distributed_backend=none,trainer.eval_interval=100,trainer.eval_num_batches=20,trainer.gradient_accumulation_steps=16,trainer.gradient_clipping=1.0,trainer.is_resume=true,trainer.learning_rate.action_model=0.0001,trainer.learning_rate.base=2e-05,trainer.learning_rate.qwen_vl_interface=1e-05,trainer.logging_frequency=10,trainer.loss_scale.vla=1.0,trainer.loss_scale.vlm=0.1,trainer.lr_scheduler_type=cosine_with_min_lr,trainer.max_grad_norm=1.0,trainer.max_train_steps=2000,trainer.num_warmup_steps=100,trainer.optimizer.betas=[0.9,
209
+ 0.95],trainer.optimizer.eps=1e-08,trainer.optimizer.fused=false,trainer.optimizer.name=AdamW,trainer.optimizer.weight_decay=1e-08,trainer.pretrained_checkpoint=/workspace/starVLA/results/Checkpoints/openvla_demon_attack_fix_latency_2/checkpoints/steps_600_state,trainer.resume_step=600,trainer.save_format=pt,trainer.save_interval=100,trainer.scheduler_specific_kwargs.min_lr=1e-06,trainer.weight_decay=0.0,wandb_entity=dongqianyu99-zhejiang-university,wandb_project=starVLA_rl_games'
210
+ id: ???
211
+ num: ???
212
+ config_name: train
213
+ env_set: {}
214
+ env_copy: []
215
+ config:
216
+ override_dirname:
217
+ kv_sep: '='
218
+ item_sep: ','
219
+ exclude_keys: []
220
+ runtime:
221
+ version: 1.3.2
222
+ version_base: '1.1'
223
+ cwd: /workspace/starVLA
224
+ config_sources:
225
+ - path: hydra.conf
226
+ schema: pkg
227
+ provider: hydra
228
+ - path: /workspace/starVLA/examples/rl_games/config
229
+ schema: file
230
+ provider: main
231
+ - path: ''
232
+ schema: structured
233
+ provider: schema
234
+ output_dir: /workspace/starVLA/results/Checkpoints/openvla_demon_attack_fix_latency_2/hydra
235
+ choices:
236
+ cross_task_setup: null
237
+ checkpoint: default
238
+ mode: single
239
+ init: bridge
240
+ env: demon_attack
241
+ model: openvla
242
+ hydra/env: default
243
+ hydra/callbacks: null
244
+ hydra/job_logging: default
245
+ hydra/hydra_logging: default
246
+ hydra/hydra_help: default
247
+ hydra/help: default
248
+ hydra/sweeper: basic
249
+ hydra/launcher: basic
250
+ hydra/output: default
251
+ verbose: false
latency-aware/demon-attack/vla/starvla-qwenoft-h1/openvla_demon_attack_fix_latency_2_small/hydra/.hydra/overrides.yaml ADDED
@@ -0,0 +1,88 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ - model=openvla
2
+ - env=demon_attack
3
+ - init=bridge
4
+ - mode=single
5
+ - run_id=openvla_demon_attack_fix_latency_2
6
+ - run_root_dir=/workspace/starVLA/results/Checkpoints
7
+ - seed=42
8
+ - wandb_entity=dongqianyu99-zhejiang-university
9
+ - wandb_project=starVLA_rl_games
10
+ - rl_games.env_eval.enabled=true
11
+ - checkpoint.sync.enabled=false
12
+ - checkpoint.sync.keep_last_n=0
13
+ - checkpoint.local.keep_last_n=1
14
+ - checkpoint.save_best_model=true
15
+ - checkpoint.save_pt_file=false
16
+ - trainer.is_resume=true
17
+ - trainer.pretrained_checkpoint=/workspace/starVLA/results/Checkpoints/openvla_demon_attack_fix_latency_2/checkpoints/steps_600_state
18
+ - trainer.resume_step=600
19
+ - trainer.max_train_steps=2000
20
+ - trainer.num_warmup_steps=100
21
+ - trainer.save_interval=100
22
+ - trainer.eval_interval=100
23
+ - trainer.eval_num_batches=20
24
+ - trainer.logging_frequency=10
25
+ - trainer.gradient_accumulation_steps=16
26
+ - trainer.distributed_backend=none
27
+ - trainer.learning_rate.base=2e-05
28
+ - trainer.learning_rate.qwen_vl_interface=1e-05
29
+ - trainer.learning_rate.action_model=0.0001
30
+ - trainer.lr_scheduler_type=cosine_with_min_lr
31
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
32
+ - trainer.loss_scale.vla=1.0
33
+ - trainer.loss_scale.vlm=0.1
34
+ - trainer.max_grad_norm=1.0
35
+ - trainer.weight_decay=0.0
36
+ - trainer.gradient_clipping=1.0
37
+ - trainer.optimizer.name=AdamW
38
+ - trainer.optimizer.betas=[0.9, 0.95]
39
+ - trainer.optimizer.eps=1e-08
40
+ - trainer.optimizer.weight_decay=1e-08
41
+ - trainer.optimizer.fused=false
42
+ - trainer.save_format=pt
43
+ - framework.name=QwenOFT
44
+ - framework.qwenvl.attn_implementation=flash_attention_2
45
+ - framework.qwenvl.enable_gradient_checkpointing=true
46
+ - framework.action_model.action_dim=7
47
+ - framework.action_model.action_env_dim=6
48
+ - framework.action_model.state_dim=7
49
+ - framework.action_model.loss_type=discrete_ce
50
+ - framework.action_model.action_horizon=1
51
+ - framework.action_model.future_action_window_size=0
52
+ - framework.action_model.past_action_window_size=0
53
+ - datasets.vla_data.include_state=true
54
+ - datasets.vla_data.action_type=discrete
55
+ - datasets.vla_data.sequential_step_sampling=false
56
+ - datasets.vla_data.per_device_batch_size=32
57
+ - datasets.vla_data.load_all_data_for_training=true
58
+ - datasets.vla_data.video_backend=torchvision_av
59
+ - datasets.vla_data.data_root_dir=/workspace/starVLA/data/demon_attack_fix_latency_2
60
+ - datasets.vla_data.data_mix=demon_attack_train__bridge
61
+ - datasets.vla_data.eval_data_mix=demon_attack_train__bridge__val
62
+ - framework.qwenvl.base_vlm=/workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
63
+ - rl_games.task=demon_attack
64
+ - rl_games.model_alias=openvla
65
+ - rl_games.initialization_mode=bridge
66
+ - rl_games.action_carrier=bridge
67
+ - rl_games.env_eval.distributed_mode=none
68
+ - rl_games.env_eval.latency.mode=single
69
+ - rl_games.env_eval.frameskip=4
70
+ - rl_games.env_eval.image_size=224
71
+ - rl_games.env_eval.seed=42
72
+ - rl_games.env_eval.fixed_episode_seeds=true
73
+ - rl_games.env_eval.latency_seed_stride=0
74
+ - rl_games.env_eval.task_seed_stride=0
75
+ - 'rl_games.env_eval.task_description=''You are playing Demon Attack from a single
76
+ game image. Choose exactly one action from: NOOP, FIRE, RIGHT, LEFT, RIGHTFIRE,
77
+ LEFTFIRE.'''
78
+ - rl_games.env_eval.latency.values=[0]
79
+ - rl_games.env_eval.latency.prompt_map_path=/workspace/starVLA/data/demon_attack_fix_latency_2/demon_attack_train__bridge/latency_prompt_map.json
80
+ - rl_games.env_eval.mid_train.enabled=true
81
+ - rl_games.env_eval.mid_train.interval_steps=100
82
+ - rl_games.env_eval.mid_train.num_episodes=5
83
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
84
+ - rl_games.env_eval.mid_train.latencies=[2]
85
+ - rl_games.env_eval.post_train.enabled=true
86
+ - rl_games.env_eval.post_train.num_episodes=5
87
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
88
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
latency-aware/demon-attack/vla/starvla-qwenoft-h1/openvla_demon_attack_fix_latency_2_small/hydra/train_starvla_hydra.log ADDED
The diff for this file is too large to render. See raw diff
 
latency-aware/demon-attack/vla/starvla-qwenoft-h1/openvla_demon_attack_fix_latency_2_small/summary.jsonl ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"steps": 100}
2
+ {"steps": 200}
3
+ {"steps": 300}
4
+ {"steps": 400}
5
+ {"steps": 500}
6
+ {"steps": 600}
7
+ {"steps": 700}
8
+ {"steps": 800}
9
+ {"steps": 900}
10
+ {"steps": 1000}
11
+ {"steps": 1100}
12
+ {"steps": 1200}
13
+ {"steps": 1300}
14
+ {"steps": 1400}
15
+ {"steps": 1500}
16
+ {"steps": 1600}
17
+ {"steps": 1700}
18
+ {"steps": 1800}
19
+ {"steps": 1900}
20
+ {"steps": 2000}
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_1_200ep/checkpoints/steps_5000_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cff93451f827160bf275aa8bc0bcab83cbd71645ef0605b2e8869bd3691a3bb6
3
+ size 9138230516
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_1_200ep/config.full.yaml ADDED
@@ -0,0 +1,198 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ name: QwenOFT
3
+ qwenvl:
4
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
5
+ attn_implementation: flash_attention_2
6
+ enable_gradient_checkpointing: true
7
+ action_model:
8
+ action_model_type: MLP
9
+ action_dim: 7
10
+ action_hidden_dim: 2560
11
+ future_action_window_size: 0
12
+ past_action_window_size: 0
13
+ loss_type: discrete_ce
14
+ state_dim: 7
15
+ action_horizon: 1
16
+ action_env_dim: 2
17
+ datasets:
18
+ vla_data:
19
+ dataset_py: lerobot_datasets
20
+ include_state: true
21
+ data_root_dir: /workspace/starVLA/data/flappy_fix_latency_1_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
+ shuffle: true
28
+ action_balance:
29
+ enabled: false
30
+ strategy: balanced_epoch
31
+ action_key: action_id
32
+ target_flap_fraction: 0.3
33
+ noop_id: 0
34
+ flap_id: 1
35
+ per_device_batch_size: 32
36
+ load_all_data_for_training: true
37
+ obs_image_size: null
38
+ video_backend: torchvision_av
39
+ dataset:
40
+ source_hf: ''
41
+ config_name: null
42
+ converted_name: flappy_train
43
+ single_source_hf: ''
44
+ mixed_source_hf: ''
45
+ single_converted_name: flappy_train
46
+ mixed_converted_name: flappy_mixed_latency_train
47
+ single_latency_filter: null
48
+ mixed_latency_filter: null
49
+ force_download: false
50
+ setup_force: false
51
+ verify_rows: 200
52
+ max_episodes: null
53
+ episodes_per_latency: null
54
+ latency_filter: null
55
+ debug_subset:
56
+ enabled: false
57
+ max_episodes: 5
58
+ suffix: debug
59
+ base_model:
60
+ repo_id: Qwen/Qwen3-VL-4B-Instruct
61
+ initialization:
62
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
63
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
64
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
65
+ trainer:
66
+ max_train_steps: 5000
67
+ num_warmup_steps: 100
68
+ save_interval: 100
69
+ eval_interval: 100
70
+ eval_num_batches: 100
71
+ eval_action_classification: true
72
+ cc_f1_tolerance: 1
73
+ learning_rate:
74
+ base: 2.0e-05
75
+ qwen_vl_interface: 1.0e-05
76
+ action_model: 0.0001
77
+ lr_scheduler_type: cosine_with_min_lr
78
+ scheduler_specific_kwargs:
79
+ min_lr: 1.0e-06
80
+ freeze_modules: ''
81
+ loss_scale:
82
+ vla: 1.0
83
+ vlm: 0.1
84
+ max_grad_norm: 1.0
85
+ weight_decay: 0.0
86
+ logging_frequency: 1
87
+ gradient_clipping: 1.0
88
+ gradient_accumulation_steps: 4
89
+ distributed_backend: none
90
+ is_resume: false
91
+ pretrained_checkpoint: null
92
+ resume_step: 0
93
+ reload_modules: null
94
+ optimizer:
95
+ name: AdamW
96
+ betas:
97
+ - 0.9
98
+ - 0.95
99
+ eps: 1.0e-08
100
+ weight_decay: 1.0e-08
101
+ fused: false
102
+ save_format: pt
103
+ workspace_dir: WORKSPACE_DIR
104
+ run_root_dir: /workspace/starVLA/results/Checkpoints
105
+ seed: 42
106
+ wandb_entity: dongqianyu99-zhejiang-university
107
+ wandb_project: starVLA_rl_games
108
+ auth:
109
+ env_file: null
110
+ hf_token_env: HF_TOKEN
111
+ wandb_api_key_env: WANDB_API_KEY
112
+ paths:
113
+ run_root_dir: results/Checkpoints
114
+ dataset_local_dir: playground/Datasets/rl_games
115
+ dataset_cache_dir: null
116
+ base_model_dir: playground/Pretrained_models/Qwen3-VL-4B-Instruct
117
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
118
+ launch:
119
+ use_accelerate: true
120
+ gpus: null
121
+ num_processes: 1
122
+ dry_run: false
123
+ conda:
124
+ enabled: true
125
+ env_name: null
126
+ rl_games:
127
+ model_alias: openvla
128
+ env_eval:
129
+ image_size: 224
130
+ frameskip: 1
131
+ seed: 42
132
+ fixed_episode_seeds: true
133
+ latency_seed_stride: 0
134
+ task_seed_stride: 0
135
+ task_description: ''
136
+ enabled: true
137
+ distributed_mode: none
138
+ vectorized:
139
+ enabled: false
140
+ batch_size: 1
141
+ latency:
142
+ prompt_map_path: null
143
+ mode: single
144
+ values:
145
+ - 0
146
+ mid_train:
147
+ enabled: false
148
+ interval_steps: 100
149
+ latencies:
150
+ - 0
151
+ num_episodes: 5
152
+ max_steps_per_episode: 3600
153
+ post_train:
154
+ enabled: false
155
+ latencies:
156
+ - 0
157
+ - 1
158
+ - 2
159
+ - 3
160
+ - 4
161
+ num_episodes: 5
162
+ max_steps_per_episode: 3600
163
+ task: flappy
164
+ initialization_mode: bridge
165
+ action_carrier: bridge
166
+ model: openvla
167
+ env: flappy
168
+ init: bridge
169
+ bridge_base_model:
170
+ repo_id:
171
+ openvla: Qwen/Qwen3-VL-4B-Instruct
172
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
173
+ pi05: Qwen/Qwen3-VL-4B-Instruct
174
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
175
+ local_dir:
176
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
177
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
178
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
179
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
180
+ mode: single
181
+ checkpoint:
182
+ load: auto
183
+ hf_repo_id: null
184
+ save_best_model: false
185
+ save_pt_file: false
186
+ local:
187
+ keep_last_n: 1
188
+ sync:
189
+ enabled: false
190
+ repo_id: null
191
+ keep_last_n: 0
192
+ sync_every_n_checkpoints: 1
193
+ resume_policy: local_latest
194
+ run_id: flappy_fix_latency_1_200ep
195
+ output_dir: /workspace/starVLA/results/Checkpoints/flappy_fix_latency_1_200ep
196
+ config_yaml: null
197
+ is_debug: false
198
+ version_id: '0.21'
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_1_200ep/config.yaml ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ obs_image_size: null
16
+ per_device_batch_size: 32
17
+ framework:
18
+ action_model:
19
+ action_dim: 7
20
+ action_env_dim: 2
21
+ action_hidden_dim: 2560
22
+ action_horizon: 1
23
+ action_model_type: MLP
24
+ loss_type: discrete_ce
25
+ name: QwenOFT
26
+ qwenvl:
27
+ attn_implementation: flash_attention_2
28
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
29
+ enable_gradient_checkpointing: true
30
+ output_dir: /workspace/starVLA/results/Checkpoints/flappy_fix_latency_1_200ep
31
+ rl_games:
32
+ env_eval:
33
+ enabled: true
34
+ mid_train:
35
+ enabled: false
36
+ interval_steps: 100
37
+ task: flappy
38
+ run_id: flappy_fix_latency_1_200ep
39
+ run_root_dir: /workspace/starVLA/results/Checkpoints
40
+ seed: 42
41
+ trainer:
42
+ distributed_backend: none
43
+ eval_action_classification: true
44
+ eval_interval: 100
45
+ eval_num_batches: 100
46
+ freeze_modules: ''
47
+ gradient_accumulation_steps: 4
48
+ gradient_clipping: 1.0
49
+ is_resume: false
50
+ learning_rate:
51
+ action_model: 0.0001
52
+ base: 2.0e-05
53
+ qwen_vl_interface: 1.0e-05
54
+ logging_frequency: 1
55
+ lr_scheduler_type: cosine_with_min_lr
56
+ max_train_steps: 5000
57
+ num_warmup_steps: 100
58
+ optimizer:
59
+ betas:
60
+ - 0.9
61
+ - 0.95
62
+ eps: 1.0e-08
63
+ fused: false
64
+ weight_decay: 1.0e-08
65
+ pretrained_checkpoint: null
66
+ reload_modules: null
67
+ save_interval: 100
68
+ scheduler_specific_kwargs:
69
+ min_lr: 1.0e-06
70
+ wandb_entity: dongqianyu99-zhejiang-university
71
+ wandb_project: starVLA_rl_games
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_1_200ep/dataset_statistics.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.6979423761367798,
6
+ 0.3020575940608978,
7
+ 0.0,
8
+ 0.0,
9
+ 0.0,
10
+ 0.0,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.4588429629802704,
15
+ 0.4588429629802704,
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": 297483,
125
+ "num_trajectories": 180
126
+ }
127
+ }
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_1_200ep/dataset_statistics_eval.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.8606944680213928,
6
+ 0.13930556178092957,
7
+ 0.0,
8
+ 0.0,
9
+ 0.0,
10
+ 0.0,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.3463059663772583,
15
+ 0.3463059663772583,
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_1_200ep/hydra/.hydra/config.yaml ADDED
@@ -0,0 +1,196 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ qwenvl:
3
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
4
+ attn_implementation: flash_attention_2
5
+ enable_gradient_checkpointing: true
6
+ action_model:
7
+ state_dim: 7
8
+ loss_type: discrete_ce
9
+ action_horizon: 1
10
+ future_action_window_size: 0
11
+ past_action_window_size: 0
12
+ action_dim: 7
13
+ action_env_dim: 2
14
+ name: QwenOFT
15
+ datasets:
16
+ vla_data:
17
+ dataset_py: lerobot_datasets
18
+ include_state: true
19
+ data_root_dir: /workspace/starVLA/data/flappy_fix_latency_1_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
+ shuffle: true
26
+ action_balance:
27
+ enabled: false
28
+ strategy: balanced_epoch
29
+ action_key: action_id
30
+ target_flap_fraction: 0.3
31
+ noop_id: 0
32
+ flap_id: 1
33
+ per_device_batch_size: 32
34
+ load_all_data_for_training: true
35
+ obs_image_size: null
36
+ video_backend: torchvision_av
37
+ dataset:
38
+ source_hf: ${dataset.single_source_hf}
39
+ config_name: null
40
+ converted_name: ${dataset.single_converted_name}
41
+ single_source_hf: ''
42
+ mixed_source_hf: ''
43
+ single_converted_name: flappy_train
44
+ mixed_converted_name: flappy_mixed_latency_train
45
+ single_latency_filter: null
46
+ mixed_latency_filter: null
47
+ force_download: false
48
+ setup_force: false
49
+ verify_rows: 200
50
+ max_episodes: null
51
+ episodes_per_latency: null
52
+ latency_filter: ${dataset.single_latency_filter}
53
+ debug_subset:
54
+ enabled: false
55
+ max_episodes: 5
56
+ suffix: debug
57
+ base_model:
58
+ repo_id: ${bridge_base_model.repo_id.${model}}
59
+ initialization:
60
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
61
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
62
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
63
+ trainer:
64
+ max_train_steps: 5000
65
+ num_warmup_steps: 100
66
+ save_interval: 100
67
+ eval_interval: 100
68
+ eval_num_batches: 100
69
+ eval_action_classification: true
70
+ cc_f1_tolerance: 1
71
+ learning_rate:
72
+ base: 2.0e-05
73
+ qwen_vl_interface: 1.0e-05
74
+ action_model: 0.0001
75
+ lr_scheduler_type: cosine_with_min_lr
76
+ scheduler_specific_kwargs:
77
+ min_lr: 1.0e-06
78
+ freeze_modules: ''
79
+ loss_scale:
80
+ vla: 1.0
81
+ vlm: 0.1
82
+ max_grad_norm: 1.0
83
+ weight_decay: 0.0
84
+ logging_frequency: 1
85
+ gradient_clipping: 1.0
86
+ gradient_accumulation_steps: 4
87
+ distributed_backend: none
88
+ is_resume: false
89
+ pretrained_checkpoint: /workspace/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
90
+ resume_step: 0
91
+ reload_modules: null
92
+ optimizer:
93
+ name: AdamW
94
+ betas:
95
+ - 0.9
96
+ - 0.95
97
+ eps: 1.0e-08
98
+ weight_decay: 1.0e-08
99
+ fused: false
100
+ save_format: pt
101
+ workspace_dir: WORKSPACE_DIR
102
+ run_root_dir: /workspace/starVLA/results/Checkpoints
103
+ seed: 42
104
+ wandb_entity: dongqianyu99-zhejiang-university
105
+ wandb_project: starVLA_rl_games
106
+ auth:
107
+ env_file: null
108
+ hf_token_env: HF_TOKEN
109
+ wandb_api_key_env: WANDB_API_KEY
110
+ paths:
111
+ run_root_dir: results/Checkpoints
112
+ dataset_local_dir: playground/Datasets/rl_games
113
+ dataset_cache_dir: null
114
+ base_model_dir: ${bridge_base_model.local_dir.${model}}
115
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
116
+ launch:
117
+ use_accelerate: true
118
+ gpus: null
119
+ num_processes: 1
120
+ dry_run: false
121
+ conda:
122
+ enabled: true
123
+ env_name: null
124
+ rl_games:
125
+ model_alias: openvla
126
+ env_eval:
127
+ image_size: 224
128
+ frameskip: 1
129
+ seed: 42
130
+ fixed_episode_seeds: true
131
+ latency_seed_stride: 0
132
+ task_seed_stride: 0
133
+ task_description: ''
134
+ enabled: true
135
+ distributed_mode: none
136
+ vectorized:
137
+ enabled: false
138
+ batch_size: 1
139
+ latency:
140
+ prompt_map_path: null
141
+ mode: single
142
+ values:
143
+ - 0
144
+ mid_train:
145
+ enabled: false
146
+ interval_steps: 100
147
+ latencies:
148
+ - 0
149
+ num_episodes: 5
150
+ max_steps_per_episode: 3600
151
+ post_train:
152
+ enabled: false
153
+ latencies:
154
+ - 0
155
+ - 1
156
+ - 2
157
+ - 3
158
+ - 4
159
+ num_episodes: 5
160
+ max_steps_per_episode: 3600
161
+ task: flappy
162
+ initialization_mode: bridge
163
+ action_carrier: bridge
164
+ model: openvla
165
+ env: flappy
166
+ init: bridge
167
+ bridge_base_model:
168
+ repo_id:
169
+ openvla: Qwen/Qwen3-VL-4B-Instruct
170
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
171
+ pi05: Qwen/Qwen3-VL-4B-Instruct
172
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
173
+ local_dir:
174
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
175
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
176
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
177
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
178
+ mode: single
179
+ checkpoint:
180
+ load: auto
181
+ hf_repo_id: null
182
+ save_best_model: false
183
+ save_pt_file: false
184
+ local:
185
+ keep_last_n: 1
186
+ sync:
187
+ enabled: false
188
+ repo_id: null
189
+ keep_last_n: 0
190
+ sync_every_n_checkpoints: 1
191
+ resume_policy: local_latest
192
+ run_id: flappy_fix_latency_1_200ep
193
+ output_dir: null
194
+ config_yaml: null
195
+ is_debug: false
196
+ version_id: '0.21'
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_1_200ep/hydra/.hydra/hydra.yaml ADDED
@@ -0,0 +1,252 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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_1_200ep
120
+ - run_root_dir=/workspace/starVLA/results/Checkpoints
121
+ - seed=42
122
+ - wandb_entity=dongqianyu99-zhejiang-university
123
+ - wandb_project=starVLA_rl_games
124
+ - rl_games.env_eval.enabled=true
125
+ - checkpoint.sync.enabled=false
126
+ - checkpoint.sync.keep_last_n=0
127
+ - checkpoint.local.keep_last_n=1
128
+ - checkpoint.save_best_model=false
129
+ - checkpoint.save_pt_file=false
130
+ - trainer.is_resume=false
131
+ - trainer.pretrained_checkpoint=/workspace/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
132
+ - trainer.resume_step=0
133
+ - trainer.max_train_steps=5000
134
+ - trainer.num_warmup_steps=100
135
+ - trainer.save_interval=100
136
+ - trainer.eval_interval=100
137
+ - trainer.eval_num_batches=100
138
+ - trainer.logging_frequency=1
139
+ - trainer.gradient_accumulation_steps=4
140
+ - trainer.distributed_backend=none
141
+ - trainer.learning_rate.base=2e-05
142
+ - trainer.learning_rate.qwen_vl_interface=1e-05
143
+ - trainer.learning_rate.action_model=0.0001
144
+ - trainer.lr_scheduler_type=cosine_with_min_lr
145
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
146
+ - trainer.loss_scale.vla=1.0
147
+ - trainer.loss_scale.vlm=0.1
148
+ - trainer.max_grad_norm=1.0
149
+ - trainer.weight_decay=0.0
150
+ - trainer.gradient_clipping=1.0
151
+ - trainer.optimizer.name=AdamW
152
+ - trainer.optimizer.betas=[0.9, 0.95]
153
+ - trainer.optimizer.eps=1e-08
154
+ - trainer.optimizer.weight_decay=1e-08
155
+ - trainer.optimizer.fused=false
156
+ - trainer.save_format=pt
157
+ - framework.name=QwenOFT
158
+ - framework.qwenvl.attn_implementation=flash_attention_2
159
+ - framework.qwenvl.enable_gradient_checkpointing=true
160
+ - framework.action_model.action_dim=7
161
+ - framework.action_model.action_env_dim=2
162
+ - framework.action_model.state_dim=7
163
+ - framework.action_model.loss_type=discrete_ce
164
+ - framework.action_model.action_horizon=1
165
+ - framework.action_model.future_action_window_size=0
166
+ - framework.action_model.past_action_window_size=0
167
+ - datasets.vla_data.include_state=true
168
+ - datasets.vla_data.action_type=discrete
169
+ - datasets.vla_data.sequential_step_sampling=false
170
+ - datasets.vla_data.shuffle=true
171
+ - datasets.vla_data.action_balance.enabled=false
172
+ - datasets.vla_data.action_balance.strategy=balanced_epoch
173
+ - datasets.vla_data.action_balance.action_key=action_id
174
+ - datasets.vla_data.action_balance.target_flap_fraction=0.3
175
+ - datasets.vla_data.action_balance.noop_id=0
176
+ - datasets.vla_data.action_balance.flap_id=1
177
+ - datasets.vla_data.per_device_batch_size=32
178
+ - datasets.vla_data.load_all_data_for_training=true
179
+ - datasets.vla_data.video_backend=torchvision_av
180
+ - datasets.vla_data.data_root_dir=/workspace/starVLA/data/flappy_fix_latency_1_200ep
181
+ - datasets.vla_data.data_mix=flappy_train__bridge
182
+ - datasets.vla_data.eval_data_mix=flappy_train__bridge__val
183
+ - framework.qwenvl.base_vlm=/workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
184
+ - rl_games.task=flappy
185
+ - rl_games.model_alias=openvla
186
+ - rl_games.initialization_mode=bridge
187
+ - rl_games.action_carrier=bridge
188
+ - rl_games.env_eval.distributed_mode=none
189
+ - rl_games.env_eval.latency.mode=single
190
+ - rl_games.env_eval.frameskip=1
191
+ - rl_games.env_eval.image_size=224
192
+ - rl_games.env_eval.seed=42
193
+ - rl_games.env_eval.fixed_episode_seeds=true
194
+ - rl_games.env_eval.latency_seed_stride=0
195
+ - rl_games.env_eval.task_seed_stride=0
196
+ - rl_games.env_eval.latency.values=[0]
197
+ - rl_games.env_eval.mid_train.enabled=false
198
+ - rl_games.env_eval.mid_train.interval_steps=100
199
+ - rl_games.env_eval.mid_train.num_episodes=5
200
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
201
+ - rl_games.env_eval.mid_train.latencies=[0]
202
+ - rl_games.env_eval.post_train.enabled=false
203
+ - rl_games.env_eval.post_train.num_episodes=5
204
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
205
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
206
+ job:
207
+ name: train_starvla_hydra
208
+ chdir: false
209
+ 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/starVLA/data/flappy_fix_latency_1_200ep,datasets.vla_data.eval_data_mix=flappy_train__bridge__val,datasets.vla_data.include_state=true,datasets.vla_data.load_all_data_for_training=true,datasets.vla_data.per_device_batch_size=32,datasets.vla_data.sequential_step_sampling=false,datasets.vla_data.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/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.values=[0],rl_games.env_eval.latency_seed_stride=0,rl_games.env_eval.mid_train.enabled=false,rl_games.env_eval.mid_train.interval_steps=100,rl_games.env_eval.mid_train.latencies=[0],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_1_200ep,run_root_dir=/workspace/starVLA/results/Checkpoints,seed=42,trainer.distributed_backend=none,trainer.eval_interval=100,trainer.eval_num_batches=100,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,
210
+ 0.95],trainer.optimizer.eps=1e-08,trainer.optimizer.fused=false,trainer.optimizer.name=AdamW,trainer.optimizer.weight_decay=1e-08,trainer.pretrained_checkpoint=/workspace/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt,trainer.resume_step=0,trainer.save_format=pt,trainer.save_interval=100,trainer.scheduler_specific_kwargs.min_lr=1e-06,trainer.weight_decay=0.0,wandb_entity=dongqianyu99-zhejiang-university,wandb_project=starVLA_rl_games
211
+ id: ???
212
+ num: ???
213
+ config_name: train
214
+ env_set: {}
215
+ env_copy: []
216
+ config:
217
+ override_dirname:
218
+ kv_sep: '='
219
+ item_sep: ','
220
+ exclude_keys: []
221
+ runtime:
222
+ version: 1.3.3
223
+ version_base: '1.1'
224
+ cwd: /workspace/starVLA
225
+ config_sources:
226
+ - path: hydra.conf
227
+ schema: pkg
228
+ provider: hydra
229
+ - path: /workspace/starVLA/examples/rl_games/config
230
+ schema: file
231
+ provider: main
232
+ - path: ''
233
+ schema: structured
234
+ provider: schema
235
+ output_dir: /workspace/starVLA/results/Checkpoints/flappy_fix_latency_1_200ep/hydra
236
+ choices:
237
+ cross_task_setup: null
238
+ checkpoint: default
239
+ mode: single
240
+ init: bridge
241
+ env: flappy
242
+ model: openvla
243
+ hydra/env: default
244
+ hydra/callbacks: null
245
+ hydra/job_logging: default
246
+ hydra/hydra_logging: default
247
+ hydra/hydra_help: default
248
+ hydra/help: default
249
+ hydra/sweeper: basic
250
+ hydra/launcher: basic
251
+ hydra/output: default
252
+ verbose: false
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_1_200ep/hydra/.hydra/overrides.yaml ADDED
@@ -0,0 +1,91 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ - model=openvla
2
+ - env=flappy
3
+ - init=bridge
4
+ - mode=single
5
+ - run_id=flappy_fix_latency_1_200ep
6
+ - run_root_dir=/workspace/starVLA/results/Checkpoints
7
+ - seed=42
8
+ - wandb_entity=dongqianyu99-zhejiang-university
9
+ - wandb_project=starVLA_rl_games
10
+ - rl_games.env_eval.enabled=true
11
+ - checkpoint.sync.enabled=false
12
+ - checkpoint.sync.keep_last_n=0
13
+ - checkpoint.local.keep_last_n=1
14
+ - checkpoint.save_best_model=false
15
+ - checkpoint.save_pt_file=false
16
+ - trainer.is_resume=false
17
+ - trainer.pretrained_checkpoint=/workspace/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
18
+ - trainer.resume_step=0
19
+ - trainer.max_train_steps=5000
20
+ - trainer.num_warmup_steps=100
21
+ - trainer.save_interval=100
22
+ - trainer.eval_interval=100
23
+ - trainer.eval_num_batches=100
24
+ - trainer.logging_frequency=1
25
+ - trainer.gradient_accumulation_steps=4
26
+ - trainer.distributed_backend=none
27
+ - trainer.learning_rate.base=2e-05
28
+ - trainer.learning_rate.qwen_vl_interface=1e-05
29
+ - trainer.learning_rate.action_model=0.0001
30
+ - trainer.lr_scheduler_type=cosine_with_min_lr
31
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
32
+ - trainer.loss_scale.vla=1.0
33
+ - trainer.loss_scale.vlm=0.1
34
+ - trainer.max_grad_norm=1.0
35
+ - trainer.weight_decay=0.0
36
+ - trainer.gradient_clipping=1.0
37
+ - trainer.optimizer.name=AdamW
38
+ - trainer.optimizer.betas=[0.9, 0.95]
39
+ - trainer.optimizer.eps=1e-08
40
+ - trainer.optimizer.weight_decay=1e-08
41
+ - trainer.optimizer.fused=false
42
+ - trainer.save_format=pt
43
+ - framework.name=QwenOFT
44
+ - framework.qwenvl.attn_implementation=flash_attention_2
45
+ - framework.qwenvl.enable_gradient_checkpointing=true
46
+ - framework.action_model.action_dim=7
47
+ - framework.action_model.action_env_dim=2
48
+ - framework.action_model.state_dim=7
49
+ - framework.action_model.loss_type=discrete_ce
50
+ - framework.action_model.action_horizon=1
51
+ - framework.action_model.future_action_window_size=0
52
+ - framework.action_model.past_action_window_size=0
53
+ - datasets.vla_data.include_state=true
54
+ - datasets.vla_data.action_type=discrete
55
+ - datasets.vla_data.sequential_step_sampling=false
56
+ - datasets.vla_data.shuffle=true
57
+ - datasets.vla_data.action_balance.enabled=false
58
+ - datasets.vla_data.action_balance.strategy=balanced_epoch
59
+ - datasets.vla_data.action_balance.action_key=action_id
60
+ - datasets.vla_data.action_balance.target_flap_fraction=0.3
61
+ - datasets.vla_data.action_balance.noop_id=0
62
+ - datasets.vla_data.action_balance.flap_id=1
63
+ - datasets.vla_data.per_device_batch_size=32
64
+ - datasets.vla_data.load_all_data_for_training=true
65
+ - datasets.vla_data.video_backend=torchvision_av
66
+ - datasets.vla_data.data_root_dir=/workspace/starVLA/data/flappy_fix_latency_1_200ep
67
+ - datasets.vla_data.data_mix=flappy_train__bridge
68
+ - datasets.vla_data.eval_data_mix=flappy_train__bridge__val
69
+ - framework.qwenvl.base_vlm=/workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
70
+ - rl_games.task=flappy
71
+ - rl_games.model_alias=openvla
72
+ - rl_games.initialization_mode=bridge
73
+ - rl_games.action_carrier=bridge
74
+ - rl_games.env_eval.distributed_mode=none
75
+ - rl_games.env_eval.latency.mode=single
76
+ - rl_games.env_eval.frameskip=1
77
+ - rl_games.env_eval.image_size=224
78
+ - rl_games.env_eval.seed=42
79
+ - rl_games.env_eval.fixed_episode_seeds=true
80
+ - rl_games.env_eval.latency_seed_stride=0
81
+ - rl_games.env_eval.task_seed_stride=0
82
+ - rl_games.env_eval.latency.values=[0]
83
+ - rl_games.env_eval.mid_train.enabled=false
84
+ - rl_games.env_eval.mid_train.interval_steps=100
85
+ - rl_games.env_eval.mid_train.num_episodes=5
86
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
87
+ - rl_games.env_eval.mid_train.latencies=[0]
88
+ - rl_games.env_eval.post_train.enabled=false
89
+ - rl_games.env_eval.post_train.num_episodes=5
90
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
91
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_1_200ep/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_1_200ep/summary.jsonl ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"steps": 100}
2
+ {"steps": 200}
3
+ {"steps": 300}
4
+ {"steps": 400}
5
+ {"steps": 500}
6
+ {"steps": 600}
7
+ {"steps": 700}
8
+ {"steps": 800}
9
+ {"steps": 900}
10
+ {"steps": 1000}
11
+ {"steps": 1100}
12
+ {"steps": 1200}
13
+ {"steps": 1300}
14
+ {"steps": 1400}
15
+ {"steps": 1500}
16
+ {"steps": 1600}
17
+ {"steps": 1700}
18
+ {"steps": 1800}
19
+ {"steps": 1900}
20
+ {"steps": 2000}
21
+ {"steps": 2100}
22
+ {"steps": 2200}
23
+ {"steps": 2300}
24
+ {"steps": 2400}
25
+ {"steps": 2500}
26
+ {"steps": 2600}
27
+ {"steps": 2700}
28
+ {"steps": 2800}
29
+ {"steps": 2900}
30
+ {"steps": 3000}
31
+ {"steps": 3100}
32
+ {"steps": 3200}
33
+ {"steps": 3300}
34
+ {"steps": 3400}
35
+ {"steps": 3500}
36
+ {"steps": 3600}
37
+ {"steps": 3700}
38
+ {"steps": 3800}
39
+ {"steps": 3900}
40
+ {"steps": 4000}
41
+ {"steps": 4100}
42
+ {"steps": 4200}
43
+ {"steps": 4300}
44
+ {"steps": 4400}
45
+ {"steps": 4500}
46
+ {"steps": 4600}
47
+ {"steps": 4700}
48
+ {"steps": 4800}
49
+ {"steps": 4900}
50
+ {"steps": 5000}
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep/checkpoints/steps_5000_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aabcc8e3a22947e22d161d324de0e613a03300f39fa1dd28bffa2df927e994cf
3
+ size 9138230516
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep/config.full.yaml ADDED
@@ -0,0 +1,199 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ name: QwenOFT
3
+ qwenvl:
4
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
5
+ attn_implementation: flash_attention_2
6
+ enable_gradient_checkpointing: true
7
+ action_model:
8
+ action_model_type: MLP
9
+ action_dim: 7
10
+ action_hidden_dim: 2560
11
+ future_action_window_size: 0
12
+ past_action_window_size: 0
13
+ loss_type: discrete_ce
14
+ state_dim: 7
15
+ action_horizon: 1
16
+ action_env_dim: 2
17
+ datasets:
18
+ vla_data:
19
+ dataset_py: lerobot_datasets
20
+ include_state: true
21
+ data_root_dir: /workspace/starVLA/data/flappy_fix_latency_2_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
+ shuffle: true
28
+ action_balance:
29
+ enabled: false
30
+ strategy: balanced_epoch
31
+ action_key: action_id
32
+ target_flap_fraction: 0.3
33
+ noop_id: 0
34
+ flap_id: 1
35
+ per_device_batch_size: 32
36
+ load_all_data_for_training: true
37
+ obs_image_size: null
38
+ video_backend: torchvision_av
39
+ dataset:
40
+ source_hf: ''
41
+ config_name: null
42
+ source_subdir: null
43
+ converted_name: flappy_train
44
+ single_source_hf: ''
45
+ mixed_source_hf: ''
46
+ single_converted_name: flappy_train
47
+ mixed_converted_name: flappy_mixed_latency_train
48
+ single_latency_filter: null
49
+ mixed_latency_filter: null
50
+ force_download: false
51
+ setup_force: false
52
+ verify_rows: 200
53
+ max_episodes: null
54
+ episodes_per_latency: null
55
+ latency_filter: null
56
+ debug_subset:
57
+ enabled: false
58
+ max_episodes: 5
59
+ suffix: debug
60
+ base_model:
61
+ repo_id: Qwen/Qwen3-VL-4B-Instruct
62
+ initialization:
63
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
64
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
65
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
66
+ trainer:
67
+ max_train_steps: 5000
68
+ num_warmup_steps: 100
69
+ save_interval: 100
70
+ eval_interval: 100
71
+ eval_num_batches: 100
72
+ eval_action_classification: true
73
+ cc_f1_tolerance: 1
74
+ learning_rate:
75
+ base: 2.0e-05
76
+ qwen_vl_interface: 1.0e-05
77
+ action_model: 0.0001
78
+ lr_scheduler_type: cosine_with_min_lr
79
+ scheduler_specific_kwargs:
80
+ min_lr: 1.0e-06
81
+ freeze_modules: ''
82
+ loss_scale:
83
+ vla: 1.0
84
+ vlm: 0.1
85
+ max_grad_norm: 1.0
86
+ weight_decay: 0.0
87
+ logging_frequency: 1
88
+ gradient_clipping: 1.0
89
+ gradient_accumulation_steps: 4
90
+ distributed_backend: none
91
+ is_resume: false
92
+ pretrained_checkpoint: null
93
+ resume_step: 0
94
+ reload_modules: null
95
+ optimizer:
96
+ name: AdamW
97
+ betas:
98
+ - 0.9
99
+ - 0.95
100
+ eps: 1.0e-08
101
+ weight_decay: 1.0e-08
102
+ fused: false
103
+ save_format: pt
104
+ workspace_dir: WORKSPACE_DIR
105
+ run_root_dir: /workspace/starVLA/results/Checkpoints
106
+ seed: 42
107
+ wandb_entity: dongqianyu99-zhejiang-university
108
+ wandb_project: starVLA_rl_games
109
+ auth:
110
+ env_file: null
111
+ hf_token_env: HF_TOKEN
112
+ wandb_api_key_env: WANDB_API_KEY
113
+ paths:
114
+ run_root_dir: results/Checkpoints
115
+ dataset_local_dir: playground/Datasets/rl_games
116
+ dataset_cache_dir: null
117
+ base_model_dir: playground/Pretrained_models/Qwen3-VL-4B-Instruct
118
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
119
+ launch:
120
+ use_accelerate: true
121
+ gpus: null
122
+ num_processes: 1
123
+ dry_run: false
124
+ conda:
125
+ enabled: true
126
+ env_name: null
127
+ rl_games:
128
+ model_alias: openvla
129
+ env_eval:
130
+ image_size: 224
131
+ frameskip: 1
132
+ seed: 42
133
+ fixed_episode_seeds: true
134
+ latency_seed_stride: 0
135
+ task_seed_stride: 0
136
+ task_description: ''
137
+ enabled: true
138
+ distributed_mode: none
139
+ vectorized:
140
+ enabled: false
141
+ batch_size: 1
142
+ latency:
143
+ prompt_map_path: null
144
+ mode: single
145
+ values:
146
+ - 0
147
+ mid_train:
148
+ enabled: false
149
+ interval_steps: 100
150
+ latencies:
151
+ - 0
152
+ num_episodes: 5
153
+ max_steps_per_episode: 3600
154
+ post_train:
155
+ enabled: false
156
+ latencies:
157
+ - 0
158
+ - 1
159
+ - 2
160
+ - 3
161
+ - 4
162
+ num_episodes: 5
163
+ max_steps_per_episode: 3600
164
+ task: flappy
165
+ initialization_mode: bridge
166
+ action_carrier: bridge
167
+ model: openvla
168
+ env: flappy
169
+ init: bridge
170
+ bridge_base_model:
171
+ repo_id:
172
+ openvla: Qwen/Qwen3-VL-4B-Instruct
173
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
174
+ pi05: Qwen/Qwen3-VL-4B-Instruct
175
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
176
+ local_dir:
177
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
178
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
179
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
180
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
181
+ mode: single
182
+ checkpoint:
183
+ load: auto
184
+ hf_repo_id: null
185
+ save_best_model: false
186
+ save_pt_file: false
187
+ local:
188
+ keep_last_n: 1
189
+ sync:
190
+ enabled: false
191
+ repo_id: null
192
+ keep_last_n: 0
193
+ sync_every_n_checkpoints: 1
194
+ resume_policy: local_latest
195
+ run_id: flappy_fix_latency_2_200ep
196
+ output_dir: /workspace/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep
197
+ config_yaml: null
198
+ is_debug: false
199
+ version_id: '0.21'
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep/config.yaml ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ obs_image_size: null
16
+ per_device_batch_size: 32
17
+ framework:
18
+ action_model:
19
+ action_dim: 7
20
+ action_env_dim: 2
21
+ action_hidden_dim: 2560
22
+ action_horizon: 1
23
+ action_model_type: MLP
24
+ loss_type: discrete_ce
25
+ name: QwenOFT
26
+ qwenvl:
27
+ attn_implementation: flash_attention_2
28
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
29
+ enable_gradient_checkpointing: true
30
+ output_dir: /workspace/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep
31
+ rl_games:
32
+ env_eval:
33
+ enabled: true
34
+ mid_train:
35
+ enabled: false
36
+ interval_steps: 100
37
+ task: flappy
38
+ run_id: flappy_fix_latency_2_200ep
39
+ run_root_dir: /workspace/starVLA/results/Checkpoints
40
+ seed: 42
41
+ trainer:
42
+ distributed_backend: none
43
+ eval_action_classification: true
44
+ eval_interval: 100
45
+ eval_num_batches: 100
46
+ freeze_modules: ''
47
+ gradient_accumulation_steps: 4
48
+ gradient_clipping: 1.0
49
+ is_resume: false
50
+ learning_rate:
51
+ action_model: 0.0001
52
+ base: 2.0e-05
53
+ qwen_vl_interface: 1.0e-05
54
+ logging_frequency: 1
55
+ lr_scheduler_type: cosine_with_min_lr
56
+ max_train_steps: 5000
57
+ num_warmup_steps: 100
58
+ optimizer:
59
+ betas:
60
+ - 0.9
61
+ - 0.95
62
+ eps: 1.0e-08
63
+ fused: false
64
+ weight_decay: 1.0e-08
65
+ pretrained_checkpoint: null
66
+ reload_modules: null
67
+ save_interval: 100
68
+ scheduler_specific_kwargs:
69
+ min_lr: 1.0e-06
70
+ wandb_entity: dongqianyu99-zhejiang-university
71
+ wandb_project: starVLA_rl_games
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep/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/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/hydra/.hydra/config.yaml ADDED
@@ -0,0 +1,197 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ qwenvl:
3
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
4
+ attn_implementation: flash_attention_2
5
+ enable_gradient_checkpointing: true
6
+ action_model:
7
+ state_dim: 7
8
+ loss_type: discrete_ce
9
+ action_horizon: 1
10
+ future_action_window_size: 0
11
+ past_action_window_size: 0
12
+ action_dim: 7
13
+ action_env_dim: 2
14
+ name: QwenOFT
15
+ datasets:
16
+ vla_data:
17
+ dataset_py: lerobot_datasets
18
+ include_state: true
19
+ data_root_dir: /workspace/starVLA/data/flappy_fix_latency_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
+ shuffle: true
26
+ action_balance:
27
+ enabled: false
28
+ strategy: balanced_epoch
29
+ action_key: action_id
30
+ target_flap_fraction: 0.3
31
+ noop_id: 0
32
+ flap_id: 1
33
+ per_device_batch_size: 32
34
+ load_all_data_for_training: true
35
+ obs_image_size: null
36
+ video_backend: torchvision_av
37
+ dataset:
38
+ source_hf: ${dataset.single_source_hf}
39
+ config_name: null
40
+ source_subdir: null
41
+ converted_name: ${dataset.single_converted_name}
42
+ single_source_hf: ''
43
+ mixed_source_hf: ''
44
+ single_converted_name: flappy_train
45
+ mixed_converted_name: flappy_mixed_latency_train
46
+ single_latency_filter: null
47
+ mixed_latency_filter: null
48
+ force_download: false
49
+ setup_force: false
50
+ verify_rows: 200
51
+ max_episodes: null
52
+ episodes_per_latency: null
53
+ latency_filter: ${dataset.single_latency_filter}
54
+ debug_subset:
55
+ enabled: false
56
+ max_episodes: 5
57
+ suffix: debug
58
+ base_model:
59
+ repo_id: ${bridge_base_model.repo_id.${model}}
60
+ initialization:
61
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
62
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
63
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
64
+ trainer:
65
+ max_train_steps: 5000
66
+ num_warmup_steps: 100
67
+ save_interval: 100
68
+ eval_interval: 100
69
+ eval_num_batches: 100
70
+ eval_action_classification: true
71
+ cc_f1_tolerance: 1
72
+ learning_rate:
73
+ base: 2.0e-05
74
+ qwen_vl_interface: 1.0e-05
75
+ action_model: 0.0001
76
+ lr_scheduler_type: cosine_with_min_lr
77
+ scheduler_specific_kwargs:
78
+ min_lr: 1.0e-06
79
+ freeze_modules: ''
80
+ loss_scale:
81
+ vla: 1.0
82
+ vlm: 0.1
83
+ max_grad_norm: 1.0
84
+ weight_decay: 0.0
85
+ logging_frequency: 1
86
+ gradient_clipping: 1.0
87
+ gradient_accumulation_steps: 4
88
+ distributed_backend: none
89
+ is_resume: false
90
+ pretrained_checkpoint: /workspace/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
91
+ resume_step: 0
92
+ reload_modules: null
93
+ optimizer:
94
+ name: AdamW
95
+ betas:
96
+ - 0.9
97
+ - 0.95
98
+ eps: 1.0e-08
99
+ weight_decay: 1.0e-08
100
+ fused: false
101
+ save_format: pt
102
+ workspace_dir: WORKSPACE_DIR
103
+ run_root_dir: /workspace/starVLA/results/Checkpoints
104
+ seed: 42
105
+ wandb_entity: dongqianyu99-zhejiang-university
106
+ wandb_project: starVLA_rl_games
107
+ auth:
108
+ env_file: null
109
+ hf_token_env: HF_TOKEN
110
+ wandb_api_key_env: WANDB_API_KEY
111
+ paths:
112
+ run_root_dir: results/Checkpoints
113
+ dataset_local_dir: playground/Datasets/rl_games
114
+ dataset_cache_dir: null
115
+ base_model_dir: ${bridge_base_model.local_dir.${model}}
116
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
117
+ launch:
118
+ use_accelerate: true
119
+ gpus: null
120
+ num_processes: 1
121
+ dry_run: false
122
+ conda:
123
+ enabled: true
124
+ env_name: null
125
+ rl_games:
126
+ model_alias: openvla
127
+ env_eval:
128
+ image_size: 224
129
+ frameskip: 1
130
+ seed: 42
131
+ fixed_episode_seeds: true
132
+ latency_seed_stride: 0
133
+ task_seed_stride: 0
134
+ task_description: ''
135
+ enabled: true
136
+ distributed_mode: none
137
+ vectorized:
138
+ enabled: false
139
+ batch_size: 1
140
+ latency:
141
+ prompt_map_path: null
142
+ mode: single
143
+ values:
144
+ - 0
145
+ mid_train:
146
+ enabled: false
147
+ interval_steps: 100
148
+ latencies:
149
+ - 0
150
+ num_episodes: 5
151
+ max_steps_per_episode: 3600
152
+ post_train:
153
+ enabled: false
154
+ latencies:
155
+ - 0
156
+ - 1
157
+ - 2
158
+ - 3
159
+ - 4
160
+ num_episodes: 5
161
+ max_steps_per_episode: 3600
162
+ task: flappy
163
+ initialization_mode: bridge
164
+ action_carrier: bridge
165
+ model: openvla
166
+ env: flappy
167
+ init: bridge
168
+ bridge_base_model:
169
+ repo_id:
170
+ openvla: Qwen/Qwen3-VL-4B-Instruct
171
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
172
+ pi05: Qwen/Qwen3-VL-4B-Instruct
173
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
174
+ local_dir:
175
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
176
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
177
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
178
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
179
+ mode: single
180
+ checkpoint:
181
+ load: auto
182
+ hf_repo_id: null
183
+ save_best_model: false
184
+ save_pt_file: false
185
+ local:
186
+ keep_last_n: 1
187
+ sync:
188
+ enabled: false
189
+ repo_id: null
190
+ keep_last_n: 0
191
+ sync_every_n_checkpoints: 1
192
+ resume_policy: local_latest
193
+ run_id: flappy_fix_latency_2_200ep
194
+ output_dir: null
195
+ config_yaml: null
196
+ is_debug: false
197
+ version_id: '0.21'
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep/hydra/.hydra/hydra.yaml ADDED
@@ -0,0 +1,252 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
120
+ - run_root_dir=/workspace/starVLA/results/Checkpoints
121
+ - seed=42
122
+ - wandb_entity=dongqianyu99-zhejiang-university
123
+ - wandb_project=starVLA_rl_games
124
+ - rl_games.env_eval.enabled=true
125
+ - checkpoint.sync.enabled=false
126
+ - checkpoint.sync.keep_last_n=0
127
+ - checkpoint.local.keep_last_n=1
128
+ - checkpoint.save_best_model=false
129
+ - checkpoint.save_pt_file=false
130
+ - trainer.is_resume=false
131
+ - trainer.pretrained_checkpoint=/workspace/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
132
+ - trainer.resume_step=0
133
+ - trainer.max_train_steps=5000
134
+ - trainer.num_warmup_steps=100
135
+ - trainer.save_interval=100
136
+ - trainer.eval_interval=100
137
+ - trainer.eval_num_batches=100
138
+ - trainer.logging_frequency=1
139
+ - trainer.gradient_accumulation_steps=4
140
+ - trainer.distributed_backend=none
141
+ - trainer.learning_rate.base=2e-05
142
+ - trainer.learning_rate.qwen_vl_interface=1e-05
143
+ - trainer.learning_rate.action_model=0.0001
144
+ - trainer.lr_scheduler_type=cosine_with_min_lr
145
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
146
+ - trainer.loss_scale.vla=1.0
147
+ - trainer.loss_scale.vlm=0.1
148
+ - trainer.max_grad_norm=1.0
149
+ - trainer.weight_decay=0.0
150
+ - trainer.gradient_clipping=1.0
151
+ - trainer.optimizer.name=AdamW
152
+ - trainer.optimizer.betas=[0.9, 0.95]
153
+ - trainer.optimizer.eps=1e-08
154
+ - trainer.optimizer.weight_decay=1e-08
155
+ - trainer.optimizer.fused=false
156
+ - trainer.save_format=pt
157
+ - framework.name=QwenOFT
158
+ - framework.qwenvl.attn_implementation=flash_attention_2
159
+ - framework.qwenvl.enable_gradient_checkpointing=true
160
+ - framework.action_model.action_dim=7
161
+ - framework.action_model.action_env_dim=2
162
+ - framework.action_model.state_dim=7
163
+ - framework.action_model.loss_type=discrete_ce
164
+ - framework.action_model.action_horizon=1
165
+ - framework.action_model.future_action_window_size=0
166
+ - framework.action_model.past_action_window_size=0
167
+ - datasets.vla_data.include_state=true
168
+ - datasets.vla_data.action_type=discrete
169
+ - datasets.vla_data.sequential_step_sampling=false
170
+ - datasets.vla_data.shuffle=true
171
+ - datasets.vla_data.action_balance.enabled=false
172
+ - datasets.vla_data.action_balance.strategy=balanced_epoch
173
+ - datasets.vla_data.action_balance.action_key=action_id
174
+ - datasets.vla_data.action_balance.target_flap_fraction=0.3
175
+ - datasets.vla_data.action_balance.noop_id=0
176
+ - datasets.vla_data.action_balance.flap_id=1
177
+ - datasets.vla_data.per_device_batch_size=32
178
+ - datasets.vla_data.load_all_data_for_training=true
179
+ - datasets.vla_data.video_backend=torchvision_av
180
+ - datasets.vla_data.data_root_dir=/workspace/starVLA/data/flappy_fix_latency_2_200ep
181
+ - datasets.vla_data.data_mix=flappy_train__bridge
182
+ - datasets.vla_data.eval_data_mix=flappy_train__bridge__val
183
+ - framework.qwenvl.base_vlm=/workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
184
+ - rl_games.task=flappy
185
+ - rl_games.model_alias=openvla
186
+ - rl_games.initialization_mode=bridge
187
+ - rl_games.action_carrier=bridge
188
+ - rl_games.env_eval.distributed_mode=none
189
+ - rl_games.env_eval.latency.mode=single
190
+ - rl_games.env_eval.frameskip=1
191
+ - rl_games.env_eval.image_size=224
192
+ - rl_games.env_eval.seed=42
193
+ - rl_games.env_eval.fixed_episode_seeds=true
194
+ - rl_games.env_eval.latency_seed_stride=0
195
+ - rl_games.env_eval.task_seed_stride=0
196
+ - rl_games.env_eval.latency.values=[0]
197
+ - rl_games.env_eval.mid_train.enabled=false
198
+ - rl_games.env_eval.mid_train.interval_steps=100
199
+ - rl_games.env_eval.mid_train.num_episodes=5
200
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
201
+ - rl_games.env_eval.mid_train.latencies=[0]
202
+ - rl_games.env_eval.post_train.enabled=false
203
+ - rl_games.env_eval.post_train.num_episodes=5
204
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
205
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
206
+ job:
207
+ name: train_starvla_hydra
208
+ chdir: false
209
+ 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/starVLA/data/flappy_fix_latency_2_200ep,datasets.vla_data.eval_data_mix=flappy_train__bridge__val,datasets.vla_data.include_state=true,datasets.vla_data.load_all_data_for_training=true,datasets.vla_data.per_device_batch_size=32,datasets.vla_data.sequential_step_sampling=false,datasets.vla_data.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/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.values=[0],rl_games.env_eval.latency_seed_stride=0,rl_games.env_eval.mid_train.enabled=false,rl_games.env_eval.mid_train.interval_steps=100,rl_games.env_eval.mid_train.latencies=[0],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,run_root_dir=/workspace/starVLA/results/Checkpoints,seed=42,trainer.distributed_backend=none,trainer.eval_interval=100,trainer.eval_num_batches=100,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,
210
+ 0.95],trainer.optimizer.eps=1e-08,trainer.optimizer.fused=false,trainer.optimizer.name=AdamW,trainer.optimizer.weight_decay=1e-08,trainer.pretrained_checkpoint=/workspace/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt,trainer.resume_step=0,trainer.save_format=pt,trainer.save_interval=100,trainer.scheduler_specific_kwargs.min_lr=1e-06,trainer.weight_decay=0.0,wandb_entity=dongqianyu99-zhejiang-university,wandb_project=starVLA_rl_games
211
+ id: ???
212
+ num: ???
213
+ config_name: train
214
+ env_set: {}
215
+ env_copy: []
216
+ config:
217
+ override_dirname:
218
+ kv_sep: '='
219
+ item_sep: ','
220
+ exclude_keys: []
221
+ runtime:
222
+ version: 1.3.3
223
+ version_base: '1.1'
224
+ cwd: /workspace/starVLA
225
+ config_sources:
226
+ - path: hydra.conf
227
+ schema: pkg
228
+ provider: hydra
229
+ - path: /workspace/starVLA/examples/rl_games/config
230
+ schema: file
231
+ provider: main
232
+ - path: ''
233
+ schema: structured
234
+ provider: schema
235
+ output_dir: /workspace/starVLA/results/Checkpoints/flappy_fix_latency_2_200ep/hydra
236
+ choices:
237
+ cross_task_setup: null
238
+ checkpoint: default
239
+ mode: single
240
+ init: bridge
241
+ env: flappy
242
+ model: openvla
243
+ hydra/env: default
244
+ hydra/callbacks: null
245
+ hydra/job_logging: default
246
+ hydra/hydra_logging: default
247
+ hydra/hydra_help: default
248
+ hydra/help: default
249
+ hydra/sweeper: basic
250
+ hydra/launcher: basic
251
+ hydra/output: default
252
+ verbose: false
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep/hydra/.hydra/overrides.yaml ADDED
@@ -0,0 +1,91 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ - model=openvla
2
+ - env=flappy
3
+ - init=bridge
4
+ - mode=single
5
+ - run_id=flappy_fix_latency_2_200ep
6
+ - run_root_dir=/workspace/starVLA/results/Checkpoints
7
+ - seed=42
8
+ - wandb_entity=dongqianyu99-zhejiang-university
9
+ - wandb_project=starVLA_rl_games
10
+ - rl_games.env_eval.enabled=true
11
+ - checkpoint.sync.enabled=false
12
+ - checkpoint.sync.keep_last_n=0
13
+ - checkpoint.local.keep_last_n=1
14
+ - checkpoint.save_best_model=false
15
+ - checkpoint.save_pt_file=false
16
+ - trainer.is_resume=false
17
+ - trainer.pretrained_checkpoint=/workspace/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
18
+ - trainer.resume_step=0
19
+ - trainer.max_train_steps=5000
20
+ - trainer.num_warmup_steps=100
21
+ - trainer.save_interval=100
22
+ - trainer.eval_interval=100
23
+ - trainer.eval_num_batches=100
24
+ - trainer.logging_frequency=1
25
+ - trainer.gradient_accumulation_steps=4
26
+ - trainer.distributed_backend=none
27
+ - trainer.learning_rate.base=2e-05
28
+ - trainer.learning_rate.qwen_vl_interface=1e-05
29
+ - trainer.learning_rate.action_model=0.0001
30
+ - trainer.lr_scheduler_type=cosine_with_min_lr
31
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
32
+ - trainer.loss_scale.vla=1.0
33
+ - trainer.loss_scale.vlm=0.1
34
+ - trainer.max_grad_norm=1.0
35
+ - trainer.weight_decay=0.0
36
+ - trainer.gradient_clipping=1.0
37
+ - trainer.optimizer.name=AdamW
38
+ - trainer.optimizer.betas=[0.9, 0.95]
39
+ - trainer.optimizer.eps=1e-08
40
+ - trainer.optimizer.weight_decay=1e-08
41
+ - trainer.optimizer.fused=false
42
+ - trainer.save_format=pt
43
+ - framework.name=QwenOFT
44
+ - framework.qwenvl.attn_implementation=flash_attention_2
45
+ - framework.qwenvl.enable_gradient_checkpointing=true
46
+ - framework.action_model.action_dim=7
47
+ - framework.action_model.action_env_dim=2
48
+ - framework.action_model.state_dim=7
49
+ - framework.action_model.loss_type=discrete_ce
50
+ - framework.action_model.action_horizon=1
51
+ - framework.action_model.future_action_window_size=0
52
+ - framework.action_model.past_action_window_size=0
53
+ - datasets.vla_data.include_state=true
54
+ - datasets.vla_data.action_type=discrete
55
+ - datasets.vla_data.sequential_step_sampling=false
56
+ - datasets.vla_data.shuffle=true
57
+ - datasets.vla_data.action_balance.enabled=false
58
+ - datasets.vla_data.action_balance.strategy=balanced_epoch
59
+ - datasets.vla_data.action_balance.action_key=action_id
60
+ - datasets.vla_data.action_balance.target_flap_fraction=0.3
61
+ - datasets.vla_data.action_balance.noop_id=0
62
+ - datasets.vla_data.action_balance.flap_id=1
63
+ - datasets.vla_data.per_device_batch_size=32
64
+ - datasets.vla_data.load_all_data_for_training=true
65
+ - datasets.vla_data.video_backend=torchvision_av
66
+ - datasets.vla_data.data_root_dir=/workspace/starVLA/data/flappy_fix_latency_2_200ep
67
+ - datasets.vla_data.data_mix=flappy_train__bridge
68
+ - datasets.vla_data.eval_data_mix=flappy_train__bridge__val
69
+ - framework.qwenvl.base_vlm=/workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
70
+ - rl_games.task=flappy
71
+ - rl_games.model_alias=openvla
72
+ - rl_games.initialization_mode=bridge
73
+ - rl_games.action_carrier=bridge
74
+ - rl_games.env_eval.distributed_mode=none
75
+ - rl_games.env_eval.latency.mode=single
76
+ - rl_games.env_eval.frameskip=1
77
+ - rl_games.env_eval.image_size=224
78
+ - rl_games.env_eval.seed=42
79
+ - rl_games.env_eval.fixed_episode_seeds=true
80
+ - rl_games.env_eval.latency_seed_stride=0
81
+ - rl_games.env_eval.task_seed_stride=0
82
+ - rl_games.env_eval.latency.values=[0]
83
+ - rl_games.env_eval.mid_train.enabled=false
84
+ - rl_games.env_eval.mid_train.interval_steps=100
85
+ - rl_games.env_eval.mid_train.num_episodes=5
86
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
87
+ - rl_games.env_eval.mid_train.latencies=[0]
88
+ - rl_games.env_eval.post_train.enabled=false
89
+ - rl_games.env_eval.post_train.num_episodes=5
90
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
91
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_2_200ep/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/summary.jsonl ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"steps": 100}
2
+ {"steps": 200}
3
+ {"steps": 300}
4
+ {"steps": 400}
5
+ {"steps": 500}
6
+ {"steps": 600}
7
+ {"steps": 700}
8
+ {"steps": 800}
9
+ {"steps": 900}
10
+ {"steps": 1000}
11
+ {"steps": 1100}
12
+ {"steps": 1200}
13
+ {"steps": 1300}
14
+ {"steps": 1400}
15
+ {"steps": 1500}
16
+ {"steps": 1600}
17
+ {"steps": 1700}
18
+ {"steps": 1800}
19
+ {"steps": 1900}
20
+ {"steps": 2000}
21
+ {"steps": 2100}
22
+ {"steps": 2200}
23
+ {"steps": 2300}
24
+ {"steps": 2400}
25
+ {"steps": 2500}
26
+ {"steps": 2600}
27
+ {"steps": 2700}
28
+ {"steps": 2800}
29
+ {"steps": 2900}
30
+ {"steps": 3000}
31
+ {"steps": 3100}
32
+ {"steps": 3200}
33
+ {"steps": 3300}
34
+ {"steps": 3400}
35
+ {"steps": 3500}
36
+ {"steps": 3600}
37
+ {"steps": 3700}
38
+ {"steps": 3800}
39
+ {"steps": 3900}
40
+ {"steps": 4000}
41
+ {"steps": 4100}
42
+ {"steps": 4200}
43
+ {"steps": 4300}
44
+ {"steps": 4400}
45
+ {"steps": 4500}
46
+ {"steps": 4600}
47
+ {"steps": 4700}
48
+ {"steps": 4800}
49
+ {"steps": 4900}
50
+ {"steps": 5000}
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_3_200ep/checkpoints/steps_5000_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5da01d6eeb7a46861da90b0a8dbc364a33f92d542d25079bcc86280b10617418
3
+ size 9138230516
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_3_200ep/config.full.yaml ADDED
@@ -0,0 +1,199 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ name: QwenOFT
3
+ qwenvl:
4
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
5
+ attn_implementation: flash_attention_2
6
+ enable_gradient_checkpointing: true
7
+ action_model:
8
+ action_model_type: MLP
9
+ action_dim: 7
10
+ action_hidden_dim: 2560
11
+ future_action_window_size: 0
12
+ past_action_window_size: 0
13
+ loss_type: discrete_ce
14
+ state_dim: 7
15
+ action_horizon: 1
16
+ action_env_dim: 2
17
+ datasets:
18
+ vla_data:
19
+ dataset_py: lerobot_datasets
20
+ include_state: true
21
+ data_root_dir: /workspace/starVLA/data/flappy_fix_latency_3_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
+ shuffle: true
28
+ action_balance:
29
+ enabled: false
30
+ strategy: balanced_epoch
31
+ action_key: action_id
32
+ target_flap_fraction: 0.3
33
+ noop_id: 0
34
+ flap_id: 1
35
+ per_device_batch_size: 32
36
+ load_all_data_for_training: true
37
+ obs_image_size: null
38
+ video_backend: torchvision_av
39
+ dataset:
40
+ source_hf: ''
41
+ config_name: null
42
+ source_subdir: null
43
+ converted_name: flappy_train
44
+ single_source_hf: ''
45
+ mixed_source_hf: ''
46
+ single_converted_name: flappy_train
47
+ mixed_converted_name: flappy_mixed_latency_train
48
+ single_latency_filter: null
49
+ mixed_latency_filter: null
50
+ force_download: false
51
+ setup_force: false
52
+ verify_rows: 200
53
+ max_episodes: null
54
+ episodes_per_latency: null
55
+ latency_filter: null
56
+ debug_subset:
57
+ enabled: false
58
+ max_episodes: 5
59
+ suffix: debug
60
+ base_model:
61
+ repo_id: Qwen/Qwen3-VL-4B-Instruct
62
+ initialization:
63
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
64
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
65
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
66
+ trainer:
67
+ max_train_steps: 5000
68
+ num_warmup_steps: 100
69
+ save_interval: 100
70
+ eval_interval: 100
71
+ eval_num_batches: 100
72
+ eval_action_classification: true
73
+ cc_f1_tolerance: 1
74
+ learning_rate:
75
+ base: 2.0e-05
76
+ qwen_vl_interface: 1.0e-05
77
+ action_model: 0.0001
78
+ lr_scheduler_type: cosine_with_min_lr
79
+ scheduler_specific_kwargs:
80
+ min_lr: 1.0e-06
81
+ freeze_modules: ''
82
+ loss_scale:
83
+ vla: 1.0
84
+ vlm: 0.1
85
+ max_grad_norm: 1.0
86
+ weight_decay: 0.0
87
+ logging_frequency: 1
88
+ gradient_clipping: 1.0
89
+ gradient_accumulation_steps: 4
90
+ distributed_backend: none
91
+ is_resume: false
92
+ pretrained_checkpoint: null
93
+ resume_step: 0
94
+ reload_modules: null
95
+ optimizer:
96
+ name: AdamW
97
+ betas:
98
+ - 0.9
99
+ - 0.95
100
+ eps: 1.0e-08
101
+ weight_decay: 1.0e-08
102
+ fused: false
103
+ save_format: pt
104
+ workspace_dir: WORKSPACE_DIR
105
+ run_root_dir: /workspace/starVLA/results/Checkpoints
106
+ seed: 42
107
+ wandb_entity: dongqianyu99-zhejiang-university
108
+ wandb_project: starVLA_rl_games
109
+ auth:
110
+ env_file: null
111
+ hf_token_env: HF_TOKEN
112
+ wandb_api_key_env: WANDB_API_KEY
113
+ paths:
114
+ run_root_dir: results/Checkpoints
115
+ dataset_local_dir: playground/Datasets/rl_games
116
+ dataset_cache_dir: null
117
+ base_model_dir: playground/Pretrained_models/Qwen3-VL-4B-Instruct
118
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
119
+ launch:
120
+ use_accelerate: true
121
+ gpus: null
122
+ num_processes: 1
123
+ dry_run: false
124
+ conda:
125
+ enabled: true
126
+ env_name: null
127
+ rl_games:
128
+ model_alias: openvla
129
+ env_eval:
130
+ image_size: 224
131
+ frameskip: 1
132
+ seed: 42
133
+ fixed_episode_seeds: true
134
+ latency_seed_stride: 0
135
+ task_seed_stride: 0
136
+ task_description: ''
137
+ enabled: true
138
+ distributed_mode: none
139
+ vectorized:
140
+ enabled: false
141
+ batch_size: 1
142
+ latency:
143
+ prompt_map_path: null
144
+ mode: single
145
+ values:
146
+ - 0
147
+ mid_train:
148
+ enabled: false
149
+ interval_steps: 100
150
+ latencies:
151
+ - 0
152
+ num_episodes: 5
153
+ max_steps_per_episode: 3600
154
+ post_train:
155
+ enabled: false
156
+ latencies:
157
+ - 0
158
+ - 1
159
+ - 2
160
+ - 3
161
+ - 4
162
+ num_episodes: 5
163
+ max_steps_per_episode: 3600
164
+ task: flappy
165
+ initialization_mode: bridge
166
+ action_carrier: bridge
167
+ model: openvla
168
+ env: flappy
169
+ init: bridge
170
+ bridge_base_model:
171
+ repo_id:
172
+ openvla: Qwen/Qwen3-VL-4B-Instruct
173
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
174
+ pi05: Qwen/Qwen3-VL-4B-Instruct
175
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
176
+ local_dir:
177
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
178
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
179
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
180
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
181
+ mode: single
182
+ checkpoint:
183
+ load: auto
184
+ hf_repo_id: null
185
+ save_best_model: false
186
+ save_pt_file: false
187
+ local:
188
+ keep_last_n: 1
189
+ sync:
190
+ enabled: false
191
+ repo_id: null
192
+ keep_last_n: 0
193
+ sync_every_n_checkpoints: 1
194
+ resume_policy: local_latest
195
+ run_id: flappy_fix_latency_3_200ep
196
+ output_dir: /workspace/starVLA/results/Checkpoints/flappy_fix_latency_3_200ep
197
+ config_yaml: null
198
+ is_debug: false
199
+ version_id: '0.21'
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_3_200ep/config.yaml ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ obs_image_size: null
16
+ per_device_batch_size: 32
17
+ framework:
18
+ action_model:
19
+ action_dim: 7
20
+ action_env_dim: 2
21
+ action_hidden_dim: 2560
22
+ action_horizon: 1
23
+ action_model_type: MLP
24
+ loss_type: discrete_ce
25
+ name: QwenOFT
26
+ qwenvl:
27
+ attn_implementation: flash_attention_2
28
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
29
+ enable_gradient_checkpointing: true
30
+ output_dir: /workspace/starVLA/results/Checkpoints/flappy_fix_latency_3_200ep
31
+ rl_games:
32
+ env_eval:
33
+ enabled: true
34
+ mid_train:
35
+ enabled: false
36
+ interval_steps: 100
37
+ task: flappy
38
+ run_id: flappy_fix_latency_3_200ep
39
+ run_root_dir: /workspace/starVLA/results/Checkpoints
40
+ seed: 42
41
+ trainer:
42
+ distributed_backend: none
43
+ eval_action_classification: true
44
+ eval_interval: 100
45
+ eval_num_batches: 100
46
+ freeze_modules: ''
47
+ gradient_accumulation_steps: 4
48
+ gradient_clipping: 1.0
49
+ is_resume: false
50
+ learning_rate:
51
+ action_model: 0.0001
52
+ base: 2.0e-05
53
+ qwen_vl_interface: 1.0e-05
54
+ logging_frequency: 1
55
+ lr_scheduler_type: cosine_with_min_lr
56
+ max_train_steps: 5000
57
+ num_warmup_steps: 100
58
+ optimizer:
59
+ betas:
60
+ - 0.9
61
+ - 0.95
62
+ eps: 1.0e-08
63
+ fused: false
64
+ weight_decay: 1.0e-08
65
+ pretrained_checkpoint: null
66
+ reload_modules: null
67
+ save_interval: 100
68
+ scheduler_specific_kwargs:
69
+ min_lr: 1.0e-06
70
+ wandb_entity: dongqianyu99-zhejiang-university
71
+ wandb_project: starVLA_rl_games
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_3_200ep/dataset_statistics.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.5971351861953735,
6
+ 0.40286484360694885,
7
+ 0.0,
8
+ 0.0,
9
+ 0.0,
10
+ 0.0,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.49107804894447327,
15
+ 0.49107804894447327,
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": 316528,
125
+ "num_trajectories": 180
126
+ }
127
+ }
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_3_200ep/dataset_statistics_eval.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.8040833473205566,
6
+ 0.19591666758060455,
7
+ 0.0,
8
+ 0.0,
9
+ 0.0,
10
+ 0.0,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.39686188101768494,
15
+ 0.39686188101768494,
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_3_200ep/hydra/.hydra/config.yaml ADDED
@@ -0,0 +1,197 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ qwenvl:
3
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
4
+ attn_implementation: flash_attention_2
5
+ enable_gradient_checkpointing: true
6
+ action_model:
7
+ state_dim: 7
8
+ loss_type: discrete_ce
9
+ action_horizon: 1
10
+ future_action_window_size: 0
11
+ past_action_window_size: 0
12
+ action_dim: 7
13
+ action_env_dim: 2
14
+ name: QwenOFT
15
+ datasets:
16
+ vla_data:
17
+ dataset_py: lerobot_datasets
18
+ include_state: true
19
+ data_root_dir: /workspace/starVLA/data/flappy_fix_latency_3_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
+ shuffle: true
26
+ action_balance:
27
+ enabled: false
28
+ strategy: balanced_epoch
29
+ action_key: action_id
30
+ target_flap_fraction: 0.3
31
+ noop_id: 0
32
+ flap_id: 1
33
+ per_device_batch_size: 32
34
+ load_all_data_for_training: true
35
+ obs_image_size: null
36
+ video_backend: torchvision_av
37
+ dataset:
38
+ source_hf: ${dataset.single_source_hf}
39
+ config_name: null
40
+ source_subdir: null
41
+ converted_name: ${dataset.single_converted_name}
42
+ single_source_hf: ''
43
+ mixed_source_hf: ''
44
+ single_converted_name: flappy_train
45
+ mixed_converted_name: flappy_mixed_latency_train
46
+ single_latency_filter: null
47
+ mixed_latency_filter: null
48
+ force_download: false
49
+ setup_force: false
50
+ verify_rows: 200
51
+ max_episodes: null
52
+ episodes_per_latency: null
53
+ latency_filter: ${dataset.single_latency_filter}
54
+ debug_subset:
55
+ enabled: false
56
+ max_episodes: 5
57
+ suffix: debug
58
+ base_model:
59
+ repo_id: ${bridge_base_model.repo_id.${model}}
60
+ initialization:
61
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
62
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
63
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
64
+ trainer:
65
+ max_train_steps: 5000
66
+ num_warmup_steps: 100
67
+ save_interval: 100
68
+ eval_interval: 100
69
+ eval_num_batches: 100
70
+ eval_action_classification: true
71
+ cc_f1_tolerance: 1
72
+ learning_rate:
73
+ base: 2.0e-05
74
+ qwen_vl_interface: 1.0e-05
75
+ action_model: 0.0001
76
+ lr_scheduler_type: cosine_with_min_lr
77
+ scheduler_specific_kwargs:
78
+ min_lr: 1.0e-06
79
+ freeze_modules: ''
80
+ loss_scale:
81
+ vla: 1.0
82
+ vlm: 0.1
83
+ max_grad_norm: 1.0
84
+ weight_decay: 0.0
85
+ logging_frequency: 1
86
+ gradient_clipping: 1.0
87
+ gradient_accumulation_steps: 4
88
+ distributed_backend: none
89
+ is_resume: false
90
+ pretrained_checkpoint: /workspace/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
91
+ resume_step: 0
92
+ reload_modules: null
93
+ optimizer:
94
+ name: AdamW
95
+ betas:
96
+ - 0.9
97
+ - 0.95
98
+ eps: 1.0e-08
99
+ weight_decay: 1.0e-08
100
+ fused: false
101
+ save_format: pt
102
+ workspace_dir: WORKSPACE_DIR
103
+ run_root_dir: /workspace/starVLA/results/Checkpoints
104
+ seed: 42
105
+ wandb_entity: dongqianyu99-zhejiang-university
106
+ wandb_project: starVLA_rl_games
107
+ auth:
108
+ env_file: null
109
+ hf_token_env: HF_TOKEN
110
+ wandb_api_key_env: WANDB_API_KEY
111
+ paths:
112
+ run_root_dir: results/Checkpoints
113
+ dataset_local_dir: playground/Datasets/rl_games
114
+ dataset_cache_dir: null
115
+ base_model_dir: ${bridge_base_model.local_dir.${model}}
116
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
117
+ launch:
118
+ use_accelerate: true
119
+ gpus: null
120
+ num_processes: 1
121
+ dry_run: false
122
+ conda:
123
+ enabled: true
124
+ env_name: null
125
+ rl_games:
126
+ model_alias: openvla
127
+ env_eval:
128
+ image_size: 224
129
+ frameskip: 1
130
+ seed: 42
131
+ fixed_episode_seeds: true
132
+ latency_seed_stride: 0
133
+ task_seed_stride: 0
134
+ task_description: ''
135
+ enabled: true
136
+ distributed_mode: none
137
+ vectorized:
138
+ enabled: false
139
+ batch_size: 1
140
+ latency:
141
+ prompt_map_path: null
142
+ mode: single
143
+ values:
144
+ - 0
145
+ mid_train:
146
+ enabled: false
147
+ interval_steps: 100
148
+ latencies:
149
+ - 0
150
+ num_episodes: 5
151
+ max_steps_per_episode: 3600
152
+ post_train:
153
+ enabled: false
154
+ latencies:
155
+ - 0
156
+ - 1
157
+ - 2
158
+ - 3
159
+ - 4
160
+ num_episodes: 5
161
+ max_steps_per_episode: 3600
162
+ task: flappy
163
+ initialization_mode: bridge
164
+ action_carrier: bridge
165
+ model: openvla
166
+ env: flappy
167
+ init: bridge
168
+ bridge_base_model:
169
+ repo_id:
170
+ openvla: Qwen/Qwen3-VL-4B-Instruct
171
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
172
+ pi05: Qwen/Qwen3-VL-4B-Instruct
173
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
174
+ local_dir:
175
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
176
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
177
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
178
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
179
+ mode: single
180
+ checkpoint:
181
+ load: auto
182
+ hf_repo_id: null
183
+ save_best_model: false
184
+ save_pt_file: false
185
+ local:
186
+ keep_last_n: 1
187
+ sync:
188
+ enabled: false
189
+ repo_id: null
190
+ keep_last_n: 0
191
+ sync_every_n_checkpoints: 1
192
+ resume_policy: local_latest
193
+ run_id: flappy_fix_latency_3_200ep
194
+ output_dir: null
195
+ config_yaml: null
196
+ is_debug: false
197
+ version_id: '0.21'
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_3_200ep/hydra/.hydra/hydra.yaml ADDED
@@ -0,0 +1,252 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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_3_200ep
120
+ - run_root_dir=/workspace/starVLA/results/Checkpoints
121
+ - seed=42
122
+ - wandb_entity=dongqianyu99-zhejiang-university
123
+ - wandb_project=starVLA_rl_games
124
+ - rl_games.env_eval.enabled=true
125
+ - checkpoint.sync.enabled=false
126
+ - checkpoint.sync.keep_last_n=0
127
+ - checkpoint.local.keep_last_n=1
128
+ - checkpoint.save_best_model=false
129
+ - checkpoint.save_pt_file=false
130
+ - trainer.is_resume=false
131
+ - trainer.pretrained_checkpoint=/workspace/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
132
+ - trainer.resume_step=0
133
+ - trainer.max_train_steps=5000
134
+ - trainer.num_warmup_steps=100
135
+ - trainer.save_interval=100
136
+ - trainer.eval_interval=100
137
+ - trainer.eval_num_batches=100
138
+ - trainer.logging_frequency=1
139
+ - trainer.gradient_accumulation_steps=4
140
+ - trainer.distributed_backend=none
141
+ - trainer.learning_rate.base=2e-05
142
+ - trainer.learning_rate.qwen_vl_interface=1e-05
143
+ - trainer.learning_rate.action_model=0.0001
144
+ - trainer.lr_scheduler_type=cosine_with_min_lr
145
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
146
+ - trainer.loss_scale.vla=1.0
147
+ - trainer.loss_scale.vlm=0.1
148
+ - trainer.max_grad_norm=1.0
149
+ - trainer.weight_decay=0.0
150
+ - trainer.gradient_clipping=1.0
151
+ - trainer.optimizer.name=AdamW
152
+ - trainer.optimizer.betas=[0.9, 0.95]
153
+ - trainer.optimizer.eps=1e-08
154
+ - trainer.optimizer.weight_decay=1e-08
155
+ - trainer.optimizer.fused=false
156
+ - trainer.save_format=pt
157
+ - framework.name=QwenOFT
158
+ - framework.qwenvl.attn_implementation=flash_attention_2
159
+ - framework.qwenvl.enable_gradient_checkpointing=true
160
+ - framework.action_model.action_dim=7
161
+ - framework.action_model.action_env_dim=2
162
+ - framework.action_model.state_dim=7
163
+ - framework.action_model.loss_type=discrete_ce
164
+ - framework.action_model.action_horizon=1
165
+ - framework.action_model.future_action_window_size=0
166
+ - framework.action_model.past_action_window_size=0
167
+ - datasets.vla_data.include_state=true
168
+ - datasets.vla_data.action_type=discrete
169
+ - datasets.vla_data.sequential_step_sampling=false
170
+ - datasets.vla_data.shuffle=true
171
+ - datasets.vla_data.action_balance.enabled=false
172
+ - datasets.vla_data.action_balance.strategy=balanced_epoch
173
+ - datasets.vla_data.action_balance.action_key=action_id
174
+ - datasets.vla_data.action_balance.target_flap_fraction=0.3
175
+ - datasets.vla_data.action_balance.noop_id=0
176
+ - datasets.vla_data.action_balance.flap_id=1
177
+ - datasets.vla_data.per_device_batch_size=32
178
+ - datasets.vla_data.load_all_data_for_training=true
179
+ - datasets.vla_data.video_backend=torchvision_av
180
+ - datasets.vla_data.data_root_dir=/workspace/starVLA/data/flappy_fix_latency_3_200ep
181
+ - datasets.vla_data.data_mix=flappy_train__bridge
182
+ - datasets.vla_data.eval_data_mix=flappy_train__bridge__val
183
+ - framework.qwenvl.base_vlm=/workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
184
+ - rl_games.task=flappy
185
+ - rl_games.model_alias=openvla
186
+ - rl_games.initialization_mode=bridge
187
+ - rl_games.action_carrier=bridge
188
+ - rl_games.env_eval.distributed_mode=none
189
+ - rl_games.env_eval.latency.mode=single
190
+ - rl_games.env_eval.frameskip=1
191
+ - rl_games.env_eval.image_size=224
192
+ - rl_games.env_eval.seed=42
193
+ - rl_games.env_eval.fixed_episode_seeds=true
194
+ - rl_games.env_eval.latency_seed_stride=0
195
+ - rl_games.env_eval.task_seed_stride=0
196
+ - rl_games.env_eval.latency.values=[0]
197
+ - rl_games.env_eval.mid_train.enabled=false
198
+ - rl_games.env_eval.mid_train.interval_steps=100
199
+ - rl_games.env_eval.mid_train.num_episodes=5
200
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
201
+ - rl_games.env_eval.mid_train.latencies=[0]
202
+ - rl_games.env_eval.post_train.enabled=false
203
+ - rl_games.env_eval.post_train.num_episodes=5
204
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
205
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
206
+ job:
207
+ name: train_starvla_hydra
208
+ chdir: false
209
+ 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/starVLA/data/flappy_fix_latency_3_200ep,datasets.vla_data.eval_data_mix=flappy_train__bridge__val,datasets.vla_data.include_state=true,datasets.vla_data.load_all_data_for_training=true,datasets.vla_data.per_device_batch_size=32,datasets.vla_data.sequential_step_sampling=false,datasets.vla_data.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/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.values=[0],rl_games.env_eval.latency_seed_stride=0,rl_games.env_eval.mid_train.enabled=false,rl_games.env_eval.mid_train.interval_steps=100,rl_games.env_eval.mid_train.latencies=[0],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_3_200ep,run_root_dir=/workspace/starVLA/results/Checkpoints,seed=42,trainer.distributed_backend=none,trainer.eval_interval=100,trainer.eval_num_batches=100,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,
210
+ 0.95],trainer.optimizer.eps=1e-08,trainer.optimizer.fused=false,trainer.optimizer.name=AdamW,trainer.optimizer.weight_decay=1e-08,trainer.pretrained_checkpoint=/workspace/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt,trainer.resume_step=0,trainer.save_format=pt,trainer.save_interval=100,trainer.scheduler_specific_kwargs.min_lr=1e-06,trainer.weight_decay=0.0,wandb_entity=dongqianyu99-zhejiang-university,wandb_project=starVLA_rl_games
211
+ id: ???
212
+ num: ???
213
+ config_name: train
214
+ env_set: {}
215
+ env_copy: []
216
+ config:
217
+ override_dirname:
218
+ kv_sep: '='
219
+ item_sep: ','
220
+ exclude_keys: []
221
+ runtime:
222
+ version: 1.3.3
223
+ version_base: '1.1'
224
+ cwd: /workspace/starVLA
225
+ config_sources:
226
+ - path: hydra.conf
227
+ schema: pkg
228
+ provider: hydra
229
+ - path: /workspace/starVLA/examples/rl_games/config
230
+ schema: file
231
+ provider: main
232
+ - path: ''
233
+ schema: structured
234
+ provider: schema
235
+ output_dir: /workspace/starVLA/results/Checkpoints/flappy_fix_latency_3_200ep/hydra
236
+ choices:
237
+ cross_task_setup: null
238
+ checkpoint: default
239
+ mode: single
240
+ init: bridge
241
+ env: flappy
242
+ model: openvla
243
+ hydra/env: default
244
+ hydra/callbacks: null
245
+ hydra/job_logging: default
246
+ hydra/hydra_logging: default
247
+ hydra/hydra_help: default
248
+ hydra/help: default
249
+ hydra/sweeper: basic
250
+ hydra/launcher: basic
251
+ hydra/output: default
252
+ verbose: false
latency-aware/flappy/vla/starvla-qwenoft-h1/flappy_fix_latency_3_200ep/hydra/.hydra/overrides.yaml ADDED
@@ -0,0 +1,91 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ - model=openvla
2
+ - env=flappy
3
+ - init=bridge
4
+ - mode=single
5
+ - run_id=flappy_fix_latency_3_200ep
6
+ - run_root_dir=/workspace/starVLA/results/Checkpoints
7
+ - seed=42
8
+ - wandb_entity=dongqianyu99-zhejiang-university
9
+ - wandb_project=starVLA_rl_games
10
+ - rl_games.env_eval.enabled=true
11
+ - checkpoint.sync.enabled=false
12
+ - checkpoint.sync.keep_last_n=0
13
+ - checkpoint.local.keep_last_n=1
14
+ - checkpoint.save_best_model=false
15
+ - checkpoint.save_pt_file=false
16
+ - trainer.is_resume=false
17
+ - trainer.pretrained_checkpoint=/workspace/starVLA/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
18
+ - trainer.resume_step=0
19
+ - trainer.max_train_steps=5000
20
+ - trainer.num_warmup_steps=100
21
+ - trainer.save_interval=100
22
+ - trainer.eval_interval=100
23
+ - trainer.eval_num_batches=100
24
+ - trainer.logging_frequency=1
25
+ - trainer.gradient_accumulation_steps=4
26
+ - trainer.distributed_backend=none
27
+ - trainer.learning_rate.base=2e-05
28
+ - trainer.learning_rate.qwen_vl_interface=1e-05
29
+ - trainer.learning_rate.action_model=0.0001
30
+ - trainer.lr_scheduler_type=cosine_with_min_lr
31
+ - trainer.scheduler_specific_kwargs.min_lr=1e-06
32
+ - trainer.loss_scale.vla=1.0
33
+ - trainer.loss_scale.vlm=0.1
34
+ - trainer.max_grad_norm=1.0
35
+ - trainer.weight_decay=0.0
36
+ - trainer.gradient_clipping=1.0
37
+ - trainer.optimizer.name=AdamW
38
+ - trainer.optimizer.betas=[0.9, 0.95]
39
+ - trainer.optimizer.eps=1e-08
40
+ - trainer.optimizer.weight_decay=1e-08
41
+ - trainer.optimizer.fused=false
42
+ - trainer.save_format=pt
43
+ - framework.name=QwenOFT
44
+ - framework.qwenvl.attn_implementation=flash_attention_2
45
+ - framework.qwenvl.enable_gradient_checkpointing=true
46
+ - framework.action_model.action_dim=7
47
+ - framework.action_model.action_env_dim=2
48
+ - framework.action_model.state_dim=7
49
+ - framework.action_model.loss_type=discrete_ce
50
+ - framework.action_model.action_horizon=1
51
+ - framework.action_model.future_action_window_size=0
52
+ - framework.action_model.past_action_window_size=0
53
+ - datasets.vla_data.include_state=true
54
+ - datasets.vla_data.action_type=discrete
55
+ - datasets.vla_data.sequential_step_sampling=false
56
+ - datasets.vla_data.shuffle=true
57
+ - datasets.vla_data.action_balance.enabled=false
58
+ - datasets.vla_data.action_balance.strategy=balanced_epoch
59
+ - datasets.vla_data.action_balance.action_key=action_id
60
+ - datasets.vla_data.action_balance.target_flap_fraction=0.3
61
+ - datasets.vla_data.action_balance.noop_id=0
62
+ - datasets.vla_data.action_balance.flap_id=1
63
+ - datasets.vla_data.per_device_batch_size=32
64
+ - datasets.vla_data.load_all_data_for_training=true
65
+ - datasets.vla_data.video_backend=torchvision_av
66
+ - datasets.vla_data.data_root_dir=/workspace/starVLA/data/flappy_fix_latency_3_200ep
67
+ - datasets.vla_data.data_mix=flappy_train__bridge
68
+ - datasets.vla_data.eval_data_mix=flappy_train__bridge__val
69
+ - framework.qwenvl.base_vlm=/workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
70
+ - rl_games.task=flappy
71
+ - rl_games.model_alias=openvla
72
+ - rl_games.initialization_mode=bridge
73
+ - rl_games.action_carrier=bridge
74
+ - rl_games.env_eval.distributed_mode=none
75
+ - rl_games.env_eval.latency.mode=single
76
+ - rl_games.env_eval.frameskip=1
77
+ - rl_games.env_eval.image_size=224
78
+ - rl_games.env_eval.seed=42
79
+ - rl_games.env_eval.fixed_episode_seeds=true
80
+ - rl_games.env_eval.latency_seed_stride=0
81
+ - rl_games.env_eval.task_seed_stride=0
82
+ - rl_games.env_eval.latency.values=[0]
83
+ - rl_games.env_eval.mid_train.enabled=false
84
+ - rl_games.env_eval.mid_train.interval_steps=100
85
+ - rl_games.env_eval.mid_train.num_episodes=5
86
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
87
+ - rl_games.env_eval.mid_train.latencies=[0]
88
+ - rl_games.env_eval.post_train.enabled=false
89
+ - rl_games.env_eval.post_train.num_episodes=5
90
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
91
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]