dongqianyu99 commited on
Commit
1fb0300
·
verified ·
1 Parent(s): 65d03c9

Upload folder using huggingface_hub

Browse files
pi05_deadly_corridor_fix_latency_0_1000ep/checkpoints/steps_500_state/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:78f85251f875a0763fda512abbefa6dcdc25423f33b943a9a985fd34860afb7b
3
+ size 11412998004
pi05_deadly_corridor_fix_latency_0_1000ep/checkpoints/steps_500_state/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e1169237449cec9abdbdcba181bab6dac0ff3fbc7f1495c31c8d1f908bd15218
3
+ size 22793150843
pi05_deadly_corridor_fix_latency_0_1000ep/checkpoints/steps_500_state/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4982813c7d8e9ab37cb1f5ae79c64f24ffa676f72ac9a387315a71881ea673fc
3
+ size 14821
pi05_deadly_corridor_fix_latency_0_1000ep/config.full.yaml ADDED
@@ -0,0 +1,237 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ name: QwenPI_v3
3
+ qwenvl:
4
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
5
+ attn_implementation: flash_attention_2
6
+ vl_hidden_dim: 2560
7
+ num_vl_layers: 36
8
+ enable_gradient_checkpointing: true
9
+ action_model:
10
+ action_model_type: LayerwiseFM
11
+ action_dim: 7
12
+ state_dim: 7
13
+ action_horizon: 1
14
+ repeated_diffusion_steps: 2
15
+ num_inference_timesteps: 4
16
+ add_pos_embed: true
17
+ max_seq_len: 1024
18
+ num_target_vision_tokens: 32
19
+ noise_beta_alpha: 1.5
20
+ noise_beta_beta: 1.0
21
+ noise_s: 0.999
22
+ num_timestep_buckets: 1000
23
+ diffusion_model_cfg:
24
+ action_dit_hidden_dim: 1024
25
+ dropout: 0.2
26
+ final_dropout: true
27
+ interleave_self_attention: true
28
+ norm_type: ada_norm
29
+ positional_embeddings: null
30
+ attention_head_dim: 64
31
+ output_dim: 1024
32
+ num_layers: 36
33
+ input_embedding_dim: 1024
34
+ cross_attention_dim: 1024
35
+ num_attention_heads: 16
36
+ loss_type: l1
37
+ future_action_window_size: 0
38
+ past_action_window_size: 0
39
+ action_env_dim: 7
40
+ datasets:
41
+ vla_data:
42
+ dataset_py: lerobot_datasets
43
+ include_state: true
44
+ data_root_dir: /workspace/starVLA/data/deadly_corridor_fix_latency_0_1000ep
45
+ data_mix: deadly_corridor_train__bridge
46
+ eval_data_mix: deadly_corridor_train__bridge__val
47
+ custom_mixtures_path: null
48
+ action_type: discrete
49
+ sequential_step_sampling: false
50
+ eval_sequential_step_sampling: null
51
+ num_workers: 4
52
+ eval_num_workers: null
53
+ shuffle: true
54
+ action_balance:
55
+ enabled: false
56
+ strategy: balanced_epoch
57
+ action_key: action_id
58
+ target_flap_fraction: 0.3
59
+ noop_id: 0
60
+ flap_id: 1
61
+ latency_curriculum:
62
+ enabled: false
63
+ strategy: exclusive
64
+ latencies: null
65
+ phase_steps: null
66
+ per_device_batch_size: 16
67
+ load_all_data_for_training: true
68
+ obs_image_size: null
69
+ video_backend: torchvision_av
70
+ dataset:
71
+ source_hf: ''
72
+ config_name: null
73
+ source_subdir: null
74
+ converted_name: deadly_corridor_train
75
+ single_source_hf: ''
76
+ mixed_source_hf: ''
77
+ single_converted_name: deadly_corridor_train
78
+ mixed_converted_name: deadly_corridor_mixed_latency_train
79
+ single_latency_filter: null
80
+ mixed_latency_filter: null
81
+ force_download: false
82
+ setup_force: false
83
+ skip_verification: false
84
+ verify_rows: 200
85
+ max_episodes: null
86
+ episodes_per_latency: null
87
+ latency_filter: null
88
+ debug_subset:
89
+ enabled: false
90
+ max_episodes: 5
91
+ suffix: debug
92
+ base_model:
93
+ repo_id: Qwen/Qwen3-VL-4B-Instruct
94
+ initialization:
95
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-PI_v3-Bridge-RT_1
96
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-PI_v3-Bridge-RT_1
97
+ checkpoint_filename: checkpoints/steps_50000_pytorch_model.pt
98
+ trainer:
99
+ max_train_steps: 500
100
+ num_warmup_steps: 100
101
+ save_interval: 100
102
+ eval_interval: 100
103
+ eval_num_batches: 100
104
+ per_latency_eval_num_batches: null
105
+ eval_action_classification: true
106
+ eval_action_classification_interval: null
107
+ cc_f1_tolerance: 1
108
+ learning_rate:
109
+ base: 5.0e-06
110
+ qwen_vl_interface: 1.0e-05
111
+ action_model: 5.0e-06
112
+ lr_scheduler_type: cosine_with_min_lr
113
+ scheduler_specific_kwargs:
114
+ min_lr: 1.0e-06
115
+ freeze_modules: ''
116
+ loss_scale:
117
+ vla: 1.0
118
+ vlm: 0.1
119
+ max_grad_norm: 1.0
120
+ weight_decay: 0.0
121
+ logging_frequency: 1
122
+ gradient_clipping: 1.0
123
+ gradient_accumulation_steps: 8
124
+ distributed_backend: none
125
+ is_resume: false
126
+ pretrained_checkpoint: null
127
+ resume_step: 0
128
+ reload_modules: null
129
+ optimizer:
130
+ name: AdamW
131
+ betas:
132
+ - 0.9
133
+ - 0.95
134
+ eps: 1.0e-08
135
+ weight_decay: 1.0e-08
136
+ fused: false
137
+ save_format: pt
138
+ workspace_dir: WORKSPACE_DIR
139
+ run_root_dir: /workspace/starVLA/results/Checkpoints
140
+ seed: 42
141
+ wandb_entity: dongqianyu99-zhejiang-university
142
+ wandb_project: starVLA_rl_games
143
+ auth:
144
+ env_file: null
145
+ hf_token_env: HF_TOKEN
146
+ wandb_api_key_env: WANDB_API_KEY
147
+ paths:
148
+ run_root_dir: results/Checkpoints
149
+ dataset_local_dir: playground/Datasets/rl_games
150
+ dataset_cache_dir: null
151
+ base_model_dir: playground/Pretrained_models/Qwen3-VL-4B-Instruct
152
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
153
+ launch:
154
+ use_accelerate: true
155
+ gpus: null
156
+ num_processes: 1
157
+ dry_run: false
158
+ conda:
159
+ enabled: true
160
+ env_name: null
161
+ rl_games:
162
+ model_alias: pi-0.5
163
+ env_eval:
164
+ image_size: 224
165
+ frameskip: 4
166
+ seed: 42
167
+ fixed_episode_seeds: true
168
+ latency_seed_stride: 0
169
+ task_seed_stride: 0
170
+ task_description: You are playing Deadly Corridor in VizDoom. Choose actions from
171
+ MOVE_FORWARD, MOVE_BACKWARD, MOVE_LEFT, MOVE_RIGHT, TURN_LEFT, TURN_RIGHT, ATTACK.
172
+ deadly:
173
+ action_layout: multibinary_7
174
+ enabled: true
175
+ distributed_mode: none
176
+ vectorized:
177
+ enabled: false
178
+ batch_size: 1
179
+ latency:
180
+ prompt_map_path: /workspace/starVLA/data/deadly_corridor_fix_latency_0_1000ep/deadly_corridor_train__bridge/latency_prompt_map.json
181
+ mode: single
182
+ values:
183
+ - 0
184
+ mid_train:
185
+ enabled: false
186
+ interval_steps: 100
187
+ latencies:
188
+ - 0
189
+ num_episodes: 5
190
+ max_steps_per_episode: 3600
191
+ post_train:
192
+ enabled: false
193
+ latencies:
194
+ - 0
195
+ - 1
196
+ - 2
197
+ - 3
198
+ - 4
199
+ num_episodes: 5
200
+ max_steps_per_episode: 3600
201
+ task: deadly_corridor
202
+ deadly_corridor_loss_type: null
203
+ initialization_mode: bridge
204
+ action_carrier: bridge
205
+ model: pi05
206
+ env: deadly_corridor
207
+ init: bridge
208
+ bridge_base_model:
209
+ repo_id:
210
+ openvla: Qwen/Qwen3-VL-4B-Instruct
211
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
212
+ pi05: Qwen/Qwen3-VL-4B-Instruct
213
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
214
+ local_dir:
215
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
216
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
217
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
218
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
219
+ mode: single
220
+ checkpoint:
221
+ load: auto
222
+ hf_repo_id: null
223
+ save_best_model: false
224
+ save_pt_file: false
225
+ local:
226
+ keep_last_n: 1
227
+ sync:
228
+ enabled: false
229
+ repo_id: null
230
+ keep_last_n: 0
231
+ sync_every_n_checkpoints: 1
232
+ resume_policy: local_latest
233
+ run_id: pi05_deadly_corridor_fix_latency_0_1000ep
234
+ output_dir: /workspace/starVLA/results/Checkpoints/pi05_deadly_corridor_fix_latency_0_1000ep
235
+ config_yaml: null
236
+ is_debug: false
237
+ version_id: '0.21'
pi05_deadly_corridor_fix_latency_0_1000ep/config.yaml ADDED
@@ -0,0 +1,99 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ checkpoint:
2
+ local:
3
+ keep_last_n: 1
4
+ save_best_model: false
5
+ save_pt_file: false
6
+ sync:
7
+ enabled: false
8
+ keep_last_n: 0
9
+ repo_id: null
10
+ datasets:
11
+ vla_data:
12
+ data_mix: deadly_corridor_train__bridge
13
+ dataset_py: lerobot_datasets
14
+ eval_data_mix: deadly_corridor_train__bridge__val
15
+ latency_curriculum:
16
+ enabled: false
17
+ obs_image_size: null
18
+ per_device_batch_size: 16
19
+ framework:
20
+ action_model:
21
+ action_dim: 7
22
+ action_env_dim: 7
23
+ action_horizon: 1
24
+ add_pos_embed: true
25
+ diffusion_model_cfg:
26
+ action_dit_hidden_dim: 1024
27
+ attention_head_dim: 64
28
+ cross_attention_dim: 1024
29
+ dropout: 0.2
30
+ final_dropout: true
31
+ input_embedding_dim: 1024
32
+ interleave_self_attention: true
33
+ norm_type: ada_norm
34
+ num_attention_heads: 16
35
+ num_layers: 36
36
+ output_dim: 1024
37
+ positional_embeddings: null
38
+ max_seq_len: 1024
39
+ noise_beta_alpha: 1.5
40
+ noise_beta_beta: 1.0
41
+ noise_s: 0.999
42
+ num_inference_timesteps: 4
43
+ num_target_vision_tokens: 32
44
+ num_timestep_buckets: 1000
45
+ state_dim: 7
46
+ name: QwenPI_v3
47
+ qwenvl:
48
+ attn_implementation: flash_attention_2
49
+ base_vlm: /workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
50
+ enable_gradient_checkpointing: true
51
+ num_vl_layers: 36
52
+ vl_hidden_dim: 2560
53
+ output_dir: /workspace/starVLA/results/Checkpoints/pi05_deadly_corridor_fix_latency_0_1000ep
54
+ rl_games:
55
+ env_eval:
56
+ deadly:
57
+ action_layout: multibinary_7
58
+ enabled: true
59
+ mid_train:
60
+ enabled: false
61
+ interval_steps: 100
62
+ task: deadly_corridor
63
+ run_id: pi05_deadly_corridor_fix_latency_0_1000ep
64
+ run_root_dir: /workspace/starVLA/results/Checkpoints
65
+ seed: 42
66
+ trainer:
67
+ cc_f1_tolerance: 1
68
+ distributed_backend: none
69
+ eval_action_classification: true
70
+ eval_action_classification_interval: null
71
+ eval_interval: 100
72
+ eval_num_batches: 100
73
+ freeze_modules: ''
74
+ gradient_accumulation_steps: 8
75
+ gradient_clipping: 1.0
76
+ is_resume: false
77
+ learning_rate:
78
+ action_model: 5.0e-06
79
+ base: 5.0e-06
80
+ qwen_vl_interface: 1.0e-05
81
+ logging_frequency: 1
82
+ lr_scheduler_type: cosine_with_min_lr
83
+ max_train_steps: 500
84
+ num_warmup_steps: 100
85
+ optimizer:
86
+ betas:
87
+ - 0.9
88
+ - 0.95
89
+ eps: 1.0e-08
90
+ fused: false
91
+ weight_decay: 1.0e-08
92
+ per_latency_eval_num_batches: null
93
+ pretrained_checkpoint: null
94
+ reload_modules: null
95
+ save_interval: 100
96
+ scheduler_specific_kwargs:
97
+ min_lr: 1.0e-06
98
+ wandb_entity: dongqianyu99-zhejiang-university
99
+ wandb_project: starVLA_rl_games
pi05_deadly_corridor_fix_latency_0_1000ep/dataset_statistics.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.924872100353241,
6
+ 0.0039678397588431835,
7
+ 0.16351675987243652,
8
+ 0.6617416739463806,
9
+ 0.23405033349990845,
10
+ 0.11331836879253387,
11
+ 0.5559674501419067
12
+ ],
13
+ "std": [
14
+ 0.26355794072151184,
15
+ 0.06286244094371796,
16
+ 0.36991649866104126,
17
+ 0.47308459877967834,
18
+ 0.42337605357170105,
19
+ 0.3170303404331207,
20
+ 0.49692174792289734
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": 38308,
125
+ "num_trajectories": 900
126
+ }
127
+ }
pi05_deadly_corridor_fix_latency_0_1000ep/dataset_statistics_eval.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.9250290989875793,
6
+ 0.004190919455140829,
7
+ 0.1720605343580246,
8
+ 0.6565774083137512,
9
+ 0.23189756274223328,
10
+ 0.11525028944015503,
11
+ 0.5634458661079407
12
+ ],
13
+ "std": [
14
+ 0.26333996653556824,
15
+ 0.06459999084472656,
16
+ 0.37743079662323,
17
+ 0.4748457670211792,
18
+ 0.4220407009124756,
19
+ 0.319315642118454,
20
+ 0.4959460198879242
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": 4295,
125
+ "num_trajectories": 100
126
+ }
127
+ }
pi05_deadly_corridor_fix_latency_0_1000ep/hydra/.hydra/config.yaml ADDED
@@ -0,0 +1,231 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: l1
9
+ action_horizon: 1
10
+ future_action_window_size: 0
11
+ past_action_window_size: 0
12
+ action_model_type: LayerwiseFM
13
+ action_dim: 7
14
+ action_env_dim: 7
15
+ repeated_diffusion_steps: 2
16
+ num_inference_timesteps: 4
17
+ num_target_vision_tokens: 32
18
+ add_pos_embed: true
19
+ max_seq_len: 1024
20
+ noise_beta_alpha: 1.5
21
+ noise_beta_beta: 1.0
22
+ noise_s: 0.999
23
+ num_timestep_buckets: 1000
24
+ diffusion_model_cfg:
25
+ action_dit_hidden_dim: 1024
26
+ output_dim: 1024
27
+ dropout: 0.2
28
+ final_dropout: true
29
+ interleave_self_attention: true
30
+ norm_type: ada_norm
31
+ positional_embeddings: null
32
+ attention_head_dim: 64
33
+ name: QwenPI_v3
34
+ datasets:
35
+ vla_data:
36
+ dataset_py: lerobot_datasets
37
+ include_state: true
38
+ data_root_dir: /workspace/starVLA/data/deadly_corridor_fix_latency_0_1000ep
39
+ data_mix: deadly_corridor_train__bridge
40
+ eval_data_mix: deadly_corridor_train__bridge__val
41
+ custom_mixtures_path: null
42
+ action_type: discrete
43
+ sequential_step_sampling: false
44
+ eval_sequential_step_sampling: null
45
+ num_workers: 4
46
+ eval_num_workers: null
47
+ shuffle: true
48
+ action_balance:
49
+ enabled: false
50
+ strategy: balanced_epoch
51
+ action_key: action_id
52
+ target_flap_fraction: 0.3
53
+ noop_id: 0
54
+ flap_id: 1
55
+ latency_curriculum:
56
+ enabled: false
57
+ strategy: exclusive
58
+ latencies: null
59
+ phase_steps: null
60
+ per_device_batch_size: 16
61
+ load_all_data_for_training: true
62
+ obs_image_size: null
63
+ video_backend: torchvision_av
64
+ dataset:
65
+ source_hf: ${dataset.single_source_hf}
66
+ config_name: null
67
+ source_subdir: null
68
+ converted_name: ${dataset.single_converted_name}
69
+ single_source_hf: ''
70
+ mixed_source_hf: ''
71
+ single_converted_name: deadly_corridor_train
72
+ mixed_converted_name: deadly_corridor_mixed_latency_train
73
+ single_latency_filter: null
74
+ mixed_latency_filter: null
75
+ force_download: false
76
+ setup_force: false
77
+ skip_verification: false
78
+ verify_rows: 200
79
+ max_episodes: null
80
+ episodes_per_latency: null
81
+ latency_filter: ${dataset.single_latency_filter}
82
+ debug_subset:
83
+ enabled: false
84
+ max_episodes: 5
85
+ suffix: debug
86
+ base_model:
87
+ repo_id: ${bridge_base_model.repo_id.${model}}
88
+ initialization:
89
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-PI_v3-Bridge-RT_1
90
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-PI_v3-Bridge-RT_1
91
+ checkpoint_filename: checkpoints/steps_50000_pytorch_model.pt
92
+ trainer:
93
+ max_train_steps: 500
94
+ num_warmup_steps: 100
95
+ save_interval: 100
96
+ eval_interval: 100
97
+ eval_num_batches: 100
98
+ per_latency_eval_num_batches: null
99
+ eval_action_classification: true
100
+ eval_action_classification_interval: null
101
+ cc_f1_tolerance: 1
102
+ learning_rate:
103
+ base: 5.0e-06
104
+ qwen_vl_interface: 1.0e-05
105
+ action_model: 5.0e-06
106
+ lr_scheduler_type: cosine_with_min_lr
107
+ scheduler_specific_kwargs:
108
+ min_lr: 1.0e-06
109
+ freeze_modules: ''
110
+ loss_scale:
111
+ vla: 1.0
112
+ vlm: 0.1
113
+ max_grad_norm: 1.0
114
+ weight_decay: 0.0
115
+ logging_frequency: 1
116
+ gradient_clipping: 1.0
117
+ gradient_accumulation_steps: 8
118
+ distributed_backend: none
119
+ is_resume: false
120
+ pretrained_checkpoint: /workspace/starVLA/playground/Pretrained_models/Qwen3VL-PI_v3-Bridge-RT_1/checkpoints/steps_50000_pytorch_model.pt
121
+ resume_step: 0
122
+ reload_modules: null
123
+ optimizer:
124
+ name: AdamW
125
+ betas:
126
+ - 0.9
127
+ - 0.95
128
+ eps: 1.0e-08
129
+ weight_decay: 1.0e-08
130
+ fused: false
131
+ save_format: pt
132
+ workspace_dir: WORKSPACE_DIR
133
+ run_root_dir: /workspace/starVLA/results/Checkpoints
134
+ seed: 42
135
+ wandb_entity: dongqianyu99-zhejiang-university
136
+ wandb_project: starVLA_rl_games
137
+ auth:
138
+ env_file: null
139
+ hf_token_env: HF_TOKEN
140
+ wandb_api_key_env: WANDB_API_KEY
141
+ paths:
142
+ run_root_dir: results/Checkpoints
143
+ dataset_local_dir: playground/Datasets/rl_games
144
+ dataset_cache_dir: null
145
+ base_model_dir: ${bridge_base_model.local_dir.${model}}
146
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
147
+ launch:
148
+ use_accelerate: true
149
+ gpus: null
150
+ num_processes: 1
151
+ dry_run: false
152
+ conda:
153
+ enabled: true
154
+ env_name: null
155
+ rl_games:
156
+ model_alias: pi-0.5
157
+ env_eval:
158
+ image_size: 224
159
+ frameskip: 4
160
+ seed: 42
161
+ fixed_episode_seeds: true
162
+ latency_seed_stride: 0
163
+ task_seed_stride: 0
164
+ task_description: You are playing Deadly Corridor in VizDoom. Choose actions from
165
+ MOVE_FORWARD, MOVE_BACKWARD, MOVE_LEFT, MOVE_RIGHT, TURN_LEFT, TURN_RIGHT, ATTACK.
166
+ deadly:
167
+ action_layout: multibinary_7
168
+ enabled: true
169
+ distributed_mode: none
170
+ vectorized:
171
+ enabled: false
172
+ batch_size: 1
173
+ latency:
174
+ prompt_map_path: /workspace/starVLA/data/deadly_corridor_fix_latency_0_1000ep/deadly_corridor_train__bridge/latency_prompt_map.json
175
+ mode: single
176
+ values:
177
+ - 0
178
+ mid_train:
179
+ enabled: false
180
+ interval_steps: 100
181
+ latencies:
182
+ - 0
183
+ num_episodes: 5
184
+ max_steps_per_episode: 3600
185
+ post_train:
186
+ enabled: false
187
+ latencies:
188
+ - 0
189
+ - 1
190
+ - 2
191
+ - 3
192
+ - 4
193
+ num_episodes: 5
194
+ max_steps_per_episode: 3600
195
+ task: deadly_corridor
196
+ deadly_corridor_loss_type: null
197
+ initialization_mode: bridge
198
+ action_carrier: bridge
199
+ model: pi05
200
+ env: deadly_corridor
201
+ init: bridge
202
+ bridge_base_model:
203
+ repo_id:
204
+ openvla: Qwen/Qwen3-VL-4B-Instruct
205
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
206
+ pi05: Qwen/Qwen3-VL-4B-Instruct
207
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
208
+ local_dir:
209
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
210
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
211
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
212
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
213
+ mode: single
214
+ checkpoint:
215
+ load: auto
216
+ hf_repo_id: null
217
+ save_best_model: false
218
+ save_pt_file: false
219
+ local:
220
+ keep_last_n: 1
221
+ sync:
222
+ enabled: false
223
+ repo_id: null
224
+ keep_last_n: 0
225
+ sync_every_n_checkpoints: 1
226
+ resume_policy: local_latest
227
+ run_id: pi05_deadly_corridor_fix_latency_0_1000ep
228
+ output_dir: null
229
+ config_yaml: null
230
+ is_debug: false
231
+ version_id: '0.21'
pi05_deadly_corridor_fix_latency_0_1000ep/hydra/.hydra/hydra.yaml ADDED
@@ -0,0 +1,277 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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=pi05
116
+ - env=deadly_corridor
117
+ - init=bridge
118
+ - mode=single
119
+ - run_id=pi05_deadly_corridor_fix_latency_0_1000ep
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-PI_v3-Bridge-RT_1/checkpoints/steps_50000_pytorch_model.pt
132
+ - trainer.resume_step=0
133
+ - trainer.max_train_steps=500
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=8
140
+ - trainer.distributed_backend=none
141
+ - trainer.learning_rate.base=5e-06
142
+ - trainer.learning_rate.qwen_vl_interface=1e-05
143
+ - trainer.learning_rate.action_model=5e-06
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=QwenPI_v3
158
+ - framework.qwenvl.attn_implementation=flash_attention_2
159
+ - framework.qwenvl.enable_gradient_checkpointing=true
160
+ - framework.action_model.action_model_type=LayerwiseFM
161
+ - framework.action_model.action_dim=7
162
+ - framework.action_model.action_env_dim=7
163
+ - framework.action_model.state_dim=7
164
+ - framework.action_model.loss_type=l1
165
+ - framework.action_model.action_horizon=1
166
+ - framework.action_model.future_action_window_size=0
167
+ - framework.action_model.past_action_window_size=0
168
+ - framework.action_model.repeated_diffusion_steps=2
169
+ - framework.action_model.num_inference_timesteps=4
170
+ - framework.action_model.num_target_vision_tokens=32
171
+ - framework.action_model.add_pos_embed=true
172
+ - framework.action_model.max_seq_len=1024
173
+ - framework.action_model.noise_beta_alpha=1.5
174
+ - framework.action_model.noise_beta_beta=1.0
175
+ - framework.action_model.noise_s=0.999
176
+ - framework.action_model.num_timestep_buckets=1000
177
+ - framework.action_model.diffusion_model_cfg.action_dit_hidden_dim=1024
178
+ - framework.action_model.diffusion_model_cfg.dropout=0.2
179
+ - framework.action_model.diffusion_model_cfg.final_dropout=true
180
+ - framework.action_model.diffusion_model_cfg.interleave_self_attention=true
181
+ - framework.action_model.diffusion_model_cfg.norm_type=ada_norm
182
+ - framework.action_model.diffusion_model_cfg.output_dim=1024
183
+ - framework.action_model.diffusion_model_cfg.attention_head_dim=64
184
+ - datasets.vla_data.include_state=true
185
+ - datasets.vla_data.action_type=discrete
186
+ - datasets.vla_data.sequential_step_sampling=false
187
+ - datasets.vla_data.shuffle=true
188
+ - datasets.vla_data.action_balance.enabled=false
189
+ - datasets.vla_data.action_balance.strategy=balanced_epoch
190
+ - datasets.vla_data.action_balance.action_key=action_id
191
+ - datasets.vla_data.action_balance.target_flap_fraction=0.3
192
+ - datasets.vla_data.action_balance.noop_id=0
193
+ - datasets.vla_data.action_balance.flap_id=1
194
+ - datasets.vla_data.latency_curriculum.enabled=false
195
+ - datasets.vla_data.latency_curriculum.strategy=exclusive
196
+ - datasets.vla_data.per_device_batch_size=16
197
+ - datasets.vla_data.load_all_data_for_training=true
198
+ - datasets.vla_data.video_backend=torchvision_av
199
+ - datasets.vla_data.data_root_dir=/workspace/starVLA/data/deadly_corridor_fix_latency_0_1000ep
200
+ - datasets.vla_data.data_mix=deadly_corridor_train__bridge
201
+ - datasets.vla_data.eval_data_mix=deadly_corridor_train__bridge__val
202
+ - framework.qwenvl.base_vlm=/workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
203
+ - rl_games.task=deadly_corridor
204
+ - rl_games.model_alias=pi-0.5
205
+ - rl_games.initialization_mode=bridge
206
+ - rl_games.action_carrier=bridge
207
+ - rl_games.env_eval.distributed_mode=none
208
+ - rl_games.env_eval.latency.mode=single
209
+ - rl_games.env_eval.frameskip=4
210
+ - rl_games.env_eval.image_size=224
211
+ - rl_games.env_eval.seed=42
212
+ - rl_games.env_eval.fixed_episode_seeds=true
213
+ - rl_games.env_eval.latency_seed_stride=0
214
+ - rl_games.env_eval.task_seed_stride=0
215
+ - rl_games.env_eval.task_description='You are playing Deadly Corridor in VizDoom.
216
+ Choose actions from MOVE_FORWARD, MOVE_BACKWARD, MOVE_LEFT, MOVE_RIGHT, TURN_LEFT,
217
+ TURN_RIGHT, ATTACK.'
218
+ - rl_games.env_eval.latency.values=[0]
219
+ - rl_games.env_eval.latency.prompt_map_path=/workspace/starVLA/data/deadly_corridor_fix_latency_0_1000ep/deadly_corridor_train__bridge/latency_prompt_map.json
220
+ - rl_games.env_eval.mid_train.enabled=false
221
+ - rl_games.env_eval.mid_train.interval_steps=100
222
+ - rl_games.env_eval.mid_train.num_episodes=5
223
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
224
+ - rl_games.env_eval.mid_train.latencies=[0]
225
+ - rl_games.env_eval.post_train.enabled=false
226
+ - rl_games.env_eval.post_train.num_episodes=5
227
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
228
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
229
+ - rl_games.env_eval.deadly.action_layout=multibinary_7
230
+ job:
231
+ name: train_starvla_hydra
232
+ chdir: false
233
+ 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=deadly_corridor_train__bridge,datasets.vla_data.data_root_dir=/workspace/starVLA/data/deadly_corridor_fix_latency_0_1000ep,datasets.vla_data.eval_data_mix=deadly_corridor_train__bridge__val,datasets.vla_data.include_state=true,datasets.vla_data.latency_curriculum.enabled=false,datasets.vla_data.latency_curriculum.strategy=exclusive,datasets.vla_data.load_all_data_for_training=true,datasets.vla_data.per_device_batch_size=16,datasets.vla_data.sequential_step_sampling=false,datasets.vla_data.shuffle=true,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.action_model_type=LayerwiseFM,framework.action_model.add_pos_embed=true,framework.action_model.diffusion_model_cfg.action_dit_hidden_dim=1024,framework.action_model.diffusion_model_cfg.attention_head_dim=64,framework.action_model.diffusion_model_cfg.dropout=0.2,framework.action_model.diffusion_model_cfg.final_dropout=true,framework.action_model.diffusion_model_cfg.interleave_self_attention=true,framework.action_model.diffusion_model_cfg.norm_type=ada_norm,framework.action_model.diffusion_model_cfg.output_dim=1024,framework.action_model.future_action_window_size=0,framework.action_model.loss_type=l1,framework.action_model.max_seq_len=1024,framework.action_model.noise_beta_alpha=1.5,framework.action_model.noise_beta_beta=1.0,framework.action_model.noise_s=0.999,framework.action_model.num_inference_timesteps=4,framework.action_model.num_target_vision_tokens=32,framework.action_model.num_timestep_buckets=1000,framework.action_model.past_action_window_size=0,framework.action_model.repeated_diffusion_steps=2,framework.action_model.state_dim=7,framework.name=QwenPI_v3,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=pi05,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=/workspace/starVLA/data/deadly_corridor_fix_latency_0_1000ep/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=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_description='You
234
+ are playing Deadly Corridor in VizDoom. Choose actions from MOVE_FORWARD, MOVE_BACKWARD,
235
+ 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=pi-0.5,rl_games.task=deadly_corridor,run_id=pi05_deadly_corridor_fix_latency_0_1000ep,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=8,trainer.gradient_clipping=1.0,trainer.is_resume=false,trainer.learning_rate.action_model=5e-06,trainer.learning_rate.base=5e-06,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,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-PI_v3-Bridge-RT_1/checkpoints/steps_50000_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
236
+ id: ???
237
+ num: ???
238
+ config_name: train
239
+ env_set: {}
240
+ env_copy: []
241
+ config:
242
+ override_dirname:
243
+ kv_sep: '='
244
+ item_sep: ','
245
+ exclude_keys: []
246
+ runtime:
247
+ version: 1.3.3
248
+ version_base: '1.1'
249
+ cwd: /workspace/starVLA
250
+ config_sources:
251
+ - path: hydra.conf
252
+ schema: pkg
253
+ provider: hydra
254
+ - path: /workspace/starVLA/examples/rl_games/config
255
+ schema: file
256
+ provider: main
257
+ - path: ''
258
+ schema: structured
259
+ provider: schema
260
+ output_dir: /workspace/starVLA/results/Checkpoints/pi05_deadly_corridor_fix_latency_0_1000ep/hydra
261
+ choices:
262
+ cross_task_setup: null
263
+ checkpoint: default
264
+ mode: single
265
+ init: bridge
266
+ env: deadly_corridor
267
+ model: pi05
268
+ hydra/env: default
269
+ hydra/callbacks: null
270
+ hydra/job_logging: default
271
+ hydra/hydra_logging: default
272
+ hydra/hydra_help: default
273
+ hydra/help: default
274
+ hydra/sweeper: basic
275
+ hydra/launcher: basic
276
+ hydra/output: default
277
+ verbose: false
pi05_deadly_corridor_fix_latency_0_1000ep/hydra/.hydra/overrides.yaml ADDED
@@ -0,0 +1,115 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ - model=pi05
2
+ - env=deadly_corridor
3
+ - init=bridge
4
+ - mode=single
5
+ - run_id=pi05_deadly_corridor_fix_latency_0_1000ep
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-PI_v3-Bridge-RT_1/checkpoints/steps_50000_pytorch_model.pt
18
+ - trainer.resume_step=0
19
+ - trainer.max_train_steps=500
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=8
26
+ - trainer.distributed_backend=none
27
+ - trainer.learning_rate.base=5e-06
28
+ - trainer.learning_rate.qwen_vl_interface=1e-05
29
+ - trainer.learning_rate.action_model=5e-06
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=QwenPI_v3
44
+ - framework.qwenvl.attn_implementation=flash_attention_2
45
+ - framework.qwenvl.enable_gradient_checkpointing=true
46
+ - framework.action_model.action_model_type=LayerwiseFM
47
+ - framework.action_model.action_dim=7
48
+ - framework.action_model.action_env_dim=7
49
+ - framework.action_model.state_dim=7
50
+ - framework.action_model.loss_type=l1
51
+ - framework.action_model.action_horizon=1
52
+ - framework.action_model.future_action_window_size=0
53
+ - framework.action_model.past_action_window_size=0
54
+ - framework.action_model.repeated_diffusion_steps=2
55
+ - framework.action_model.num_inference_timesteps=4
56
+ - framework.action_model.num_target_vision_tokens=32
57
+ - framework.action_model.add_pos_embed=true
58
+ - framework.action_model.max_seq_len=1024
59
+ - framework.action_model.noise_beta_alpha=1.5
60
+ - framework.action_model.noise_beta_beta=1.0
61
+ - framework.action_model.noise_s=0.999
62
+ - framework.action_model.num_timestep_buckets=1000
63
+ - framework.action_model.diffusion_model_cfg.action_dit_hidden_dim=1024
64
+ - framework.action_model.diffusion_model_cfg.dropout=0.2
65
+ - framework.action_model.diffusion_model_cfg.final_dropout=true
66
+ - framework.action_model.diffusion_model_cfg.interleave_self_attention=true
67
+ - framework.action_model.diffusion_model_cfg.norm_type=ada_norm
68
+ - framework.action_model.diffusion_model_cfg.output_dim=1024
69
+ - framework.action_model.diffusion_model_cfg.attention_head_dim=64
70
+ - datasets.vla_data.include_state=true
71
+ - datasets.vla_data.action_type=discrete
72
+ - datasets.vla_data.sequential_step_sampling=false
73
+ - datasets.vla_data.shuffle=true
74
+ - datasets.vla_data.action_balance.enabled=false
75
+ - datasets.vla_data.action_balance.strategy=balanced_epoch
76
+ - datasets.vla_data.action_balance.action_key=action_id
77
+ - datasets.vla_data.action_balance.target_flap_fraction=0.3
78
+ - datasets.vla_data.action_balance.noop_id=0
79
+ - datasets.vla_data.action_balance.flap_id=1
80
+ - datasets.vla_data.latency_curriculum.enabled=false
81
+ - datasets.vla_data.latency_curriculum.strategy=exclusive
82
+ - datasets.vla_data.per_device_batch_size=16
83
+ - datasets.vla_data.load_all_data_for_training=true
84
+ - datasets.vla_data.video_backend=torchvision_av
85
+ - datasets.vla_data.data_root_dir=/workspace/starVLA/data/deadly_corridor_fix_latency_0_1000ep
86
+ - datasets.vla_data.data_mix=deadly_corridor_train__bridge
87
+ - datasets.vla_data.eval_data_mix=deadly_corridor_train__bridge__val
88
+ - framework.qwenvl.base_vlm=/workspace/starVLA/playground/Pretrained_models/Qwen3-VL-4B-Instruct
89
+ - rl_games.task=deadly_corridor
90
+ - rl_games.model_alias=pi-0.5
91
+ - rl_games.initialization_mode=bridge
92
+ - rl_games.action_carrier=bridge
93
+ - rl_games.env_eval.distributed_mode=none
94
+ - rl_games.env_eval.latency.mode=single
95
+ - rl_games.env_eval.frameskip=4
96
+ - rl_games.env_eval.image_size=224
97
+ - rl_games.env_eval.seed=42
98
+ - rl_games.env_eval.fixed_episode_seeds=true
99
+ - rl_games.env_eval.latency_seed_stride=0
100
+ - rl_games.env_eval.task_seed_stride=0
101
+ - rl_games.env_eval.task_description='You are playing Deadly Corridor in VizDoom.
102
+ Choose actions from MOVE_FORWARD, MOVE_BACKWARD, MOVE_LEFT, MOVE_RIGHT, TURN_LEFT,
103
+ TURN_RIGHT, ATTACK.'
104
+ - rl_games.env_eval.latency.values=[0]
105
+ - rl_games.env_eval.latency.prompt_map_path=/workspace/starVLA/data/deadly_corridor_fix_latency_0_1000ep/deadly_corridor_train__bridge/latency_prompt_map.json
106
+ - rl_games.env_eval.mid_train.enabled=false
107
+ - rl_games.env_eval.mid_train.interval_steps=100
108
+ - rl_games.env_eval.mid_train.num_episodes=5
109
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
110
+ - rl_games.env_eval.mid_train.latencies=[0]
111
+ - rl_games.env_eval.post_train.enabled=false
112
+ - rl_games.env_eval.post_train.num_episodes=5
113
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
114
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
115
+ - rl_games.env_eval.deadly.action_layout=multibinary_7
pi05_deadly_corridor_fix_latency_0_1000ep/hydra/train_starvla_hydra.log ADDED
The diff for this file is too large to render. See raw diff
 
pi05_deadly_corridor_fix_latency_0_1000ep/summary.jsonl ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {"steps": 100}
2
+ {"steps": 200}
3
+ {"steps": 300}
4
+ {"steps": 400}
5
+ {"steps": 500}