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

Consolidate paper experiment assets by task and run (151-300)

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/small-policy/memory_rollout_teacher:deadly_corridor:teacher:fixed_l6:fs4:obs8p75:stride4:seed0/train/memory_rollout_teacher:deadly_corridor:teacher:fixed_l6:fs4:obs8p75:stride4:seed0/checkpoint_p0/checkpoint_000048836_25004032.pth +3 -0
  2. latency-aware/deadly-corridor/small-policy/memory_rollout_teacher:deadly_corridor:teacher:fixed_l6:fs4:obs8p75:stride4:seed0/train/memory_rollout_teacher:deadly_corridor:teacher:fixed_l6:fs4:obs8p75:stride4:seed0/config.json +260 -0
  3. latency-aware/deadly-corridor/vla/starvla-wanoft-h8/wan_oft_deadly_corridor_fix_latency_6_context5_standard_sft_1000_effbs128_224_currentbce/checkpoints/steps_1000_pytorch_model.pt +3 -0
  4. latency-aware/deadly-corridor/vla/starvla-wanoft-h8/wan_oft_deadly_corridor_fix_latency_6_context5_standard_sft_1000_effbs128_224_currentbce/config.full.yaml +286 -0
  5. latency-aware/deadly-corridor/vla/starvla-wanoft-h8/wan_oft_deadly_corridor_fix_latency_6_context5_standard_sft_1000_effbs128_224_currentbce/config.yaml +83 -0
  6. latency-aware/deadly-corridor/vla/starvla-wanoft-h8/wan_oft_deadly_corridor_fix_latency_6_context5_standard_sft_1000_effbs128_224_currentbce/dataset_statistics.json +127 -0
  7. latency-aware/demon-attack/small-policy/memory_rollout_teacher:demon_attack:teacher:fixed_l6:fs4:obs15:stride4:seed0/train/memory_rollout_teacher:demon_attack:teacher:fixed_l6:fs4:obs15:stride4:seed0/checkpoint_p0/checkpoint_000048880_25026560.pth +3 -0
  8. latency-aware/demon-attack/small-policy/memory_rollout_teacher:demon_attack:teacher:fixed_l6:fs4:obs15:stride4:seed0/train/memory_rollout_teacher:demon_attack:teacher:fixed_l6:fs4:obs15:stride4:seed0/config.json +255 -0
  9. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/checkpoints/steps_7000_model.safetensors +3 -0
  10. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/config.full.yaml +209 -0
  11. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/config.yaml +75 -0
  12. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/dataset_statistics.json +127 -0
  13. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/dataset_statistics_eval.json +127 -0
  14. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/hydra/.hydra/config.yaml +207 -0
  15. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/hydra/.hydra/hydra.yaml +259 -0
  16. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/hydra/.hydra/overrides.yaml +97 -0
  17. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/hydra/train_starvla_hydra.log +0 -0
  18. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/summary.jsonl +70 -0
  19. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/checkpoints/steps_7000_model.safetensors +3 -0
  20. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/config.full.yaml +209 -0
  21. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/config.yaml +75 -0
  22. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/dataset_statistics.json +127 -0
  23. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/dataset_statistics_eval.json +127 -0
  24. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/hydra/.hydra/config.yaml +207 -0
  25. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/hydra/.hydra/hydra.yaml +259 -0
  26. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/hydra/.hydra/overrides.yaml +97 -0
  27. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/hydra/train_starvla_hydra.log +0 -0
  28. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/summary.jsonl +70 -0
  29. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/checkpoints/steps_7000_model.safetensors +3 -0
  30. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/config.full.yaml +209 -0
  31. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/config.yaml +75 -0
  32. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/dataset_statistics.json +127 -0
  33. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/dataset_statistics_eval.json +127 -0
  34. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/hydra/.hydra/config.yaml +207 -0
  35. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/hydra/.hydra/hydra.yaml +259 -0
  36. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/hydra/.hydra/overrides.yaml +97 -0
  37. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/hydra/train_starvla_hydra.log +0 -0
  38. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/summary.jsonl +70 -0
  39. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/checkpoints/steps_7000_model.safetensors +3 -0
  40. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/config.full.yaml +209 -0
  41. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/config.yaml +75 -0
  42. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/dataset_statistics.json +127 -0
  43. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/dataset_statistics_eval.json +127 -0
  44. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/hydra/.hydra/config.yaml +207 -0
  45. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/hydra/.hydra/hydra.yaml +259 -0
  46. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/hydra/.hydra/overrides.yaml +97 -0
  47. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/hydra/train_starvla_hydra.log +0 -0
  48. latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/summary.jsonl +70 -0
  49. latency-aware/demon-attack/vla/starvla-wanoft-h8/wan_oft_demon_attack_fix_latency_6_context5_standard_sft_5000_effbs128_224_currentce/checkpoints/steps_5000_pytorch_model.pt +3 -0
  50. latency-aware/demon-attack/vla/starvla-wanoft-h8/wan_oft_demon_attack_fix_latency_6_context5_standard_sft_5000_effbs128_224_currentce/config.full.yaml +280 -0
