Consolidate paper experiment assets by task and run (151-300)
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/checkpoints/steps_7000_model.safetensors +3 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/config.full.yaml +209 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/config.yaml +75 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/dataset_statistics.json +127 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/dataset_statistics_eval.json +127 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/hydra/.hydra/config.yaml +207 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/hydra/.hydra/hydra.yaml +259 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/hydra/.hydra/overrides.yaml +97 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/hydra/train_starvla_hydra.log +0 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_2_200ep/summary.jsonl +70 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/checkpoints/steps_7000_model.safetensors +3 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/config.full.yaml +209 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/config.yaml +75 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/dataset_statistics.json +127 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/dataset_statistics_eval.json +127 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/hydra/.hydra/config.yaml +207 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/hydra/.hydra/hydra.yaml +259 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/hydra/.hydra/overrides.yaml +97 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/hydra/train_starvla_hydra.log +0 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_4_200ep/summary.jsonl +70 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/checkpoints/steps_7000_model.safetensors +3 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/config.full.yaml +209 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/config.yaml +75 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/dataset_statistics.json +127 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/dataset_statistics_eval.json +127 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/hydra/.hydra/config.yaml +207 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/hydra/.hydra/hydra.yaml +259 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/hydra/.hydra/overrides.yaml +97 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/hydra/train_starvla_hydra.log +0 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_6_200ep/summary.jsonl +70 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/checkpoints/steps_7000_model.safetensors +3 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/config.full.yaml +209 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/config.yaml +75 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/dataset_statistics.json +127 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/dataset_statistics_eval.json +127 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/hydra/.hydra/config.yaml +207 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/hydra/.hydra/hydra.yaml +259 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/hydra/.hydra/overrides.yaml +97 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/hydra/train_starvla_hydra.log +0 -0
- latency-aware/demon-attack/vla/starvla-qwenoft-h1/demon_attack_fix_latency_8_200ep/summary.jsonl +70 -0
- 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
- 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'
|