latency-aware/deadly-corridor/small-policy/memory_rollout_teacher:deadly_corridor:teacher:fixed_l6:fs4:obs8p75:stride4:seed0/train/memory_rollout_teacher:deadly_corridor:teacher:fixed_l6:fs4:obs8p75:stride4:seed0/checkpoint_p0/checkpoint_000048836_25004032.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6786d84ce663078e7cf02e7032035c4d899843fe06ca3d6d07e666c8d606644f
3
+ size 13352656
latency-aware/deadly-corridor/small-policy/memory_rollout_teacher:deadly_corridor:teacher:fixed_l6:fs4:obs8p75:stride4:seed0/train/memory_rollout_teacher:deadly_corridor:teacher:fixed_l6:fs4:obs8p75:stride4:seed0/config.json ADDED
@@ -0,0 +1,260 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "help": false,
3
+ "algo": "APPO",
4
+ "env": "doom_deadly_corridor",
5
+ "experiment": "deadly_corridor_teacher_fixed_l6_obs8p75_seed0",
6
+ "train_dir": "results/checkpoints/memory/rollout_teacher",
7
+ "restart_behavior": "resume",
8
+ "device": "gpu",
9
+ "seed": 0,
10
+ "num_policies": 1,
11
+ "async_rl": true,
12
+ "serial_mode": false,
13
+ "batched_sampling": false,
14
+ "num_batches_to_accumulate": 2,
15
+ "worker_num_splits": 2,
16
+ "policy_workers_per_policy": 1,
17
+ "max_policy_lag": 1000,
18
+ "num_workers": 32,
19
+ "num_envs_per_worker": 4,
20
+ "batch_size": 1024,
21
+ "num_batches_per_epoch": 2,
22
+ "num_epochs": 2,
23
+ "rollout": 128,
24
+ "recurrence": 128,
25
+ "shuffle_minibatches": false,
26
+ "gamma": 0.99,
27
+ "reward_scale": 1.0,
28
+ "reward_clip": 1000.0,
29
+ "value_bootstrap": false,
30
+ "normalize_returns": true,
31
+ "exploration_loss_coeff": 0.001,
32
+ "value_loss_coeff": 0.5,
33
+ "kl_loss_coeff": 0.0,
34
+ "exploration_loss": "symmetric_kl",
35
+ "gae_lambda": 0.95,
36
+ "ppo_clip_ratio": 0.1,
37
+ "ppo_clip_value": 0.2,
38
+ "with_vtrace": false,
39
+ "vtrace_rho": 1.0,
40
+ "vtrace_c": 1.0,
41
+ "optimizer": "adam",
42
+ "adam_eps": 1e-06,
43
+ "adam_beta1": 0.9,
44
+ "adam_beta2": 0.999,
45
+ "max_grad_norm": 4.0,
46
+ "learning_rate": 0.0001,
47
+ "lr_schedule": "constant",
48
+ "lr_schedule_kl_threshold": 0.008,
49
+ "lr_adaptive_min": 1e-06,
50
+ "lr_adaptive_max": 0.01,
51
+ "obs_subtract_mean": 0.0,
52
+ "obs_scale": 255.0,
53
+ "normalize_input": true,
54
+ "normalize_input_keys": null,
55
+ "decorrelate_experience_max_seconds": 0,
56
+ "decorrelate_envs_on_one_worker": true,
57
+ "actor_worker_gpus": [],
58
+ "set_workers_cpu_affinity": true,
59
+ "force_envs_single_thread": false,
60
+ "default_niceness": 0,
61
+ "log_to_file": true,
62
+ "experiment_summaries_interval": 1,
63
+ "flush_summaries_interval": 30,
64
+ "stats_avg": 100,
65
+ "summaries_use_frameskip": true,
66
+ "heartbeat_interval": 20,
67
+ "heartbeat_reporting_interval": 600,
68
+ "train_for_env_steps": 25000000,
69
+ "train_for_seconds": 10000000000,
70
+ "save_every_sec": 600,
71
+ "keep_checkpoints": 5,
72
+ "load_checkpoint_kind": "latest",
73
+ "save_milestones_sec": -1,
74
+ "save_best_every_sec": 5,
75
+ "save_best_metric": "reward",
76
+ "save_best_after": 100000,
77
+ "benchmark": false,
78
+ "encoder_mlp_layers": [
79
+ 512,
80
+ 512
81
+ ],
82
+ "encoder_conv_architecture": "convnet_simple",
83
+ "encoder_conv_mlp_layers": [
84
+ 512
85
+ ],
86
+ "use_rnn": true,
87
+ "rnn_size": 512,
88
+ "rnn_type": "gru",
89
+ "rnn_num_layers": 1,
90
+ "decoder_mlp_layers": [],
91
+ "nonlinearity": "elu",
92
+ "policy_initialization": "orthogonal",
93
+ "policy_init_gain": 1.0,
94
+ "actor_critic_share_weights": true,
95
+ "adaptive_stddev": true,
96
+ "continuous_tanh_scale": 0.0,
97
+ "initial_stddev": 1.0,
98
+ "use_env_info_cache": false,
99
+ "env_gpu_actions": false,
100
+ "env_gpu_observations": true,
101
+ "env_frameskip": 1,
102
+ "env_framestack": 1,
103
+ "pixel_format": "CHW",
104
+ "use_record_episode_statistics": false,
105
+ "with_wandb": true,
106
+ "wandb_user": null,
107
+ "wandb_project": "latency-sensitive-bench",
108
+ "wandb_group": "deadly-corridor-obs8p75-fixed_l6",
109
+ "wandb_job_type": "sample_factory",
110
+ "wandb_tags": [
111
+ "memory_rollout_teacher",
112
+ "deadly-corridor",
113
+ "teacher",
114
+ "fixed",
115
+ "fixed_l6",
116
+ "obs8p75",
117
+ "seed0"
118
+ ],
119
+ "with_pbt": false,
120
+ "pbt_mix_policies_in_one_env": true,
121
+ "pbt_period_env_steps": 5000000,
122
+ "pbt_start_mutation": 20000000,
123
+ "pbt_replace_fraction": 0.3,
124
+ "pbt_mutation_rate": 0.15,
125
+ "pbt_replace_reward_gap": 0.1,
126
+ "pbt_replace_reward_gap_absolute": 1e-06,
127
+ "pbt_optimize_gamma": false,
128
+ "pbt_target_objective": "true_objective",
129
+ "pbt_perturb_min": 1.1,
130
+ "pbt_perturb_max": 1.5,
131
+ "fps": 35,
132
+ "eval_env_frameskip": 1,
133
+ "no_render": false,
134
+ "save_video": false,
135
+ "video_frames": 1000000000.0,
136
+ "video_name": null,
137
+ "max_num_frames": 1000000000.0,
138
+ "max_num_episodes": 1000000000.0,
139
+ "push_to_hub": false,
140
+ "hf_repository": null,
141
+ "policy_index": 0,
142
+ "eval_deterministic": true,
143
+ "train_script": null,
144
+ "enjoy_script": null,
145
+ "num_agents": -1,
146
+ "num_humans": 0,
147
+ "num_bots": -1,
148
+ "start_bot_difficulty": null,
149
+ "timelimit": null,
150
+ "res_w": 128,
151
+ "res_h": 72,
152
+ "wide_aspect_ratio": false,
153
+ "record_to": null,
154
+ "mode": "train",
155
+ "env_fps": 35.0,
156
+ "obs_fps": 8.75,
157
+ "latency_type": "fixed",
158
+ "fixed_latency_ms": 171.42857142857144,
159
+ "mean_latency_ms": null,
160
+ "std_latency_ms": null,
161
+ "min_latency_ms": null,
162
+ "max_latency_ms": null,
163
+ "latency_seed": null,
164
+ "add_latency_info": false,
165
+ "max_pending_actions": null,
166
+ "hold_policy": "hold",
167
+ "ordering_policy": "latest_ready",
168
+ "eval_episodes": 100,
169
+ "eval_parallel_envs": 1,
170
+ "eval_latency_raw_frame_values": "0,1,2,3,4,5,10,15,40",
171
+ "eval_max_steps": 30000,
172
+ "eval_raw_reward": true,
173
+ "disable_reward_scaling": true,
174
+ "frame_stack": 4,
175
+ "max_episode_steps": 0,
176
+ "simulator": "cpu",
177
+ "episode_metrics_path": "results/checkpoints/memory/rollout_teacher/deadly_corridor_teacher_fixed_l6_obs8p75_seed0/episode_metrics.jsonl",
178
+ "vizdoom_init_max_parallel": 10,
179
+ "vizdoom_init_lock_timeout_sec": 5.0,
180
+ "vizdoom_init_lock_max_wait_sec": 60.0,
181
+ "vizdoom_init_retry_jitter_sec": 0.25,
182
+ "command_line": "--mode train --algo APPO --env doom_deadly_corridor --experiment deadly_corridor_teacher_fixed_l6_obs8p75_seed0 --train_dir results/checkpoints/memory/rollout_teacher --restart_behavior resume --device gpu --seed 0 --episode_metrics_path results/checkpoints/memory/rollout_teacher/deadly_corridor_teacher_fixed_l6_obs8p75_seed0/episode_metrics.jsonl --train_for_env_steps 25000000 --num_workers 32 --num_envs_per_worker 4 --num_policies 1 --batch_size 1024 --rollout 128 --recurrence 128 --num_epochs 2 --num_batches_per_epoch 2 --worker_num_splits 2 --learning_rate 0.0001 --gamma 0.99 --gae_lambda 0.95 --ppo_clip_ratio 0.1 --ppo_clip_value 0.2 --exploration_loss symmetric_kl --exploration_loss_coeff 0.001 --value_loss_coeff 0.5 --max_grad_norm 4.0 --rnn_type gru --rnn_size 512 --save_every_sec 600 --keep_checkpoints 5 --stats_avg 100 --experiment_summaries_interval 1 --async_rl True --use_rnn True --normalize_returns True --normalize_input True --latency-type fixed --fixed-latency-ms 171.42857142857144 --add-latency-info False --eval-episodes 100 --eval-parallel-envs 1 --eval-max-steps 30000 --eval-deterministic True --eval-raw-reward --with_wandb True --wandb_project latency-sensitive-bench --wandb_group deadly-corridor-obs8p75-fixed_l6 --wandb_job_type sample_factory --wandb_tags memory_rollout_teacher deadly-corridor teacher fixed fixed_l6 obs8p75 seed0 --env-fps 35 --obs-fps 8.75 --res_w 128 --res_h 72 --wide_aspect_ratio False --frame_stack 4 --simulator cpu --disable-reward-scaling True --hold-policy hold --ordering-policy latest_ready",
183
+ "cli_args": {
184
+ "algo": "APPO",
185
+ "env": "doom_deadly_corridor",
186
+ "experiment": "deadly_corridor_teacher_fixed_l6_obs8p75_seed0",
187
+ "train_dir": "results/checkpoints/memory/rollout_teacher",
188
+ "restart_behavior": "resume",
189
+ "device": "gpu",
190
+ "seed": 0,
191
+ "num_policies": 1,
192
+ "async_rl": true,
193
+ "worker_num_splits": 2,
194
+ "num_workers": 32,
195
+ "num_envs_per_worker": 4,
196
+ "batch_size": 1024,
197
+ "num_batches_per_epoch": 2,
198
+ "num_epochs": 2,
199
+ "rollout": 128,
200
+ "recurrence": 128,
201
+ "gamma": 0.99,
202
+ "normalize_returns": true,
203
+ "exploration_loss_coeff": 0.001,
204
+ "value_loss_coeff": 0.5,
205
+ "exploration_loss": "symmetric_kl",
206
+ "gae_lambda": 0.95,
207
+ "ppo_clip_ratio": 0.1,
208
+ "ppo_clip_value": 0.2,
209
+ "max_grad_norm": 4.0,
210
+ "learning_rate": 0.0001,
211
+ "normalize_input": true,
212
+ "experiment_summaries_interval": 1,
213
+ "stats_avg": 100,
214
+ "train_for_env_steps": 25000000,
215
+ "save_every_sec": 600,
216
+ "keep_checkpoints": 5,
217
+ "use_rnn": true,
218
+ "rnn_size": 512,
219
+ "rnn_type": "gru",
220
+ "with_wandb": true,
221
+ "wandb_project": "latency-sensitive-bench",
222
+ "wandb_group": "deadly-corridor-obs8p75-fixed_l6",
223
+ "wandb_job_type": "sample_factory",
224
+ "wandb_tags": [
225
+ "memory_rollout_teacher",
226
+ "deadly-corridor",
227
+ "teacher",
228
+ "fixed",
229
+ "fixed_l6",
230
+ "obs8p75",
231
+ "seed0"
232
+ ],
233
+ "eval_deterministic": true,
234
+ "res_w": 128,
235
+ "res_h": 72,
236
+ "wide_aspect_ratio": false,
237
+ "mode": "train",
238
+ "env_fps": 35.0,
239
+ "obs_fps": 8.75,
240
+ "latency_type": "fixed",
241
+ "fixed_latency_ms": 171.42857142857144,
242
+ "add_latency_info": false,
243
+ "hold_policy": "hold",
244
+ "ordering_policy": "latest_ready",
245
+ "eval_episodes": 100,
246
+ "eval_parallel_envs": 1,
247
+ "eval_max_steps": 30000,
248
+ "eval_raw_reward": true,
249
+ "disable_reward_scaling": true,
250
+ "frame_stack": 4,
251
+ "simulator": "cpu",
252
+ "episode_metrics_path": "results/checkpoints/memory/rollout_teacher/deadly_corridor_teacher_fixed_l6_obs8p75_seed0/episode_metrics.jsonl",
253
+ "env_frameskip": 1,
254
+ "eval_env_frameskip": 1
255
+ },
256
+ "git_hash": "bedb62f3a97309f6dacdb6a2953e3312d9776765",
257
+ "git_repo_name": "git@github.com:ZihanWang314/latency-sensitive-bench.git",
258
+ "output_dir": "outputs/memory/rollout_teacher/deadly_corridor/train/teacher/fixed_l6/obs8p75/seed_0",
259
+ "wandb_unique_id": "deadly-corridor-obs8p75-fixed_l6-s0"
260
+ }
latency-aware/deadly-corridor/vla/starvla-wanoft-h8/wan_oft_deadly_corridor_fix_latency_6_context5_standard_sft_1000_effbs128_224_currentbce/checkpoints/steps_1000_pytorch_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a8bd1415990de5a8616a203b15d58f105dbcc57e5ac10ffff0ea70e2220d9c5c
3
+ size 25211531190
latency-aware/deadly-corridor/vla/starvla-wanoft-h8/wan_oft_deadly_corridor_fix_latency_6_context5_standard_sft_1000_effbs128_224_currentbce/config.full.yaml ADDED
@@ -0,0 +1,286 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ name: WanOFT
3
+ world_model:
4
+ base_wm: /lustre/fsw/portfolios/nvr/projects/nvr_lacr_llm/users/zihwang/wanoft_bench/third_party/starVLA/playground/Pretrained_models/Wan-AI/Wan2.2-TI2V-5B-Diffusers
5
+ extract_layers:
6
+ - -1
7
+ num_frames: 5
8
+ qwenvl:
9
+ base_vlm: /lustre/fsw/portfolios/nvr/projects/nvr_lacr_llm/users/zihwang/wanoft_bench/third_party/starVLA/playground/Pretrained_models/Wan-AI/Wan2.2-TI2V-5B-Diffusers
10
+ attn_implementation: sdpa
11
+ flex_backend: triton
12
+ enable_gradient_checkpointing: true
13
+ action_model:
14
+ action_model_type: MLP
15
+ action_dim: 7
16
+ action_hidden_dim: 3072
17
+ action_horizon: 8
18
+ future_action_window_size: 7
19
+ past_action_window_size: 0
20
+ loss_type: current_multibinary_bce
21
+ class_weights: null
22
+ future_loss_weight: null
23
+ action_query_source: mean
24
+ action_query_num_heads: 24
25
+ state_dim: 7
26
+ action_env_dim: 7
27
+ kv_memory:
28
+ enabled: false
29
+ window: 4
30
+ rollout_len: 8
31
+ packed_train: false
32
+ rebased_sink: true
33
+ datasets:
34
+ vla_data:
35
+ dataset_py: lerobot_datasets
36
+ include_state: true
37
+ data_root_dir: /lustre/fsw/portfolios/nvr/projects/nvr_lacr_llm/users/zihwang/wanoft_bench/third_party/starVLA/data/deadly_corridor_fix_latency_6_1000ep_context5
38
+ data_mix: deadly_corridor_train__bridge
39
+ eval_data_mix: null
40
+ custom_mixtures_path: null
41
+ action_type: discrete
42
+ sequential_step_sampling: false
43
+ eval_sequential_step_sampling: null
44
+ num_workers: 8
45
+ eval_num_workers: 8
46
+ prefetch_factor: 4
47
+ persistent_workers: true
48
+ pin_memory: true
49
+ shuffle: true
50
+ action_balance:
51
+ enabled: false
52
+ strategy: balanced_epoch
53
+ action_key: action_id
54
+ target_flap_fraction: 0.3
55
+ noop_id: 0
56
+ flap_id: 1
57
+ latency_curriculum:
58
+ enabled: false
59
+ strategy: exclusive
60
+ latencies: null
61
+ phase_steps: null
62
+ phase_distributions: null
63
+ new_latency_passes: 1.0
64
+ replay_passes: 0.25
65
+ target_total_passes: 2.0
66
+ final_equalization: true
67
+ step_budget_mode: auto
68
+ eval_at_phase_end: false
69
+ save_at_phase_end: false
70
+ computed_plan: null
71
+ per_device_batch_size: 1
72
+ load_all_data_for_training: true
73
+ num_obs_frames: 1
74
+ image_mode: single
75
+ prompt_mode: raw
76
+ stitch_grid:
77
+ - 2
78
+ - 2
79
+ obs_image_size:
80
+ - 224
81
+ - 224
82
+ video_backend: torchvision_av
83
+ observation_indices:
84
+ - -4
85
+ - -3
86
+ - -2
87
+ - -1
88
+ - 0
89
+ language_indices:
90
+ - 0
91
+ state_indices:
92
+ - 0
93
+ action_indices:
94
+ - 0
95
+ - 1
96
+ - 2
97
+ - 3
98
+ - 4
99
+ - 5
100
+ - 6
101
+ - 7
102
+ pack_image_sequence: true
103
+ image_sequence_length: 5
104
+ context_images_column: observation.context_images
105
+ dataset:
106
+ source_hf: latency-sensitive-bench/memory-data
107
+ config_name: deadly_corridor_fixed_latency_6_1000ep_7k2steps
108
+ source_subdir: null
109
+ converted_name: deadly_corridor_train__bridge
110
+ single_source_hf: ''
111
+ mixed_source_hf: ''
112
+ single_converted_name: deadly_corridor_train__bridge
113
+ mixed_converted_name: deadly_corridor_mixed_latency_train__bridge
114
+ single_latency_filter: null
115
+ mixed_latency_filter: null
116
+ force_download: false
117
+ setup_force: false
118
+ skip_verification: false
119
+ target_latency_unit: observation_steps
120
+ verify_rows: 200
121
+ max_episodes: 1000
122
+ episodes_per_latency: null
123
+ latency_filter: null
124
+ debug_subset:
125
+ enabled: false
126
+ max_episodes: 5
127
+ suffix: debug
128
+ base_model:
129
+ repo_id: Wan-AI/Wan2.2-TI2V-5B-Diffusers
130
+ initialization:
131
+ checkpoint_local_dir: playground/Pretrained_models/WM4A-Wan2d2-OFT-LIBERO-4in1
132
+ checkpoint_hf_repo_id: StarVLA/WM4A-Wan2d2-OFT-LIBERO-4in1
133
+ checkpoint_filename: checkpoints/steps_60000_pytorch_model.pt
134
+ trainer:
135
+ max_train_steps: 1000
136
+ num_warmup_steps: 0
137
+ save_interval: 500
138
+ eval_interval: 100
139
+ eval_num_batches: 100
140
+ per_latency_eval_num_batches: null
141
+ eval_action_classification: false
142
+ eval_action_classification_interval: null
143
+ cc_f1_tolerance: 1
144
+ learning_rate:
145
+ base: 2.5e-05
146
+ qwen_vl_interface: 1.0e-05
147
+ action_model: 0.0001
148
+ action_query_proj: 0.0001
149
+ lr_scheduler_type: cosine_with_min_lr
150
+ scheduler_specific_kwargs:
151
+ min_lr: 1.0e-06
152
+ freeze_modules: ''
153
+ freeze_vit: false
154
+ freeze_tied_embedding: false
155
+ freeze_llm_layers: []
156
+ loss_scale:
157
+ vla: 1.0
158
+ vlm: 0.1
159
+ max_grad_norm: 1.0
160
+ weight_decay: 0.0
161
+ logging_frequency: 1
162
+ profile_timing:
163
+ enabled: false
164
+ log_interval: 10
165
+ gradient_clipping: 1.0
166
+ gradient_accumulation_steps: 16
167
+ distributed_backend: deepspeed
168
+ is_resume: false
169
+ pretrained_checkpoint: null
170
+ resume_step: 0
171
+ reload_modules: null
172
+ optimizer:
173
+ name: AdamW
174
+ betas:
175
+ - 0.9
176
+ - 0.95
177
+ eps: 1.0e-08
178
+ weight_decay: 1.0e-08
179
+ fused: true
180
+ save_format: pt
181
+ workspace_dir: WORKSPACE_DIR
182
+ run_root_dir: /lustre/fsw/portfolios/nvr/projects/nvr_lacr_llm/users/zihwang/wanoft_bench/third_party/starVLA/results/Checkpoints
183
+ seed: 42
184
+ wandb_entity: zihanwang-ai-northwestern-university
185
+ wandb_project: starVLA_rl_games
186
+ auth:
187
+ env_file: null
188
+ hf_token_env: HF_TOKEN
189
+ wandb_api_key_env: WANDB_API_KEY
190
+ paths:
191
+ run_root_dir: results/Checkpoints
192
+ dataset_local_dir: data/deadly_corridor_fix_latency_6_1000ep_context5
193
+ dataset_cache_dir: null
194
+ base_model_dir: playground/Pretrained_models/Wan-AI/Wan2.2-TI2V-5B-Diffusers
195
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
196
+ launch:
197
+ use_accelerate: true
198
+ gpus: null
199
+ num_processes: 1
200
+ dry_run: false
201
+ conda:
202
+ enabled: true
203
+ env_name: null
204
+ rl_games:
205
+ model_alias: wan_oft
206
+ env_eval:
207
+ image_size: 224
208
+ frameskip: 4
209
+ image_transform: raw_rgb
210
+ prompt_mode: raw
211
+ ghost_trail:
212
+ history_frames: 5
213
+ gamma: 1.3
214
+ min_alpha: 35
215
+ scroll_px_per_step: 4.0
216
+ ground_fraction: 0.22
217
+ seed: 42
218
+ fixed_episode_seeds: true
219
+ latency_seed_stride: 0
220
+ task_seed_stride: 0
221
+ task_description: You are playing Deadly Corridor in VizDoom. Choose actions from
222
+ MOVE_FORWARD, MOVE_BACKWARD, MOVE_LEFT, MOVE_RIGHT, TURN_LEFT, TURN_RIGHT, ATTACK.
223
+ eval_parallel_envs: 5
224
+ action_chunk_execution:
225
+ enabled: false
226
+ chunk_size: null
227
+ deadly:
228
+ action_layout: multibinary_7
229
+ multibinary_threshold: 0.0
230
+ enabled: false
231
+ eval_backend: eval_core
232
+ distributed_mode: rank_sharded
233
+ vectorized:
234
+ enabled: false
235
+ batch_size: 1
236
+ latency:
237
+ prompt_map_path: data/deadly_corridor_fix_latency_6_1000ep_context5/deadly_corridor_train__bridge/latency_prompt_map.json
238
+ mode: single
239
+ values:
240
+ - 0
241
+ mid_train:
242
+ enabled: false
243
+ interval_steps: 100
244
+ latencies:
245
+ - 0
246
+ num_episodes: 5
247
+ max_steps_per_episode: 3600
248
+ post_train:
249
+ enabled: false
250
+ latencies:
251
+ - 0
252
+ - 1
253
+ - 2
254
+ - 3
255
+ - 4
256
+ num_episodes: 5
257
+ max_steps_per_episode: 3600
258
+ task: deadly_corridor
259
+ deadly_corridor_loss_type: current_multibinary_bce
260
+ initialization_mode: bridge
261
+ action_carrier: bridge
262
+ model: wan_oft
263
+ env: deadly_corridor
264
+ init: wan_oft_libero
265
+ mode: single
266
+ checkpoint:
267
+ load: auto
268
+ hf_repo_id: null
269
+ save_best_model: false
270
+ save_final_model: true
271
+ save_pt_file: true
272
+ save_training_state: true
273
+ save_safetensors_file: false
274
+ local:
275
+ keep_last_n: 1
276
+ sync:
277
+ enabled: false
278
+ repo_id: null
279
+ keep_last_n: 0
280
+ sync_every_n_checkpoints: 1
281
+ resume_policy: local_latest
282
+ run_id: wan_oft_deadly_corridor_fix_latency_6_context5_standard_sft_1000_effbs128_224_currentbce
283
+ output_dir: /lustre/fsw/portfolios/nvr/projects/nvr_lacr_llm/users/zihwang/wanoft_bench/third_party/starVLA/results/Checkpoints/wan_oft_deadly_corridor_fix_latency_6_context5_standard_sft_1000_effbs128_224_currentbce
284
+ config_yaml: null
285
+ is_debug: false
286
+ version_id: '0.21'
latency-aware/deadly-corridor/vla/starvla-wanoft-h8/wan_oft_deadly_corridor_fix_latency_6_context5_standard_sft_1000_effbs128_224_currentbce/config.yaml ADDED
@@ -0,0 +1,83 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ checkpoint:
2
+ local:
3
+ keep_last_n: 1
4
+ save_best_model: false
5
+ save_final_model: true
6
+ save_pt_file: true
7
+ save_safetensors_file: false
8
+ save_training_state: true
9
+ sync:
10
+ enabled: false
11
+ keep_last_n: 0
12
+ repo_id: null
13
+ datasets:
14
+ vla_data:
15
+ data_mix: deadly_corridor_train__bridge
16
+ dataset_py: lerobot_datasets
17
+ eval_data_mix: null
18
+ latency_curriculum:
19
+ enabled: false
20
+ obs_image_size:
21
+ - 224
22
+ - 224
23
+ per_device_batch_size: 1
24
+ framework:
25
+ action_model:
26
+ action_dim: 7
27
+ action_env_dim: 7
28
+ action_hidden_dim: 3072
29
+ action_horizon: 8
30
+ action_model_type: MLP
31
+ action_query_source: mean
32
+ loss_type: current_multibinary_bce
33
+ name: WanOFT
34
+ qwenvl:
35
+ base_vlm: /workspace/runtime/models/Wan2.2-TI2V-5B-Diffusers
36
+ world_model:
37
+ base_wm: /workspace/runtime/models/Wan2.2-TI2V-5B-Diffusers
38
+ extract_layers:
39
+ - -1
40
+ num_frames: 5
41
+ output_dir: /lustre/fsw/portfolios/nvr/projects/nvr_lacr_llm/users/zihwang/wanoft_bench/third_party/starVLA/results/Checkpoints/wan_oft_deadly_corridor_fix_latency_6_context5_standard_sft_1000_effbs128_224_currentbce
42
+ rl_games:
43
+ env_eval:
44
+ enabled: false
45
+ run_id: wan_oft_deadly_corridor_fix_latency_6_context5_standard_sft_1000_effbs128_224_currentbce
46
+ run_root_dir: /lustre/fsw/portfolios/nvr/projects/nvr_lacr_llm/users/zihwang/wanoft_bench/third_party/starVLA/results/Checkpoints
47
+ seed: 42
48
+ trainer:
49
+ distributed_backend: deepspeed
50
+ eval_action_classification: false
51
+ eval_action_classification_interval: null
52
+ eval_interval: 100
53
+ freeze_llm_layers: []
54
+ freeze_modules: ''
55
+ freeze_tied_embedding: false
56
+ freeze_vit: false
57
+ gradient_accumulation_steps: 16
58
+ is_resume: false
59
+ learning_rate:
60
+ action_model: 0.0001
61
+ action_query_proj: 0.0001
62
+ base: 2.5e-05
63
+ qwen_vl_interface: 1.0e-05
64
+ logging_frequency: 1
65
+ lr_scheduler_type: cosine_with_min_lr
66
+ max_train_steps: 1000
67
+ num_warmup_steps: 0
68
+ optimizer:
69
+ betas:
70
+ - 0.9
71
+ - 0.95
72
+ eps: 1.0e-08
73
+ fused: true
74
+ weight_decay: 1.0e-08
75
+ pretrained_checkpoint: null
76
+ profile_timing:
77
+ enabled: false
78
+ reload_modules: null
79
+ save_interval: 500
80
+ scheduler_specific_kwargs:
81
+ min_lr: 1.0e-06
82
+ wandb_entity: zihanwang-ai-northwestern-university
83
+ wandb_project: starVLA_rl_games
latency-aware/deadly-corridor/vla/starvla-wanoft-h8/wan_oft_deadly_corridor_fix_latency_6_context5_standard_sft_1000_effbs128_224_currentbce/dataset_statistics.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.9434846043586731,
6
+ 0.023302890360355377,
7
+ 0.21685810387134552,
8
+ 0.5315648317337036,
9
+ 0.18562282621860504,
10
+ 0.09055173397064209,
11
+ 0.8273231983184814
12
+ ],
13
+ "std": [
14
+ 0.2308816909790039,
15
+ 0.15088047087192535,
16
+ 0.4120523929595947,
17
+ 0.4989914894104004,
18
+ 0.388867050409317,
19
+ 0.287028968334198,
20
+ 0.37795618176460266
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
+ 1.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": 42484,
125
+ "num_trajectories": 990
126
+ }
127
+ }
latency-aware/demon-attack/small-policy/memory_rollout_teacher:demon_attack:teacher:fixed_l6:fs4:obs15:stride4:seed0/train/memory_rollout_teacher:demon_attack:teacher:fixed_l6:fs4:obs15:stride4:seed0/checkpoint_p0/checkpoint_000048880_25026560.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:131b3bc73aab996f9154c1fb40b632b77afd46fe1cf1cf9733b11d29870234bd
3
+ size 7210813
latency-aware/demon-attack/small-policy/memory_rollout_teacher:demon_attack:teacher:fixed_l6:fs4:obs15:stride4:seed0/train/memory_rollout_teacher:demon_attack:teacher:fixed_l6:fs4:obs15:stride4:seed0/config.json ADDED
@@ -0,0 +1,255 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "help": false,
3
+ "algo": "APPO",
4
+ "env": "latency_demon_attack",
5
+ "experiment": "demon_attack_teacher_fixed_l6_obs15_seed0",
6
+ "train_dir": "results/checkpoints/memory/rollout_teacher",
7
+ "restart_behavior": "resume",
8
+ "device": "gpu",
9
+ "seed": 0,
10
+ "num_policies": 1,
11
+ "async_rl": true,
12
+ "serial_mode": false,
13
+ "batched_sampling": true,
14
+ "num_batches_to_accumulate": 2,
15
+ "worker_num_splits": 1,
16
+ "policy_workers_per_policy": 1,
17
+ "max_policy_lag": 300,
18
+ "num_workers": 1,
19
+ "num_envs_per_worker": 1,
20
+ "batch_size": 1024,
21
+ "num_batches_per_epoch": 8,
22
+ "num_epochs": 2,
23
+ "rollout": 64,
24
+ "recurrence": 32,
25
+ "shuffle_minibatches": false,
26
+ "gamma": 0.99,
27
+ "reward_scale": 1.0,
28
+ "reward_clip": 1000.0,
29
+ "value_bootstrap": false,
30
+ "normalize_returns": true,
31
+ "exploration_loss_coeff": 0.003,
32
+ "value_loss_coeff": 0.5,
33
+ "kl_loss_coeff": 0.0,
34
+ "exploration_loss": "entropy",
35
+ "gae_lambda": 0.95,
36
+ "ppo_clip_ratio": 0.1,
37
+ "ppo_clip_value": 0.2,
38
+ "with_vtrace": false,
39
+ "vtrace_rho": 1.0,
40
+ "vtrace_c": 1.0,
41
+ "optimizer": "adam",
42
+ "adam_eps": 1e-05,
43
+ "adam_beta1": 0.9,
44
+ "adam_beta2": 0.999,
45
+ "max_grad_norm": 0.0,
46
+ "learning_rate": 0.000303,
47
+ "lr_schedule": "linear_decay",
48
+ "lr_schedule_kl_threshold": 0.008,
49
+ "lr_adaptive_min": 1e-06,
50
+ "lr_adaptive_max": 0.01,
51
+ "obs_subtract_mean": 0.0,
52
+ "obs_scale": 255.0,
53
+ "normalize_input": true,
54
+ "normalize_input_keys": null,
55
+ "decorrelate_experience_max_seconds": 0,
56
+ "decorrelate_envs_on_one_worker": true,
57
+ "actor_worker_gpus": [
58
+ 0
59
+ ],
60
+ "set_workers_cpu_affinity": true,
61
+ "force_envs_single_thread": false,
62
+ "default_niceness": 0,
63
+ "log_to_file": true,
64
+ "experiment_summaries_interval": 1,
65
+ "flush_summaries_interval": 30,
66
+ "stats_avg": 100,
67
+ "summaries_use_frameskip": true,
68
+ "heartbeat_interval": 20,
69
+ "heartbeat_reporting_interval": 180,
70
+ "train_for_env_steps": 25000000,
71
+ "train_for_seconds": 10000000000,
72
+ "save_every_sec": 600,
73
+ "keep_checkpoints": 5,
74
+ "load_checkpoint_kind": "latest",
75
+ "save_milestones_sec": -1,
76
+ "save_best_every_sec": 5,
77
+ "save_best_metric": "reward",
78
+ "save_best_after": 100000,
79
+ "benchmark": false,
80
+ "encoder_mlp_layers": [
81
+ 512,
82
+ 512
83
+ ],
84
+ "encoder_conv_architecture": "convnet_atari",
85
+ "encoder_conv_mlp_layers": [
86
+ 512
87
+ ],
88
+ "use_rnn": false,
89
+ "rnn_size": 512,
90
+ "rnn_type": "gru",
91
+ "rnn_num_layers": 1,
92
+ "decoder_mlp_layers": [],
93
+ "nonlinearity": "elu",
94
+ "policy_initialization": "orthogonal",
95
+ "policy_init_gain": 1.0,
96
+ "actor_critic_share_weights": true,
97
+ "adaptive_stddev": true,
98
+ "continuous_tanh_scale": 0.0,
99
+ "initial_stddev": 1.0,
100
+ "use_env_info_cache": false,
101
+ "env_gpu_actions": true,
102
+ "env_gpu_observations": true,
103
+ "env_frameskip": 1,
104
+ "env_framestack": 1,
105
+ "pixel_format": "CHW",
106
+ "use_record_episode_statistics": false,
107
+ "with_wandb": true,
108
+ "wandb_user": null,
109
+ "wandb_project": "latency-sensitive-bench",
110
+ "wandb_group": "demon-attack-obs15-fixed_l6",
111
+ "wandb_job_type": "sample_factory",
112
+ "wandb_tags": [
113
+ "memory_rollout_teacher",
114
+ "demon-attack",
115
+ "teacher",
116
+ "fixed",
117
+ "fixed_l6",
118
+ "obs15",
119
+ "seed0"
120
+ ],
121
+ "with_pbt": false,
122
+ "pbt_mix_policies_in_one_env": true,
123
+ "pbt_period_env_steps": 5000000,
124
+ "pbt_start_mutation": 20000000,
125
+ "pbt_replace_fraction": 0.3,
126
+ "pbt_mutation_rate": 0.15,
127
+ "pbt_replace_reward_gap": 0.1,
128
+ "pbt_replace_reward_gap_absolute": 1e-06,
129
+ "pbt_optimize_gamma": false,
130
+ "pbt_target_objective": "true_objective",
131
+ "pbt_perturb_min": 1.1,
132
+ "pbt_perturb_max": 1.5,
133
+ "env_id": "ALE/DemonAttack-v5",
134
+ "env_fps": 60.0,
135
+ "obs_fps": 15.0,
136
+ "screen_size": 84,
137
+ "noop_max": 30,
138
+ "reset_mode": "fire",
139
+ "frame_stack": 4,
140
+ "terminal_on_life_loss": true,
141
+ "clip_reward": true,
142
+ "repeat_action_probability": 0.0,
143
+ "full_action_space": false,
144
+ "max_episode_steps": 25000,
145
+ "simulator": "gpu",
146
+ "gpu_env_device": "auto",
147
+ "gpu_env_batch_size": 512,
148
+ "export_env_raw_rgb_frames": false,
149
+ "mode": "train",
150
+ "latency_type": "fixed",
151
+ "fixed_latency_ms": 100.0,
152
+ "mean_latency_ms": null,
153
+ "std_latency_ms": null,
154
+ "min_latency_ms": null,
155
+ "max_latency_ms": null,
156
+ "latency_seed": null,
157
+ "add_latency_info": false,
158
+ "max_pending_actions": null,
159
+ "hold_policy": "hold",
160
+ "ordering_policy": "latest_ready",
161
+ "eval_episodes": 100,
162
+ "eval_parallel_envs": 50,
163
+ "eval_latency_raw_frame_values": "0,1,2,3,4,5",
164
+ "eval_max_steps": 7200,
165
+ "eval_deterministic": true,
166
+ "eval_raw_reward": true,
167
+ "episode_metrics_path": "results/checkpoints/memory/rollout_teacher/demon_attack_teacher_fixed_l6_obs15_seed0/episode_metrics.jsonl",
168
+ "command_line": "--mode train --algo APPO --env latency_demon_attack --experiment demon_attack_teacher_fixed_l6_obs15_seed0 --train_dir results/checkpoints/memory/rollout_teacher --restart_behavior resume --device gpu --actor_worker_gpus 0 --env_gpu_observations True --env_gpu_actions True --gpu-env-batch-size 512 --seed 0 --episode_metrics_path results/checkpoints/memory/rollout_teacher/demon_attack_teacher_fixed_l6_obs15_seed0/episode_metrics.jsonl --train_for_env_steps 25000000 --num_workers 1 --num_envs_per_worker 1 --num_policies 1 --batch_size 1024 --rollout 64 --recurrence 32 --num_epochs 2 --num_batches_per_epoch 8 --worker_num_splits 1 --max_policy_lag 300 --learning_rate 0.000303 --gamma 0.99 --gae_lambda 0.95 --ppo_clip_ratio 0.1 --ppo_clip_value 0.2 --value_loss_coeff 0.5 --max_grad_norm 0.0 --save_every_sec 600 --keep_checkpoints 5 --stats_avg 100 --experiment_summaries_interval 1 --batched_sampling True --async_rl True --batched_sampling True --use_rnn False --normalize_returns True --normalize_input True --latency-type fixed --fixed-latency-ms 100.0 --add-latency-info False --eval-episodes 100 --eval-parallel-envs 50 --eval-max-steps 7200 --eval-deterministic True --eval-raw-reward --with_wandb True --wandb_project latency-sensitive-bench --wandb_group demon-attack-obs15-fixed_l6 --wandb_job_type sample_factory --wandb_tags memory_rollout_teacher demon-attack teacher fixed fixed_l6 obs15 seed0 --env_id ALE/DemonAttack-v5 --env-fps 60 --obs-fps 15.0 --screen_size 84 --noop_max 30 --reset-mode fire --frame_stack 4 --terminal_on_life_loss True --clip_reward True --repeat_action_probability 0.0 --full_action_space False --max_episode_steps 25000 --simulator gpu --gpu-env-device auto --hold-policy hold --ordering-policy latest_ready",
169
+ "cli_args": {
170
+ "algo": "APPO",
171
+ "env": "latency_demon_attack",
172
+ "experiment": "demon_attack_teacher_fixed_l6_obs15_seed0",
173
+ "train_dir": "results/checkpoints/memory/rollout_teacher",
174
+ "restart_behavior": "resume",
175
+ "device": "gpu",
176
+ "seed": 0,
177
+ "num_policies": 1,
178
+ "async_rl": true,
179
+ "batched_sampling": true,
180
+ "worker_num_splits": 1,
181
+ "max_policy_lag": 300,
182
+ "num_workers": 1,
183
+ "num_envs_per_worker": 1,
184
+ "batch_size": 1024,
185
+ "num_batches_per_epoch": 8,
186
+ "num_epochs": 2,
187
+ "rollout": 64,
188
+ "recurrence": 32,
189
+ "gamma": 0.99,
190
+ "normalize_returns": true,
191
+ "value_loss_coeff": 0.5,
192
+ "gae_lambda": 0.95,
193
+ "ppo_clip_ratio": 0.1,
194
+ "ppo_clip_value": 0.2,
195
+ "max_grad_norm": 0.0,
196
+ "learning_rate": 0.000303,
197
+ "normalize_input": true,
198
+ "actor_worker_gpus": [
199
+ 0
200
+ ],
201
+ "experiment_summaries_interval": 1,
202
+ "stats_avg": 100,
203
+ "train_for_env_steps": 25000000,
204
+ "save_every_sec": 600,
205
+ "keep_checkpoints": 5,
206
+ "use_rnn": false,
207
+ "env_gpu_actions": true,
208
+ "env_gpu_observations": true,
209
+ "with_wandb": true,
210
+ "wandb_project": "latency-sensitive-bench",
211
+ "wandb_group": "demon-attack-obs15-fixed_l6",
212
+ "wandb_job_type": "sample_factory",
213
+ "wandb_tags": [
214
+ "memory_rollout_teacher",
215
+ "demon-attack",
216
+ "teacher",
217
+ "fixed",
218
+ "fixed_l6",
219
+ "obs15",
220
+ "seed0"
221
+ ],
222
+ "env_id": "ALE/DemonAttack-v5",
223
+ "env_fps": 60.0,
224
+ "obs_fps": 15.0,
225
+ "screen_size": 84,
226
+ "noop_max": 30,
227
+ "reset_mode": "fire",
228
+ "frame_stack": 4,
229
+ "terminal_on_life_loss": true,
230
+ "clip_reward": true,
231
+ "repeat_action_probability": 0.0,
232
+ "full_action_space": false,
233
+ "max_episode_steps": 25000,
234
+ "simulator": "gpu",
235
+ "gpu_env_device": "auto",
236
+ "gpu_env_batch_size": 512,
237
+ "mode": "train",
238
+ "latency_type": "fixed",
239
+ "fixed_latency_ms": 100.0,
240
+ "add_latency_info": false,
241
+ "hold_policy": "hold",
242
+ "ordering_policy": "latest_ready",
243
+ "eval_episodes": 100,
244
+ "eval_parallel_envs": 50,
245
+ "eval_max_steps": 7200,
246
+ "eval_deterministic": true,
247
+ "eval_raw_reward": true,
248
+ "episode_metrics_path": "results/checkpoints/memory/rollout_teacher/demon_attack_teacher_fixed_l6_obs15_seed0/episode_metrics.jsonl"
249
+ },
250
+ "git_hash": "bedb62f3a97309f6dacdb6a2953e3312d9776765",
251
+ "git_repo_name": "git@github.com:ZihanWang314/latency-sensitive-bench.git",
252
+ "eval_env_frameskip": 1,
253
+ "output_dir": "outputs/memory/rollout_teacher/demon_attack/train/teacher/fixed_l6/obs15/seed_0",
254
+ "wandb_unique_id": "demon-attack-obs15-fixed_l6-s0"
255
+ }
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/checkpoints/steps_7000_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2ab6d4d4eb795a9a09da9c63d489771039440689877642e3f9ec48dd9112119a
3
+ size 9138230516
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/config.full.yaml ADDED
@@ -0,0 +1,209 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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_200ep
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
+ 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
+ latency_curriculum:
36
+ enabled: false
37
+ strategy: exclusive
38
+ latencies: null
39
+ phase_steps: null
40
+ per_device_batch_size: 32
41
+ load_all_data_for_training: true
42
+ obs_image_size: null
43
+ video_backend: torchvision_av
44
+ dataset:
45
+ source_hf: ''
46
+ config_name: null
47
+ source_subdir: null
48
+ converted_name: demon_attack_train
49
+ single_source_hf: ''
50
+ mixed_source_hf: ''
51
+ single_converted_name: demon_attack_train
52
+ mixed_converted_name: demon_attack_mixed_latency_train
53
+ single_latency_filter: null
54
+ mixed_latency_filter: null
55
+ force_download: false
56
+ setup_force: false
57
+ skip_verification: false
58
+ verify_rows: 200
59
+ max_episodes: null
60
+ episodes_per_latency: null
61
+ latency_filter: null
62
+ debug_subset:
63
+ enabled: false
64
+ max_episodes: 5
65
+ suffix: debug
66
+ base_model:
67
+ repo_id: Qwen/Qwen3-VL-4B-Instruct
68
+ initialization:
69
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
70
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
71
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
72
+ trainer:
73
+ max_train_steps: 7000
74
+ num_warmup_steps: 100
75
+ save_interval: 100
76
+ eval_interval: 100
77
+ eval_num_batches: 100
78
+ per_latency_eval_num_batches: null
79
+ eval_action_classification: true
80
+ cc_f1_tolerance: 1
81
+ learning_rate:
82
+ base: 2.0e-05
83
+ qwen_vl_interface: 1.0e-05
84
+ action_model: 0.0001
85
+ lr_scheduler_type: cosine_with_min_lr
86
+ scheduler_specific_kwargs:
87
+ min_lr: 1.0e-06
88
+ freeze_modules: ''
89
+ loss_scale:
90
+ vla: 1.0
91
+ vlm: 0.1
92
+ max_grad_norm: 1.0
93
+ weight_decay: 0.0
94
+ logging_frequency: 1
95
+ gradient_clipping: 1.0
96
+ gradient_accumulation_steps: 4
97
+ distributed_backend: none
98
+ is_resume: false
99
+ pretrained_checkpoint: null
100
+ resume_step: 0
101
+ reload_modules: null
102
+ optimizer:
103
+ name: AdamW
104
+ betas:
105
+ - 0.9
106
+ - 0.95
107
+ eps: 1.0e-08
108
+ weight_decay: 1.0e-08
109
+ fused: false
110
+ save_format: pt
111
+ workspace_dir: WORKSPACE_DIR
112
+ run_root_dir: /workspace/starVLA/results/Checkpoints
113
+ seed: 42
114
+ wandb_entity: dongqianyu99-zhejiang-university
115
+ wandb_project: starVLA_rl_games
116
+ auth:
117
+ env_file: null
118
+ hf_token_env: HF_TOKEN
119
+ wandb_api_key_env: WANDB_API_KEY
120
+ paths:
121
+ run_root_dir: results/Checkpoints
122
+ dataset_local_dir: playground/Datasets/rl_games
123
+ dataset_cache_dir: null
124
+ base_model_dir: playground/Pretrained_models/Qwen3-VL-4B-Instruct
125
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
126
+ launch:
127
+ use_accelerate: true
128
+ gpus: null
129
+ num_processes: 1
130
+ dry_run: false
131
+ conda:
132
+ enabled: true
133
+ env_name: null
134
+ rl_games:
135
+ model_alias: openvla
136
+ env_eval:
137
+ image_size: 224
138
+ frameskip: 4
139
+ seed: 42
140
+ fixed_episode_seeds: true
141
+ latency_seed_stride: 0
142
+ task_seed_stride: 0
143
+ task_description: 'You are playing Demon Attack from a single game image. Choose
144
+ exactly one action from: NOOP, FIRE, RIGHT, LEFT, RIGHTFIRE, LEFTFIRE.'
145
+ demon_attack:
146
+ noop_max: 30
147
+ enabled: true
148
+ distributed_mode: none
149
+ vectorized:
150
+ enabled: false
151
+ batch_size: 1
152
+ latency:
153
+ prompt_map_path: /workspace/starVLA/data/demon_attack_fix_latency_2_200ep/demon_attack_train__bridge/latency_prompt_map.json
154
+ mode: single
155
+ values:
156
+ - 0
157
+ mid_train:
158
+ enabled: false
159
+ interval_steps: 100
160
+ latencies:
161
+ - 0
162
+ num_episodes: 5
163
+ max_steps_per_episode: 3600
164
+ post_train:
165
+ enabled: false
166
+ latencies:
167
+ - 0
168
+ - 1
169
+ - 2
170
+ - 3
171
+ - 4
172
+ num_episodes: 5
173
+ max_steps_per_episode: 3600
174
+ task: demon_attack
175
+ initialization_mode: bridge
176
+ action_carrier: bridge
177
+ model: openvla
178
+ env: demon_attack
179
+ init: bridge
180
+ bridge_base_model:
181
+ repo_id:
182
+ openvla: Qwen/Qwen3-VL-4B-Instruct
183
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
184
+ pi05: Qwen/Qwen3-VL-4B-Instruct
185
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
186
+ local_dir:
187
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
188
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
189
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
190
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
191
+ mode: single
192
+ checkpoint:
193
+ load: auto
194
+ hf_repo_id: null
195
+ save_best_model: false
196
+ save_pt_file: false
197
+ local:
198
+ keep_last_n: 1
199
+ sync:
200
+ enabled: false
201
+ repo_id: null
202
+ keep_last_n: 0
203
+ sync_every_n_checkpoints: 1
204
+ resume_policy: local_latest
205
+ run_id: demon_attack_fix_latency_2_200ep
206
+ output_dir: /workspace/starVLA/results/Checkpoints/demon_attack_fix_latency_2_200ep
207
+ config_yaml: null
208
+ is_debug: false
209
+ version_id: '0.21'
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/config.yaml ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: demon_attack_train__bridge
13
+ dataset_py: lerobot_datasets
14
+ eval_data_mix: demon_attack_train__bridge__val
15
+ latency_curriculum:
16
+ enabled: false
17
+ obs_image_size: null
18
+ per_device_batch_size: 32
19
+ framework:
20
+ action_model:
21
+ action_dim: 7
22
+ action_env_dim: 6
23
+ action_hidden_dim: 2560
24
+ action_horizon: 1
25
+ action_model_type: MLP
26
+ loss_type: discrete_ce
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/demon_attack_fix_latency_2_200ep
33
+ rl_games:
34
+ env_eval:
35
+ enabled: true
36
+ mid_train:
37
+ enabled: false
38
+ interval_steps: 100
39
+ task: demon_attack
40
+ run_id: demon_attack_fix_latency_2_200ep
41
+ run_root_dir: /workspace/starVLA/results/Checkpoints
42
+ seed: 42
43
+ trainer:
44
+ cc_f1_tolerance: 1
45
+ distributed_backend: none
46
+ eval_action_classification: true
47
+ eval_interval: 100
48
+ eval_num_batches: 100
49
+ freeze_modules: ''
50
+ gradient_accumulation_steps: 4
51
+ gradient_clipping: 1.0
52
+ is_resume: false
53
+ learning_rate:
54
+ action_model: 0.0001
55
+ base: 2.0e-05
56
+ qwen_vl_interface: 1.0e-05
57
+ logging_frequency: 1
58
+ lr_scheduler_type: cosine_with_min_lr
59
+ max_train_steps: 7000
60
+ num_warmup_steps: 100
61
+ optimizer:
62
+ betas:
63
+ - 0.9
64
+ - 0.95
65
+ eps: 1.0e-08
66
+ fused: false
67
+ weight_decay: 1.0e-08
68
+ per_latency_eval_num_batches: null
69
+ pretrained_checkpoint: null
70
+ reload_modules: null
71
+ save_interval: 100
72
+ scheduler_specific_kwargs:
73
+ min_lr: 1.0e-06
74
+ wandb_entity: dongqianyu99-zhejiang-university
75
+ wandb_project: starVLA_rl_games
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/dataset_statistics.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.06905864179134369,
6
+ 0.37893569469451904,
7
+ 0.13702447712421417,
8
+ 0.1011248454451561,
9
+ 0.12969808280467987,
10
+ 0.18415828049182892,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.2532638907432556,
15
+ 0.4849342703819275,
16
+ 0.3443855047225952,
17
+ 0.3010394871234894,
18
+ 0.3360588848590851,
19
+ 0.3878043293952942,
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": 349558,
125
+ "num_trajectories": 180
126
+ }
127
+ }
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/dataset_statistics_eval.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.07371096312999725,
6
+ 0.3668539524078369,
7
+ 0.13910865783691406,
8
+ 0.1111111119389534,
9
+ 0.13123494386672974,
10
+ 0.17798034846782684,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.2612755298614502,
15
+ 0.48196175694465637,
16
+ 0.3461320102214813,
17
+ 0.3142721354961395,
18
+ 0.3376607596874237,
19
+ 0.38254332542419434,
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": 52326,
125
+ "num_trajectories": 20
126
+ }
127
+ }
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/hydra/.hydra/config.yaml ADDED
@@ -0,0 +1,207 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: 6
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/demon_attack_fix_latency_2_200ep
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
+ 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
+ latency_curriculum:
34
+ enabled: false
35
+ strategy: exclusive
36
+ latencies: null
37
+ phase_steps: null
38
+ per_device_batch_size: 32
39
+ load_all_data_for_training: true
40
+ obs_image_size: null
41
+ video_backend: torchvision_av
42
+ dataset:
43
+ source_hf: ${dataset.single_source_hf}
44
+ config_name: null
45
+ source_subdir: null
46
+ converted_name: ${dataset.single_converted_name}
47
+ single_source_hf: ''
48
+ mixed_source_hf: ''
49
+ single_converted_name: demon_attack_train
50
+ mixed_converted_name: demon_attack_mixed_latency_train
51
+ single_latency_filter: null
52
+ mixed_latency_filter: null
53
+ force_download: false
54
+ setup_force: false
55
+ skip_verification: false
56
+ verify_rows: 200
57
+ max_episodes: null
58
+ episodes_per_latency: null
59
+ latency_filter: ${dataset.single_latency_filter}
60
+ debug_subset:
61
+ enabled: false
62
+ max_episodes: 5
63
+ suffix: debug
64
+ base_model:
65
+ repo_id: ${bridge_base_model.repo_id.${model}}
66
+ initialization:
67
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
68
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
69
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
70
+ trainer:
71
+ max_train_steps: 7000
72
+ num_warmup_steps: 100
73
+ save_interval: 100
74
+ eval_interval: 100
75
+ eval_num_batches: 100
76
+ per_latency_eval_num_batches: null
77
+ eval_action_classification: true
78
+ cc_f1_tolerance: 1
79
+ learning_rate:
80
+ base: 2.0e-05
81
+ qwen_vl_interface: 1.0e-05
82
+ action_model: 0.0001
83
+ lr_scheduler_type: cosine_with_min_lr
84
+ scheduler_specific_kwargs:
85
+ min_lr: 1.0e-06
86
+ freeze_modules: ''
87
+ loss_scale:
88
+ vla: 1.0
89
+ vlm: 0.1
90
+ max_grad_norm: 1.0
91
+ weight_decay: 0.0
92
+ logging_frequency: 1
93
+ gradient_clipping: 1.0
94
+ gradient_accumulation_steps: 4
95
+ distributed_backend: none
96
+ is_resume: false
97
+ pretrained_checkpoint: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
98
+ resume_step: 0
99
+ reload_modules: null
100
+ optimizer:
101
+ name: AdamW
102
+ betas:
103
+ - 0.9
104
+ - 0.95
105
+ eps: 1.0e-08
106
+ weight_decay: 1.0e-08
107
+ fused: false
108
+ save_format: pt
109
+ workspace_dir: WORKSPACE_DIR
110
+ run_root_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints
111
+ seed: 42
112
+ wandb_entity: dongqianyu99-zhejiang-university
113
+ wandb_project: starVLA_rl_games
114
+ auth:
115
+ env_file: null
116
+ hf_token_env: HF_TOKEN
117
+ wandb_api_key_env: WANDB_API_KEY
118
+ paths:
119
+ run_root_dir: results/Checkpoints
120
+ dataset_local_dir: playground/Datasets/rl_games
121
+ dataset_cache_dir: null
122
+ base_model_dir: ${bridge_base_model.local_dir.${model}}
123
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
124
+ launch:
125
+ use_accelerate: true
126
+ gpus: null
127
+ num_processes: 1
128
+ dry_run: false
129
+ conda:
130
+ enabled: true
131
+ env_name: null
132
+ rl_games:
133
+ model_alias: openvla
134
+ env_eval:
135
+ image_size: 224
136
+ frameskip: 4
137
+ seed: 42
138
+ fixed_episode_seeds: true
139
+ latency_seed_stride: 0
140
+ task_seed_stride: 0
141
+ task_description: 'You are playing Demon Attack from a single game image. Choose
142
+ exactly one action from: NOOP, FIRE, RIGHT, LEFT, RIGHTFIRE, LEFTFIRE.'
143
+ demon_attack:
144
+ noop_max: 30
145
+ enabled: true
146
+ distributed_mode: none
147
+ vectorized:
148
+ enabled: false
149
+ batch_size: 1
150
+ latency:
151
+ prompt_map_path: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_2_200ep/demon_attack_train__bridge/latency_prompt_map.json
152
+ mode: single
153
+ values:
154
+ - 0
155
+ mid_train:
156
+ enabled: false
157
+ interval_steps: 100
158
+ latencies:
159
+ - 0
160
+ num_episodes: 5
161
+ max_steps_per_episode: 3600
162
+ post_train:
163
+ enabled: false
164
+ latencies:
165
+ - 0
166
+ - 1
167
+ - 2
168
+ - 3
169
+ - 4
170
+ num_episodes: 5
171
+ max_steps_per_episode: 3600
172
+ task: demon_attack
173
+ initialization_mode: bridge
174
+ action_carrier: bridge
175
+ model: openvla
176
+ env: demon_attack
177
+ init: bridge
178
+ bridge_base_model:
179
+ repo_id:
180
+ openvla: Qwen/Qwen3-VL-4B-Instruct
181
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
182
+ pi05: Qwen/Qwen3-VL-4B-Instruct
183
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
184
+ local_dir:
185
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
186
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
187
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
188
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
189
+ mode: single
190
+ checkpoint:
191
+ load: auto
192
+ hf_repo_id: null
193
+ save_best_model: false
194
+ save_pt_file: false
195
+ local:
196
+ keep_last_n: 1
197
+ sync:
198
+ enabled: false
199
+ repo_id: null
200
+ keep_last_n: 0
201
+ sync_every_n_checkpoints: 1
202
+ resume_policy: local_latest
203
+ run_id: demon_attack_fix_latency_2_200ep
204
+ output_dir: null
205
+ config_yaml: null
206
+ is_debug: false
207
+ version_id: '0.21'
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/hydra/.hydra/hydra.yaml ADDED
@@ -0,0 +1,259 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ hydra:
2
+ run:
3
+ dir: ${run_root_dir}/${run_id}/hydra
4
+ sweep:
5
+ dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
6
+ subdir: ${hydra.job.num}
7
+ launcher:
8
+ _target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
9
+ sweeper:
10
+ _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
11
+ max_batch_size: null
12
+ params: null
13
+ help:
14
+ app_name: ${hydra.job.name}
15
+ header: '${hydra.help.app_name} is powered by Hydra.
16
+
17
+ '
18
+ footer: 'Powered by Hydra (https://hydra.cc)
19
+
20
+ Use --hydra-help to view Hydra specific help
21
+
22
+ '
23
+ template: '${hydra.help.header}
24
+
25
+ == Configuration groups ==
26
+
27
+ Compose your configuration from those groups (group=option)
28
+
29
+
30
+ $APP_CONFIG_GROUPS
31
+
32
+
33
+ == Config ==
34
+
35
+ Override anything in the config (foo.bar=value)
36
+
37
+
38
+ $CONFIG
39
+
40
+
41
+ ${hydra.help.footer}
42
+
43
+ '
44
+ hydra_help:
45
+ template: 'Hydra (${hydra.runtime.version})
46
+
47
+ See https://hydra.cc for more info.
48
+
49
+
50
+ == Flags ==
51
+
52
+ $FLAGS_HELP
53
+
54
+
55
+ == Configuration groups ==
56
+
57
+ Compose your configuration from those groups (For example, append hydra/job_logging=disabled
58
+ to command line)
59
+
60
+
61
+ $HYDRA_CONFIG_GROUPS
62
+
63
+
64
+ Use ''--cfg hydra'' to Show the Hydra config.
65
+
66
+ '
67
+ hydra_help: ???
68
+ hydra_logging:
69
+ version: 1
70
+ formatters:
71
+ simple:
72
+ format: '[%(asctime)s][HYDRA] %(message)s'
73
+ handlers:
74
+ console:
75
+ class: logging.StreamHandler
76
+ formatter: simple
77
+ stream: ext://sys.stdout
78
+ root:
79
+ level: INFO
80
+ handlers:
81
+ - console
82
+ loggers:
83
+ logging_example:
84
+ level: DEBUG
85
+ disable_existing_loggers: false
86
+ job_logging:
87
+ version: 1
88
+ formatters:
89
+ simple:
90
+ format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
91
+ handlers:
92
+ console:
93
+ class: logging.StreamHandler
94
+ formatter: simple
95
+ stream: ext://sys.stdout
96
+ file:
97
+ class: logging.FileHandler
98
+ formatter: simple
99
+ filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
100
+ root:
101
+ level: INFO
102
+ handlers:
103
+ - console
104
+ - file
105
+ disable_existing_loggers: false
106
+ env: {}
107
+ mode: RUN
108
+ searchpath: []
109
+ callbacks: {}
110
+ output_subdir: .hydra
111
+ overrides:
112
+ hydra:
113
+ - hydra.mode=RUN
114
+ task:
115
+ - model=openvla
116
+ - env=demon_attack
117
+ - init=bridge
118
+ - mode=single
119
+ - run_id=demon_attack_fix_latency_2_200ep
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=false
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=7000
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=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.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.latency_curriculum.enabled=false
178
+ - datasets.vla_data.latency_curriculum.strategy=exclusive
179
+ - datasets.vla_data.per_device_batch_size=32
180
+ - datasets.vla_data.load_all_data_for_training=true
181
+ - datasets.vla_data.video_backend=torchvision_av
182
+ - datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_2_200ep
183
+ - datasets.vla_data.data_mix=demon_attack_train__bridge
184
+ - datasets.vla_data.eval_data_mix=demon_attack_train__bridge__val
185
+ - framework.qwenvl.base_vlm=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct
186
+ - rl_games.task=demon_attack
187
+ - rl_games.model_alias=openvla
188
+ - rl_games.initialization_mode=bridge
189
+ - rl_games.action_carrier=bridge
190
+ - rl_games.env_eval.distributed_mode=none
191
+ - rl_games.env_eval.latency.mode=single
192
+ - rl_games.env_eval.frameskip=4
193
+ - rl_games.env_eval.image_size=224
194
+ - rl_games.env_eval.seed=42
195
+ - rl_games.env_eval.fixed_episode_seeds=true
196
+ - rl_games.env_eval.latency_seed_stride=0
197
+ - rl_games.env_eval.task_seed_stride=0
198
+ - 'rl_games.env_eval.task_description=''You are playing Demon Attack from a single
199
+ game image. Choose exactly one action from: NOOP, FIRE, RIGHT, LEFT, RIGHTFIRE,
200
+ LEFTFIRE.'''
201
+ - rl_games.env_eval.latency.values=[0]
202
+ - rl_games.env_eval.latency.prompt_map_path=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_2_200ep/demon_attack_train__bridge/latency_prompt_map.json
203
+ - rl_games.env_eval.mid_train.enabled=false
204
+ - rl_games.env_eval.mid_train.interval_steps=100
205
+ - rl_games.env_eval.mid_train.num_episodes=5
206
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
207
+ - rl_games.env_eval.mid_train.latencies=[0]
208
+ - rl_games.env_eval.post_train.enabled=false
209
+ - rl_games.env_eval.post_train.num_episodes=5
210
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
211
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
212
+ job:
213
+ name: train_starvla_hydra
214
+ chdir: false
215
+ 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=demon_attack_train__bridge,datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_2_200ep,datasets.vla_data.eval_data_mix=demon_attack_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=32,datasets.vla_data.sequential_step_sampling=false,datasets.vla_data.shuffle=true,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=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct,framework.qwenvl.enable_gradient_checkpointing=true,init=bridge,mode=single,model=openvla,rl_games.action_carrier=bridge,rl_games.env_eval.distributed_mode=none,rl_games.env_eval.enabled=true,rl_games.env_eval.fixed_episode_seeds=true,rl_games.env_eval.frameskip=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/demon_attack_fix_latency_2_200ep/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=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
216
+ are playing Demon Attack from a single game image. Choose exactly one action
217
+ 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=demon_attack_fix_latency_2_200ep,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=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=7000,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=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt,trainer.resume_step=0,trainer.save_format=pt,trainer.save_interval=100,trainer.scheduler_specific_kwargs.min_lr=1e-06,trainer.weight_decay=0.0,wandb_entity=dongqianyu99-zhejiang-university,wandb_project=starVLA_rl_games'
218
+ id: ???
219
+ num: ???
220
+ config_name: train
221
+ env_set: {}
222
+ env_copy: []
223
+ config:
224
+ override_dirname:
225
+ kv_sep: '='
226
+ item_sep: ','
227
+ exclude_keys: []
228
+ runtime:
229
+ version: 1.3.2
230
+ version_base: '1.1'
231
+ cwd: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA
232
+ config_sources:
233
+ - path: hydra.conf
234
+ schema: pkg
235
+ provider: hydra
236
+ - path: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA/examples/rl_games/config
237
+ schema: file
238
+ provider: main
239
+ - path: ''
240
+ schema: structured
241
+ provider: schema
242
+ output_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints/demon_attack_fix_latency_2_200ep/hydra
243
+ choices:
244
+ cross_task_setup: null
245
+ checkpoint: default
246
+ mode: single
247
+ init: bridge
248
+ env: demon_attack
249
+ model: openvla
250
+ hydra/env: default
251
+ hydra/callbacks: null
252
+ hydra/job_logging: default
253
+ hydra/hydra_logging: default
254
+ hydra/hydra_help: default
255
+ hydra/help: default
256
+ hydra/sweeper: basic
257
+ hydra/launcher: basic
258
+ hydra/output: default
259
+ verbose: false
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/hydra/.hydra/overrides.yaml ADDED
@@ -0,0 +1,97 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ - model=openvla
2
+ - env=demon_attack
3
+ - init=bridge
4
+ - mode=single
5
+ - run_id=demon_attack_fix_latency_2_200ep
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=false
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=7000
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=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.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.latency_curriculum.enabled=false
64
+ - datasets.vla_data.latency_curriculum.strategy=exclusive
65
+ - datasets.vla_data.per_device_batch_size=32
66
+ - datasets.vla_data.load_all_data_for_training=true
67
+ - datasets.vla_data.video_backend=torchvision_av
68
+ - datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_2_200ep
69
+ - datasets.vla_data.data_mix=demon_attack_train__bridge
70
+ - datasets.vla_data.eval_data_mix=demon_attack_train__bridge__val
71
+ - framework.qwenvl.base_vlm=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct
72
+ - rl_games.task=demon_attack
73
+ - rl_games.model_alias=openvla
74
+ - rl_games.initialization_mode=bridge
75
+ - rl_games.action_carrier=bridge
76
+ - rl_games.env_eval.distributed_mode=none
77
+ - rl_games.env_eval.latency.mode=single
78
+ - rl_games.env_eval.frameskip=4
79
+ - rl_games.env_eval.image_size=224
80
+ - rl_games.env_eval.seed=42
81
+ - rl_games.env_eval.fixed_episode_seeds=true
82
+ - rl_games.env_eval.latency_seed_stride=0
83
+ - rl_games.env_eval.task_seed_stride=0
84
+ - 'rl_games.env_eval.task_description=''You are playing Demon Attack from a single
85
+ game image. Choose exactly one action from: NOOP, FIRE, RIGHT, LEFT, RIGHTFIRE,
86
+ LEFTFIRE.'''
87
+ - rl_games.env_eval.latency.values=[0]
88
+ - rl_games.env_eval.latency.prompt_map_path=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_2_200ep/demon_attack_train__bridge/latency_prompt_map.json
89
+ - rl_games.env_eval.mid_train.enabled=false
90
+ - rl_games.env_eval.mid_train.interval_steps=100
91
+ - rl_games.env_eval.mid_train.num_episodes=5
92
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
93
+ - rl_games.env_eval.mid_train.latencies=[0]
94
+ - rl_games.env_eval.post_train.enabled=false
95
+ - rl_games.env_eval.post_train.num_episodes=5
96
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
97
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_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/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/summary.jsonl ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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}
51
+ {"steps": 5100}
52
+ {"steps": 5200}
53
+ {"steps": 5300}
54
+ {"steps": 5400}
55
+ {"steps": 5500}
56
+ {"steps": 5600}
57
+ {"steps": 5700}
58
+ {"steps": 5800}
59
+ {"steps": 5900}
60
+ {"steps": 6000}
61
+ {"steps": 6100}
62
+ {"steps": 6200}
63
+ {"steps": 6300}
64
+ {"steps": 6400}
65
+ {"steps": 6500}
66
+ {"steps": 6600}
67
+ {"steps": 6700}
68
+ {"steps": 6800}
69
+ {"steps": 6900}
70
+ {"steps": 7000}
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/checkpoints/steps_7000_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:62e9b2a2b38f2b52cedc248c6e135b2593b212d1965fc3efae42e8130b57b373
3
+ size 9138230516
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/config.full.yaml ADDED
@@ -0,0 +1,209 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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_4_200ep
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
+ 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
+ latency_curriculum:
36
+ enabled: false
37
+ strategy: exclusive
38
+ latencies: null
39
+ phase_steps: null
40
+ per_device_batch_size: 32
41
+ load_all_data_for_training: true
42
+ obs_image_size: null
43
+ video_backend: torchvision_av
44
+ dataset:
45
+ source_hf: ''
46
+ config_name: null
47
+ source_subdir: null
48
+ converted_name: demon_attack_train
49
+ single_source_hf: ''
50
+ mixed_source_hf: ''
51
+ single_converted_name: demon_attack_train
52
+ mixed_converted_name: demon_attack_mixed_latency_train
53
+ single_latency_filter: null
54
+ mixed_latency_filter: null
55
+ force_download: false
56
+ setup_force: false
57
+ skip_verification: false
58
+ verify_rows: 200
59
+ max_episodes: null
60
+ episodes_per_latency: null
61
+ latency_filter: null
62
+ debug_subset:
63
+ enabled: false
64
+ max_episodes: 5
65
+ suffix: debug
66
+ base_model:
67
+ repo_id: Qwen/Qwen3-VL-4B-Instruct
68
+ initialization:
69
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
70
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
71
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
72
+ trainer:
73
+ max_train_steps: 7000
74
+ num_warmup_steps: 100
75
+ save_interval: 100
76
+ eval_interval: 100
77
+ eval_num_batches: 100
78
+ per_latency_eval_num_batches: null
79
+ eval_action_classification: true
80
+ cc_f1_tolerance: 1
81
+ learning_rate:
82
+ base: 2.0e-05
83
+ qwen_vl_interface: 1.0e-05
84
+ action_model: 0.0001
85
+ lr_scheduler_type: cosine_with_min_lr
86
+ scheduler_specific_kwargs:
87
+ min_lr: 1.0e-06
88
+ freeze_modules: ''
89
+ loss_scale:
90
+ vla: 1.0
91
+ vlm: 0.1
92
+ max_grad_norm: 1.0
93
+ weight_decay: 0.0
94
+ logging_frequency: 1
95
+ gradient_clipping: 1.0
96
+ gradient_accumulation_steps: 4
97
+ distributed_backend: none
98
+ is_resume: false
99
+ pretrained_checkpoint: null
100
+ resume_step: 0
101
+ reload_modules: null
102
+ optimizer:
103
+ name: AdamW
104
+ betas:
105
+ - 0.9
106
+ - 0.95
107
+ eps: 1.0e-08
108
+ weight_decay: 1.0e-08
109
+ fused: false
110
+ save_format: pt
111
+ workspace_dir: WORKSPACE_DIR
112
+ run_root_dir: /workspace/starVLA/results/Checkpoints
113
+ seed: 42
114
+ wandb_entity: dongqianyu99-zhejiang-university
115
+ wandb_project: starVLA_rl_games
116
+ auth:
117
+ env_file: null
118
+ hf_token_env: HF_TOKEN
119
+ wandb_api_key_env: WANDB_API_KEY
120
+ paths:
121
+ run_root_dir: results/Checkpoints
122
+ dataset_local_dir: playground/Datasets/rl_games
123
+ dataset_cache_dir: null
124
+ base_model_dir: playground/Pretrained_models/Qwen3-VL-4B-Instruct
125
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
126
+ launch:
127
+ use_accelerate: true
128
+ gpus: null
129
+ num_processes: 1
130
+ dry_run: false
131
+ conda:
132
+ enabled: true
133
+ env_name: null
134
+ rl_games:
135
+ model_alias: openvla
136
+ env_eval:
137
+ image_size: 224
138
+ frameskip: 4
139
+ seed: 42
140
+ fixed_episode_seeds: true
141
+ latency_seed_stride: 0
142
+ task_seed_stride: 0
143
+ task_description: 'You are playing Demon Attack from a single game image. Choose
144
+ exactly one action from: NOOP, FIRE, RIGHT, LEFT, RIGHTFIRE, LEFTFIRE.'
145
+ demon_attack:
146
+ noop_max: 30
147
+ enabled: true
148
+ distributed_mode: none
149
+ vectorized:
150
+ enabled: false
151
+ batch_size: 1
152
+ latency:
153
+ prompt_map_path: /workspace/starVLA/data/demon_attack_fix_latency_4_200ep/demon_attack_train__bridge/latency_prompt_map.json
154
+ mode: single
155
+ values:
156
+ - 0
157
+ mid_train:
158
+ enabled: false
159
+ interval_steps: 100
160
+ latencies:
161
+ - 0
162
+ num_episodes: 5
163
+ max_steps_per_episode: 3600
164
+ post_train:
165
+ enabled: false
166
+ latencies:
167
+ - 0
168
+ - 1
169
+ - 2
170
+ - 3
171
+ - 4
172
+ num_episodes: 5
173
+ max_steps_per_episode: 3600
174
+ task: demon_attack
175
+ initialization_mode: bridge
176
+ action_carrier: bridge
177
+ model: openvla
178
+ env: demon_attack
179
+ init: bridge
180
+ bridge_base_model:
181
+ repo_id:
182
+ openvla: Qwen/Qwen3-VL-4B-Instruct
183
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
184
+ pi05: Qwen/Qwen3-VL-4B-Instruct
185
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
186
+ local_dir:
187
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
188
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
189
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
190
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
191
+ mode: single
192
+ checkpoint:
193
+ load: auto
194
+ hf_repo_id: null
195
+ save_best_model: false
196
+ save_pt_file: false
197
+ local:
198
+ keep_last_n: 1
199
+ sync:
200
+ enabled: false
201
+ repo_id: null
202
+ keep_last_n: 0
203
+ sync_every_n_checkpoints: 1
204
+ resume_policy: local_latest
205
+ run_id: demon_attack_fix_latency_4_200ep
206
+ output_dir: /workspace/starVLA/results/Checkpoints/demon_attack_fix_latency_4_200ep
207
+ config_yaml: null
208
+ is_debug: false
209
+ version_id: '0.21'
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/config.yaml ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: demon_attack_train__bridge
13
+ dataset_py: lerobot_datasets
14
+ eval_data_mix: demon_attack_train__bridge__val
15
+ latency_curriculum:
16
+ enabled: false
17
+ obs_image_size: null
18
+ per_device_batch_size: 32
19
+ framework:
20
+ action_model:
21
+ action_dim: 7
22
+ action_env_dim: 6
23
+ action_hidden_dim: 2560
24
+ action_horizon: 1
25
+ action_model_type: MLP
26
+ loss_type: discrete_ce
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/demon_attack_fix_latency_2_200ep
33
+ rl_games:
34
+ env_eval:
35
+ enabled: true
36
+ mid_train:
37
+ enabled: false
38
+ interval_steps: 100
39
+ task: demon_attack
40
+ run_id: demon_attack_fix_latency_4_200ep
41
+ run_root_dir: /workspace/starVLA/results/Checkpoints
42
+ seed: 42
43
+ trainer:
44
+ cc_f1_tolerance: 1
45
+ distributed_backend: none
46
+ eval_action_classification: true
47
+ eval_interval: 100
48
+ eval_num_batches: 100
49
+ freeze_modules: ''
50
+ gradient_accumulation_steps: 4
51
+ gradient_clipping: 1.0
52
+ is_resume: false
53
+ learning_rate:
54
+ action_model: 0.0001
55
+ base: 2.0e-05
56
+ qwen_vl_interface: 1.0e-05
57
+ logging_frequency: 1
58
+ lr_scheduler_type: cosine_with_min_lr
59
+ max_train_steps: 7000
60
+ num_warmup_steps: 100
61
+ optimizer:
62
+ betas:
63
+ - 0.9
64
+ - 0.95
65
+ eps: 1.0e-08
66
+ fused: false
67
+ weight_decay: 1.0e-08
68
+ per_latency_eval_num_batches: null
69
+ pretrained_checkpoint: null
70
+ reload_modules: null
71
+ save_interval: 100
72
+ scheduler_specific_kwargs:
73
+ min_lr: 1.0e-06
74
+ wandb_entity: dongqianyu99-zhejiang-university
75
+ wandb_project: starVLA_rl_games
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/dataset_statistics.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.02783012203872204,
6
+ 0.6656418442726135,
7
+ 0.008087566122412682,
8
+ 0.061184193938970566,
9
+ 0.1591275930404663,
10
+ 0.07812871038913727,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.16479234397411346,
15
+ 0.47177085280418396,
16
+ 0.0895964726805687,
17
+ 0.23992209136486053,
18
+ 0.36527156829833984,
19
+ 0.26814261078834534,
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
+ 0.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": 332733,
125
+ "num_trajectories": 180
126
+ }
127
+ }
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/dataset_statistics_eval.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.028564849868416786,
6
+ 0.675146758556366,
7
+ 0.00799223780632019,
8
+ 0.06372412294149399,
9
+ 0.15450015664100647,
10
+ 0.07007186114788055,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.1666077822446823,
15
+ 0.46833404898643494,
16
+ 0.08902717381715775,
17
+ 0.24433407187461853,
18
+ 0.36147764325141907,
19
+ 0.2552240490913391,
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
+ 0.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": 60809,
125
+ "num_trajectories": 20
126
+ }
127
+ }
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/hydra/.hydra/config.yaml ADDED
@@ -0,0 +1,207 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: 6
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/demon_attack_fix_latency_4_200ep
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
+ 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
+ latency_curriculum:
34
+ enabled: false
35
+ strategy: exclusive
36
+ latencies: null
37
+ phase_steps: null
38
+ per_device_batch_size: 32
39
+ load_all_data_for_training: true
40
+ obs_image_size: null
41
+ video_backend: torchvision_av
42
+ dataset:
43
+ source_hf: ${dataset.single_source_hf}
44
+ config_name: null
45
+ source_subdir: null
46
+ converted_name: ${dataset.single_converted_name}
47
+ single_source_hf: ''
48
+ mixed_source_hf: ''
49
+ single_converted_name: demon_attack_train
50
+ mixed_converted_name: demon_attack_mixed_latency_train
51
+ single_latency_filter: null
52
+ mixed_latency_filter: null
53
+ force_download: false
54
+ setup_force: false
55
+ skip_verification: false
56
+ verify_rows: 200
57
+ max_episodes: null
58
+ episodes_per_latency: null
59
+ latency_filter: ${dataset.single_latency_filter}
60
+ debug_subset:
61
+ enabled: false
62
+ max_episodes: 5
63
+ suffix: debug
64
+ base_model:
65
+ repo_id: ${bridge_base_model.repo_id.${model}}
66
+ initialization:
67
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
68
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
69
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
70
+ trainer:
71
+ max_train_steps: 7000
72
+ num_warmup_steps: 100
73
+ save_interval: 100
74
+ eval_interval: 100
75
+ eval_num_batches: 100
76
+ per_latency_eval_num_batches: null
77
+ eval_action_classification: true
78
+ cc_f1_tolerance: 1
79
+ learning_rate:
80
+ base: 2.0e-05
81
+ qwen_vl_interface: 1.0e-05
82
+ action_model: 0.0001
83
+ lr_scheduler_type: cosine_with_min_lr
84
+ scheduler_specific_kwargs:
85
+ min_lr: 1.0e-06
86
+ freeze_modules: ''
87
+ loss_scale:
88
+ vla: 1.0
89
+ vlm: 0.1
90
+ max_grad_norm: 1.0
91
+ weight_decay: 0.0
92
+ logging_frequency: 1
93
+ gradient_clipping: 1.0
94
+ gradient_accumulation_steps: 4
95
+ distributed_backend: none
96
+ is_resume: false
97
+ pretrained_checkpoint: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
98
+ resume_step: 0
99
+ reload_modules: null
100
+ optimizer:
101
+ name: AdamW
102
+ betas:
103
+ - 0.9
104
+ - 0.95
105
+ eps: 1.0e-08
106
+ weight_decay: 1.0e-08
107
+ fused: false
108
+ save_format: pt
109
+ workspace_dir: WORKSPACE_DIR
110
+ run_root_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints
111
+ seed: 42
112
+ wandb_entity: dongqianyu99-zhejiang-university
113
+ wandb_project: starVLA_rl_games
114
+ auth:
115
+ env_file: null
116
+ hf_token_env: HF_TOKEN
117
+ wandb_api_key_env: WANDB_API_KEY
118
+ paths:
119
+ run_root_dir: results/Checkpoints
120
+ dataset_local_dir: playground/Datasets/rl_games
121
+ dataset_cache_dir: null
122
+ base_model_dir: ${bridge_base_model.local_dir.${model}}
123
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
124
+ launch:
125
+ use_accelerate: true
126
+ gpus: null
127
+ num_processes: 1
128
+ dry_run: false
129
+ conda:
130
+ enabled: true
131
+ env_name: null
132
+ rl_games:
133
+ model_alias: openvla
134
+ env_eval:
135
+ image_size: 224
136
+ frameskip: 4
137
+ seed: 42
138
+ fixed_episode_seeds: true
139
+ latency_seed_stride: 0
140
+ task_seed_stride: 0
141
+ task_description: 'You are playing Demon Attack from a single game image. Choose
142
+ exactly one action from: NOOP, FIRE, RIGHT, LEFT, RIGHTFIRE, LEFTFIRE.'
143
+ demon_attack:
144
+ noop_max: 30
145
+ enabled: true
146
+ distributed_mode: none
147
+ vectorized:
148
+ enabled: false
149
+ batch_size: 1
150
+ latency:
151
+ prompt_map_path: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_4_200ep/demon_attack_train__bridge/latency_prompt_map.json
152
+ mode: single
153
+ values:
154
+ - 0
155
+ mid_train:
156
+ enabled: false
157
+ interval_steps: 100
158
+ latencies:
159
+ - 0
160
+ num_episodes: 5
161
+ max_steps_per_episode: 3600
162
+ post_train:
163
+ enabled: false
164
+ latencies:
165
+ - 0
166
+ - 1
167
+ - 2
168
+ - 3
169
+ - 4
170
+ num_episodes: 5
171
+ max_steps_per_episode: 3600
172
+ task: demon_attack
173
+ initialization_mode: bridge
174
+ action_carrier: bridge
175
+ model: openvla
176
+ env: demon_attack
177
+ init: bridge
178
+ bridge_base_model:
179
+ repo_id:
180
+ openvla: Qwen/Qwen3-VL-4B-Instruct
181
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
182
+ pi05: Qwen/Qwen3-VL-4B-Instruct
183
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
184
+ local_dir:
185
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
186
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
187
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
188
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
189
+ mode: single
190
+ checkpoint:
191
+ load: auto
192
+ hf_repo_id: null
193
+ save_best_model: false
194
+ save_pt_file: false
195
+ local:
196
+ keep_last_n: 1
197
+ sync:
198
+ enabled: false
199
+ repo_id: null
200
+ keep_last_n: 0
201
+ sync_every_n_checkpoints: 1
202
+ resume_policy: local_latest
203
+ run_id: demon_attack_fix_latency_4_200ep
204
+ output_dir: null
205
+ config_yaml: null
206
+ is_debug: false
207
+ version_id: '0.21'
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/hydra/.hydra/hydra.yaml ADDED
@@ -0,0 +1,259 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ hydra:
2
+ run:
3
+ dir: ${run_root_dir}/${run_id}/hydra
4
+ sweep:
5
+ dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
6
+ subdir: ${hydra.job.num}
7
+ launcher:
8
+ _target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
9
+ sweeper:
10
+ _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
11
+ max_batch_size: null
12
+ params: null
13
+ help:
14
+ app_name: ${hydra.job.name}
15
+ header: '${hydra.help.app_name} is powered by Hydra.
16
+
17
+ '
18
+ footer: 'Powered by Hydra (https://hydra.cc)
19
+
20
+ Use --hydra-help to view Hydra specific help
21
+
22
+ '
23
+ template: '${hydra.help.header}
24
+
25
+ == Configuration groups ==
26
+
27
+ Compose your configuration from those groups (group=option)
28
+
29
+
30
+ $APP_CONFIG_GROUPS
31
+
32
+
33
+ == Config ==
34
+
35
+ Override anything in the config (foo.bar=value)
36
+
37
+
38
+ $CONFIG
39
+
40
+
41
+ ${hydra.help.footer}
42
+
43
+ '
44
+ hydra_help:
45
+ template: 'Hydra (${hydra.runtime.version})
46
+
47
+ See https://hydra.cc for more info.
48
+
49
+
50
+ == Flags ==
51
+
52
+ $FLAGS_HELP
53
+
54
+
55
+ == Configuration groups ==
56
+
57
+ Compose your configuration from those groups (For example, append hydra/job_logging=disabled
58
+ to command line)
59
+
60
+
61
+ $HYDRA_CONFIG_GROUPS
62
+
63
+
64
+ Use ''--cfg hydra'' to Show the Hydra config.
65
+
66
+ '
67
+ hydra_help: ???
68
+ hydra_logging:
69
+ version: 1
70
+ formatters:
71
+ simple:
72
+ format: '[%(asctime)s][HYDRA] %(message)s'
73
+ handlers:
74
+ console:
75
+ class: logging.StreamHandler
76
+ formatter: simple
77
+ stream: ext://sys.stdout
78
+ root:
79
+ level: INFO
80
+ handlers:
81
+ - console
82
+ loggers:
83
+ logging_example:
84
+ level: DEBUG
85
+ disable_existing_loggers: false
86
+ job_logging:
87
+ version: 1
88
+ formatters:
89
+ simple:
90
+ format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
91
+ handlers:
92
+ console:
93
+ class: logging.StreamHandler
94
+ formatter: simple
95
+ stream: ext://sys.stdout
96
+ file:
97
+ class: logging.FileHandler
98
+ formatter: simple
99
+ filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
100
+ root:
101
+ level: INFO
102
+ handlers:
103
+ - console
104
+ - file
105
+ disable_existing_loggers: false
106
+ env: {}
107
+ mode: RUN
108
+ searchpath: []
109
+ callbacks: {}
110
+ output_subdir: .hydra
111
+ overrides:
112
+ hydra:
113
+ - hydra.mode=RUN
114
+ task:
115
+ - model=openvla
116
+ - env=demon_attack
117
+ - init=bridge
118
+ - mode=single
119
+ - run_id=demon_attack_fix_latency_4_200ep
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=false
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=7000
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=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.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.latency_curriculum.enabled=false
178
+ - datasets.vla_data.latency_curriculum.strategy=exclusive
179
+ - datasets.vla_data.per_device_batch_size=32
180
+ - datasets.vla_data.load_all_data_for_training=true
181
+ - datasets.vla_data.video_backend=torchvision_av
182
+ - datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_4_200ep
183
+ - datasets.vla_data.data_mix=demon_attack_train__bridge
184
+ - datasets.vla_data.eval_data_mix=demon_attack_train__bridge__val
185
+ - framework.qwenvl.base_vlm=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct
186
+ - rl_games.task=demon_attack
187
+ - rl_games.model_alias=openvla
188
+ - rl_games.initialization_mode=bridge
189
+ - rl_games.action_carrier=bridge
190
+ - rl_games.env_eval.distributed_mode=none
191
+ - rl_games.env_eval.latency.mode=single
192
+ - rl_games.env_eval.frameskip=4
193
+ - rl_games.env_eval.image_size=224
194
+ - rl_games.env_eval.seed=42
195
+ - rl_games.env_eval.fixed_episode_seeds=true
196
+ - rl_games.env_eval.latency_seed_stride=0
197
+ - rl_games.env_eval.task_seed_stride=0
198
+ - 'rl_games.env_eval.task_description=''You are playing Demon Attack from a single
199
+ game image. Choose exactly one action from: NOOP, FIRE, RIGHT, LEFT, RIGHTFIRE,
200
+ LEFTFIRE.'''
201
+ - rl_games.env_eval.latency.values=[0]
202
+ - rl_games.env_eval.latency.prompt_map_path=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_4_200ep/demon_attack_train__bridge/latency_prompt_map.json
203
+ - rl_games.env_eval.mid_train.enabled=false
204
+ - rl_games.env_eval.mid_train.interval_steps=100
205
+ - rl_games.env_eval.mid_train.num_episodes=5
206
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
207
+ - rl_games.env_eval.mid_train.latencies=[0]
208
+ - rl_games.env_eval.post_train.enabled=false
209
+ - rl_games.env_eval.post_train.num_episodes=5
210
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
211
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
212
+ job:
213
+ name: train_starvla_hydra
214
+ chdir: false
215
+ 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=demon_attack_train__bridge,datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_4_200ep,datasets.vla_data.eval_data_mix=demon_attack_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=32,datasets.vla_data.sequential_step_sampling=false,datasets.vla_data.shuffle=true,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=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct,framework.qwenvl.enable_gradient_checkpointing=true,init=bridge,mode=single,model=openvla,rl_games.action_carrier=bridge,rl_games.env_eval.distributed_mode=none,rl_games.env_eval.enabled=true,rl_games.env_eval.fixed_episode_seeds=true,rl_games.env_eval.frameskip=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/demon_attack_fix_latency_4_200ep/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=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
216
+ are playing Demon Attack from a single game image. Choose exactly one action
217
+ 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=demon_attack_fix_latency_4_200ep,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=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=7000,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=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt,trainer.resume_step=0,trainer.save_format=pt,trainer.save_interval=100,trainer.scheduler_specific_kwargs.min_lr=1e-06,trainer.weight_decay=0.0,wandb_entity=dongqianyu99-zhejiang-university,wandb_project=starVLA_rl_games'
218
+ id: ???
219
+ num: ???
220
+ config_name: train
221
+ env_set: {}
222
+ env_copy: []
223
+ config:
224
+ override_dirname:
225
+ kv_sep: '='
226
+ item_sep: ','
227
+ exclude_keys: []
228
+ runtime:
229
+ version: 1.3.2
230
+ version_base: '1.1'
231
+ cwd: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA
232
+ config_sources:
233
+ - path: hydra.conf
234
+ schema: pkg
235
+ provider: hydra
236
+ - path: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA/examples/rl_games/config
237
+ schema: file
238
+ provider: main
239
+ - path: ''
240
+ schema: structured
241
+ provider: schema
242
+ output_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints/demon_attack_fix_latency_4_200ep/hydra
243
+ choices:
244
+ cross_task_setup: null
245
+ checkpoint: default
246
+ mode: single
247
+ init: bridge
248
+ env: demon_attack
249
+ model: openvla
250
+ hydra/env: default
251
+ hydra/callbacks: null
252
+ hydra/job_logging: default
253
+ hydra/hydra_logging: default
254
+ hydra/hydra_help: default
255
+ hydra/help: default
256
+ hydra/sweeper: basic
257
+ hydra/launcher: basic
258
+ hydra/output: default
259
+ verbose: false
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/hydra/.hydra/overrides.yaml ADDED
@@ -0,0 +1,97 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ - model=openvla
2
+ - env=demon_attack
3
+ - init=bridge
4
+ - mode=single
5
+ - run_id=demon_attack_fix_latency_4_200ep
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=false
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=7000
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=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.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.latency_curriculum.enabled=false
64
+ - datasets.vla_data.latency_curriculum.strategy=exclusive
65
+ - datasets.vla_data.per_device_batch_size=32
66
+ - datasets.vla_data.load_all_data_for_training=true
67
+ - datasets.vla_data.video_backend=torchvision_av
68
+ - datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_4_200ep
69
+ - datasets.vla_data.data_mix=demon_attack_train__bridge
70
+ - datasets.vla_data.eval_data_mix=demon_attack_train__bridge__val
71
+ - framework.qwenvl.base_vlm=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct
72
+ - rl_games.task=demon_attack
73
+ - rl_games.model_alias=openvla
74
+ - rl_games.initialization_mode=bridge
75
+ - rl_games.action_carrier=bridge
76
+ - rl_games.env_eval.distributed_mode=none
77
+ - rl_games.env_eval.latency.mode=single
78
+ - rl_games.env_eval.frameskip=4
79
+ - rl_games.env_eval.image_size=224
80
+ - rl_games.env_eval.seed=42
81
+ - rl_games.env_eval.fixed_episode_seeds=true
82
+ - rl_games.env_eval.latency_seed_stride=0
83
+ - rl_games.env_eval.task_seed_stride=0
84
+ - 'rl_games.env_eval.task_description=''You are playing Demon Attack from a single
85
+ game image. Choose exactly one action from: NOOP, FIRE, RIGHT, LEFT, RIGHTFIRE,
86
+ LEFTFIRE.'''
87
+ - rl_games.env_eval.latency.values=[0]
88
+ - rl_games.env_eval.latency.prompt_map_path=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_4_200ep/demon_attack_train__bridge/latency_prompt_map.json
89
+ - rl_games.env_eval.mid_train.enabled=false
90
+ - rl_games.env_eval.mid_train.interval_steps=100
91
+ - rl_games.env_eval.mid_train.num_episodes=5
92
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
93
+ - rl_games.env_eval.mid_train.latencies=[0]
94
+ - rl_games.env_eval.post_train.enabled=false
95
+ - rl_games.env_eval.post_train.num_episodes=5
96
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
97
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/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/demon_attack_fix_latency_4_200ep/summary.jsonl ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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}
51
+ {"steps": 5100}
52
+ {"steps": 5200}
53
+ {"steps": 5300}
54
+ {"steps": 5400}
55
+ {"steps": 5500}
56
+ {"steps": 5600}
57
+ {"steps": 5700}
58
+ {"steps": 5800}
59
+ {"steps": 5900}
60
+ {"steps": 6000}
61
+ {"steps": 6100}
62
+ {"steps": 6200}
63
+ {"steps": 6300}
64
+ {"steps": 6400}
65
+ {"steps": 6500}
66
+ {"steps": 6600}
67
+ {"steps": 6700}
68
+ {"steps": 6800}
69
+ {"steps": 6900}
70
+ {"steps": 7000}
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/checkpoints/steps_7000_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:375ff9f24479e219e8197bbc2eee3e71e61e1853e0c80eefe438ca947d5c5b1b
3
+ size 9138230516
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/config.full.yaml ADDED
@@ -0,0 +1,209 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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_6_200ep
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
+ 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
+ latency_curriculum:
36
+ enabled: false
37
+ strategy: exclusive
38
+ latencies: null
39
+ phase_steps: null
40
+ per_device_batch_size: 32
41
+ load_all_data_for_training: true
42
+ obs_image_size: null
43
+ video_backend: torchvision_av
44
+ dataset:
45
+ source_hf: ''
46
+ config_name: null
47
+ source_subdir: null
48
+ converted_name: demon_attack_train
49
+ single_source_hf: ''
50
+ mixed_source_hf: ''
51
+ single_converted_name: demon_attack_train
52
+ mixed_converted_name: demon_attack_mixed_latency_train
53
+ single_latency_filter: null
54
+ mixed_latency_filter: null
55
+ force_download: false
56
+ setup_force: false
57
+ skip_verification: false
58
+ verify_rows: 200
59
+ max_episodes: null
60
+ episodes_per_latency: null
61
+ latency_filter: null
62
+ debug_subset:
63
+ enabled: false
64
+ max_episodes: 5
65
+ suffix: debug
66
+ base_model:
67
+ repo_id: Qwen/Qwen3-VL-4B-Instruct
68
+ initialization:
69
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
70
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
71
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
72
+ trainer:
73
+ max_train_steps: 7000
74
+ num_warmup_steps: 100
75
+ save_interval: 100
76
+ eval_interval: 100
77
+ eval_num_batches: 100
78
+ per_latency_eval_num_batches: null
79
+ eval_action_classification: true
80
+ cc_f1_tolerance: 1
81
+ learning_rate:
82
+ base: 2.0e-05
83
+ qwen_vl_interface: 1.0e-05
84
+ action_model: 0.0001
85
+ lr_scheduler_type: cosine_with_min_lr
86
+ scheduler_specific_kwargs:
87
+ min_lr: 1.0e-06
88
+ freeze_modules: ''
89
+ loss_scale:
90
+ vla: 1.0
91
+ vlm: 0.1
92
+ max_grad_norm: 1.0
93
+ weight_decay: 0.0
94
+ logging_frequency: 1
95
+ gradient_clipping: 1.0
96
+ gradient_accumulation_steps: 4
97
+ distributed_backend: none
98
+ is_resume: false
99
+ pretrained_checkpoint: null
100
+ resume_step: 0
101
+ reload_modules: null
102
+ optimizer:
103
+ name: AdamW
104
+ betas:
105
+ - 0.9
106
+ - 0.95
107
+ eps: 1.0e-08
108
+ weight_decay: 1.0e-08
109
+ fused: false
110
+ save_format: pt
111
+ workspace_dir: WORKSPACE_DIR
112
+ run_root_dir: /workspace/starVLA/results/Checkpoints
113
+ seed: 42
114
+ wandb_entity: dongqianyu99-zhejiang-university
115
+ wandb_project: starVLA_rl_games
116
+ auth:
117
+ env_file: null
118
+ hf_token_env: HF_TOKEN
119
+ wandb_api_key_env: WANDB_API_KEY
120
+ paths:
121
+ run_root_dir: results/Checkpoints
122
+ dataset_local_dir: playground/Datasets/rl_games
123
+ dataset_cache_dir: null
124
+ base_model_dir: playground/Pretrained_models/Qwen3-VL-4B-Instruct
125
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
126
+ launch:
127
+ use_accelerate: true
128
+ gpus: null
129
+ num_processes: 1
130
+ dry_run: false
131
+ conda:
132
+ enabled: true
133
+ env_name: null
134
+ rl_games:
135
+ model_alias: openvla
136
+ env_eval:
137
+ image_size: 224
138
+ frameskip: 4
139
+ seed: 42
140
+ fixed_episode_seeds: true
141
+ latency_seed_stride: 0
142
+ task_seed_stride: 0
143
+ task_description: 'You are playing Demon Attack from a single game image. Choose
144
+ exactly one action from: NOOP, FIRE, RIGHT, LEFT, RIGHTFIRE, LEFTFIRE.'
145
+ demon_attack:
146
+ noop_max: 30
147
+ enabled: true
148
+ distributed_mode: none
149
+ vectorized:
150
+ enabled: false
151
+ batch_size: 1
152
+ latency:
153
+ prompt_map_path: /workspace/starVLA/data/demon_attack_fix_latency_6_200ep/demon_attack_train__bridge/latency_prompt_map.json
154
+ mode: single
155
+ values:
156
+ - 0
157
+ mid_train:
158
+ enabled: false
159
+ interval_steps: 100
160
+ latencies:
161
+ - 0
162
+ num_episodes: 5
163
+ max_steps_per_episode: 3600
164
+ post_train:
165
+ enabled: false
166
+ latencies:
167
+ - 0
168
+ - 1
169
+ - 2
170
+ - 3
171
+ - 4
172
+ num_episodes: 5
173
+ max_steps_per_episode: 3600
174
+ task: demon_attack
175
+ initialization_mode: bridge
176
+ action_carrier: bridge
177
+ model: openvla
178
+ env: demon_attack
179
+ init: bridge
180
+ bridge_base_model:
181
+ repo_id:
182
+ openvla: Qwen/Qwen3-VL-4B-Instruct
183
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
184
+ pi05: Qwen/Qwen3-VL-4B-Instruct
185
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
186
+ local_dir:
187
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
188
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
189
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
190
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
191
+ mode: single
192
+ checkpoint:
193
+ load: auto
194
+ hf_repo_id: null
195
+ save_best_model: false
196
+ save_pt_file: false
197
+ local:
198
+ keep_last_n: 1
199
+ sync:
200
+ enabled: false
201
+ repo_id: null
202
+ keep_last_n: 0
203
+ sync_every_n_checkpoints: 1
204
+ resume_policy: local_latest
205
+ run_id: demon_attack_fix_latency_6_200ep
206
+ output_dir: /workspace/starVLA/results/Checkpoints/demon_attack_fix_latency_6_200ep
207
+ config_yaml: null
208
+ is_debug: false
209
+ version_id: '0.21'
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/config.yaml ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: demon_attack_train__bridge
13
+ dataset_py: lerobot_datasets
14
+ eval_data_mix: demon_attack_train__bridge__val
15
+ latency_curriculum:
16
+ enabled: false
17
+ obs_image_size: null
18
+ per_device_batch_size: 32
19
+ framework:
20
+ action_model:
21
+ action_dim: 7
22
+ action_env_dim: 6
23
+ action_hidden_dim: 2560
24
+ action_horizon: 1
25
+ action_model_type: MLP
26
+ loss_type: discrete_ce
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/demon_attack_fix_latency_6_200ep
33
+ rl_games:
34
+ env_eval:
35
+ enabled: true
36
+ mid_train:
37
+ enabled: false
38
+ interval_steps: 100
39
+ task: demon_attack
40
+ run_id: demon_attack_fix_latency_6_200ep
41
+ run_root_dir: /workspace/starVLA/results/Checkpoints
42
+ seed: 42
43
+ trainer:
44
+ cc_f1_tolerance: 1
45
+ distributed_backend: none
46
+ eval_action_classification: true
47
+ eval_interval: 100
48
+ eval_num_batches: 100
49
+ freeze_modules: ''
50
+ gradient_accumulation_steps: 4
51
+ gradient_clipping: 1.0
52
+ is_resume: false
53
+ learning_rate:
54
+ action_model: 0.0001
55
+ base: 2.0e-05
56
+ qwen_vl_interface: 1.0e-05
57
+ logging_frequency: 1
58
+ lr_scheduler_type: cosine_with_min_lr
59
+ max_train_steps: 7000
60
+ num_warmup_steps: 100
61
+ optimizer:
62
+ betas:
63
+ - 0.9
64
+ - 0.95
65
+ eps: 1.0e-08
66
+ fused: false
67
+ weight_decay: 1.0e-08
68
+ per_latency_eval_num_batches: null
69
+ pretrained_checkpoint: null
70
+ reload_modules: null
71
+ save_interval: 100
72
+ scheduler_specific_kwargs:
73
+ min_lr: 1.0e-06
74
+ wandb_entity: dongqianyu99-zhejiang-university
75
+ wandb_project: starVLA_rl_games
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/dataset_statistics.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.06990445405244827,
6
+ 0.6139481067657471,
7
+ 0.008142709732055664,
8
+ 0.061076752841472626,
9
+ 0.1516113430261612,
10
+ 0.09531665593385696,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.2553316354751587,
15
+ 0.48614779114723206,
16
+ 0.08989176899194717,
17
+ 0.23972871899604797,
18
+ 0.3591064512729645,
19
+ 0.29394957423210144,
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
+ 0.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": 310953,
125
+ "num_trajectories": 180
126
+ }
127
+ }
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/dataset_statistics_eval.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.07470002770423889,
6
+ 0.6113184094429016,
7
+ 0.008066286332905293,
8
+ 0.058384548872709274,
9
+ 0.15318627655506134,
10
+ 0.09434445202350616,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.2629486620426178,
15
+ 0.48751503229141235,
16
+ 0.08943457156419754,
17
+ 0.23448707163333893,
18
+ 0.36014819145202637,
19
+ 0.2922261357307434,
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
+ 0.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": 54672,
125
+ "num_trajectories": 20
126
+ }
127
+ }
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/hydra/.hydra/config.yaml ADDED
@@ -0,0 +1,207 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: 6
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/demon_attack_fix_latency_6_200ep
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
+ 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
+ latency_curriculum:
34
+ enabled: false
35
+ strategy: exclusive
36
+ latencies: null
37
+ phase_steps: null
38
+ per_device_batch_size: 32
39
+ load_all_data_for_training: true
40
+ obs_image_size: null
41
+ video_backend: torchvision_av
42
+ dataset:
43
+ source_hf: ${dataset.single_source_hf}
44
+ config_name: null
45
+ source_subdir: null
46
+ converted_name: ${dataset.single_converted_name}
47
+ single_source_hf: ''
48
+ mixed_source_hf: ''
49
+ single_converted_name: demon_attack_train
50
+ mixed_converted_name: demon_attack_mixed_latency_train
51
+ single_latency_filter: null
52
+ mixed_latency_filter: null
53
+ force_download: false
54
+ setup_force: false
55
+ skip_verification: false
56
+ verify_rows: 200
57
+ max_episodes: null
58
+ episodes_per_latency: null
59
+ latency_filter: ${dataset.single_latency_filter}
60
+ debug_subset:
61
+ enabled: false
62
+ max_episodes: 5
63
+ suffix: debug
64
+ base_model:
65
+ repo_id: ${bridge_base_model.repo_id.${model}}
66
+ initialization:
67
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
68
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
69
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
70
+ trainer:
71
+ max_train_steps: 7000
72
+ num_warmup_steps: 100
73
+ save_interval: 100
74
+ eval_interval: 100
75
+ eval_num_batches: 100
76
+ per_latency_eval_num_batches: null
77
+ eval_action_classification: true
78
+ cc_f1_tolerance: 1
79
+ learning_rate:
80
+ base: 2.0e-05
81
+ qwen_vl_interface: 1.0e-05
82
+ action_model: 0.0001
83
+ lr_scheduler_type: cosine_with_min_lr
84
+ scheduler_specific_kwargs:
85
+ min_lr: 1.0e-06
86
+ freeze_modules: ''
87
+ loss_scale:
88
+ vla: 1.0
89
+ vlm: 0.1
90
+ max_grad_norm: 1.0
91
+ weight_decay: 0.0
92
+ logging_frequency: 1
93
+ gradient_clipping: 1.0
94
+ gradient_accumulation_steps: 4
95
+ distributed_backend: none
96
+ is_resume: false
97
+ pretrained_checkpoint: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
98
+ resume_step: 0
99
+ reload_modules: null
100
+ optimizer:
101
+ name: AdamW
102
+ betas:
103
+ - 0.9
104
+ - 0.95
105
+ eps: 1.0e-08
106
+ weight_decay: 1.0e-08
107
+ fused: false
108
+ save_format: pt
109
+ workspace_dir: WORKSPACE_DIR
110
+ run_root_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints
111
+ seed: 42
112
+ wandb_entity: dongqianyu99-zhejiang-university
113
+ wandb_project: starVLA_rl_games
114
+ auth:
115
+ env_file: null
116
+ hf_token_env: HF_TOKEN
117
+ wandb_api_key_env: WANDB_API_KEY
118
+ paths:
119
+ run_root_dir: results/Checkpoints
120
+ dataset_local_dir: playground/Datasets/rl_games
121
+ dataset_cache_dir: null
122
+ base_model_dir: ${bridge_base_model.local_dir.${model}}
123
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
124
+ launch:
125
+ use_accelerate: true
126
+ gpus: null
127
+ num_processes: 1
128
+ dry_run: false
129
+ conda:
130
+ enabled: true
131
+ env_name: null
132
+ rl_games:
133
+ model_alias: openvla
134
+ env_eval:
135
+ image_size: 224
136
+ frameskip: 4
137
+ seed: 42
138
+ fixed_episode_seeds: true
139
+ latency_seed_stride: 0
140
+ task_seed_stride: 0
141
+ task_description: 'You are playing Demon Attack from a single game image. Choose
142
+ exactly one action from: NOOP, FIRE, RIGHT, LEFT, RIGHTFIRE, LEFTFIRE.'
143
+ demon_attack:
144
+ noop_max: 30
145
+ enabled: true
146
+ distributed_mode: none
147
+ vectorized:
148
+ enabled: false
149
+ batch_size: 1
150
+ latency:
151
+ prompt_map_path: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_6_200ep/demon_attack_train__bridge/latency_prompt_map.json
152
+ mode: single
153
+ values:
154
+ - 0
155
+ mid_train:
156
+ enabled: false
157
+ interval_steps: 100
158
+ latencies:
159
+ - 0
160
+ num_episodes: 5
161
+ max_steps_per_episode: 3600
162
+ post_train:
163
+ enabled: false
164
+ latencies:
165
+ - 0
166
+ - 1
167
+ - 2
168
+ - 3
169
+ - 4
170
+ num_episodes: 5
171
+ max_steps_per_episode: 3600
172
+ task: demon_attack
173
+ initialization_mode: bridge
174
+ action_carrier: bridge
175
+ model: openvla
176
+ env: demon_attack
177
+ init: bridge
178
+ bridge_base_model:
179
+ repo_id:
180
+ openvla: Qwen/Qwen3-VL-4B-Instruct
181
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
182
+ pi05: Qwen/Qwen3-VL-4B-Instruct
183
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
184
+ local_dir:
185
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
186
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
187
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
188
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
189
+ mode: single
190
+ checkpoint:
191
+ load: auto
192
+ hf_repo_id: null
193
+ save_best_model: false
194
+ save_pt_file: false
195
+ local:
196
+ keep_last_n: 1
197
+ sync:
198
+ enabled: false
199
+ repo_id: null
200
+ keep_last_n: 0
201
+ sync_every_n_checkpoints: 1
202
+ resume_policy: local_latest
203
+ run_id: demon_attack_fix_latency_6_200ep
204
+ output_dir: null
205
+ config_yaml: null
206
+ is_debug: false
207
+ version_id: '0.21'
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/hydra/.hydra/hydra.yaml ADDED
@@ -0,0 +1,259 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ hydra:
2
+ run:
3
+ dir: ${run_root_dir}/${run_id}/hydra
4
+ sweep:
5
+ dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
6
+ subdir: ${hydra.job.num}
7
+ launcher:
8
+ _target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
9
+ sweeper:
10
+ _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
11
+ max_batch_size: null
12
+ params: null
13
+ help:
14
+ app_name: ${hydra.job.name}
15
+ header: '${hydra.help.app_name} is powered by Hydra.
16
+
17
+ '
18
+ footer: 'Powered by Hydra (https://hydra.cc)
19
+
20
+ Use --hydra-help to view Hydra specific help
21
+
22
+ '
23
+ template: '${hydra.help.header}
24
+
25
+ == Configuration groups ==
26
+
27
+ Compose your configuration from those groups (group=option)
28
+
29
+
30
+ $APP_CONFIG_GROUPS
31
+
32
+
33
+ == Config ==
34
+
35
+ Override anything in the config (foo.bar=value)
36
+
37
+
38
+ $CONFIG
39
+
40
+
41
+ ${hydra.help.footer}
42
+
43
+ '
44
+ hydra_help:
45
+ template: 'Hydra (${hydra.runtime.version})
46
+
47
+ See https://hydra.cc for more info.
48
+
49
+
50
+ == Flags ==
51
+
52
+ $FLAGS_HELP
53
+
54
+
55
+ == Configuration groups ==
56
+
57
+ Compose your configuration from those groups (For example, append hydra/job_logging=disabled
58
+ to command line)
59
+
60
+
61
+ $HYDRA_CONFIG_GROUPS
62
+
63
+
64
+ Use ''--cfg hydra'' to Show the Hydra config.
65
+
66
+ '
67
+ hydra_help: ???
68
+ hydra_logging:
69
+ version: 1
70
+ formatters:
71
+ simple:
72
+ format: '[%(asctime)s][HYDRA] %(message)s'
73
+ handlers:
74
+ console:
75
+ class: logging.StreamHandler
76
+ formatter: simple
77
+ stream: ext://sys.stdout
78
+ root:
79
+ level: INFO
80
+ handlers:
81
+ - console
82
+ loggers:
83
+ logging_example:
84
+ level: DEBUG
85
+ disable_existing_loggers: false
86
+ job_logging:
87
+ version: 1
88
+ formatters:
89
+ simple:
90
+ format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
91
+ handlers:
92
+ console:
93
+ class: logging.StreamHandler
94
+ formatter: simple
95
+ stream: ext://sys.stdout
96
+ file:
97
+ class: logging.FileHandler
98
+ formatter: simple
99
+ filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
100
+ root:
101
+ level: INFO
102
+ handlers:
103
+ - console
104
+ - file
105
+ disable_existing_loggers: false
106
+ env: {}
107
+ mode: RUN
108
+ searchpath: []
109
+ callbacks: {}
110
+ output_subdir: .hydra
111
+ overrides:
112
+ hydra:
113
+ - hydra.mode=RUN
114
+ task:
115
+ - model=openvla
116
+ - env=demon_attack
117
+ - init=bridge
118
+ - mode=single
119
+ - run_id=demon_attack_fix_latency_6_200ep
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=false
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=7000
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=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.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.latency_curriculum.enabled=false
178
+ - datasets.vla_data.latency_curriculum.strategy=exclusive
179
+ - datasets.vla_data.per_device_batch_size=32
180
+ - datasets.vla_data.load_all_data_for_training=true
181
+ - datasets.vla_data.video_backend=torchvision_av
182
+ - datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_6_200ep
183
+ - datasets.vla_data.data_mix=demon_attack_train__bridge
184
+ - datasets.vla_data.eval_data_mix=demon_attack_train__bridge__val
185
+ - framework.qwenvl.base_vlm=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct
186
+ - rl_games.task=demon_attack
187
+ - rl_games.model_alias=openvla
188
+ - rl_games.initialization_mode=bridge
189
+ - rl_games.action_carrier=bridge
190
+ - rl_games.env_eval.distributed_mode=none
191
+ - rl_games.env_eval.latency.mode=single
192
+ - rl_games.env_eval.frameskip=4
193
+ - rl_games.env_eval.image_size=224
194
+ - rl_games.env_eval.seed=42
195
+ - rl_games.env_eval.fixed_episode_seeds=true
196
+ - rl_games.env_eval.latency_seed_stride=0
197
+ - rl_games.env_eval.task_seed_stride=0
198
+ - 'rl_games.env_eval.task_description=''You are playing Demon Attack from a single
199
+ game image. Choose exactly one action from: NOOP, FIRE, RIGHT, LEFT, RIGHTFIRE,
200
+ LEFTFIRE.'''
201
+ - rl_games.env_eval.latency.values=[0]
202
+ - rl_games.env_eval.latency.prompt_map_path=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_6_200ep/demon_attack_train__bridge/latency_prompt_map.json
203
+ - rl_games.env_eval.mid_train.enabled=false
204
+ - rl_games.env_eval.mid_train.interval_steps=100
205
+ - rl_games.env_eval.mid_train.num_episodes=5
206
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
207
+ - rl_games.env_eval.mid_train.latencies=[0]
208
+ - rl_games.env_eval.post_train.enabled=false
209
+ - rl_games.env_eval.post_train.num_episodes=5
210
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
211
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
212
+ job:
213
+ name: train_starvla_hydra
214
+ chdir: false
215
+ 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=demon_attack_train__bridge,datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_6_200ep,datasets.vla_data.eval_data_mix=demon_attack_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=32,datasets.vla_data.sequential_step_sampling=false,datasets.vla_data.shuffle=true,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=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct,framework.qwenvl.enable_gradient_checkpointing=true,init=bridge,mode=single,model=openvla,rl_games.action_carrier=bridge,rl_games.env_eval.distributed_mode=none,rl_games.env_eval.enabled=true,rl_games.env_eval.fixed_episode_seeds=true,rl_games.env_eval.frameskip=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/demon_attack_fix_latency_6_200ep/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=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
216
+ are playing Demon Attack from a single game image. Choose exactly one action
217
+ 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=demon_attack_fix_latency_6_200ep,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=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=7000,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=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt,trainer.resume_step=0,trainer.save_format=pt,trainer.save_interval=100,trainer.scheduler_specific_kwargs.min_lr=1e-06,trainer.weight_decay=0.0,wandb_entity=dongqianyu99-zhejiang-university,wandb_project=starVLA_rl_games'
218
+ id: ???
219
+ num: ???
220
+ config_name: train
221
+ env_set: {}
222
+ env_copy: []
223
+ config:
224
+ override_dirname:
225
+ kv_sep: '='
226
+ item_sep: ','
227
+ exclude_keys: []
228
+ runtime:
229
+ version: 1.3.2
230
+ version_base: '1.1'
231
+ cwd: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA
232
+ config_sources:
233
+ - path: hydra.conf
234
+ schema: pkg
235
+ provider: hydra
236
+ - path: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA/examples/rl_games/config
237
+ schema: file
238
+ provider: main
239
+ - path: ''
240
+ schema: structured
241
+ provider: schema
242
+ output_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints/demon_attack_fix_latency_6_200ep/hydra
243
+ choices:
244
+ cross_task_setup: null
245
+ checkpoint: default
246
+ mode: single
247
+ init: bridge
248
+ env: demon_attack
249
+ model: openvla
250
+ hydra/env: default
251
+ hydra/callbacks: null
252
+ hydra/job_logging: default
253
+ hydra/hydra_logging: default
254
+ hydra/hydra_help: default
255
+ hydra/help: default
256
+ hydra/sweeper: basic
257
+ hydra/launcher: basic
258
+ hydra/output: default
259
+ verbose: false
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/hydra/.hydra/overrides.yaml ADDED
@@ -0,0 +1,97 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ - model=openvla
2
+ - env=demon_attack
3
+ - init=bridge
4
+ - mode=single
5
+ - run_id=demon_attack_fix_latency_6_200ep
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=false
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=7000
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=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.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.latency_curriculum.enabled=false
64
+ - datasets.vla_data.latency_curriculum.strategy=exclusive
65
+ - datasets.vla_data.per_device_batch_size=32
66
+ - datasets.vla_data.load_all_data_for_training=true
67
+ - datasets.vla_data.video_backend=torchvision_av
68
+ - datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_6_200ep
69
+ - datasets.vla_data.data_mix=demon_attack_train__bridge
70
+ - datasets.vla_data.eval_data_mix=demon_attack_train__bridge__val
71
+ - framework.qwenvl.base_vlm=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct
72
+ - rl_games.task=demon_attack
73
+ - rl_games.model_alias=openvla
74
+ - rl_games.initialization_mode=bridge
75
+ - rl_games.action_carrier=bridge
76
+ - rl_games.env_eval.distributed_mode=none
77
+ - rl_games.env_eval.latency.mode=single
78
+ - rl_games.env_eval.frameskip=4
79
+ - rl_games.env_eval.image_size=224
80
+ - rl_games.env_eval.seed=42
81
+ - rl_games.env_eval.fixed_episode_seeds=true
82
+ - rl_games.env_eval.latency_seed_stride=0
83
+ - rl_games.env_eval.task_seed_stride=0
84
+ - 'rl_games.env_eval.task_description=''You are playing Demon Attack from a single
85
+ game image. Choose exactly one action from: NOOP, FIRE, RIGHT, LEFT, RIGHTFIRE,
86
+ LEFTFIRE.'''
87
+ - rl_games.env_eval.latency.values=[0]
88
+ - rl_games.env_eval.latency.prompt_map_path=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_6_200ep/demon_attack_train__bridge/latency_prompt_map.json
89
+ - rl_games.env_eval.mid_train.enabled=false
90
+ - rl_games.env_eval.mid_train.interval_steps=100
91
+ - rl_games.env_eval.mid_train.num_episodes=5
92
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
93
+ - rl_games.env_eval.mid_train.latencies=[0]
94
+ - rl_games.env_eval.post_train.enabled=false
95
+ - rl_games.env_eval.post_train.num_episodes=5
96
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
97
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/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/demon_attack_fix_latency_6_200ep/summary.jsonl ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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}
51
+ {"steps": 5100}
52
+ {"steps": 5200}
53
+ {"steps": 5300}
54
+ {"steps": 5400}
55
+ {"steps": 5500}
56
+ {"steps": 5600}
57
+ {"steps": 5700}
58
+ {"steps": 5800}
59
+ {"steps": 5900}
60
+ {"steps": 6000}
61
+ {"steps": 6100}
62
+ {"steps": 6200}
63
+ {"steps": 6300}
64
+ {"steps": 6400}
65
+ {"steps": 6500}
66
+ {"steps": 6600}
67
+ {"steps": 6700}
68
+ {"steps": 6800}
69
+ {"steps": 6900}
70
+ {"steps": 7000}
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/checkpoints/steps_7000_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0d52d791412c41f17f607087c988d99267ef09ac097022b3fc3d97ebbb5b0b00
3
+ size 9138230516
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/config.full.yaml ADDED
@@ -0,0 +1,209 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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_8_200ep
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
+ 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
+ latency_curriculum:
36
+ enabled: false
37
+ strategy: exclusive
38
+ latencies: null
39
+ phase_steps: null
40
+ per_device_batch_size: 32
41
+ load_all_data_for_training: true
42
+ obs_image_size: null
43
+ video_backend: torchvision_av
44
+ dataset:
45
+ source_hf: ''
46
+ config_name: null
47
+ source_subdir: null
48
+ converted_name: demon_attack_train
49
+ single_source_hf: ''
50
+ mixed_source_hf: ''
51
+ single_converted_name: demon_attack_train
52
+ mixed_converted_name: demon_attack_mixed_latency_train
53
+ single_latency_filter: null
54
+ mixed_latency_filter: null
55
+ force_download: false
56
+ setup_force: false
57
+ skip_verification: false
58
+ verify_rows: 200
59
+ max_episodes: null
60
+ episodes_per_latency: null
61
+ latency_filter: null
62
+ debug_subset:
63
+ enabled: false
64
+ max_episodes: 5
65
+ suffix: debug
66
+ base_model:
67
+ repo_id: Qwen/Qwen3-VL-4B-Instruct
68
+ initialization:
69
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
70
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
71
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
72
+ trainer:
73
+ max_train_steps: 7000
74
+ num_warmup_steps: 100
75
+ save_interval: 100
76
+ eval_interval: 100
77
+ eval_num_batches: 100
78
+ per_latency_eval_num_batches: null
79
+ eval_action_classification: true
80
+ cc_f1_tolerance: 1
81
+ learning_rate:
82
+ base: 2.0e-05
83
+ qwen_vl_interface: 1.0e-05
84
+ action_model: 0.0001
85
+ lr_scheduler_type: cosine_with_min_lr
86
+ scheduler_specific_kwargs:
87
+ min_lr: 1.0e-06
88
+ freeze_modules: ''
89
+ loss_scale:
90
+ vla: 1.0
91
+ vlm: 0.1
92
+ max_grad_norm: 1.0
93
+ weight_decay: 0.0
94
+ logging_frequency: 1
95
+ gradient_clipping: 1.0
96
+ gradient_accumulation_steps: 4
97
+ distributed_backend: none
98
+ is_resume: false
99
+ pretrained_checkpoint: null
100
+ resume_step: 0
101
+ reload_modules: null
102
+ optimizer:
103
+ name: AdamW
104
+ betas:
105
+ - 0.9
106
+ - 0.95
107
+ eps: 1.0e-08
108
+ weight_decay: 1.0e-08
109
+ fused: false
110
+ save_format: pt
111
+ workspace_dir: WORKSPACE_DIR
112
+ run_root_dir: /workspace/starVLA/results/Checkpoints
113
+ seed: 42
114
+ wandb_entity: dongqianyu99-zhejiang-university
115
+ wandb_project: starVLA_rl_games
116
+ auth:
117
+ env_file: null
118
+ hf_token_env: HF_TOKEN
119
+ wandb_api_key_env: WANDB_API_KEY
120
+ paths:
121
+ run_root_dir: results/Checkpoints
122
+ dataset_local_dir: playground/Datasets/rl_games
123
+ dataset_cache_dir: null
124
+ base_model_dir: playground/Pretrained_models/Qwen3-VL-4B-Instruct
125
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
126
+ launch:
127
+ use_accelerate: true
128
+ gpus: null
129
+ num_processes: 1
130
+ dry_run: false
131
+ conda:
132
+ enabled: true
133
+ env_name: null
134
+ rl_games:
135
+ model_alias: openvla
136
+ env_eval:
137
+ image_size: 224
138
+ frameskip: 4
139
+ seed: 42
140
+ fixed_episode_seeds: true
141
+ latency_seed_stride: 0
142
+ task_seed_stride: 0
143
+ task_description: 'You are playing Demon Attack from a single game image. Choose
144
+ exactly one action from: NOOP, FIRE, RIGHT, LEFT, RIGHTFIRE, LEFTFIRE.'
145
+ demon_attack:
146
+ noop_max: 30
147
+ enabled: true
148
+ distributed_mode: none
149
+ vectorized:
150
+ enabled: false
151
+ batch_size: 1
152
+ latency:
153
+ prompt_map_path: /workspace/starVLA/data/demon_attack_fix_latency_8_200ep/demon_attack_train__bridge/latency_prompt_map.json
154
+ mode: single
155
+ values:
156
+ - 0
157
+ mid_train:
158
+ enabled: false
159
+ interval_steps: 100
160
+ latencies:
161
+ - 0
162
+ num_episodes: 5
163
+ max_steps_per_episode: 3600
164
+ post_train:
165
+ enabled: false
166
+ latencies:
167
+ - 0
168
+ - 1
169
+ - 2
170
+ - 3
171
+ - 4
172
+ num_episodes: 5
173
+ max_steps_per_episode: 3600
174
+ task: demon_attack
175
+ initialization_mode: bridge
176
+ action_carrier: bridge
177
+ model: openvla
178
+ env: demon_attack
179
+ init: bridge
180
+ bridge_base_model:
181
+ repo_id:
182
+ openvla: Qwen/Qwen3-VL-4B-Instruct
183
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
184
+ pi05: Qwen/Qwen3-VL-4B-Instruct
185
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
186
+ local_dir:
187
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
188
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
189
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
190
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
191
+ mode: single
192
+ checkpoint:
193
+ load: auto
194
+ hf_repo_id: null
195
+ save_best_model: false
196
+ save_pt_file: false
197
+ local:
198
+ keep_last_n: 1
199
+ sync:
200
+ enabled: false
201
+ repo_id: null
202
+ keep_last_n: 0
203
+ sync_every_n_checkpoints: 1
204
+ resume_policy: local_latest
205
+ run_id: demon_attack_fix_latency_8_200ep
206
+ output_dir: /workspace/starVLA/results/Checkpoints/demon_attack_fix_latency_8_200ep
207
+ config_yaml: null
208
+ is_debug: false
209
+ version_id: '0.21'
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/config.yaml ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: demon_attack_train__bridge
13
+ dataset_py: lerobot_datasets
14
+ eval_data_mix: demon_attack_train__bridge__val
15
+ latency_curriculum:
16
+ enabled: false
17
+ obs_image_size: null
18
+ per_device_batch_size: 32
19
+ framework:
20
+ action_model:
21
+ action_dim: 7
22
+ action_env_dim: 6
23
+ action_hidden_dim: 2560
24
+ action_horizon: 1
25
+ action_model_type: MLP
26
+ loss_type: discrete_ce
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/demon_attack_fix_latency_8_200ep
33
+ rl_games:
34
+ env_eval:
35
+ enabled: true
36
+ mid_train:
37
+ enabled: false
38
+ interval_steps: 100
39
+ task: demon_attack
40
+ run_id: demon_attack_fix_latency_8_200ep
41
+ run_root_dir: /workspace/starVLA/results/Checkpoints
42
+ seed: 42
43
+ trainer:
44
+ cc_f1_tolerance: 1
45
+ distributed_backend: none
46
+ eval_action_classification: true
47
+ eval_interval: 100
48
+ eval_num_batches: 100
49
+ freeze_modules: ''
50
+ gradient_accumulation_steps: 4
51
+ gradient_clipping: 1.0
52
+ is_resume: false
53
+ learning_rate:
54
+ action_model: 0.0001
55
+ base: 2.0e-05
56
+ qwen_vl_interface: 1.0e-05
57
+ logging_frequency: 1
58
+ lr_scheduler_type: cosine_with_min_lr
59
+ max_train_steps: 7000
60
+ num_warmup_steps: 100
61
+ optimizer:
62
+ betas:
63
+ - 0.9
64
+ - 0.95
65
+ eps: 1.0e-08
66
+ fused: false
67
+ weight_decay: 1.0e-08
68
+ per_latency_eval_num_batches: null
69
+ pretrained_checkpoint: null
70
+ reload_modules: null
71
+ save_interval: 100
72
+ scheduler_specific_kwargs:
73
+ min_lr: 1.0e-06
74
+ wandb_entity: dongqianyu99-zhejiang-university
75
+ wandb_project: starVLA_rl_games
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/dataset_statistics.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.08639267086982727,
6
+ 0.627116858959198,
7
+ 0.006486095488071442,
8
+ 0.005628371611237526,
9
+ 0.148070827126503,
10
+ 0.12630519270896912,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.28125613927841187,
15
+ 0.48417720198631287,
16
+ 0.08015890419483185,
17
+ 0.07480393350124359,
18
+ 0.35502511262893677,
19
+ 0.33193621039390564,
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
+ 0.0,
53
+ 0.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": 282142,
125
+ "num_trajectories": 180
126
+ }
127
+ }
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/dataset_statistics_eval.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.09224364906549454,
6
+ 0.6070592999458313,
7
+ 0.007906598970293999,
8
+ 0.0057144262827932835,
9
+ 0.1609153002500534,
10
+ 0.12616074085235596,
11
+ 0.0
12
+ ],
13
+ "std": [
14
+ 0.2894008159637451,
15
+ 0.48834389448165894,
16
+ 0.08855940401554108,
17
+ 0.07536990940570831,
18
+ 0.36744311451911926,
19
+ 0.33201834559440613,
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
+ 0.0,
53
+ 0.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": 40599,
125
+ "num_trajectories": 20
126
+ }
127
+ }
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/hydra/.hydra/config.yaml ADDED
@@ -0,0 +1,207 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: 6
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/demon_attack_fix_latency_8_200ep
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
+ 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
+ latency_curriculum:
34
+ enabled: false
35
+ strategy: exclusive
36
+ latencies: null
37
+ phase_steps: null
38
+ per_device_batch_size: 32
39
+ load_all_data_for_training: true
40
+ obs_image_size: null
41
+ video_backend: torchvision_av
42
+ dataset:
43
+ source_hf: ${dataset.single_source_hf}
44
+ config_name: null
45
+ source_subdir: null
46
+ converted_name: ${dataset.single_converted_name}
47
+ single_source_hf: ''
48
+ mixed_source_hf: ''
49
+ single_converted_name: demon_attack_train
50
+ mixed_converted_name: demon_attack_mixed_latency_train
51
+ single_latency_filter: null
52
+ mixed_latency_filter: null
53
+ force_download: false
54
+ setup_force: false
55
+ skip_verification: false
56
+ verify_rows: 200
57
+ max_episodes: null
58
+ episodes_per_latency: null
59
+ latency_filter: ${dataset.single_latency_filter}
60
+ debug_subset:
61
+ enabled: false
62
+ max_episodes: 5
63
+ suffix: debug
64
+ base_model:
65
+ repo_id: ${bridge_base_model.repo_id.${model}}
66
+ initialization:
67
+ checkpoint_local_dir: playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1
68
+ checkpoint_hf_repo_id: StarVLA/Qwen3VL-OFT-Bridge-RT-1
69
+ checkpoint_filename: checkpoints/steps_5000_pytorch_model.pt
70
+ trainer:
71
+ max_train_steps: 7000
72
+ num_warmup_steps: 100
73
+ save_interval: 100
74
+ eval_interval: 100
75
+ eval_num_batches: 100
76
+ per_latency_eval_num_batches: null
77
+ eval_action_classification: true
78
+ cc_f1_tolerance: 1
79
+ learning_rate:
80
+ base: 2.0e-05
81
+ qwen_vl_interface: 1.0e-05
82
+ action_model: 0.0001
83
+ lr_scheduler_type: cosine_with_min_lr
84
+ scheduler_specific_kwargs:
85
+ min_lr: 1.0e-06
86
+ freeze_modules: ''
87
+ loss_scale:
88
+ vla: 1.0
89
+ vlm: 0.1
90
+ max_grad_norm: 1.0
91
+ weight_decay: 0.0
92
+ logging_frequency: 1
93
+ gradient_clipping: 1.0
94
+ gradient_accumulation_steps: 4
95
+ distributed_backend: none
96
+ is_resume: false
97
+ pretrained_checkpoint: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt
98
+ resume_step: 0
99
+ reload_modules: null
100
+ optimizer:
101
+ name: AdamW
102
+ betas:
103
+ - 0.9
104
+ - 0.95
105
+ eps: 1.0e-08
106
+ weight_decay: 1.0e-08
107
+ fused: false
108
+ save_format: pt
109
+ workspace_dir: WORKSPACE_DIR
110
+ run_root_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints
111
+ seed: 42
112
+ wandb_entity: dongqianyu99-zhejiang-university
113
+ wandb_project: starVLA_rl_games
114
+ auth:
115
+ env_file: null
116
+ hf_token_env: HF_TOKEN
117
+ wandb_api_key_env: WANDB_API_KEY
118
+ paths:
119
+ run_root_dir: results/Checkpoints
120
+ dataset_local_dir: playground/Datasets/rl_games
121
+ dataset_cache_dir: null
122
+ base_model_dir: ${bridge_base_model.local_dir.${model}}
123
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
124
+ launch:
125
+ use_accelerate: true
126
+ gpus: null
127
+ num_processes: 1
128
+ dry_run: false
129
+ conda:
130
+ enabled: true
131
+ env_name: null
132
+ rl_games:
133
+ model_alias: openvla
134
+ env_eval:
135
+ image_size: 224
136
+ frameskip: 4
137
+ seed: 42
138
+ fixed_episode_seeds: true
139
+ latency_seed_stride: 0
140
+ task_seed_stride: 0
141
+ task_description: 'You are playing Demon Attack from a single game image. Choose
142
+ exactly one action from: NOOP, FIRE, RIGHT, LEFT, RIGHTFIRE, LEFTFIRE.'
143
+ demon_attack:
144
+ noop_max: 30
145
+ enabled: true
146
+ distributed_mode: none
147
+ vectorized:
148
+ enabled: false
149
+ batch_size: 1
150
+ latency:
151
+ prompt_map_path: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_8_200ep/demon_attack_train__bridge/latency_prompt_map.json
152
+ mode: single
153
+ values:
154
+ - 0
155
+ mid_train:
156
+ enabled: false
157
+ interval_steps: 100
158
+ latencies:
159
+ - 0
160
+ num_episodes: 5
161
+ max_steps_per_episode: 3600
162
+ post_train:
163
+ enabled: false
164
+ latencies:
165
+ - 0
166
+ - 1
167
+ - 2
168
+ - 3
169
+ - 4
170
+ num_episodes: 5
171
+ max_steps_per_episode: 3600
172
+ task: demon_attack
173
+ initialization_mode: bridge
174
+ action_carrier: bridge
175
+ model: openvla
176
+ env: demon_attack
177
+ init: bridge
178
+ bridge_base_model:
179
+ repo_id:
180
+ openvla: Qwen/Qwen3-VL-4B-Instruct
181
+ pi0: StarVLA/Qwen2.5-VL-3B-Instruct-Action
182
+ pi05: Qwen/Qwen3-VL-4B-Instruct
183
+ gr00t: Qwen/Qwen3-VL-4B-Instruct
184
+ local_dir:
185
+ openvla: playground/Pretrained_models/Qwen3-VL-4B-Instruct
186
+ pi0: playground/Pretrained_models/Qwen2.5-VL-3B-Instruct-Action
187
+ pi05: playground/Pretrained_models/Qwen3-VL-4B-Instruct
188
+ gr00t: playground/Pretrained_models/Qwen3-VL-4B-Instruct
189
+ mode: single
190
+ checkpoint:
191
+ load: auto
192
+ hf_repo_id: null
193
+ save_best_model: false
194
+ save_pt_file: false
195
+ local:
196
+ keep_last_n: 1
197
+ sync:
198
+ enabled: false
199
+ repo_id: null
200
+ keep_last_n: 0
201
+ sync_every_n_checkpoints: 1
202
+ resume_policy: local_latest
203
+ run_id: demon_attack_fix_latency_8_200ep
204
+ output_dir: null
205
+ config_yaml: null
206
+ is_debug: false
207
+ version_id: '0.21'
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/hydra/.hydra/hydra.yaml ADDED
@@ -0,0 +1,259 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ hydra:
2
+ run:
3
+ dir: ${run_root_dir}/${run_id}/hydra
4
+ sweep:
5
+ dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
6
+ subdir: ${hydra.job.num}
7
+ launcher:
8
+ _target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
9
+ sweeper:
10
+ _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
11
+ max_batch_size: null
12
+ params: null
13
+ help:
14
+ app_name: ${hydra.job.name}
15
+ header: '${hydra.help.app_name} is powered by Hydra.
16
+
17
+ '
18
+ footer: 'Powered by Hydra (https://hydra.cc)
19
+
20
+ Use --hydra-help to view Hydra specific help
21
+
22
+ '
23
+ template: '${hydra.help.header}
24
+
25
+ == Configuration groups ==
26
+
27
+ Compose your configuration from those groups (group=option)
28
+
29
+
30
+ $APP_CONFIG_GROUPS
31
+
32
+
33
+ == Config ==
34
+
35
+ Override anything in the config (foo.bar=value)
36
+
37
+
38
+ $CONFIG
39
+
40
+
41
+ ${hydra.help.footer}
42
+
43
+ '
44
+ hydra_help:
45
+ template: 'Hydra (${hydra.runtime.version})
46
+
47
+ See https://hydra.cc for more info.
48
+
49
+
50
+ == Flags ==
51
+
52
+ $FLAGS_HELP
53
+
54
+
55
+ == Configuration groups ==
56
+
57
+ Compose your configuration from those groups (For example, append hydra/job_logging=disabled
58
+ to command line)
59
+
60
+
61
+ $HYDRA_CONFIG_GROUPS
62
+
63
+
64
+ Use ''--cfg hydra'' to Show the Hydra config.
65
+
66
+ '
67
+ hydra_help: ???
68
+ hydra_logging:
69
+ version: 1
70
+ formatters:
71
+ simple:
72
+ format: '[%(asctime)s][HYDRA] %(message)s'
73
+ handlers:
74
+ console:
75
+ class: logging.StreamHandler
76
+ formatter: simple
77
+ stream: ext://sys.stdout
78
+ root:
79
+ level: INFO
80
+ handlers:
81
+ - console
82
+ loggers:
83
+ logging_example:
84
+ level: DEBUG
85
+ disable_existing_loggers: false
86
+ job_logging:
87
+ version: 1
88
+ formatters:
89
+ simple:
90
+ format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
91
+ handlers:
92
+ console:
93
+ class: logging.StreamHandler
94
+ formatter: simple
95
+ stream: ext://sys.stdout
96
+ file:
97
+ class: logging.FileHandler
98
+ formatter: simple
99
+ filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
100
+ root:
101
+ level: INFO
102
+ handlers:
103
+ - console
104
+ - file
105
+ disable_existing_loggers: false
106
+ env: {}
107
+ mode: RUN
108
+ searchpath: []
109
+ callbacks: {}
110
+ output_subdir: .hydra
111
+ overrides:
112
+ hydra:
113
+ - hydra.mode=RUN
114
+ task:
115
+ - model=openvla
116
+ - env=demon_attack
117
+ - init=bridge
118
+ - mode=single
119
+ - run_id=demon_attack_fix_latency_8_200ep
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=false
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=7000
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=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.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.latency_curriculum.enabled=false
178
+ - datasets.vla_data.latency_curriculum.strategy=exclusive
179
+ - datasets.vla_data.per_device_batch_size=32
180
+ - datasets.vla_data.load_all_data_for_training=true
181
+ - datasets.vla_data.video_backend=torchvision_av
182
+ - datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_8_200ep
183
+ - datasets.vla_data.data_mix=demon_attack_train__bridge
184
+ - datasets.vla_data.eval_data_mix=demon_attack_train__bridge__val
185
+ - framework.qwenvl.base_vlm=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct
186
+ - rl_games.task=demon_attack
187
+ - rl_games.model_alias=openvla
188
+ - rl_games.initialization_mode=bridge
189
+ - rl_games.action_carrier=bridge
190
+ - rl_games.env_eval.distributed_mode=none
191
+ - rl_games.env_eval.latency.mode=single
192
+ - rl_games.env_eval.frameskip=4
193
+ - rl_games.env_eval.image_size=224
194
+ - rl_games.env_eval.seed=42
195
+ - rl_games.env_eval.fixed_episode_seeds=true
196
+ - rl_games.env_eval.latency_seed_stride=0
197
+ - rl_games.env_eval.task_seed_stride=0
198
+ - 'rl_games.env_eval.task_description=''You are playing Demon Attack from a single
199
+ game image. Choose exactly one action from: NOOP, FIRE, RIGHT, LEFT, RIGHTFIRE,
200
+ LEFTFIRE.'''
201
+ - rl_games.env_eval.latency.values=[0]
202
+ - rl_games.env_eval.latency.prompt_map_path=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_8_200ep/demon_attack_train__bridge/latency_prompt_map.json
203
+ - rl_games.env_eval.mid_train.enabled=false
204
+ - rl_games.env_eval.mid_train.interval_steps=100
205
+ - rl_games.env_eval.mid_train.num_episodes=5
206
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
207
+ - rl_games.env_eval.mid_train.latencies=[0]
208
+ - rl_games.env_eval.post_train.enabled=false
209
+ - rl_games.env_eval.post_train.num_episodes=5
210
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
211
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
212
+ job:
213
+ name: train_starvla_hydra
214
+ chdir: false
215
+ 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=demon_attack_train__bridge,datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_8_200ep,datasets.vla_data.eval_data_mix=demon_attack_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=32,datasets.vla_data.sequential_step_sampling=false,datasets.vla_data.shuffle=true,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=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct,framework.qwenvl.enable_gradient_checkpointing=true,init=bridge,mode=single,model=openvla,rl_games.action_carrier=bridge,rl_games.env_eval.distributed_mode=none,rl_games.env_eval.enabled=true,rl_games.env_eval.fixed_episode_seeds=true,rl_games.env_eval.frameskip=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/demon_attack_fix_latency_8_200ep/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=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
216
+ are playing Demon Attack from a single game image. Choose exactly one action
217
+ 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=demon_attack_fix_latency_8_200ep,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=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=7000,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=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3VL-OFT-Bridge-RT-1/checkpoints/steps_5000_pytorch_model.pt,trainer.resume_step=0,trainer.save_format=pt,trainer.save_interval=100,trainer.scheduler_specific_kwargs.min_lr=1e-06,trainer.weight_decay=0.0,wandb_entity=dongqianyu99-zhejiang-university,wandb_project=starVLA_rl_games'
218
+ id: ???
219
+ num: ???
220
+ config_name: train
221
+ env_set: {}
222
+ env_copy: []
223
+ config:
224
+ override_dirname:
225
+ kv_sep: '='
226
+ item_sep: ','
227
+ exclude_keys: []
228
+ runtime:
229
+ version: 1.3.2
230
+ version_base: '1.1'
231
+ cwd: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA
232
+ config_sources:
233
+ - path: hydra.conf
234
+ schema: pkg
235
+ provider: hydra
236
+ - path: /inspire/hdd/project/spatialintelligence/public/lzj/starVLA/examples/rl_games/config
237
+ schema: file
238
+ provider: main
239
+ - path: ''
240
+ schema: structured
241
+ provider: schema
242
+ output_dir: /inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/results/Checkpoints/demon_attack_fix_latency_8_200ep/hydra
243
+ choices:
244
+ cross_task_setup: null
245
+ checkpoint: default
246
+ mode: single
247
+ init: bridge
248
+ env: demon_attack
249
+ model: openvla
250
+ hydra/env: default
251
+ hydra/callbacks: null
252
+ hydra/job_logging: default
253
+ hydra/hydra_logging: default
254
+ hydra/hydra_help: default
255
+ hydra/help: default
256
+ hydra/sweeper: basic
257
+ hydra/launcher: basic
258
+ hydra/output: default
259
+ verbose: false
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/hydra/.hydra/overrides.yaml ADDED
@@ -0,0 +1,97 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ - model=openvla
2
+ - env=demon_attack
3
+ - init=bridge
4
+ - mode=single
5
+ - run_id=demon_attack_fix_latency_8_200ep
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=false
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=7000
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=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.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.latency_curriculum.enabled=false
64
+ - datasets.vla_data.latency_curriculum.strategy=exclusive
65
+ - datasets.vla_data.per_device_batch_size=32
66
+ - datasets.vla_data.load_all_data_for_training=true
67
+ - datasets.vla_data.video_backend=torchvision_av
68
+ - datasets.vla_data.data_root_dir=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_8_200ep
69
+ - datasets.vla_data.data_mix=demon_attack_train__bridge
70
+ - datasets.vla_data.eval_data_mix=demon_attack_train__bridge__val
71
+ - framework.qwenvl.base_vlm=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/playground/Pretrained_models/Qwen3-VL-4B-Instruct
72
+ - rl_games.task=demon_attack
73
+ - rl_games.model_alias=openvla
74
+ - rl_games.initialization_mode=bridge
75
+ - rl_games.action_carrier=bridge
76
+ - rl_games.env_eval.distributed_mode=none
77
+ - rl_games.env_eval.latency.mode=single
78
+ - rl_games.env_eval.frameskip=4
79
+ - rl_games.env_eval.image_size=224
80
+ - rl_games.env_eval.seed=42
81
+ - rl_games.env_eval.fixed_episode_seeds=true
82
+ - rl_games.env_eval.latency_seed_stride=0
83
+ - rl_games.env_eval.task_seed_stride=0
84
+ - 'rl_games.env_eval.task_description=''You are playing Demon Attack from a single
85
+ game image. Choose exactly one action from: NOOP, FIRE, RIGHT, LEFT, RIGHTFIRE,
86
+ LEFTFIRE.'''
87
+ - rl_games.env_eval.latency.values=[0]
88
+ - rl_games.env_eval.latency.prompt_map_path=/inspire/hdd/global_user/liumingyu-253208120284/lzj/starvla/data/demon_attack_fix_latency_8_200ep/demon_attack_train__bridge/latency_prompt_map.json
89
+ - rl_games.env_eval.mid_train.enabled=false
90
+ - rl_games.env_eval.mid_train.interval_steps=100
91
+ - rl_games.env_eval.mid_train.num_episodes=5
92
+ - rl_games.env_eval.mid_train.max_steps_per_episode=3600
93
+ - rl_games.env_eval.mid_train.latencies=[0]
94
+ - rl_games.env_eval.post_train.enabled=false
95
+ - rl_games.env_eval.post_train.num_episodes=5
96
+ - rl_games.env_eval.post_train.max_steps_per_episode=3600
97
+ - rl_games.env_eval.post_train.latencies=[0,1,2,3,4]
latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/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/demon_attack_fix_latency_8_200ep/summary.jsonl ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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}
51
+ {"steps": 5100}
52
+ {"steps": 5200}
53
+ {"steps": 5300}
54
+ {"steps": 5400}
55
+ {"steps": 5500}
56
+ {"steps": 5600}
57
+ {"steps": 5700}
58
+ {"steps": 5800}
59
+ {"steps": 5900}
60
+ {"steps": 6000}
61
+ {"steps": 6100}
62
+ {"steps": 6200}
63
+ {"steps": 6300}
64
+ {"steps": 6400}
65
+ {"steps": 6500}
66
+ {"steps": 6600}
67
+ {"steps": 6700}
68
+ {"steps": 6800}
69
+ {"steps": 6900}
70
+ {"steps": 7000}
latency-aware/demon-attack/vla/starvla-wanoft-h8/wan_oft_demon_attack_fix_latency_6_context5_standard_sft_5000_effbs128_224_currentce/checkpoints/steps_5000_pytorch_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:791c66ee0313063e3eeec01a3eeb788e642f8e22952e232caa6562998d0754dd
3
+ size 25211531190
latency-aware/demon-attack/vla/starvla-wanoft-h8/wan_oft_demon_attack_fix_latency_6_context5_standard_sft_5000_effbs128_224_currentce/config.full.yaml ADDED
@@ -0,0 +1,280 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ framework:
2
+ name: WanOFT
3
+ world_model:
4
+ base_wm: /lustre/fsw/portfolios/nvr/projects/nvr_lacr_llm/users/zihwang/wanoft_bench/third_party/starVLA/playground/Pretrained_models/Wan-AI/Wan2.2-TI2V-5B-Diffusers
5
+ extract_layers:
6
+ - -1
7
+ num_frames: 5
8
+ qwenvl:
9
+ base_vlm: /lustre/fsw/portfolios/nvr/projects/nvr_lacr_llm/users/zihwang/wanoft_bench/third_party/starVLA/playground/Pretrained_models/Wan-AI/Wan2.2-TI2V-5B-Diffusers
10
+ attn_implementation: sdpa
11
+ flex_backend: triton
12
+ enable_gradient_checkpointing: true
13
+ action_model:
14
+ action_model_type: MLP
15
+ action_dim: 7
16
+ action_hidden_dim: 3072
17
+ action_horizon: 8
18
+ future_action_window_size: 7
19
+ past_action_window_size: 0
20
+ loss_type: current_discrete_ce
21
+ class_weights: null
22
+ future_loss_weight: null
23
+ action_query_source: mean
24
+ action_query_num_heads: 24
25
+ state_dim: 7
26
+ action_env_dim: 6
27
+ kv_memory:
28
+ enabled: false
29
+ window: 4
30
+ rollout_len: 8
31
+ packed_train: false
32
+ rebased_sink: true
33
+ datasets:
34
+ vla_data:
35
+ dataset_py: lerobot_datasets
36
+ include_state: true
37
+ data_root_dir: /lustre/fsw/portfolios/nvr/projects/nvr_lacr_llm/users/zihwang/wanoft_bench/third_party/starVLA/data/demon_attack_fix_latency_6_200ep_context5
38
+ data_mix: demon_attack_train__bridge
39
+ eval_data_mix: null
40
+ custom_mixtures_path: null
41
+ action_type: discrete
42
+ sequential_step_sampling: false
43
+ eval_sequential_step_sampling: null
44
+ num_workers: 8
45
+ eval_num_workers: 8
46
+ prefetch_factor: 4
47
+ persistent_workers: true
48
+ pin_memory: true
49
+ shuffle: true
50
+ action_balance:
51
+ enabled: false
52
+ strategy: balanced_epoch
53
+ action_key: action_id
54
+ target_flap_fraction: 0.3
55
+ noop_id: 0
56
+ flap_id: 1
57
+ latency_curriculum:
58
+ enabled: false
59
+ strategy: exclusive
60
+ latencies: null
61
+ phase_steps: null
62
+ phase_distributions: null
63
+ new_latency_passes: 1.0
64
+ replay_passes: 0.25
65
+ target_total_passes: 2.0
66
+ final_equalization: true
67
+ step_budget_mode: auto
68
+ eval_at_phase_end: false
69
+ save_at_phase_end: false
70
+ computed_plan: null
71
+ per_device_batch_size: 1
72
+ load_all_data_for_training: true
73
+ num_obs_frames: 1
74
+ image_mode: single
75
+ prompt_mode: raw
76
+ stitch_grid:
77
+ - 2
78
+ - 2
79
+ obs_image_size:
80
+ - 224
81
+ - 224
82
+ video_backend: torchvision_av
83
+ observation_indices:
84
+ - -4
85
+ - -3
86
+ - -2
87
+ - -1
88
+ - 0
89
+ language_indices:
90
+ - 0
91
+ state_indices:
92
+ - 0
93
+ action_indices:
94
+ - 0
95
+ - 1
96
+ - 2
97
+ - 3
98
+ - 4
99
+ - 5
100
+ - 6
101
+ - 7
102
+ pack_image_sequence: true
103
+ image_sequence_length: 5
104
+ context_images_column: observation.context_images
105
+ dataset:
106
+ source_hf: latency-sensitive-bench/memory-data
107
+ config_name: demon_attack_fixed_latency_6_200ep_7k2steps
108
+ source_subdir: null
109
+ converted_name: demon_attack_train__bridge
110
+ single_source_hf: ''
111
+ mixed_source_hf: ''
112
+ single_converted_name: demon_attack_train__bridge
113
+ mixed_converted_name: demon_attack_mixed_latency_train__bridge
114
+ single_latency_filter: null
115
+ mixed_latency_filter: null
116
+ force_download: false
117
+ setup_force: false
118
+ skip_verification: false
119
+ target_latency_unit: observation_steps
120
+ verify_rows: 200
121
+ max_episodes: 200
122
+ episodes_per_latency: null
123
+ latency_filter: null
124
+ debug_subset:
125
+ enabled: false
126
+ max_episodes: 5
127
+ suffix: debug
128
+ base_model:
129
+ repo_id: Wan-AI/Wan2.2-TI2V-5B-Diffusers
130
+ initialization:
131
+ checkpoint_local_dir: playground/Pretrained_models/WM4A-Wan2d2-OFT-LIBERO-4in1
132
+ checkpoint_hf_repo_id: StarVLA/WM4A-Wan2d2-OFT-LIBERO-4in1
133
+ checkpoint_filename: checkpoints/steps_60000_pytorch_model.pt
134
+ trainer:
135
+ max_train_steps: 5000
136
+ num_warmup_steps: 0
137
+ save_interval: 500
138
+ eval_interval: 100
139
+ eval_num_batches: 100
140
+ per_latency_eval_num_batches: null
141
+ eval_action_classification: false
142
+ eval_action_classification_interval: null
143
+ cc_f1_tolerance: 1
144
+ learning_rate:
145
+ base: 2.5e-05
146
+ qwen_vl_interface: 1.0e-05
147
+ action_model: 0.0001
148
+ action_query_proj: 0.0001
149
+ lr_scheduler_type: cosine_with_min_lr
150
+ scheduler_specific_kwargs:
151
+ min_lr: 1.0e-06
152
+ freeze_modules: ''
153
+ freeze_vit: false
154
+ freeze_tied_embedding: false
155
+ freeze_llm_layers: []
156
+ loss_scale:
157
+ vla: 1.0
158
+ vlm: 0.1
159
+ max_grad_norm: 1.0
160
+ weight_decay: 0.0
161
+ logging_frequency: 1
162
+ profile_timing:
163
+ enabled: false
164
+ log_interval: 10
165
+ gradient_clipping: 1.0
166
+ gradient_accumulation_steps: 16
167
+ distributed_backend: deepspeed
168
+ is_resume: true
169
+ pretrained_checkpoint: /lustre/fsw/portfolios/nvr/projects/nvr_lacr_llm/users/zihwang/wanoft_bench/third_party/starVLA/results/Checkpoints/wan_oft_demon_attack_fix_latency_6_context5_standard_sft_5000_effbs128_224_currentce/checkpoints/steps_4000_state
170
+ resume_step: 4000
171
+ reload_modules: null
172
+ optimizer:
173
+ name: AdamW
174
+ betas:
175
+ - 0.9
176
+ - 0.95
177
+ eps: 1.0e-08
178
+ weight_decay: 1.0e-08
179
+ fused: true
180
+ save_format: pt
181
+ workspace_dir: WORKSPACE_DIR
182
+ run_root_dir: /lustre/fsw/portfolios/nvr/projects/nvr_lacr_llm/users/zihwang/wanoft_bench/third_party/starVLA/results/Checkpoints
183
+ seed: 42
184
+ wandb_entity: zihanwang-ai-northwestern-university
185
+ wandb_project: starVLA_rl_games
186
+ auth:
187
+ env_file: null
188
+ hf_token_env: HF_TOKEN
189
+ wandb_api_key_env: WANDB_API_KEY
190
+ paths:
191
+ run_root_dir: results/Checkpoints
192
+ dataset_local_dir: data/demon_attack_fix_latency_6_200ep_context5
193
+ dataset_cache_dir: null
194
+ base_model_dir: playground/Pretrained_models/Wan-AI/Wan2.2-TI2V-5B-Diffusers
195
+ accelerate_config: starVLA/config/deepseeds/deepspeed_zero2.yaml
196
+ launch:
197
+ use_accelerate: true
198
+ gpus: null
199
+ num_processes: 1
200
+ dry_run: false
201
+ conda:
202
+ enabled: true
203
+ env_name: null
204
+ rl_games:
205
+ model_alias: wan_oft
206
+ env_eval:
207
+ image_size: 224
208
+ frameskip: 4
209
+ image_transform: raw_rgb
210
+ prompt_mode: raw
211
+ ghost_trail:
212
+ history_frames: 5
213
+ gamma: 1.3
214
+ min_alpha: 35
215
+ scroll_px_per_step: 4.0
216
+ ground_fraction: 0.22
217
+ seed: 42
218
+ fixed_episode_seeds: true
219
+ latency_seed_stride: 0
220
+ task_seed_stride: 0
221
+ task_description: 'You are playing Demon Attack from a single game image. Choose
222
+ exactly one action from: NOOP, FIRE, RIGHT, LEFT, RIGHTFIRE, LEFTFIRE.'
223
+ eval_parallel_envs: 5
224
+ action_chunk_execution:
225
+ enabled: false
226
+ chunk_size: null
227
+ demon_attack:
228
+ noop_max: 30
229
+ enabled: true
230
+ eval_backend: eval_core
231
+ distributed_mode: rank_sharded
232
+ vectorized:
233
+ enabled: false
234
+ batch_size: 1
235
+ latency:
236
+ prompt_map_path: data/demon_attack_fix_latency_6_200ep_context5/demon_attack_train__bridge/latency_prompt_map.json
237
+ mode: single
238
+ values:
239
+ - 0
240
+ mid_train:
241
+ enabled: false
242
+ interval_steps: 100
243
+ latencies:
244
+ - 0
245
+ num_episodes: 5
246
+ max_steps_per_episode: 3600
247
+ post_train:
248
+ enabled: false
249
+ latencies:
250
+ - 6
251
+ num_episodes: 20
252
+ max_steps_per_episode: 3600
253
+ task: demon_attack
254
+ initialization_mode: bridge
255
+ action_carrier: bridge
256
+ model: wan_oft
257
+ env: demon_attack
258
+ init: wan_oft_libero
259
+ mode: single
260
+ checkpoint:
261
+ load: auto
262
+ hf_repo_id: null
263
+ save_best_model: false
264
+ save_final_model: true
265
+ save_pt_file: true
266
+ save_training_state: true
267
+ save_safetensors_file: false
268
+ local:
269
+ keep_last_n: 1
270
+ sync:
271
+ enabled: false
272
+ repo_id: null
273
+ keep_last_n: 0
274
+ sync_every_n_checkpoints: 1
275
+ resume_policy: local_latest
276
+ run_id: wan_oft_demon_attack_fix_latency_6_context5_standard_sft_5000_effbs128_224_currentce
277
+ output_dir: /lustre/fsw/portfolios/nvr/projects/nvr_lacr_llm/users/zihwang/wanoft_bench/third_party/starVLA/results/Checkpoints/wan_oft_demon_attack_fix_latency_6_context5_standard_sft_5000_effbs128_224_currentce
278
+ config_yaml: null
279
+ is_debug: false
280
+ version_id: '0.21'