Upload settings
Browse files- .gitattributes +5 -0
- settings/florence_bridge/frameskip/config.full.yaml +116 -0
- settings/florence_bridge/frameskip/config.yaml +71 -0
- settings/florence_bridge/frameskip/dataset_statistics.json +264 -0
- settings/florence_bridge/frameskip/external_resume_certificate.json +132 -0
- settings/florence_bridge/frameskip/initialization_manifest.json +6 -0
- settings/florence_bridge/frameskip/valid_anchor_manifest.json +3 -0
- settings/florence_bridge/keyframe/config.full.yaml +117 -0
- settings/florence_bridge/keyframe/config.yaml +70 -0
- settings/florence_bridge/keyframe/dataset_statistics.json +264 -0
- settings/florence_bridge/loss_is/config.full.yaml +123 -0
- settings/florence_bridge/loss_is/config.yaml +75 -0
- settings/florence_bridge/loss_is/dataset_statistics.json +264 -0
- settings/florence_robotwin/common_start/config.full.yaml +107 -0
- settings/florence_robotwin/common_start/config.yaml +61 -0
- settings/florence_robotwin/common_start/dataset_statistics.json +218 -0
- settings/florence_robotwin/common_start/initialization_manifest.json +6 -0
- settings/florence_robotwin/uniform/config.full.yaml +113 -0
- settings/florence_robotwin/uniform/config.yaml +68 -0
- settings/florence_robotwin/uniform/dataset_statistics.json +218 -0
- settings/florence_robotwin/uniform/external_resume_certificate.json +138 -0
- settings/florence_robotwin/uniform/initialization_manifest.json +6 -0
- settings/florence_robotwin/uniform/valid_anchor_manifest.json +3 -0
- settings/qwen_robotwin/common_start/config.full.yaml +242 -0
- settings/qwen_robotwin/common_start/config.yaml +64 -0
- settings/qwen_robotwin/common_start/dataset_statistics.json +218 -0
- settings/qwen_robotwin/common_start/initialization_manifest.json +6 -0
- settings/qwen_robotwin/common_start/valid_anchor_manifest.json +3 -0
- settings/qwen_robotwin/residual/config.full.yaml +112 -0
- settings/qwen_robotwin/residual/config.yaml +65 -0
- settings/qwen_robotwin/residual/dataset_statistics.json +218 -0
- settings/qwen_robotwin/residual/external_resume_certificate.json +138 -0
- settings/qwen_robotwin/residual/initialization_manifest.json +6 -0
- settings/qwen_robotwin/residual/valid_anchor_manifest.json +3 -0
- settings/qwen_robotwin/uniform/config.full.yaml +108 -0
- settings/qwen_robotwin/uniform/config.yaml +64 -0
- settings/qwen_robotwin/uniform/dataset_statistics.json +218 -0
- settings/qwen_robotwin/uniform/external_resume_certificate.json +138 -0
- settings/qwen_robotwin/uniform/initialization_manifest.json +6 -0
- settings/qwen_robotwin/uniform/valid_anchor_manifest.json +3 -0
.gitattributes
CHANGED
|
@@ -36,3 +36,8 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 36 |
proposals/residual_temporal_proposal_k8_alpha0.2.json filter=lfs diff=lfs merge=lfs -text
|
| 37 |
manifests/content_fingerprints.json filter=lfs diff=lfs merge=lfs -text
|
| 38 |
manifests/valid_anchor_manifest_florence_scratch_v1.json filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 36 |
proposals/residual_temporal_proposal_k8_alpha0.2.json filter=lfs diff=lfs merge=lfs -text
|
| 37 |
manifests/content_fingerprints.json filter=lfs diff=lfs merge=lfs -text
|
| 38 |
manifests/valid_anchor_manifest_florence_scratch_v1.json filter=lfs diff=lfs merge=lfs -text
|
| 39 |
+
settings/florence_bridge/frameskip/valid_anchor_manifest.json filter=lfs diff=lfs merge=lfs -text
|
| 40 |
+
settings/florence_robotwin/uniform/valid_anchor_manifest.json filter=lfs diff=lfs merge=lfs -text
|
| 41 |
+
settings/qwen_robotwin/common_start/valid_anchor_manifest.json filter=lfs diff=lfs merge=lfs -text
|
| 42 |
+
settings/qwen_robotwin/residual/valid_anchor_manifest.json filter=lfs diff=lfs merge=lfs -text
|
| 43 |
+
settings/qwen_robotwin/uniform/valid_anchor_manifest.json filter=lfs diff=lfs merge=lfs -text
|
settings/florence_bridge/frameskip/config.full.yaml
ADDED
|
@@ -0,0 +1,116 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
run_id: frameskip-adapted-gaussian-tpi-r02-seed-117465
|
| 2 |
+
run_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/external-baselines/frameskip-adapted-gaussian-tpi-r02-20260825
|
| 3 |
+
seed: 117465
|
| 4 |
+
wandb_entity: null
|
| 5 |
+
wandb_project: InfoFlow-StarVLA-BridgeRT
|
| 6 |
+
is_debug: false
|
| 7 |
+
version_id: '0.21'
|
| 8 |
+
framework:
|
| 9 |
+
name: QwenGR00T
|
| 10 |
+
qwenvl:
|
| 11 |
+
base_vlm: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Pretrained_models/Florence-2-large
|
| 12 |
+
attn_implementation: flash_attention_2
|
| 13 |
+
vl_hidden_dim: 2048
|
| 14 |
+
action_model:
|
| 15 |
+
action_model_type: DiT-B
|
| 16 |
+
action_hidden_dim: 1024
|
| 17 |
+
hidden_size: 1024
|
| 18 |
+
add_pos_embed: true
|
| 19 |
+
max_seq_len: 1024
|
| 20 |
+
action_dim: 7
|
| 21 |
+
state_dim: 7
|
| 22 |
+
action_horizon: 16
|
| 23 |
+
repeated_diffusion_steps: 8
|
| 24 |
+
noise_beta_alpha: 1.5
|
| 25 |
+
noise_beta_beta: 1.0
|
| 26 |
+
noise_s: 0.999
|
| 27 |
+
num_timestep_buckets: 1000
|
| 28 |
+
num_inference_timesteps: 4
|
| 29 |
+
num_target_vision_tokens: 32
|
| 30 |
+
diffusion_model_cfg:
|
| 31 |
+
cross_attention_dim: 1024
|
| 32 |
+
dropout: 0.2
|
| 33 |
+
final_dropout: true
|
| 34 |
+
interleave_self_attention: true
|
| 35 |
+
norm_type: ada_norm
|
| 36 |
+
num_layers: 16
|
| 37 |
+
output_dim: 1024
|
| 38 |
+
positional_embeddings: null
|
| 39 |
+
future_action_window_size: 15
|
| 40 |
+
past_action_window_size: 0
|
| 41 |
+
dino:
|
| 42 |
+
dino_backbone: dinov2_vits14
|
| 43 |
+
datasets:
|
| 44 |
+
vla_data:
|
| 45 |
+
dataset_py: lerobot_datasets
|
| 46 |
+
data_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Datasets/OXE_LEROBOT_DATASET
|
| 47 |
+
data_mix: bridge_rt_infoflow
|
| 48 |
+
action_type: delta_ee
|
| 49 |
+
action_mode: abs
|
| 50 |
+
precomputed_statistics_path: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Pretrained_models/Florence-GR00T-Bridge-RT-1/dataset_statistics.json
|
| 51 |
+
precomputed_statistics_key_map:
|
| 52 |
+
bridge_orig_lerobot: oxe_bridge
|
| 53 |
+
fractal20220817_data_lerobot: oxe_rt1
|
| 54 |
+
CoT_prompt: Your task is {instruction}. To identify the key objects for your task.
|
| 55 |
+
Locate their bounding boxes in [x1,y1,x2,y2] format.
|
| 56 |
+
per_device_batch_size: 4
|
| 57 |
+
load_all_data_for_training: true
|
| 58 |
+
obs_image_size:
|
| 59 |
+
- 224
|
| 60 |
+
- 224
|
| 61 |
+
video_backend: torchvision_av
|
| 62 |
+
num_workers: 4
|
| 63 |
+
balance_dataset_weights: false
|
| 64 |
+
balance_trajectory_weights: false
|
| 65 |
+
traceable_sampling: true
|
| 66 |
+
expected_world_size: 2
|
| 67 |
+
dataset_provenance_path: /home/miaochengyu/starVLA-infoflow-codex/protocols/bridgert/data_provenance.json
|
| 68 |
+
dataset_content_fingerprint_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/bridgert/content_fingerprints.json
|
| 69 |
+
valid_anchor_manifest_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/bridgert/valid_anchor_manifest_florence_scratch_v1.json
|
| 70 |
+
valid_anchor_manifest_migration_path: /home/miaochengyu/starVLA-infoflow-codex/external_baselines/frameskip_adapted/preflight/valid_anchor_manifest_migration_20260825.json
|
| 71 |
+
temporal_proposal_path: null
|
| 72 |
+
transform_rng_seed: 298226
|
| 73 |
+
trainer:
|
| 74 |
+
use_deepspeed: true
|
| 75 |
+
max_train_steps: 100000
|
| 76 |
+
num_warmup_steps: 5000
|
| 77 |
+
save_interval: null
|
| 78 |
+
save_steps:
|
| 79 |
+
- 20500
|
| 80 |
+
evaluation_snapshot_interval: 1000
|
| 81 |
+
eval_interval: 100001
|
| 82 |
+
repeated_diffusion_steps: 4
|
| 83 |
+
pretrained_checkpoint: null
|
| 84 |
+
is_resume: true
|
| 85 |
+
allow_model_only_resume: false
|
| 86 |
+
learning_rate:
|
| 87 |
+
base: 3.0e-05
|
| 88 |
+
qwen_vl_interface: 1.0e-05
|
| 89 |
+
action_model: 0.0001
|
| 90 |
+
lr_scheduler_type: cosine_with_min_lr
|
| 91 |
+
scheduler_specific_kwargs:
|
| 92 |
+
min_lr: 5.0e-07
|
| 93 |
+
freeze_modules: ''
|
| 94 |
+
trainable_modules: ''
|
| 95 |
+
frozen_modules_eval: false
|
| 96 |
+
logging_frequency: 50
|
| 97 |
+
gradient_clipping: 1.0
|
| 98 |
+
gradient_accumulation_steps: 8
|
| 99 |
+
save_format: pt
|
| 100 |
+
save_final_checkpoint: false
|
| 101 |
+
optimizer:
|
| 102 |
+
name: AdamW
|
| 103 |
+
betas:
|
| 104 |
+
- 0.9
|
| 105 |
+
- 0.95
|
| 106 |
+
eps: 1.0e-08
|
| 107 |
+
weight_decay: 1.0e-08
|
| 108 |
+
resume_accelerate_checkpoint: /mnt/miaochengyu/starvla-infoflow-runtime/runs/florence-scratch-official-100k/florence-scratch-uniform-seed-42-official100k-mb4-eval1k-20260726/checkpoints/steps_20000_accelerate
|
| 109 |
+
resume_checkpoint_integrity_manifest: /home/miaochengyu/starVLA-infoflow-codex/protocols/bridgert/common_start_accelerate_integrity_20260730.json
|
| 110 |
+
resume_checkpoint_integrity_manifest_sha256: 899d2cef3f73191e11f7a1628afc05b60c80af5816b8fd77c1c69e90d5c9ff39
|
| 111 |
+
branch_parent_seed: 42
|
| 112 |
+
branch_rng_seed: 117465
|
| 113 |
+
model_rng_seed: 588643
|
| 114 |
+
stop_after_steps: 20500
|
| 115 |
+
config_yaml: /home/miaochengyu/starVLA-infoflow-codex/external_baselines/frameskip_adapted/configs/florence_bridge_frameskip_gaussian_r02_seed117465.yaml
|
| 116 |
+
output_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/external-baselines/frameskip-adapted-gaussian-tpi-r02-20260825/frameskip-adapted-gaussian-tpi-r02-seed-117465
|
settings/florence_bridge/frameskip/config.yaml
ADDED
|
@@ -0,0 +1,71 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
datasets:
|
| 2 |
+
vla_data:
|
| 3 |
+
action_mode: abs
|
| 4 |
+
balance_dataset_weights: false
|
| 5 |
+
balance_trajectory_weights: false
|
| 6 |
+
data_mix: bridge_rt_infoflow
|
| 7 |
+
data_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Datasets/OXE_LEROBOT_DATASET
|
| 8 |
+
dataset_content_fingerprint_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/bridgert/content_fingerprints.json
|
| 9 |
+
dataset_provenance_path: /home/miaochengyu/starVLA-infoflow-codex/protocols/bridgert/data_provenance.json
|
| 10 |
+
dataset_py: lerobot_datasets
|
| 11 |
+
expected_world_size: 2
|
| 12 |
+
num_workers: 4
|
| 13 |
+
per_device_batch_size: 4
|
| 14 |
+
precomputed_statistics_key_map:
|
| 15 |
+
bridge_orig_lerobot: oxe_bridge
|
| 16 |
+
fractal20220817_data_lerobot: oxe_rt1
|
| 17 |
+
precomputed_statistics_path: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Pretrained_models/Florence-GR00T-Bridge-RT-1/dataset_statistics.json
|
| 18 |
+
temporal_proposal_path: null
|
| 19 |
+
traceable_sampling: true
|
| 20 |
+
valid_anchor_manifest_migration_path: /home/miaochengyu/starVLA-infoflow-codex/external_baselines/frameskip_adapted/preflight/valid_anchor_manifest_migration_20260825.json
|
| 21 |
+
valid_anchor_manifest_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/bridgert/valid_anchor_manifest_florence_scratch_v1.json
|
| 22 |
+
video_backend: torchvision_av
|
| 23 |
+
framework:
|
| 24 |
+
action_model:
|
| 25 |
+
action_dim: 7
|
| 26 |
+
action_horizon: 16
|
| 27 |
+
action_model_type: DiT-B
|
| 28 |
+
add_pos_embed: true
|
| 29 |
+
diffusion_model_cfg:
|
| 30 |
+
cross_attention_dim: 1024
|
| 31 |
+
dropout: 0.2
|
| 32 |
+
final_dropout: true
|
| 33 |
+
interleave_self_attention: true
|
| 34 |
+
norm_type: ada_norm
|
| 35 |
+
num_layers: 16
|
| 36 |
+
output_dim: 1024
|
| 37 |
+
positional_embeddings: null
|
| 38 |
+
hidden_size: 1024
|
| 39 |
+
max_seq_len: 1024
|
| 40 |
+
noise_beta_alpha: 1.5
|
| 41 |
+
noise_beta_beta: 1.0
|
| 42 |
+
num_inference_timesteps: 4
|
| 43 |
+
num_target_vision_tokens: 32
|
| 44 |
+
num_timestep_buckets: 1000
|
| 45 |
+
state_dim: 7
|
| 46 |
+
name: QwenGR00T
|
| 47 |
+
qwenvl:
|
| 48 |
+
base_vlm: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Pretrained_models/Florence-2-large
|
| 49 |
+
output_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/external-baselines/frameskip-adapted-gaussian-tpi-r02-20260825/frameskip-adapted-gaussian-tpi-r02-seed-117465
|
| 50 |
+
run_id: frameskip-adapted-gaussian-tpi-r02-seed-117465
|
| 51 |
+
run_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/external-baselines/frameskip-adapted-gaussian-tpi-r02-20260825
|
| 52 |
+
seed: 117465
|
| 53 |
+
trainer:
|
| 54 |
+
freeze_modules: ''
|
| 55 |
+
gradient_accumulation_steps: 8
|
| 56 |
+
learning_rate:
|
| 57 |
+
action_model: 0.0001
|
| 58 |
+
base: 3.0e-05
|
| 59 |
+
qwen_vl_interface: 1.0e-05
|
| 60 |
+
lr_scheduler_type: cosine_with_min_lr
|
| 61 |
+
max_train_steps: 100000
|
| 62 |
+
num_warmup_steps: 5000
|
| 63 |
+
optimizer:
|
| 64 |
+
betas:
|
| 65 |
+
- 0.9
|
| 66 |
+
- 0.95
|
| 67 |
+
eps: 1.0e-08
|
| 68 |
+
name: AdamW
|
| 69 |
+
weight_decay: 1.0e-08
|
| 70 |
+
scheduler_specific_kwargs:
|
| 71 |
+
min_lr: 5.0e-07
|
settings/florence_bridge/frameskip/dataset_statistics.json
ADDED
|
@@ -0,0 +1,264 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"oxe_bridge": {
|
| 3 |
+
"action": {
|
| 4 |
+
"mean": [
|
| 5 |
+
5.682967821485363e-05,
|
| 6 |
+
3.278055373812094e-05,
|
| 7 |
+
-3.1604828109266236e-05,
|
| 8 |
+
-3.6026765883434564e-05,
|
| 9 |
+
-9.757651423569769e-05,
|
| 10 |
+
6.01583014940843e-05,
|
| 11 |
+
0.14414729177951813
|
| 12 |
+
],
|
| 13 |
+
"std": [
|
| 14 |
+
0.004886276229987901,
|
| 15 |
+
0.006847766456005231,
|
| 16 |
+
0.006337809696791937,
|
| 17 |
+
0.014227778584428946,
|
| 18 |
+
0.015261554008856799,
|
| 19 |
+
0.038695293138416895,
|
| 20 |
+
0.3521380562074972
|
| 21 |
+
],
|
| 22 |
+
"max": [
|
| 23 |
+
0.41691166162490845,
|
| 24 |
+
0.25864794850349426,
|
| 25 |
+
0.21218234300613403,
|
| 26 |
+
3.122201919555664,
|
| 27 |
+
1.8618112802505493,
|
| 28 |
+
6.272472858428955,
|
| 29 |
+
1.0
|
| 30 |
+
],
|
| 31 |
+
"min": [
|
| 32 |
+
-0.4007510244846344,
|
| 33 |
+
-0.13874775171279907,
|
| 34 |
+
-0.22553899884223938,
|
| 35 |
+
-3.2010786533355713,
|
| 36 |
+
-1.8618112802505493,
|
| 37 |
+
-6.279075622558594,
|
| 38 |
+
0.0
|
| 39 |
+
],
|
| 40 |
+
"q01": [
|
| 41 |
+
-0.02875255048274994,
|
| 42 |
+
-0.04170213546603918,
|
| 43 |
+
-0.026096721179783344,
|
| 44 |
+
-0.08052874729037285,
|
| 45 |
+
-0.09249906800687313,
|
| 46 |
+
-0.20738555490970612,
|
| 47 |
+
0.0
|
| 48 |
+
],
|
| 49 |
+
"q99": [
|
| 50 |
+
0.028306663036346436,
|
| 51 |
+
0.04089853074401617,
|
| 52 |
+
0.0401805154979229,
|
| 53 |
+
0.08173403143882751,
|
| 54 |
+
0.07760760560631752,
|
| 55 |
+
0.2038465365767479,
|
| 56 |
+
1.0
|
| 57 |
+
],
|
| 58 |
+
"mask": [
|
| 59 |
+
true,
|
| 60 |
+
true,
|
| 61 |
+
true,
|
| 62 |
+
true,
|
| 63 |
+
true,
|
| 64 |
+
true,
|
| 65 |
+
false
|
| 66 |
+
]
|
| 67 |
+
},
|
| 68 |
+
"state": {
|
| 69 |
+
"mean": [
|
| 70 |
+
0.07735871523618698,
|
| 71 |
+
0.007681275252252817,
|
| 72 |
+
0.016110140830278397,
|
| 73 |
+
0.0016226600855588913,
|
| 74 |
+
-0.019300073385238647,
|
| 75 |
+
0.02691473439335823,
|
| 76 |
+
0.0,
|
| 77 |
+
0.17703326046466827
|
| 78 |
+
],
|
| 79 |
+
"std": [
|
| 80 |
+
0.13736907523153707,
|
| 81 |
+
0.047840646787850445,
|
| 82 |
+
0.03799073317180484,
|
| 83 |
+
0.06560660645733722,
|
| 84 |
+
0.09093375673607122,
|
| 85 |
+
0.292728734025679,
|
| 86 |
+
0.0,
|
| 87 |
+
0.3536876512395259
|
| 88 |
+
],
|
| 89 |
+
"max": [
|
| 90 |
+
0.5862360596656799,
|
| 91 |
+
0.4034728705883026,
|
| 92 |
+
0.3568263053894043,
|
| 93 |
+
1.3517684936523438,
|
| 94 |
+
1.570796251296997,
|
| 95 |
+
3.141204357147217,
|
| 96 |
+
0.0,
|
| 97 |
+
1.1121242046356201
|
| 98 |
+
],
|
| 99 |
+
"min": [
|
| 100 |
+
-0.04167502000927925,
|
| 101 |
+
-0.3563207685947418,
|
| 102 |
+
-0.15537554025650024,
|
| 103 |
+
-3.141592502593994,
|
| 104 |
+
-1.4992541074752808,
|
| 105 |
+
-3.14153790473938,
|
| 106 |
+
0.0,
|
| 107 |
+
0.04637829214334488
|
| 108 |
+
],
|
| 109 |
+
"q01": [
|
| 110 |
+
0.17102622985839844,
|
| 111 |
+
-0.1698118858039379,
|
| 112 |
+
-0.055632163770496845,
|
| 113 |
+
-0.36493172496557236,
|
| 114 |
+
-0.541871190071106,
|
| 115 |
+
-1.354275494813919,
|
| 116 |
+
0.0,
|
| 117 |
+
0.052190229296684265
|
| 118 |
+
],
|
| 119 |
+
"q99": [
|
| 120 |
+
0.45322076976299286,
|
| 121 |
+
0.2354845181107521,
|
| 122 |
+
0.19489620998501778,
|
| 123 |
+
0.3780156075954437,
|
| 124 |
+
0.27568644285202026,
|
| 125 |
+
1.8500566184520721,
|
| 126 |
+
0.0,
|
| 127 |
+
1.0105689764022827
|
| 128 |
+
]
|
| 129 |
+
},
|
| 130 |
+
"num_transitions": 1305714,
|
| 131 |
+
"num_trajectories": 53192
|
| 132 |
+
},
|
| 133 |
+
"oxe_rt1": {
|
| 134 |
+
"action": {
|
| 135 |
+
"mean": [
|
| 136 |
+
0.0017468753503635526,
|
| 137 |
+
0.0015664808452129364,
|
| 138 |
+
-0.00315628363750875,
|
| 139 |
+
0.010832978412508965,
|
| 140 |
+
-0.0014390418073162436,
|
| 141 |
+
0.00022827448265161365,
|
| 142 |
+
0.13385511934757233
|
| 143 |
+
],
|
| 144 |
+
"std": [
|
| 145 |
+
0.034738154754287556,
|
| 146 |
+
0.029950575875189896,
|
| 147 |
+
0.03716987948198757,
|
| 148 |
+
0.08027394458720655,
|
| 149 |
+
0.06586824211259418,
|
| 150 |
+
0.07296740097742303,
|
| 151 |
+
0.33989797865933385
|
| 152 |
+
],
|
| 153 |
+
"max": [
|
| 154 |
+
2.9984593391418457,
|
| 155 |
+
22.09052848815918,
|
| 156 |
+
2.7507524490356445,
|
| 157 |
+
1.570636510848999,
|
| 158 |
+
1.5321086645126343,
|
| 159 |
+
1.5691522359848022,
|
| 160 |
+
1.0
|
| 161 |
+
],
|
| 162 |
+
"min": [
|
| 163 |
+
-2.0204520225524902,
|
| 164 |
+
-5.497899532318115,
|
| 165 |
+
-2.031663417816162,
|
| 166 |
+
-1.569917917251587,
|
| 167 |
+
-1.569892168045044,
|
| 168 |
+
-1.570419430732727,
|
| 169 |
+
0.0
|
| 170 |
+
],
|
| 171 |
+
"q01": [
|
| 172 |
+
-0.22453527510166169,
|
| 173 |
+
-0.14820013284683228,
|
| 174 |
+
-0.231589707583189,
|
| 175 |
+
-0.3517994859814644,
|
| 176 |
+
-0.4193011274933815,
|
| 177 |
+
-0.43643461108207704,
|
| 178 |
+
0.0
|
| 179 |
+
],
|
| 180 |
+
"q99": [
|
| 181 |
+
0.17824687153100965,
|
| 182 |
+
0.14938379630446405,
|
| 183 |
+
0.21842354819178575,
|
| 184 |
+
0.5892666035890578,
|
| 185 |
+
0.35272657424211445,
|
| 186 |
+
0.44796681255102094,
|
| 187 |
+
1.0
|
| 188 |
+
],
|
| 189 |
+
"mask": [
|
| 190 |
+
true,
|
| 191 |
+
true,
|
| 192 |
+
true,
|
| 193 |
+
true,
|
| 194 |
+
true,
|
| 195 |
+
true,
|
| 196 |
+
false
|
| 197 |
+
]
|
| 198 |
+
},
|
| 199 |
+
"state": {
|
| 200 |
+
"mean": [
|
| 201 |
+
0.1399736851453781,
|
| 202 |
+
-0.02083534561097622,
|
| 203 |
+
0.19427375495433807,
|
| 204 |
+
0.10653364658355713,
|
| 205 |
+
-0.06201138719916344,
|
| 206 |
+
0.12378455698490143,
|
| 207 |
+
0.023165317252278328,
|
| 208 |
+
0.052437249571084976
|
| 209 |
+
],
|
| 210 |
+
"std": [
|
| 211 |
+
0.2502841726583427,
|
| 212 |
+
0.06813404469049046,
|
| 213 |
+
0.358260305088155,
|
| 214 |
+
0.2931043604956031,
|
| 215 |
+
0.2779312663188148,
|
| 216 |
+
0.33769767306558707,
|
| 217 |
+
0.09232656626252377,
|
| 218 |
+
0.16498523970985932
|
| 219 |
+
],
|
| 220 |
+
"max": [
|
| 221 |
+
1.0534898042678833,
|
| 222 |
+
0.48018959164619446,
|
| 223 |
+
1.6896663904190063,
|
| 224 |
+
1.0,
|
| 225 |
+
0.9999993443489075,
|
| 226 |
+
0.9999874830245972,
|
| 227 |
+
0.9554369449615479,
|
| 228 |
+
0.9914546012878418
|
| 229 |
+
],
|
| 230 |
+
"min": [
|
| 231 |
+
-0.4436439275741577,
|
| 232 |
+
-0.9970501065254211,
|
| 233 |
+
-0.006579156965017319,
|
| 234 |
+
0.0,
|
| 235 |
+
-0.8643477559089661,
|
| 236 |
+
-0.7079970240592957,
|
| 237 |
+
-0.7688722014427185,
|
| 238 |
+
-0.4999994933605194
|
| 239 |
+
],
|
| 240 |
+
"q01": [
|
| 241 |
+
0.32481380939483645,
|
| 242 |
+
-0.28334290891885755,
|
| 243 |
+
0.14107070609927178,
|
| 244 |
+
0.0,
|
| 245 |
+
-0.686474204659462,
|
| 246 |
+
-0.6808923494815826,
|
| 247 |
+
-0.36045596331357954,
|
| 248 |
+
-0.454380963742733
|
| 249 |
+
],
|
| 250 |
+
"q99": [
|
| 251 |
+
0.8750156319141384,
|
| 252 |
+
0.21247054174542404,
|
| 253 |
+
1.0727112340927123,
|
| 254 |
+
1.0,
|
| 255 |
+
0.9377871316671368,
|
| 256 |
+
0.9563051050901409,
|
| 257 |
+
0.45990042358636823,
|
| 258 |
+
0.7216041100025177
|
| 259 |
+
]
|
| 260 |
+
},
|
| 261 |
+
"num_transitions": 3786152,
|
| 262 |
+
"num_trajectories": 87212
|
| 263 |
+
}
|
| 264 |
+
}
|
settings/florence_bridge/frameskip/external_resume_certificate.json
ADDED
|
@@ -0,0 +1,132 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"branch_resume_identity": {
|
| 3 |
+
"config_sha256": "3c6b4479da73caa22b9f37b00f88d319a0657aee51d1e8c313df225dba1c15b0",
|
| 4 |
+
"identity_schema_version": 2,
|
| 5 |
+
"manifest_sha256": "a1627df8ff110c7298eaf025ffcd4f48d9ca17117cfcfb3926b962d0bb19c0ca",
|
| 6 |
+
"model_structure_sha256": "e055e3fa0085edc6a9bf6cdb1605a05e61ae289957e24e37cfe2cfce2dc94c39",
|
| 7 |
+
"proposal_alpha": null,
|
| 8 |
+
"proposal_sha256": null,
|
| 9 |
+
"seed": 117465,
|
| 10 |
+
"total_batch_size": 64
|
| 11 |
+
},
|
| 12 |
+
"completed_steps": 20000,
|
| 13 |
+
"optimizer_param_groups": [
|
| 14 |
+
{
|
| 15 |
+
"betas": [
|
| 16 |
+
0.9,
|
| 17 |
+
0.95
|
| 18 |
+
],
|
| 19 |
+
"eps": 1e-08,
|
| 20 |
+
"initial_lr": 1e-05,
|
| 21 |
+
"lr": 9.407412610098572e-06,
|
| 22 |
+
"name": "qwen_vl_interface",
|
| 23 |
+
"weight_decay": 1e-08
|
| 24 |
+
},
|
| 25 |
+
{
|
| 26 |
+
"betas": [
|
| 27 |
+
0.9,
|
| 28 |
+
0.95
|
| 29 |
+
],
|
| 30 |
+
"eps": 1e-08,
|
| 31 |
+
"initial_lr": 0.0001,
|
| 32 |
+
"lr": 9.407412610098571e-05,
|
| 33 |
+
"name": "action_model",
|
| 34 |
+
"weight_decay": 1e-08
|
| 35 |
+
}
|
| 36 |
+
],
|
| 37 |
+
"optimizer_state_entries": 4,
|
| 38 |
+
"parent_checkpoint": "/mnt/miaochengyu/starvla-infoflow-runtime/runs/florence-scratch-official-100k/florence-scratch-uniform-seed-42-official100k-mb4-eval1k-20260726/checkpoints/steps_20000_accelerate",
|
| 39 |
+
"parent_checkpoint_commit_sha256": "caaaae0c1c2bc45c0b3b5c0ac8ef6f3bc60a5e189b0bcc9979dad1ba371ef3ad",
|
| 40 |
+
"parent_checkpoint_files": [
|
| 41 |
+
{
|
| 42 |
+
"bytes": 13,
|
| 43 |
+
"name": "latest",
|
| 44 |
+
"sha256": "68d0ca762047fd48fdd6a06a5e6465209289c028bfd63f5557a110ff1ebc15ac"
|
| 45 |
+
},
|
| 46 |
+
{
|
| 47 |
+
"bytes": 5514449457,
|
| 48 |
+
"name": "pytorch_model/bf16_zero_pp_rank_0_mp_rank_00_optim_states.pt",
|
| 49 |
+
"sha256": "42263278dd16848dd50fd159f430bb40efa541d8b62aba5d608531d3af90f1d4"
|
| 50 |
+
},
|
| 51 |
+
{
|
| 52 |
+
"bytes": 5514463601,
|
| 53 |
+
"name": "pytorch_model/bf16_zero_pp_rank_1_mp_rank_00_optim_states.pt",
|
| 54 |
+
"sha256": "c6e21653b454605b3cb3db62bf9b1bae90ddf63eea1389cc0a0dd351f1f0839c"
|
| 55 |
+
},
|
| 56 |
+
{
|
| 57 |
+
"bytes": 1839089981,
|
| 58 |
+
"name": "pytorch_model/mp_rank_00_model_states.pt",
|
| 59 |
+
"sha256": "822f82ffa3ebb7f8e1e9cb15f7c24dafb4054d9fb2b63646fb81ba63172c09b6"
|
| 60 |
+
},
|
| 61 |
+
{
|
| 62 |
+
"bytes": 2154184469,
|
| 63 |
+
"name": "pytorch_model.pt",
|
| 64 |
+
"sha256": "c4f39cd339271347b1ab83ddc9c8dae7e07e315bea8cf86812b2e0b6772abd42"
|
| 65 |
+
},
|
| 66 |
+
{
|
| 67 |
+
"bytes": 15017,
|
| 68 |
+
"name": "random_states_0.pkl",
|
| 69 |
+
"sha256": "29e1b8cf5838519c544dfe660c71c064b59cd35d8f5c5ae9fa1e63e9cfd3e773"
|
| 70 |
+
},
|
| 71 |
+
{
|
| 72 |
+
"bytes": 15081,
|
| 73 |
+
"name": "random_states_1.pkl",
|
| 74 |
+
"sha256": "fed527bb36c43639d19411c7da8f3eb3eca04543ed7eb06f62c1fa978f3bdf31"
|
| 75 |
+
},
|
| 76 |
+
{
|
| 77 |
+
"bytes": 33270,
|
| 78 |
+
"name": "zero_to_fp32.py",
|
| 79 |
+
"sha256": "e8a2553b434ac043ec63b022db1e768277acf8d4e70f10a2842db6d238977135"
|
| 80 |
+
}
|
| 81 |
+
],
|
| 82 |
+
"parent_checkpoint_files_sha256": "223f8085e17142c10429b17ba5be21fee02fb5ae49bd409dd4f13f6b9caab867",
|
| 83 |
+
"parent_checkpoint_integrity_manifest_sha256": "899d2cef3f73191e11f7a1628afc05b60c80af5816b8fd77c1c69e90d5c9ff39",
|
| 84 |
+
"parent_config_sha256": "c2ea33c16faf5b550708a494df5600198ef7e8d15bece8dffe83609efa748993",
|
| 85 |
+
"parent_resume_identity": {
|
| 86 |
+
"config_sha256": "1c394b6043c323133b0512d3efd761ae8332e5098969864f410b71651aa56f44",
|
| 87 |
+
"manifest_sha256": "a1627df8ff110c7298eaf025ffcd4f48d9ca17117cfcfb3926b962d0bb19c0ca",
|
| 88 |
+
"model_structure_sha256": "e055e3fa0085edc6a9bf6cdb1605a05e61ae289957e24e37cfe2cfce2dc94c39",
|
| 89 |
+
"proposal_alpha": null,
|
| 90 |
+
"proposal_sha256": null,
|
| 91 |
+
"seed": 42,
|
| 92 |
+
"total_batch_size": 64
|
| 93 |
+
},
|
| 94 |
+
"proposal_sha256": null,
|
| 95 |
+
"rank_state_summaries": [
|
| 96 |
+
{
|
| 97 |
+
"optimizer_state_entries": 2,
|
| 98 |
+
"optimizer_state_sha256": "704f68d9f3c71b48c4f4ded7dd46cc6396324adad50611368f6b37f2df38e2f3",
|
| 99 |
+
"rank": 0,
|
| 100 |
+
"trainable_state_sha256": "e6b7d1c26f4ecb3d761ca3f1eaa21b7cf4be87def2c5dcef71b2860ff69fb09e"
|
| 101 |
+
},
|
| 102 |
+
{
|
| 103 |
+
"optimizer_state_entries": 2,
|
| 104 |
+
"optimizer_state_sha256": "96c9f5c983020c067f752234588992862a29feba7cc51804b9baaf3a25dcf98c",
|
| 105 |
+
"rank": 1,
|
| 106 |
+
"trainable_state_sha256": "e6b7d1c26f4ecb3d761ca3f1eaa21b7cf4be87def2c5dcef71b2860ff69fb09e"
|
| 107 |
+
}
|
| 108 |
+
],
|
| 109 |
+
"repeated_diffusion_steps": 4,
|
| 110 |
+
"rng_streams": {
|
| 111 |
+
"model_base": 588643,
|
| 112 |
+
"sampler": 117465,
|
| 113 |
+
"transform": 298226
|
| 114 |
+
},
|
| 115 |
+
"scheduler": {
|
| 116 |
+
"class": "LambdaLR",
|
| 117 |
+
"last_epoch": 20000,
|
| 118 |
+
"last_lr": [
|
| 119 |
+
9.407412610098572e-06,
|
| 120 |
+
9.407412610098571e-05
|
| 121 |
+
],
|
| 122 |
+
"step_count": 20001
|
| 123 |
+
},
|
| 124 |
+
"scheduler_resume_mode": "reconstructed_from_completed_steps",
|
| 125 |
+
"schema_version": 1,
|
| 126 |
+
"status": "exact_external_resume_loaded",
|
| 127 |
+
"total_batch_size": 64,
|
| 128 |
+
"trainable_parameter_elements": 919063815,
|
| 129 |
+
"trainable_parameter_sha256": "f2572c4f7913235cc889bdf27eca7e605bdaa4d68e92c0a32148bbf6decbd998",
|
| 130 |
+
"trainable_parameter_tensors": 1164,
|
| 131 |
+
"world_size": 2
|
| 132 |
+
}
|
settings/florence_bridge/frameskip/initialization_manifest.json
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"schema_version": 1,
|
| 3 |
+
"initialization_seed": 117465,
|
| 4 |
+
"action_model_sha256": "fbe26e8054c77b62d41353672000d33e47bad277cc0f368bb3f0db571e847682",
|
| 5 |
+
"action_model_parameter_elements": 142598407
|
| 6 |
+
}
|
settings/florence_bridge/frameskip/valid_anchor_manifest.json
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7f2123722a291f54ca3414d640f4d9e504ec510521ac5ca620b4745e713cdb8c
|
| 3 |
+
size 101950249
|
settings/florence_bridge/keyframe/config.full.yaml
ADDED
|
@@ -0,0 +1,117 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
run_id: keyframe-focused-adapted-seed-117465-20k-to-20500
|
| 2 |
+
run_root_dir: /home/liuyu/student-workspace/outputs/keyframe_focused_adapted/runs
|
| 3 |
+
seed: 117465
|
| 4 |
+
wandb_entity: null
|
| 5 |
+
wandb_project: Keyframe-Focused-adapted-BridgeRT
|
| 6 |
+
is_debug: false
|
| 7 |
+
version_id: '0.21'
|
| 8 |
+
framework:
|
| 9 |
+
name: QwenGR00T
|
| 10 |
+
qwenvl:
|
| 11 |
+
base_vlm: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Pretrained_models/Florence-2-large
|
| 12 |
+
attn_implementation: flash_attention_2
|
| 13 |
+
vl_hidden_dim: 2048
|
| 14 |
+
action_model:
|
| 15 |
+
action_model_type: DiT-B
|
| 16 |
+
action_hidden_dim: 1024
|
| 17 |
+
hidden_size: 1024
|
| 18 |
+
add_pos_embed: true
|
| 19 |
+
max_seq_len: 1024
|
| 20 |
+
action_dim: 7
|
| 21 |
+
state_dim: 7
|
| 22 |
+
action_horizon: 16
|
| 23 |
+
repeated_diffusion_steps: 8
|
| 24 |
+
noise_beta_alpha: 1.5
|
| 25 |
+
noise_beta_beta: 1.0
|
| 26 |
+
noise_s: 0.999
|
| 27 |
+
num_timestep_buckets: 1000
|
| 28 |
+
num_inference_timesteps: 4
|
| 29 |
+
num_target_vision_tokens: 32
|
| 30 |
+
diffusion_model_cfg:
|
| 31 |
+
cross_attention_dim: 1024
|
| 32 |
+
dropout: 0.2
|
| 33 |
+
final_dropout: true
|
| 34 |
+
interleave_self_attention: true
|
| 35 |
+
norm_type: ada_norm
|
| 36 |
+
num_layers: 16
|
| 37 |
+
output_dim: 1024
|
| 38 |
+
positional_embeddings: null
|
| 39 |
+
future_action_window_size: 15
|
| 40 |
+
past_action_window_size: 0
|
| 41 |
+
dino:
|
| 42 |
+
dino_backbone: dinov2_vits14
|
| 43 |
+
datasets:
|
| 44 |
+
vla_data:
|
| 45 |
+
dataset_py: lerobot_datasets
|
| 46 |
+
data_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Datasets/OXE_LEROBOT_DATASET
|
| 47 |
+
data_mix: bridge_rt_infoflow
|
| 48 |
+
action_type: delta_ee
|
| 49 |
+
action_mode: abs
|
| 50 |
+
precomputed_statistics_path: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Pretrained_models/Florence-GR00T-Bridge-RT-1/dataset_statistics.json
|
| 51 |
+
precomputed_statistics_key_map:
|
| 52 |
+
bridge_orig_lerobot: oxe_bridge
|
| 53 |
+
fractal20220817_data_lerobot: oxe_rt1
|
| 54 |
+
CoT_prompt: Your task is {instruction}. To identify the key objects for your task.
|
| 55 |
+
Locate their bounding boxes in [x1,y1,x2,y2] format.
|
| 56 |
+
per_device_batch_size: 4
|
| 57 |
+
load_all_data_for_training: true
|
| 58 |
+
obs_image_size:
|
| 59 |
+
- 224
|
| 60 |
+
- 224
|
| 61 |
+
video_backend: torchvision_av
|
| 62 |
+
num_workers: 4
|
| 63 |
+
balance_dataset_weights: false
|
| 64 |
+
balance_trajectory_weights: false
|
| 65 |
+
traceable_sampling: true
|
| 66 |
+
expected_world_size: 2
|
| 67 |
+
dataset_provenance_path: /home/liuyu/student-workspace/code/starVLA-infoflow-codex/protocols/bridgert/data_provenance.json
|
| 68 |
+
dataset_content_fingerprint_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/bridgert/content_fingerprints.json
|
| 69 |
+
valid_anchor_manifest_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/bridgert/valid_anchor_manifest_florence_scratch_v1.json
|
| 70 |
+
valid_anchor_manifest_migration_path: null
|
| 71 |
+
temporal_proposal_path: null
|
| 72 |
+
transform_rng_seed: 298226
|
| 73 |
+
trainer:
|
| 74 |
+
use_deepspeed: true
|
| 75 |
+
max_train_steps: 100000
|
| 76 |
+
num_warmup_steps: 5000
|
| 77 |
+
save_interval: null
|
| 78 |
+
save_steps:
|
| 79 |
+
- 20500
|
| 80 |
+
evaluation_snapshot_interval: 1000
|
| 81 |
+
eval_interval: 100001
|
| 82 |
+
repeated_diffusion_steps: 4
|
| 83 |
+
pretrained_checkpoint: null
|
| 84 |
+
is_resume: true
|
| 85 |
+
allow_model_only_resume: false
|
| 86 |
+
learning_rate:
|
| 87 |
+
base: 3.0e-05
|
| 88 |
+
qwen_vl_interface: 1.0e-05
|
| 89 |
+
action_model: 0.0001
|
| 90 |
+
lr_scheduler_type: cosine_with_min_lr
|
| 91 |
+
scheduler_specific_kwargs:
|
| 92 |
+
min_lr: 5.0e-07
|
| 93 |
+
freeze_modules: ''
|
| 94 |
+
trainable_modules: ''
|
| 95 |
+
frozen_modules_eval: false
|
| 96 |
+
logging_frequency: 50
|
| 97 |
+
gradient_clipping: 1.0
|
| 98 |
+
gradient_accumulation_steps: 8
|
| 99 |
+
save_format: pt
|
| 100 |
+
save_final_checkpoint: false
|
| 101 |
+
optimizer:
|
| 102 |
+
name: AdamW
|
| 103 |
+
betas:
|
| 104 |
+
- 0.9
|
| 105 |
+
- 0.95
|
| 106 |
+
eps: 1.0e-08
|
| 107 |
+
weight_decay: 1.0e-08
|
| 108 |
+
resume_accelerate_checkpoint: /mnt/miaochengyu/starvla-infoflow-runtime/runs/florence-scratch-official-100k/florence-scratch-uniform-seed-42-official100k-mb4-eval1k-20260726/checkpoints/steps_20000_accelerate
|
| 109 |
+
resume_checkpoint_integrity_manifest: /home/liuyu/student-workspace/code/starVLA-infoflow-codex/protocols/bridgert/common_start_accelerate_integrity_20260730.json
|
| 110 |
+
resume_checkpoint_integrity_manifest_sha256: 899d2cef3f73191e11f7a1628afc05b60c80af5816b8fd77c1c69e90d5c9ff39
|
| 111 |
+
branch_parent_seed: 42
|
| 112 |
+
branch_rng_seed: 117465
|
| 113 |
+
model_rng_seed: 588643
|
| 114 |
+
sample_loss_weight_key: keyframe_weight
|
| 115 |
+
stop_after_steps: 20500
|
| 116 |
+
config_yaml: /home/liuyu/student-workspace/code/starVLA-infoflow-codex/external_baselines/keyframe_focused_adapted/configs/keyframe_focused_seed117465_20k_to_20500.yaml
|
| 117 |
+
output_dir: /home/liuyu/student-workspace/outputs/keyframe_focused_adapted/runs/keyframe-focused-adapted-seed-117465-20k-to-20500
|
settings/florence_bridge/keyframe/config.yaml
ADDED
|
@@ -0,0 +1,70 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
datasets:
|
| 2 |
+
vla_data:
|
| 3 |
+
action_mode: abs
|
| 4 |
+
balance_dataset_weights: false
|
| 5 |
+
balance_trajectory_weights: false
|
| 6 |
+
data_mix: bridge_rt_infoflow
|
| 7 |
+
data_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Datasets/OXE_LEROBOT_DATASET
|
| 8 |
+
dataset_content_fingerprint_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/bridgert/content_fingerprints.json
|
| 9 |
+
dataset_provenance_path: /home/liuyu/student-workspace/code/starVLA-infoflow-codex/protocols/bridgert/data_provenance.json
|
| 10 |
+
dataset_py: lerobot_datasets
|
| 11 |
+
expected_world_size: 2
|
| 12 |
+
num_workers: 4
|
| 13 |
+
per_device_batch_size: 4
|
| 14 |
+
precomputed_statistics_key_map:
|
| 15 |
+
bridge_orig_lerobot: oxe_bridge
|
| 16 |
+
fractal20220817_data_lerobot: oxe_rt1
|
| 17 |
+
precomputed_statistics_path: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Pretrained_models/Florence-GR00T-Bridge-RT-1/dataset_statistics.json
|
| 18 |
+
temporal_proposal_path: null
|
| 19 |
+
traceable_sampling: true
|
| 20 |
+
valid_anchor_manifest_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/bridgert/valid_anchor_manifest_florence_scratch_v1.json
|
| 21 |
+
video_backend: torchvision_av
|
| 22 |
+
framework:
|
| 23 |
+
action_model:
|
| 24 |
+
action_dim: 7
|
| 25 |
+
action_horizon: 16
|
| 26 |
+
action_model_type: DiT-B
|
| 27 |
+
add_pos_embed: true
|
| 28 |
+
diffusion_model_cfg:
|
| 29 |
+
cross_attention_dim: 1024
|
| 30 |
+
dropout: 0.2
|
| 31 |
+
final_dropout: true
|
| 32 |
+
interleave_self_attention: true
|
| 33 |
+
norm_type: ada_norm
|
| 34 |
+
num_layers: 16
|
| 35 |
+
output_dim: 1024
|
| 36 |
+
positional_embeddings: null
|
| 37 |
+
hidden_size: 1024
|
| 38 |
+
max_seq_len: 1024
|
| 39 |
+
noise_beta_alpha: 1.5
|
| 40 |
+
noise_beta_beta: 1.0
|
| 41 |
+
num_inference_timesteps: 4
|
| 42 |
+
num_target_vision_tokens: 32
|
| 43 |
+
num_timestep_buckets: 1000
|
| 44 |
+
state_dim: 7
|
| 45 |
+
name: QwenGR00T
|
| 46 |
+
qwenvl:
|
| 47 |
+
base_vlm: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Pretrained_models/Florence-2-large
|
| 48 |
+
output_dir: /home/liuyu/student-workspace/outputs/keyframe_focused_adapted/runs/keyframe-focused-adapted-seed-117465-20k-to-20500
|
| 49 |
+
run_id: keyframe-focused-adapted-seed-117465-20k-to-20500
|
| 50 |
+
run_root_dir: /home/liuyu/student-workspace/outputs/keyframe_focused_adapted/runs
|
| 51 |
+
seed: 117465
|
| 52 |
+
trainer:
|
| 53 |
+
freeze_modules: ''
|
| 54 |
+
gradient_accumulation_steps: 8
|
| 55 |
+
learning_rate:
|
| 56 |
+
action_model: 0.0001
|
| 57 |
+
base: 3.0e-05
|
| 58 |
+
qwen_vl_interface: 1.0e-05
|
| 59 |
+
lr_scheduler_type: cosine_with_min_lr
|
| 60 |
+
max_train_steps: 100000
|
| 61 |
+
num_warmup_steps: 5000
|
| 62 |
+
optimizer:
|
| 63 |
+
betas:
|
| 64 |
+
- 0.9
|
| 65 |
+
- 0.95
|
| 66 |
+
eps: 1.0e-08
|
| 67 |
+
name: AdamW
|
| 68 |
+
weight_decay: 1.0e-08
|
| 69 |
+
scheduler_specific_kwargs:
|
| 70 |
+
min_lr: 5.0e-07
|
settings/florence_bridge/keyframe/dataset_statistics.json
ADDED
|
@@ -0,0 +1,264 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"oxe_bridge": {
|
| 3 |
+
"action": {
|
| 4 |
+
"mean": [
|
| 5 |
+
5.682967821485363e-05,
|
| 6 |
+
3.278055373812094e-05,
|
| 7 |
+
-3.1604828109266236e-05,
|
| 8 |
+
-3.6026765883434564e-05,
|
| 9 |
+
-9.757651423569769e-05,
|
| 10 |
+
6.01583014940843e-05,
|
| 11 |
+
0.14414729177951813
|
| 12 |
+
],
|
| 13 |
+
"std": [
|
| 14 |
+
0.004886276229987901,
|
| 15 |
+
0.006847766456005231,
|
| 16 |
+
0.006337809696791937,
|
| 17 |
+
0.014227778584428946,
|
| 18 |
+
0.015261554008856799,
|
| 19 |
+
0.038695293138416895,
|
| 20 |
+
0.3521380562074972
|
| 21 |
+
],
|
| 22 |
+
"max": [
|
| 23 |
+
0.41691166162490845,
|
| 24 |
+
0.25864794850349426,
|
| 25 |
+
0.21218234300613403,
|
| 26 |
+
3.122201919555664,
|
| 27 |
+
1.8618112802505493,
|
| 28 |
+
6.272472858428955,
|
| 29 |
+
1.0
|
| 30 |
+
],
|
| 31 |
+
"min": [
|
| 32 |
+
-0.4007510244846344,
|
| 33 |
+
-0.13874775171279907,
|
| 34 |
+
-0.22553899884223938,
|
| 35 |
+
-3.2010786533355713,
|
| 36 |
+
-1.8618112802505493,
|
| 37 |
+
-6.279075622558594,
|
| 38 |
+
0.0
|
| 39 |
+
],
|
| 40 |
+
"q01": [
|
| 41 |
+
-0.02875255048274994,
|
| 42 |
+
-0.04170213546603918,
|
| 43 |
+
-0.026096721179783344,
|
| 44 |
+
-0.08052874729037285,
|
| 45 |
+
-0.09249906800687313,
|
| 46 |
+
-0.20738555490970612,
|
| 47 |
+
0.0
|
| 48 |
+
],
|
| 49 |
+
"q99": [
|
| 50 |
+
0.028306663036346436,
|
| 51 |
+
0.04089853074401617,
|
| 52 |
+
0.0401805154979229,
|
| 53 |
+
0.08173403143882751,
|
| 54 |
+
0.07760760560631752,
|
| 55 |
+
0.2038465365767479,
|
| 56 |
+
1.0
|
| 57 |
+
],
|
| 58 |
+
"mask": [
|
| 59 |
+
true,
|
| 60 |
+
true,
|
| 61 |
+
true,
|
| 62 |
+
true,
|
| 63 |
+
true,
|
| 64 |
+
true,
|
| 65 |
+
false
|
| 66 |
+
]
|
| 67 |
+
},
|
| 68 |
+
"state": {
|
| 69 |
+
"mean": [
|
| 70 |
+
0.07735871523618698,
|
| 71 |
+
0.007681275252252817,
|
| 72 |
+
0.016110140830278397,
|
| 73 |
+
0.0016226600855588913,
|
| 74 |
+
-0.019300073385238647,
|
| 75 |
+
0.02691473439335823,
|
| 76 |
+
0.0,
|
| 77 |
+
0.17703326046466827
|
| 78 |
+
],
|
| 79 |
+
"std": [
|
| 80 |
+
0.13736907523153707,
|
| 81 |
+
0.047840646787850445,
|
| 82 |
+
0.03799073317180484,
|
| 83 |
+
0.06560660645733722,
|
| 84 |
+
0.09093375673607122,
|
| 85 |
+
0.292728734025679,
|
| 86 |
+
0.0,
|
| 87 |
+
0.3536876512395259
|
| 88 |
+
],
|
| 89 |
+
"max": [
|
| 90 |
+
0.5862360596656799,
|
| 91 |
+
0.4034728705883026,
|
| 92 |
+
0.3568263053894043,
|
| 93 |
+
1.3517684936523438,
|
| 94 |
+
1.570796251296997,
|
| 95 |
+
3.141204357147217,
|
| 96 |
+
0.0,
|
| 97 |
+
1.1121242046356201
|
| 98 |
+
],
|
| 99 |
+
"min": [
|
| 100 |
+
-0.04167502000927925,
|
| 101 |
+
-0.3563207685947418,
|
| 102 |
+
-0.15537554025650024,
|
| 103 |
+
-3.141592502593994,
|
| 104 |
+
-1.4992541074752808,
|
| 105 |
+
-3.14153790473938,
|
| 106 |
+
0.0,
|
| 107 |
+
0.04637829214334488
|
| 108 |
+
],
|
| 109 |
+
"q01": [
|
| 110 |
+
0.17102622985839844,
|
| 111 |
+
-0.1698118858039379,
|
| 112 |
+
-0.055632163770496845,
|
| 113 |
+
-0.36493172496557236,
|
| 114 |
+
-0.541871190071106,
|
| 115 |
+
-1.354275494813919,
|
| 116 |
+
0.0,
|
| 117 |
+
0.052190229296684265
|
| 118 |
+
],
|
| 119 |
+
"q99": [
|
| 120 |
+
0.45322076976299286,
|
| 121 |
+
0.2354845181107521,
|
| 122 |
+
0.19489620998501778,
|
| 123 |
+
0.3780156075954437,
|
| 124 |
+
0.27568644285202026,
|
| 125 |
+
1.8500566184520721,
|
| 126 |
+
0.0,
|
| 127 |
+
1.0105689764022827
|
| 128 |
+
]
|
| 129 |
+
},
|
| 130 |
+
"num_transitions": 1305714,
|
| 131 |
+
"num_trajectories": 53192
|
| 132 |
+
},
|
| 133 |
+
"oxe_rt1": {
|
| 134 |
+
"action": {
|
| 135 |
+
"mean": [
|
| 136 |
+
0.0017468753503635526,
|
| 137 |
+
0.0015664808452129364,
|
| 138 |
+
-0.00315628363750875,
|
| 139 |
+
0.010832978412508965,
|
| 140 |
+
-0.0014390418073162436,
|
| 141 |
+
0.00022827448265161365,
|
| 142 |
+
0.13385511934757233
|
| 143 |
+
],
|
| 144 |
+
"std": [
|
| 145 |
+
0.034738154754287556,
|
| 146 |
+
0.029950575875189896,
|
| 147 |
+
0.03716987948198757,
|
| 148 |
+
0.08027394458720655,
|
| 149 |
+
0.06586824211259418,
|
| 150 |
+
0.07296740097742303,
|
| 151 |
+
0.33989797865933385
|
| 152 |
+
],
|
| 153 |
+
"max": [
|
| 154 |
+
2.9984593391418457,
|
| 155 |
+
22.09052848815918,
|
| 156 |
+
2.7507524490356445,
|
| 157 |
+
1.570636510848999,
|
| 158 |
+
1.5321086645126343,
|
| 159 |
+
1.5691522359848022,
|
| 160 |
+
1.0
|
| 161 |
+
],
|
| 162 |
+
"min": [
|
| 163 |
+
-2.0204520225524902,
|
| 164 |
+
-5.497899532318115,
|
| 165 |
+
-2.031663417816162,
|
| 166 |
+
-1.569917917251587,
|
| 167 |
+
-1.569892168045044,
|
| 168 |
+
-1.570419430732727,
|
| 169 |
+
0.0
|
| 170 |
+
],
|
| 171 |
+
"q01": [
|
| 172 |
+
-0.22453527510166169,
|
| 173 |
+
-0.14820013284683228,
|
| 174 |
+
-0.231589707583189,
|
| 175 |
+
-0.3517994859814644,
|
| 176 |
+
-0.4193011274933815,
|
| 177 |
+
-0.43643461108207704,
|
| 178 |
+
0.0
|
| 179 |
+
],
|
| 180 |
+
"q99": [
|
| 181 |
+
0.17824687153100965,
|
| 182 |
+
0.14938379630446405,
|
| 183 |
+
0.21842354819178575,
|
| 184 |
+
0.5892666035890578,
|
| 185 |
+
0.35272657424211445,
|
| 186 |
+
0.44796681255102094,
|
| 187 |
+
1.0
|
| 188 |
+
],
|
| 189 |
+
"mask": [
|
| 190 |
+
true,
|
| 191 |
+
true,
|
| 192 |
+
true,
|
| 193 |
+
true,
|
| 194 |
+
true,
|
| 195 |
+
true,
|
| 196 |
+
false
|
| 197 |
+
]
|
| 198 |
+
},
|
| 199 |
+
"state": {
|
| 200 |
+
"mean": [
|
| 201 |
+
0.1399736851453781,
|
| 202 |
+
-0.02083534561097622,
|
| 203 |
+
0.19427375495433807,
|
| 204 |
+
0.10653364658355713,
|
| 205 |
+
-0.06201138719916344,
|
| 206 |
+
0.12378455698490143,
|
| 207 |
+
0.023165317252278328,
|
| 208 |
+
0.052437249571084976
|
| 209 |
+
],
|
| 210 |
+
"std": [
|
| 211 |
+
0.2502841726583427,
|
| 212 |
+
0.06813404469049046,
|
| 213 |
+
0.358260305088155,
|
| 214 |
+
0.2931043604956031,
|
| 215 |
+
0.2779312663188148,
|
| 216 |
+
0.33769767306558707,
|
| 217 |
+
0.09232656626252377,
|
| 218 |
+
0.16498523970985932
|
| 219 |
+
],
|
| 220 |
+
"max": [
|
| 221 |
+
1.0534898042678833,
|
| 222 |
+
0.48018959164619446,
|
| 223 |
+
1.6896663904190063,
|
| 224 |
+
1.0,
|
| 225 |
+
0.9999993443489075,
|
| 226 |
+
0.9999874830245972,
|
| 227 |
+
0.9554369449615479,
|
| 228 |
+
0.9914546012878418
|
| 229 |
+
],
|
| 230 |
+
"min": [
|
| 231 |
+
-0.4436439275741577,
|
| 232 |
+
-0.9970501065254211,
|
| 233 |
+
-0.006579156965017319,
|
| 234 |
+
0.0,
|
| 235 |
+
-0.8643477559089661,
|
| 236 |
+
-0.7079970240592957,
|
| 237 |
+
-0.7688722014427185,
|
| 238 |
+
-0.4999994933605194
|
| 239 |
+
],
|
| 240 |
+
"q01": [
|
| 241 |
+
0.32481380939483645,
|
| 242 |
+
-0.28334290891885755,
|
| 243 |
+
0.14107070609927178,
|
| 244 |
+
0.0,
|
| 245 |
+
-0.686474204659462,
|
| 246 |
+
-0.6808923494815826,
|
| 247 |
+
-0.36045596331357954,
|
| 248 |
+
-0.454380963742733
|
| 249 |
+
],
|
| 250 |
+
"q99": [
|
| 251 |
+
0.8750156319141384,
|
| 252 |
+
0.21247054174542404,
|
| 253 |
+
1.0727112340927123,
|
| 254 |
+
1.0,
|
| 255 |
+
0.9377871316671368,
|
| 256 |
+
0.9563051050901409,
|
| 257 |
+
0.45990042358636823,
|
| 258 |
+
0.7216041100025177
|
| 259 |
+
]
|
| 260 |
+
},
|
| 261 |
+
"num_transitions": 3786152,
|
| 262 |
+
"num_trajectories": 87212
|
| 263 |
+
}
|
| 264 |
+
}
|
settings/florence_bridge/loss_is/config.full.yaml
ADDED
|
@@ -0,0 +1,123 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
run_id: loss-is-online-seed-117465-20k-to-20500
|
| 2 |
+
run_root_dir: /home/liuyu/student-workspace/outputs/loss_is_adapted/runs
|
| 3 |
+
seed: 117465
|
| 4 |
+
wandb_entity: null
|
| 5 |
+
wandb_project: Loss-IS-adapted-BridgeRT
|
| 6 |
+
is_debug: false
|
| 7 |
+
version_id: '0.21'
|
| 8 |
+
framework:
|
| 9 |
+
name: QwenGR00T
|
| 10 |
+
qwenvl:
|
| 11 |
+
base_vlm: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Pretrained_models/Florence-2-large
|
| 12 |
+
attn_implementation: flash_attention_2
|
| 13 |
+
vl_hidden_dim: 2048
|
| 14 |
+
action_model:
|
| 15 |
+
action_model_type: DiT-B
|
| 16 |
+
action_hidden_dim: 1024
|
| 17 |
+
hidden_size: 1024
|
| 18 |
+
add_pos_embed: true
|
| 19 |
+
max_seq_len: 1024
|
| 20 |
+
action_dim: 7
|
| 21 |
+
state_dim: 7
|
| 22 |
+
action_horizon: 16
|
| 23 |
+
repeated_diffusion_steps: 8
|
| 24 |
+
noise_beta_alpha: 1.5
|
| 25 |
+
noise_beta_beta: 1.0
|
| 26 |
+
noise_s: 0.999
|
| 27 |
+
num_timestep_buckets: 1000
|
| 28 |
+
num_inference_timesteps: 4
|
| 29 |
+
num_target_vision_tokens: 32
|
| 30 |
+
diffusion_model_cfg:
|
| 31 |
+
cross_attention_dim: 1024
|
| 32 |
+
dropout: 0.2
|
| 33 |
+
final_dropout: true
|
| 34 |
+
interleave_self_attention: true
|
| 35 |
+
norm_type: ada_norm
|
| 36 |
+
num_layers: 16
|
| 37 |
+
output_dim: 1024
|
| 38 |
+
positional_embeddings: null
|
| 39 |
+
future_action_window_size: 15
|
| 40 |
+
past_action_window_size: 0
|
| 41 |
+
dino:
|
| 42 |
+
dino_backbone: dinov2_vits14
|
| 43 |
+
datasets:
|
| 44 |
+
vla_data:
|
| 45 |
+
dataset_py: lerobot_datasets
|
| 46 |
+
data_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Datasets/OXE_LEROBOT_DATASET
|
| 47 |
+
data_mix: bridge_rt_infoflow
|
| 48 |
+
action_type: delta_ee
|
| 49 |
+
action_mode: abs
|
| 50 |
+
precomputed_statistics_path: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Pretrained_models/Florence-GR00T-Bridge-RT-1/dataset_statistics.json
|
| 51 |
+
precomputed_statistics_key_map:
|
| 52 |
+
bridge_orig_lerobot: oxe_bridge
|
| 53 |
+
fractal20220817_data_lerobot: oxe_rt1
|
| 54 |
+
CoT_prompt: Your task is {instruction}. To identify the key objects for your task.
|
| 55 |
+
Locate their bounding boxes in [x1,y1,x2,y2] format.
|
| 56 |
+
per_device_batch_size: 4
|
| 57 |
+
load_all_data_for_training: true
|
| 58 |
+
obs_image_size:
|
| 59 |
+
- 224
|
| 60 |
+
- 224
|
| 61 |
+
video_backend: torchvision_av
|
| 62 |
+
num_workers: 4
|
| 63 |
+
balance_dataset_weights: false
|
| 64 |
+
balance_trajectory_weights: false
|
| 65 |
+
traceable_sampling: true
|
| 66 |
+
expected_world_size: 2
|
| 67 |
+
dataset_provenance_path: /home/liuyu/student-workspace/code/starVLA-infoflow-codex/protocols/bridgert/data_provenance.json
|
| 68 |
+
dataset_content_fingerprint_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/bridgert/content_fingerprints.json
|
| 69 |
+
valid_anchor_manifest_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/bridgert/valid_anchor_manifest_florence_scratch_v1.json
|
| 70 |
+
valid_anchor_manifest_migration_path: null
|
| 71 |
+
temporal_proposal_path: null
|
| 72 |
+
transform_rng_seed: 298226
|
| 73 |
+
trainer:
|
| 74 |
+
use_deepspeed: true
|
| 75 |
+
max_train_steps: 100000
|
| 76 |
+
num_warmup_steps: 5000
|
| 77 |
+
save_interval: null
|
| 78 |
+
save_steps:
|
| 79 |
+
- 20500
|
| 80 |
+
evaluation_snapshot_interval: 1000
|
| 81 |
+
eval_interval: 100001
|
| 82 |
+
repeated_diffusion_steps: 4
|
| 83 |
+
pretrained_checkpoint: null
|
| 84 |
+
is_resume: true
|
| 85 |
+
allow_model_only_resume: false
|
| 86 |
+
learning_rate:
|
| 87 |
+
base: 3.0e-05
|
| 88 |
+
qwen_vl_interface: 1.0e-05
|
| 89 |
+
action_model: 0.0001
|
| 90 |
+
lr_scheduler_type: cosine_with_min_lr
|
| 91 |
+
scheduler_specific_kwargs:
|
| 92 |
+
min_lr: 5.0e-07
|
| 93 |
+
freeze_modules: ''
|
| 94 |
+
trainable_modules: ''
|
| 95 |
+
frozen_modules_eval: false
|
| 96 |
+
logging_frequency: 50
|
| 97 |
+
gradient_clipping: 1.0
|
| 98 |
+
gradient_accumulation_steps: 8
|
| 99 |
+
save_format: pt
|
| 100 |
+
save_final_checkpoint: false
|
| 101 |
+
optimizer:
|
| 102 |
+
name: AdamW
|
| 103 |
+
betas:
|
| 104 |
+
- 0.9
|
| 105 |
+
- 0.95
|
| 106 |
+
eps: 1.0e-08
|
| 107 |
+
weight_decay: 1.0e-08
|
| 108 |
+
resume_accelerate_checkpoint: /mnt/miaochengyu/starvla-infoflow-runtime/runs/florence-scratch-official-100k/florence-scratch-uniform-seed-42-official100k-mb4-eval1k-20260726/checkpoints/steps_20000_accelerate
|
| 109 |
+
resume_checkpoint_integrity_manifest: /home/liuyu/student-workspace/code/starVLA-infoflow-codex/protocols/bridgert/common_start_accelerate_integrity_20260730.json
|
| 110 |
+
resume_checkpoint_integrity_manifest_sha256: 899d2cef3f73191e11f7a1628afc05b60c80af5816b8fd77c1c69e90d5c9ff39
|
| 111 |
+
branch_parent_seed: 42
|
| 112 |
+
branch_rng_seed: 117465
|
| 113 |
+
model_rng_seed: 588643
|
| 114 |
+
model_rng_trace_key: loss_is_occurrence_id
|
| 115 |
+
sample_loss_weight_key: loss_is_importance_weight
|
| 116 |
+
stop_after_steps: 20500
|
| 117 |
+
loss_is:
|
| 118 |
+
candidate_factor: 3
|
| 119 |
+
seed: 117465
|
| 120 |
+
condition_threshold: 2.0
|
| 121 |
+
condition_momentum: 0.9
|
| 122 |
+
config_yaml: /home/liuyu/student-workspace/code/starVLA-loss-is-adapted/external_baselines/loss_is_adapted/configs/loss_is_seed117465_20k_to_20500.yaml
|
| 123 |
+
output_dir: /home/liuyu/student-workspace/outputs/loss_is_adapted/runs/loss-is-online-seed-117465-20k-to-20500
|
settings/florence_bridge/loss_is/config.yaml
ADDED
|
@@ -0,0 +1,75 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
datasets:
|
| 2 |
+
vla_data:
|
| 3 |
+
action_mode: abs
|
| 4 |
+
balance_dataset_weights: false
|
| 5 |
+
balance_trajectory_weights: false
|
| 6 |
+
data_mix: bridge_rt_infoflow
|
| 7 |
+
data_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Datasets/OXE_LEROBOT_DATASET
|
| 8 |
+
dataset_content_fingerprint_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/bridgert/content_fingerprints.json
|
| 9 |
+
dataset_provenance_path: /home/liuyu/student-workspace/code/starVLA-infoflow-codex/protocols/bridgert/data_provenance.json
|
| 10 |
+
dataset_py: lerobot_datasets
|
| 11 |
+
expected_world_size: 2
|
| 12 |
+
num_workers: 4
|
| 13 |
+
per_device_batch_size: 4
|
| 14 |
+
precomputed_statistics_key_map:
|
| 15 |
+
bridge_orig_lerobot: oxe_bridge
|
| 16 |
+
fractal20220817_data_lerobot: oxe_rt1
|
| 17 |
+
precomputed_statistics_path: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Pretrained_models/Florence-GR00T-Bridge-RT-1/dataset_statistics.json
|
| 18 |
+
temporal_proposal_path: null
|
| 19 |
+
traceable_sampling: true
|
| 20 |
+
valid_anchor_manifest_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/bridgert/valid_anchor_manifest_florence_scratch_v1.json
|
| 21 |
+
video_backend: torchvision_av
|
| 22 |
+
framework:
|
| 23 |
+
action_model:
|
| 24 |
+
action_dim: 7
|
| 25 |
+
action_horizon: 16
|
| 26 |
+
action_model_type: DiT-B
|
| 27 |
+
add_pos_embed: true
|
| 28 |
+
diffusion_model_cfg:
|
| 29 |
+
cross_attention_dim: 1024
|
| 30 |
+
dropout: 0.2
|
| 31 |
+
final_dropout: true
|
| 32 |
+
interleave_self_attention: true
|
| 33 |
+
norm_type: ada_norm
|
| 34 |
+
num_layers: 16
|
| 35 |
+
output_dim: 1024
|
| 36 |
+
positional_embeddings: null
|
| 37 |
+
hidden_size: 1024
|
| 38 |
+
max_seq_len: 1024
|
| 39 |
+
noise_beta_alpha: 1.5
|
| 40 |
+
noise_beta_beta: 1.0
|
| 41 |
+
num_inference_timesteps: 4
|
| 42 |
+
num_target_vision_tokens: 32
|
| 43 |
+
num_timestep_buckets: 1000
|
| 44 |
+
state_dim: 7
|
| 45 |
+
name: QwenGR00T
|
| 46 |
+
qwenvl:
|
| 47 |
+
base_vlm: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Pretrained_models/Florence-2-large
|
| 48 |
+
output_dir: /home/liuyu/student-workspace/outputs/loss_is_adapted/runs/loss-is-online-seed-117465-20k-to-20500
|
| 49 |
+
run_id: loss-is-online-seed-117465-20k-to-20500
|
| 50 |
+
run_root_dir: /home/liuyu/student-workspace/outputs/loss_is_adapted/runs
|
| 51 |
+
seed: 117465
|
| 52 |
+
trainer:
|
| 53 |
+
freeze_modules: ''
|
| 54 |
+
gradient_accumulation_steps: 8
|
| 55 |
+
learning_rate:
|
| 56 |
+
action_model: 0.0001
|
| 57 |
+
base: 3.0e-05
|
| 58 |
+
qwen_vl_interface: 1.0e-05
|
| 59 |
+
loss_is:
|
| 60 |
+
candidate_factor: 3
|
| 61 |
+
condition_momentum: 0.9
|
| 62 |
+
condition_threshold: 2.0
|
| 63 |
+
seed: 117465
|
| 64 |
+
lr_scheduler_type: cosine_with_min_lr
|
| 65 |
+
max_train_steps: 100000
|
| 66 |
+
num_warmup_steps: 5000
|
| 67 |
+
optimizer:
|
| 68 |
+
betas:
|
| 69 |
+
- 0.9
|
| 70 |
+
- 0.95
|
| 71 |
+
eps: 1.0e-08
|
| 72 |
+
name: AdamW
|
| 73 |
+
weight_decay: 1.0e-08
|
| 74 |
+
scheduler_specific_kwargs:
|
| 75 |
+
min_lr: 5.0e-07
|
settings/florence_bridge/loss_is/dataset_statistics.json
ADDED
|
@@ -0,0 +1,264 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"oxe_bridge": {
|
| 3 |
+
"action": {
|
| 4 |
+
"mean": [
|
| 5 |
+
5.682967821485363e-05,
|
| 6 |
+
3.278055373812094e-05,
|
| 7 |
+
-3.1604828109266236e-05,
|
| 8 |
+
-3.6026765883434564e-05,
|
| 9 |
+
-9.757651423569769e-05,
|
| 10 |
+
6.01583014940843e-05,
|
| 11 |
+
0.14414729177951813
|
| 12 |
+
],
|
| 13 |
+
"std": [
|
| 14 |
+
0.004886276229987901,
|
| 15 |
+
0.006847766456005231,
|
| 16 |
+
0.006337809696791937,
|
| 17 |
+
0.014227778584428946,
|
| 18 |
+
0.015261554008856799,
|
| 19 |
+
0.038695293138416895,
|
| 20 |
+
0.3521380562074972
|
| 21 |
+
],
|
| 22 |
+
"max": [
|
| 23 |
+
0.41691166162490845,
|
| 24 |
+
0.25864794850349426,
|
| 25 |
+
0.21218234300613403,
|
| 26 |
+
3.122201919555664,
|
| 27 |
+
1.8618112802505493,
|
| 28 |
+
6.272472858428955,
|
| 29 |
+
1.0
|
| 30 |
+
],
|
| 31 |
+
"min": [
|
| 32 |
+
-0.4007510244846344,
|
| 33 |
+
-0.13874775171279907,
|
| 34 |
+
-0.22553899884223938,
|
| 35 |
+
-3.2010786533355713,
|
| 36 |
+
-1.8618112802505493,
|
| 37 |
+
-6.279075622558594,
|
| 38 |
+
0.0
|
| 39 |
+
],
|
| 40 |
+
"q01": [
|
| 41 |
+
-0.02875255048274994,
|
| 42 |
+
-0.04170213546603918,
|
| 43 |
+
-0.026096721179783344,
|
| 44 |
+
-0.08052874729037285,
|
| 45 |
+
-0.09249906800687313,
|
| 46 |
+
-0.20738555490970612,
|
| 47 |
+
0.0
|
| 48 |
+
],
|
| 49 |
+
"q99": [
|
| 50 |
+
0.028306663036346436,
|
| 51 |
+
0.04089853074401617,
|
| 52 |
+
0.0401805154979229,
|
| 53 |
+
0.08173403143882751,
|
| 54 |
+
0.07760760560631752,
|
| 55 |
+
0.2038465365767479,
|
| 56 |
+
1.0
|
| 57 |
+
],
|
| 58 |
+
"mask": [
|
| 59 |
+
true,
|
| 60 |
+
true,
|
| 61 |
+
true,
|
| 62 |
+
true,
|
| 63 |
+
true,
|
| 64 |
+
true,
|
| 65 |
+
false
|
| 66 |
+
]
|
| 67 |
+
},
|
| 68 |
+
"state": {
|
| 69 |
+
"mean": [
|
| 70 |
+
0.07735871523618698,
|
| 71 |
+
0.007681275252252817,
|
| 72 |
+
0.016110140830278397,
|
| 73 |
+
0.0016226600855588913,
|
| 74 |
+
-0.019300073385238647,
|
| 75 |
+
0.02691473439335823,
|
| 76 |
+
0.0,
|
| 77 |
+
0.17703326046466827
|
| 78 |
+
],
|
| 79 |
+
"std": [
|
| 80 |
+
0.13736907523153707,
|
| 81 |
+
0.047840646787850445,
|
| 82 |
+
0.03799073317180484,
|
| 83 |
+
0.06560660645733722,
|
| 84 |
+
0.09093375673607122,
|
| 85 |
+
0.292728734025679,
|
| 86 |
+
0.0,
|
| 87 |
+
0.3536876512395259
|
| 88 |
+
],
|
| 89 |
+
"max": [
|
| 90 |
+
0.5862360596656799,
|
| 91 |
+
0.4034728705883026,
|
| 92 |
+
0.3568263053894043,
|
| 93 |
+
1.3517684936523438,
|
| 94 |
+
1.570796251296997,
|
| 95 |
+
3.141204357147217,
|
| 96 |
+
0.0,
|
| 97 |
+
1.1121242046356201
|
| 98 |
+
],
|
| 99 |
+
"min": [
|
| 100 |
+
-0.04167502000927925,
|
| 101 |
+
-0.3563207685947418,
|
| 102 |
+
-0.15537554025650024,
|
| 103 |
+
-3.141592502593994,
|
| 104 |
+
-1.4992541074752808,
|
| 105 |
+
-3.14153790473938,
|
| 106 |
+
0.0,
|
| 107 |
+
0.04637829214334488
|
| 108 |
+
],
|
| 109 |
+
"q01": [
|
| 110 |
+
0.17102622985839844,
|
| 111 |
+
-0.1698118858039379,
|
| 112 |
+
-0.055632163770496845,
|
| 113 |
+
-0.36493172496557236,
|
| 114 |
+
-0.541871190071106,
|
| 115 |
+
-1.354275494813919,
|
| 116 |
+
0.0,
|
| 117 |
+
0.052190229296684265
|
| 118 |
+
],
|
| 119 |
+
"q99": [
|
| 120 |
+
0.45322076976299286,
|
| 121 |
+
0.2354845181107521,
|
| 122 |
+
0.19489620998501778,
|
| 123 |
+
0.3780156075954437,
|
| 124 |
+
0.27568644285202026,
|
| 125 |
+
1.8500566184520721,
|
| 126 |
+
0.0,
|
| 127 |
+
1.0105689764022827
|
| 128 |
+
]
|
| 129 |
+
},
|
| 130 |
+
"num_transitions": 1305714,
|
| 131 |
+
"num_trajectories": 53192
|
| 132 |
+
},
|
| 133 |
+
"oxe_rt1": {
|
| 134 |
+
"action": {
|
| 135 |
+
"mean": [
|
| 136 |
+
0.0017468753503635526,
|
| 137 |
+
0.0015664808452129364,
|
| 138 |
+
-0.00315628363750875,
|
| 139 |
+
0.010832978412508965,
|
| 140 |
+
-0.0014390418073162436,
|
| 141 |
+
0.00022827448265161365,
|
| 142 |
+
0.13385511934757233
|
| 143 |
+
],
|
| 144 |
+
"std": [
|
| 145 |
+
0.034738154754287556,
|
| 146 |
+
0.029950575875189896,
|
| 147 |
+
0.03716987948198757,
|
| 148 |
+
0.08027394458720655,
|
| 149 |
+
0.06586824211259418,
|
| 150 |
+
0.07296740097742303,
|
| 151 |
+
0.33989797865933385
|
| 152 |
+
],
|
| 153 |
+
"max": [
|
| 154 |
+
2.9984593391418457,
|
| 155 |
+
22.09052848815918,
|
| 156 |
+
2.7507524490356445,
|
| 157 |
+
1.570636510848999,
|
| 158 |
+
1.5321086645126343,
|
| 159 |
+
1.5691522359848022,
|
| 160 |
+
1.0
|
| 161 |
+
],
|
| 162 |
+
"min": [
|
| 163 |
+
-2.0204520225524902,
|
| 164 |
+
-5.497899532318115,
|
| 165 |
+
-2.031663417816162,
|
| 166 |
+
-1.569917917251587,
|
| 167 |
+
-1.569892168045044,
|
| 168 |
+
-1.570419430732727,
|
| 169 |
+
0.0
|
| 170 |
+
],
|
| 171 |
+
"q01": [
|
| 172 |
+
-0.22453527510166169,
|
| 173 |
+
-0.14820013284683228,
|
| 174 |
+
-0.231589707583189,
|
| 175 |
+
-0.3517994859814644,
|
| 176 |
+
-0.4193011274933815,
|
| 177 |
+
-0.43643461108207704,
|
| 178 |
+
0.0
|
| 179 |
+
],
|
| 180 |
+
"q99": [
|
| 181 |
+
0.17824687153100965,
|
| 182 |
+
0.14938379630446405,
|
| 183 |
+
0.21842354819178575,
|
| 184 |
+
0.5892666035890578,
|
| 185 |
+
0.35272657424211445,
|
| 186 |
+
0.44796681255102094,
|
| 187 |
+
1.0
|
| 188 |
+
],
|
| 189 |
+
"mask": [
|
| 190 |
+
true,
|
| 191 |
+
true,
|
| 192 |
+
true,
|
| 193 |
+
true,
|
| 194 |
+
true,
|
| 195 |
+
true,
|
| 196 |
+
false
|
| 197 |
+
]
|
| 198 |
+
},
|
| 199 |
+
"state": {
|
| 200 |
+
"mean": [
|
| 201 |
+
0.1399736851453781,
|
| 202 |
+
-0.02083534561097622,
|
| 203 |
+
0.19427375495433807,
|
| 204 |
+
0.10653364658355713,
|
| 205 |
+
-0.06201138719916344,
|
| 206 |
+
0.12378455698490143,
|
| 207 |
+
0.023165317252278328,
|
| 208 |
+
0.052437249571084976
|
| 209 |
+
],
|
| 210 |
+
"std": [
|
| 211 |
+
0.2502841726583427,
|
| 212 |
+
0.06813404469049046,
|
| 213 |
+
0.358260305088155,
|
| 214 |
+
0.2931043604956031,
|
| 215 |
+
0.2779312663188148,
|
| 216 |
+
0.33769767306558707,
|
| 217 |
+
0.09232656626252377,
|
| 218 |
+
0.16498523970985932
|
| 219 |
+
],
|
| 220 |
+
"max": [
|
| 221 |
+
1.0534898042678833,
|
| 222 |
+
0.48018959164619446,
|
| 223 |
+
1.6896663904190063,
|
| 224 |
+
1.0,
|
| 225 |
+
0.9999993443489075,
|
| 226 |
+
0.9999874830245972,
|
| 227 |
+
0.9554369449615479,
|
| 228 |
+
0.9914546012878418
|
| 229 |
+
],
|
| 230 |
+
"min": [
|
| 231 |
+
-0.4436439275741577,
|
| 232 |
+
-0.9970501065254211,
|
| 233 |
+
-0.006579156965017319,
|
| 234 |
+
0.0,
|
| 235 |
+
-0.8643477559089661,
|
| 236 |
+
-0.7079970240592957,
|
| 237 |
+
-0.7688722014427185,
|
| 238 |
+
-0.4999994933605194
|
| 239 |
+
],
|
| 240 |
+
"q01": [
|
| 241 |
+
0.32481380939483645,
|
| 242 |
+
-0.28334290891885755,
|
| 243 |
+
0.14107070609927178,
|
| 244 |
+
0.0,
|
| 245 |
+
-0.686474204659462,
|
| 246 |
+
-0.6808923494815826,
|
| 247 |
+
-0.36045596331357954,
|
| 248 |
+
-0.454380963742733
|
| 249 |
+
],
|
| 250 |
+
"q99": [
|
| 251 |
+
0.8750156319141384,
|
| 252 |
+
0.21247054174542404,
|
| 253 |
+
1.0727112340927123,
|
| 254 |
+
1.0,
|
| 255 |
+
0.9377871316671368,
|
| 256 |
+
0.9563051050901409,
|
| 257 |
+
0.45990042358636823,
|
| 258 |
+
0.7216041100025177
|
| 259 |
+
]
|
| 260 |
+
},
|
| 261 |
+
"num_transitions": 3786152,
|
| 262 |
+
"num_trajectories": 87212
|
| 263 |
+
}
|
| 264 |
+
}
|
settings/florence_robotwin/common_start/config.full.yaml
ADDED
|
@@ -0,0 +1,107 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
run_id: florence-groot-robotwin-xvla3view-official150k-seed42-20260804
|
| 2 |
+
run_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/florence-robotwin-official150k
|
| 3 |
+
seed: 42
|
| 4 |
+
wandb_entity: null
|
| 5 |
+
wandb_project: InfoFlow-StarVLA-RoboTwin
|
| 6 |
+
is_debug: false
|
| 7 |
+
version_id: '0.21'
|
| 8 |
+
framework:
|
| 9 |
+
name: QwenGR00T
|
| 10 |
+
qwenvl:
|
| 11 |
+
base_vlm: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Pretrained_models/Florence-2-large
|
| 12 |
+
attn_implementation: eager
|
| 13 |
+
vl_hidden_dim: 1024
|
| 14 |
+
selected_view_indices:
|
| 15 |
+
- 0
|
| 16 |
+
- 1
|
| 17 |
+
- 2
|
| 18 |
+
compile_vision_tower: true
|
| 19 |
+
action_model:
|
| 20 |
+
action_model_type: DiT-B
|
| 21 |
+
action_hidden_dim: 1024
|
| 22 |
+
hidden_size: 1024
|
| 23 |
+
add_pos_embed: true
|
| 24 |
+
max_seq_len: 1024
|
| 25 |
+
action_dim: 14
|
| 26 |
+
state_dim: 0
|
| 27 |
+
action_horizon: 50
|
| 28 |
+
repeated_diffusion_steps: 8
|
| 29 |
+
noise_beta_alpha: 1.5
|
| 30 |
+
noise_beta_beta: 1.0
|
| 31 |
+
noise_s: 0.999
|
| 32 |
+
num_timestep_buckets: 1000
|
| 33 |
+
num_inference_timesteps: 4
|
| 34 |
+
num_target_vision_tokens: 32
|
| 35 |
+
diffusion_model_cfg:
|
| 36 |
+
cross_attention_dim: 1024
|
| 37 |
+
dropout: 0.2
|
| 38 |
+
final_dropout: true
|
| 39 |
+
interleave_self_attention: true
|
| 40 |
+
norm_type: ada_norm
|
| 41 |
+
num_layers: 16
|
| 42 |
+
output_dim: 1024
|
| 43 |
+
positional_embeddings: null
|
| 44 |
+
future_action_window_size: 49
|
| 45 |
+
past_action_window_size: 0
|
| 46 |
+
dino:
|
| 47 |
+
dino_backbone: dinov2_vits14
|
| 48 |
+
datasets:
|
| 49 |
+
vla_data:
|
| 50 |
+
dataset_py: lerobot_datasets
|
| 51 |
+
data_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Datasets/RoboTwin
|
| 52 |
+
data_mix: robotwin_all_50
|
| 53 |
+
action_type: abs_qpos
|
| 54 |
+
action_mode: abs
|
| 55 |
+
include_state: false
|
| 56 |
+
sequential_step_sampling: false
|
| 57 |
+
per_device_batch_size: 2
|
| 58 |
+
load_all_data_for_training: true
|
| 59 |
+
image_size:
|
| 60 |
+
- 224
|
| 61 |
+
- 224
|
| 62 |
+
video_backend: pyav
|
| 63 |
+
num_workers: 4
|
| 64 |
+
expected_world_size: 2
|
| 65 |
+
trainer:
|
| 66 |
+
use_deepspeed: true
|
| 67 |
+
max_train_steps: 150000
|
| 68 |
+
num_warmup_steps: 5000
|
| 69 |
+
save_interval: null
|
| 70 |
+
save_steps:
|
| 71 |
+
- 10000
|
| 72 |
+
- 11000
|
| 73 |
+
- 12000
|
| 74 |
+
- 13000
|
| 75 |
+
- 14000
|
| 76 |
+
- 15000
|
| 77 |
+
- 16000
|
| 78 |
+
- 17000
|
| 79 |
+
- 18000
|
| 80 |
+
- 19000
|
| 81 |
+
- 20000
|
| 82 |
+
eval_interval: null
|
| 83 |
+
repeated_diffusion_steps: 4
|
| 84 |
+
learning_rate:
|
| 85 |
+
base: 1.0e-05
|
| 86 |
+
qwen_vl_interface: 1.0e-05
|
| 87 |
+
action_model: 0.0001
|
| 88 |
+
lr_scheduler_type: cosine_with_min_lr
|
| 89 |
+
scheduler_specific_kwargs:
|
| 90 |
+
min_lr: 5.0e-07
|
| 91 |
+
freeze_modules: ''
|
| 92 |
+
trainable_modules: ''
|
| 93 |
+
frozen_modules_eval: false
|
| 94 |
+
logging_frequency: 50
|
| 95 |
+
gradient_clipping: 1.0
|
| 96 |
+
gradient_accumulation_steps: 16
|
| 97 |
+
save_format: pt
|
| 98 |
+
save_final_checkpoint: false
|
| 99 |
+
optimizer:
|
| 100 |
+
name: AdamW
|
| 101 |
+
betas:
|
| 102 |
+
- 0.9
|
| 103 |
+
- 0.95
|
| 104 |
+
eps: 1.0e-08
|
| 105 |
+
weight_decay: 1.0e-08
|
| 106 |
+
config_yaml: /home/miaochengyu/starVLA-infoflow-codex/examples/InfoFlow/train_files/server/florence_groot_robotwin_official150k_2gpu.yaml
|
| 107 |
+
output_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/florence-robotwin-official150k/florence-groot-robotwin-xvla3view-official150k-seed42-20260804
|
settings/florence_robotwin/common_start/config.yaml
ADDED
|
@@ -0,0 +1,61 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
datasets:
|
| 2 |
+
vla_data:
|
| 3 |
+
action_mode: abs
|
| 4 |
+
data_mix: robotwin_all_50
|
| 5 |
+
data_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Datasets/RoboTwin
|
| 6 |
+
dataset_py: lerobot_datasets
|
| 7 |
+
num_workers: 4
|
| 8 |
+
per_device_batch_size: 2
|
| 9 |
+
video_backend: pyav
|
| 10 |
+
framework:
|
| 11 |
+
action_model:
|
| 12 |
+
action_dim: 14
|
| 13 |
+
action_horizon: 50
|
| 14 |
+
action_model_type: DiT-B
|
| 15 |
+
add_pos_embed: true
|
| 16 |
+
diffusion_model_cfg:
|
| 17 |
+
cross_attention_dim: 1024
|
| 18 |
+
dropout: 0.2
|
| 19 |
+
final_dropout: true
|
| 20 |
+
interleave_self_attention: true
|
| 21 |
+
norm_type: ada_norm
|
| 22 |
+
num_layers: 16
|
| 23 |
+
output_dim: 1024
|
| 24 |
+
positional_embeddings: null
|
| 25 |
+
hidden_size: 1024
|
| 26 |
+
max_seq_len: 1024
|
| 27 |
+
noise_beta_alpha: 1.5
|
| 28 |
+
noise_beta_beta: 1.0
|
| 29 |
+
num_inference_timesteps: 4
|
| 30 |
+
num_target_vision_tokens: 32
|
| 31 |
+
num_timestep_buckets: 1000
|
| 32 |
+
state_dim: 0
|
| 33 |
+
name: QwenGR00T
|
| 34 |
+
qwenvl:
|
| 35 |
+
base_vlm: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Pretrained_models/Florence-2-large
|
| 36 |
+
compile_vision_tower: true
|
| 37 |
+
selected_view_indices:
|
| 38 |
+
- 0
|
| 39 |
+
- 1
|
| 40 |
+
- 2
|
| 41 |
+
output_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/florence-robotwin-official150k/florence-groot-robotwin-xvla3view-official150k-seed42-20260804
|
| 42 |
+
run_id: florence-groot-robotwin-xvla3view-official150k-seed42-20260804
|
| 43 |
+
run_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/florence-robotwin-official150k
|
| 44 |
+
seed: 42
|
| 45 |
+
trainer:
|
| 46 |
+
freeze_modules: ''
|
| 47 |
+
learning_rate:
|
| 48 |
+
action_model: 0.0001
|
| 49 |
+
base: 1.0e-05
|
| 50 |
+
qwen_vl_interface: 1.0e-05
|
| 51 |
+
lr_scheduler_type: cosine_with_min_lr
|
| 52 |
+
max_train_steps: 150000
|
| 53 |
+
num_warmup_steps: 5000
|
| 54 |
+
optimizer:
|
| 55 |
+
betas:
|
| 56 |
+
- 0.9
|
| 57 |
+
- 0.95
|
| 58 |
+
eps: 1.0e-08
|
| 59 |
+
weight_decay: 1.0e-08
|
| 60 |
+
scheduler_specific_kwargs:
|
| 61 |
+
min_lr: 5.0e-07
|
settings/florence_robotwin/common_start/dataset_statistics.json
ADDED
|
@@ -0,0 +1,218 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"new_embodiment": {
|
| 3 |
+
"action": {
|
| 4 |
+
"mean": [
|
| 5 |
+
-0.22423998869955541,
|
| 6 |
+
1.0729362124204638,
|
| 7 |
+
0.7750796097517013,
|
| 8 |
+
-0.3270166309736669,
|
| 9 |
+
0.061595864617647746,
|
| 10 |
+
-0.06420705380151048,
|
| 11 |
+
0.22430071420967582,
|
| 12 |
+
1.1274866509437562,
|
| 13 |
+
0.8317736595869064,
|
| 14 |
+
-0.3638282522559167,
|
| 15 |
+
-0.024959841337949915,
|
| 16 |
+
0.053022032135631905,
|
| 17 |
+
0.6774904182553296,
|
| 18 |
+
0.659867690205574
|
| 19 |
+
],
|
| 20 |
+
"std": [
|
| 21 |
+
0.39622931737429135,
|
| 22 |
+
1.0035904493395147,
|
| 23 |
+
0.7811611351334437,
|
| 24 |
+
0.6707485714474851,
|
| 25 |
+
0.26869082049839216,
|
| 26 |
+
0.6492019642472783,
|
| 27 |
+
0.3473367940092324,
|
| 28 |
+
1.0176226676795117,
|
| 29 |
+
0.8039915805960643,
|
| 30 |
+
0.7067249045233707,
|
| 31 |
+
0.2625964517715169,
|
| 32 |
+
0.6808759458519373,
|
| 33 |
+
0.448215115111442,
|
| 34 |
+
0.45399216711602836
|
| 35 |
+
],
|
| 36 |
+
"max": [
|
| 37 |
+
6.45659065246582,
|
| 38 |
+
4.179152488708496,
|
| 39 |
+
5.346591472625732,
|
| 40 |
+
1.7942548990249634,
|
| 41 |
+
1.8604620695114136,
|
| 42 |
+
5.43813943862915,
|
| 43 |
+
7.142920017242432,
|
| 44 |
+
4.157068729400635,
|
| 45 |
+
5.672450065612793,
|
| 46 |
+
1.95806884765625,
|
| 47 |
+
1.5663840770721436,
|
| 48 |
+
5.278968811035156,
|
| 49 |
+
1.0,
|
| 50 |
+
1.0
|
| 51 |
+
],
|
| 52 |
+
"min": [
|
| 53 |
+
-7.552278995513916,
|
| 54 |
+
-0.36354875564575195,
|
| 55 |
+
-0.18577136099338531,
|
| 56 |
+
-1.956291913986206,
|
| 57 |
+
-1.6801013946533203,
|
| 58 |
+
-7.678869724273682,
|
| 59 |
+
-8.539926528930664,
|
| 60 |
+
-0.6294453740119934,
|
| 61 |
+
-0.07775841653347015,
|
| 62 |
+
-2.1328067779541016,
|
| 63 |
+
-2.1285502910614014,
|
| 64 |
+
-8.228925704956055,
|
| 65 |
+
0.0,
|
| 66 |
+
0.0
|
| 67 |
+
],
|
| 68 |
+
"q01": [
|
| 69 |
+
-7.334665780067444,
|
| 70 |
+
-5.257390398583084e-07,
|
| 71 |
+
-2.8215323254698887e-05,
|
| 72 |
+
-1.841589093208313,
|
| 73 |
+
-1.2851550579071045,
|
| 74 |
+
-6.268352918624878,
|
| 75 |
+
-5.995517673492431,
|
| 76 |
+
-0.572821855545044,
|
| 77 |
+
-2.81171942333458e-05,
|
| 78 |
+
-1.8467556238174438,
|
| 79 |
+
-1.4545854091644288,
|
| 80 |
+
-5.945669202804565,
|
| 81 |
+
0.0,
|
| 82 |
+
0.0
|
| 83 |
+
],
|
| 84 |
+
"q99": [
|
| 85 |
+
5.016749763488781,
|
| 86 |
+
3.4792869091033936,
|
| 87 |
+
4.225283479690551,
|
| 88 |
+
1.7064905214309687,
|
| 89 |
+
1.4829185009002717,
|
| 90 |
+
3.7177388882636553,
|
| 91 |
+
1.258314609527588,
|
| 92 |
+
2.967935800552368,
|
| 93 |
+
3.2566452693939207,
|
| 94 |
+
1.766038613319397,
|
| 95 |
+
1.4066662788391113,
|
| 96 |
+
3.5830023288726807,
|
| 97 |
+
1.0,
|
| 98 |
+
1.0
|
| 99 |
+
],
|
| 100 |
+
"mask": [
|
| 101 |
+
true,
|
| 102 |
+
true,
|
| 103 |
+
true,
|
| 104 |
+
true,
|
| 105 |
+
true,
|
| 106 |
+
true,
|
| 107 |
+
true,
|
| 108 |
+
true,
|
| 109 |
+
true,
|
| 110 |
+
true,
|
| 111 |
+
true,
|
| 112 |
+
true,
|
| 113 |
+
false,
|
| 114 |
+
false
|
| 115 |
+
]
|
| 116 |
+
},
|
| 117 |
+
"state": {
|
| 118 |
+
"mean": [
|
| 119 |
+
-0.22281882727518681,
|
| 120 |
+
1.0668349397182468,
|
| 121 |
+
0.7706326398253446,
|
| 122 |
+
-0.32523305794224144,
|
| 123 |
+
0.061181390470028435,
|
| 124 |
+
-0.06316296239456282,
|
| 125 |
+
0.2225376246869564,
|
| 126 |
+
1.1201750242710118,
|
| 127 |
+
0.8263923591375355,
|
| 128 |
+
-0.3615948969544843,
|
| 129 |
+
-0.024702461528722783,
|
| 130 |
+
0.05203246331657284,
|
| 131 |
+
0.6793262696266178,
|
| 132 |
+
0.6616973677277567
|
| 133 |
+
],
|
| 134 |
+
"std": [
|
| 135 |
+
0.3952287088673167,
|
| 136 |
+
1.004018577033247,
|
| 137 |
+
0.7809103811570286,
|
| 138 |
+
0.6688088363177989,
|
| 139 |
+
0.2677572381895678,
|
| 140 |
+
0.6463212583712936,
|
| 141 |
+
0.3464848832136367,
|
| 142 |
+
1.018232630032839,
|
| 143 |
+
0.803879839768273,
|
| 144 |
+
0.7045612787595747,
|
| 145 |
+
0.2614341421313462,
|
| 146 |
+
0.6774619870452645,
|
| 147 |
+
0.4474922049702762,
|
| 148 |
+
0.45333916022037923
|
| 149 |
+
],
|
| 150 |
+
"max": [
|
| 151 |
+
6.45659065246582,
|
| 152 |
+
4.179152488708496,
|
| 153 |
+
5.346591472625732,
|
| 154 |
+
1.7942548990249634,
|
| 155 |
+
1.8604620695114136,
|
| 156 |
+
5.43813943862915,
|
| 157 |
+
7.142920017242432,
|
| 158 |
+
4.157068729400635,
|
| 159 |
+
5.672450065612793,
|
| 160 |
+
1.95806884765625,
|
| 161 |
+
1.5663840770721436,
|
| 162 |
+
5.278968811035156,
|
| 163 |
+
1.0,
|
| 164 |
+
1.0
|
| 165 |
+
],
|
| 166 |
+
"min": [
|
| 167 |
+
-7.552278995513916,
|
| 168 |
+
-0.36354875564575195,
|
| 169 |
+
-0.18577136099338531,
|
| 170 |
+
-1.956291913986206,
|
| 171 |
+
-1.6801013946533203,
|
| 172 |
+
-7.678869724273682,
|
| 173 |
+
-8.539926528930664,
|
| 174 |
+
-0.6294453740119934,
|
| 175 |
+
-0.07775841653347015,
|
| 176 |
+
-2.1328067779541016,
|
| 177 |
+
-2.1285502910614014,
|
| 178 |
+
-8.228925704956055,
|
| 179 |
+
0.0,
|
| 180 |
+
0.0
|
| 181 |
+
],
|
| 182 |
+
"q01": [
|
| 183 |
+
-7.334665780067444,
|
| 184 |
+
-5.257390398583084e-07,
|
| 185 |
+
-2.8215323254698887e-05,
|
| 186 |
+
-1.841589093208313,
|
| 187 |
+
-1.2851550579071045,
|
| 188 |
+
-6.268109321594238,
|
| 189 |
+
-5.912737417221069,
|
| 190 |
+
-0.572821855545044,
|
| 191 |
+
-2.81171942333458e-05,
|
| 192 |
+
-1.8467556238174438,
|
| 193 |
+
-1.440324923992157,
|
| 194 |
+
-5.8623305034637445,
|
| 195 |
+
0.0,
|
| 196 |
+
0.0
|
| 197 |
+
],
|
| 198 |
+
"q99": [
|
| 199 |
+
4.902149534225477,
|
| 200 |
+
3.4775466918945312,
|
| 201 |
+
4.224453401565551,
|
| 202 |
+
1.7064905214309687,
|
| 203 |
+
1.480757713317871,
|
| 204 |
+
3.609884884357428,
|
| 205 |
+
1.258314609527588,
|
| 206 |
+
2.967935800552368,
|
| 207 |
+
3.2566452693939207,
|
| 208 |
+
1.7433889162540437,
|
| 209 |
+
1.4066662788391113,
|
| 210 |
+
3.581243851184845,
|
| 211 |
+
1.0,
|
| 212 |
+
1.0
|
| 213 |
+
]
|
| 214 |
+
},
|
| 215 |
+
"num_transitions": 6075103,
|
| 216 |
+
"num_trajectories": 27500
|
| 217 |
+
}
|
| 218 |
+
}
|
settings/florence_robotwin/common_start/initialization_manifest.json
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"schema_version": 1,
|
| 3 |
+
"initialization_seed": 42,
|
| 4 |
+
"action_model_sha256": "915f157d03a58d059818a774e9150aa2028dd054752b513a5b4f3c85313bc383",
|
| 5 |
+
"action_model_parameter_elements": 141815566
|
| 6 |
+
}
|
settings/florence_robotwin/uniform/config.full.yaml
ADDED
|
@@ -0,0 +1,113 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
run_id: florence-robotwin-uniform-resume10k-12k-20260809
|
| 2 |
+
run_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/florence-robotwin-uniform-resume10k-12k-20260809
|
| 3 |
+
seed: 484370
|
| 4 |
+
wandb_entity: null
|
| 5 |
+
wandb_project: InfoFlow-StarVLA-RoboTwin
|
| 6 |
+
is_debug: false
|
| 7 |
+
version_id: '0.21'
|
| 8 |
+
framework:
|
| 9 |
+
name: QwenGR00T
|
| 10 |
+
qwenvl:
|
| 11 |
+
base_vlm: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Pretrained_models/Florence-2-large
|
| 12 |
+
attn_implementation: eager
|
| 13 |
+
vl_hidden_dim: 1024
|
| 14 |
+
selected_view_indices:
|
| 15 |
+
- 0
|
| 16 |
+
- 1
|
| 17 |
+
- 2
|
| 18 |
+
compile_vision_tower: true
|
| 19 |
+
action_model:
|
| 20 |
+
action_model_type: DiT-B
|
| 21 |
+
action_hidden_dim: 1024
|
| 22 |
+
hidden_size: 1024
|
| 23 |
+
add_pos_embed: true
|
| 24 |
+
max_seq_len: 1024
|
| 25 |
+
action_dim: 14
|
| 26 |
+
state_dim: 0
|
| 27 |
+
action_horizon: 50
|
| 28 |
+
repeated_diffusion_steps: 8
|
| 29 |
+
noise_beta_alpha: 1.5
|
| 30 |
+
noise_beta_beta: 1.0
|
| 31 |
+
noise_s: 0.999
|
| 32 |
+
num_timestep_buckets: 1000
|
| 33 |
+
num_inference_timesteps: 4
|
| 34 |
+
num_target_vision_tokens: 32
|
| 35 |
+
diffusion_model_cfg:
|
| 36 |
+
cross_attention_dim: 1024
|
| 37 |
+
dropout: 0.2
|
| 38 |
+
final_dropout: true
|
| 39 |
+
interleave_self_attention: true
|
| 40 |
+
norm_type: ada_norm
|
| 41 |
+
num_layers: 16
|
| 42 |
+
output_dim: 1024
|
| 43 |
+
positional_embeddings: null
|
| 44 |
+
future_action_window_size: 49
|
| 45 |
+
past_action_window_size: 0
|
| 46 |
+
dino:
|
| 47 |
+
dino_backbone: dinov2_vits14
|
| 48 |
+
datasets:
|
| 49 |
+
vla_data:
|
| 50 |
+
dataset_py: lerobot_datasets
|
| 51 |
+
data_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Datasets/RoboTwin
|
| 52 |
+
data_mix: robotwin_all_50
|
| 53 |
+
action_type: abs_qpos
|
| 54 |
+
action_mode: abs
|
| 55 |
+
include_state: false
|
| 56 |
+
sequential_step_sampling: false
|
| 57 |
+
per_device_batch_size: 2
|
| 58 |
+
load_all_data_for_training: true
|
| 59 |
+
image_size:
|
| 60 |
+
- 224
|
| 61 |
+
- 224
|
| 62 |
+
video_backend: pyav
|
| 63 |
+
num_workers: 4
|
| 64 |
+
expected_world_size: 2
|
| 65 |
+
traceable_sampling: true
|
| 66 |
+
dataset_provenance_path: /home/miaochengyu/starVLA-infoflow-codex/protocols/robotwin/data_provenance.json
|
| 67 |
+
dataset_content_fingerprint_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/robotwin/content_fingerprints.json
|
| 68 |
+
valid_anchor_manifest_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/robotwin/valid_anchor_manifest.json
|
| 69 |
+
temporal_proposal_path: null
|
| 70 |
+
transform_rng_seed: 289726
|
| 71 |
+
trainer:
|
| 72 |
+
use_deepspeed: true
|
| 73 |
+
max_train_steps: 150000
|
| 74 |
+
num_warmup_steps: 5000
|
| 75 |
+
save_interval: null
|
| 76 |
+
save_steps:
|
| 77 |
+
- 10500
|
| 78 |
+
- 11000
|
| 79 |
+
- 11500
|
| 80 |
+
- 12000
|
| 81 |
+
eval_interval: null
|
| 82 |
+
repeated_diffusion_steps: 4
|
| 83 |
+
learning_rate:
|
| 84 |
+
base: 1.0e-05
|
| 85 |
+
qwen_vl_interface: 1.0e-05
|
| 86 |
+
action_model: 0.0001
|
| 87 |
+
lr_scheduler_type: cosine_with_min_lr
|
| 88 |
+
scheduler_specific_kwargs:
|
| 89 |
+
min_lr: 5.0e-07
|
| 90 |
+
freeze_modules: ''
|
| 91 |
+
trainable_modules: ''
|
| 92 |
+
frozen_modules_eval: false
|
| 93 |
+
logging_frequency: 50
|
| 94 |
+
gradient_clipping: 1.0
|
| 95 |
+
gradient_accumulation_steps: 16
|
| 96 |
+
save_format: pt
|
| 97 |
+
save_final_checkpoint: false
|
| 98 |
+
optimizer:
|
| 99 |
+
name: AdamW
|
| 100 |
+
betas:
|
| 101 |
+
- 0.9
|
| 102 |
+
- 0.95
|
| 103 |
+
eps: 1.0e-08
|
| 104 |
+
weight_decay: 1.0e-08
|
| 105 |
+
stop_after_steps: 12000
|
| 106 |
+
is_resume: true
|
| 107 |
+
allow_model_only_resume: false
|
| 108 |
+
resume_accelerate_checkpoint: /mnt/miaochengyu/starvla-infoflow-runtime/runs/florence-robotwin-official150k/florence-groot-robotwin-xvla3view-official150k-seed42-20260804/checkpoints/steps_10000_accelerate
|
| 109 |
+
branch_parent_seed: 42
|
| 110 |
+
branch_rng_seed: 484370
|
| 111 |
+
model_rng_seed: 721927
|
| 112 |
+
config_yaml: /home/miaochengyu/starVLA-infoflow-codex/examples/InfoFlow/train_files/server/florence_groot_robotwin_official150k_traceable_10k.yaml
|
| 113 |
+
output_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/florence-robotwin-uniform-resume10k-12k-20260809/florence-robotwin-uniform-resume10k-12k-20260809
|
settings/florence_robotwin/uniform/config.yaml
ADDED
|
@@ -0,0 +1,68 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
datasets:
|
| 2 |
+
vla_data:
|
| 3 |
+
action_mode: abs
|
| 4 |
+
data_mix: robotwin_all_50
|
| 5 |
+
data_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Datasets/RoboTwin
|
| 6 |
+
dataset_content_fingerprint_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/robotwin/content_fingerprints.json
|
| 7 |
+
dataset_provenance_path: /home/miaochengyu/starVLA-infoflow-codex/protocols/robotwin/data_provenance.json
|
| 8 |
+
dataset_py: lerobot_datasets
|
| 9 |
+
expected_world_size: 2
|
| 10 |
+
num_workers: 4
|
| 11 |
+
per_device_batch_size: 2
|
| 12 |
+
temporal_proposal_path: null
|
| 13 |
+
traceable_sampling: true
|
| 14 |
+
valid_anchor_manifest_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/robotwin/valid_anchor_manifest.json
|
| 15 |
+
video_backend: pyav
|
| 16 |
+
framework:
|
| 17 |
+
action_model:
|
| 18 |
+
action_dim: 14
|
| 19 |
+
action_horizon: 50
|
| 20 |
+
action_model_type: DiT-B
|
| 21 |
+
add_pos_embed: true
|
| 22 |
+
diffusion_model_cfg:
|
| 23 |
+
cross_attention_dim: 1024
|
| 24 |
+
dropout: 0.2
|
| 25 |
+
final_dropout: true
|
| 26 |
+
interleave_self_attention: true
|
| 27 |
+
norm_type: ada_norm
|
| 28 |
+
num_layers: 16
|
| 29 |
+
output_dim: 1024
|
| 30 |
+
positional_embeddings: null
|
| 31 |
+
hidden_size: 1024
|
| 32 |
+
max_seq_len: 1024
|
| 33 |
+
noise_beta_alpha: 1.5
|
| 34 |
+
noise_beta_beta: 1.0
|
| 35 |
+
num_inference_timesteps: 4
|
| 36 |
+
num_target_vision_tokens: 32
|
| 37 |
+
num_timestep_buckets: 1000
|
| 38 |
+
state_dim: 0
|
| 39 |
+
name: QwenGR00T
|
| 40 |
+
qwenvl:
|
| 41 |
+
base_vlm: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Pretrained_models/Florence-2-large
|
| 42 |
+
compile_vision_tower: true
|
| 43 |
+
selected_view_indices:
|
| 44 |
+
- 0
|
| 45 |
+
- 1
|
| 46 |
+
- 2
|
| 47 |
+
output_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/florence-robotwin-uniform-resume10k-12k-20260809/florence-robotwin-uniform-resume10k-12k-20260809
|
| 48 |
+
run_id: florence-robotwin-uniform-resume10k-12k-20260809
|
| 49 |
+
run_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/florence-robotwin-uniform-resume10k-12k-20260809
|
| 50 |
+
seed: 484370
|
| 51 |
+
trainer:
|
| 52 |
+
freeze_modules: ''
|
| 53 |
+
gradient_accumulation_steps: 16
|
| 54 |
+
learning_rate:
|
| 55 |
+
action_model: 0.0001
|
| 56 |
+
base: 1.0e-05
|
| 57 |
+
qwen_vl_interface: 1.0e-05
|
| 58 |
+
lr_scheduler_type: cosine_with_min_lr
|
| 59 |
+
max_train_steps: 150000
|
| 60 |
+
num_warmup_steps: 5000
|
| 61 |
+
optimizer:
|
| 62 |
+
betas:
|
| 63 |
+
- 0.9
|
| 64 |
+
- 0.95
|
| 65 |
+
eps: 1.0e-08
|
| 66 |
+
weight_decay: 1.0e-08
|
| 67 |
+
scheduler_specific_kwargs:
|
| 68 |
+
min_lr: 5.0e-07
|
settings/florence_robotwin/uniform/dataset_statistics.json
ADDED
|
@@ -0,0 +1,218 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"new_embodiment": {
|
| 3 |
+
"action": {
|
| 4 |
+
"mean": [
|
| 5 |
+
-0.22423998869955536,
|
| 6 |
+
1.0729362124204633,
|
| 7 |
+
0.775079609751701,
|
| 8 |
+
-0.32701663097366684,
|
| 9 |
+
0.06159586461764771,
|
| 10 |
+
-0.06420705380151046,
|
| 11 |
+
0.22430071420967568,
|
| 12 |
+
1.1274866509437556,
|
| 13 |
+
0.831773659586906,
|
| 14 |
+
-0.3638282522559164,
|
| 15 |
+
-0.024959841337949897,
|
| 16 |
+
0.05302203213563182,
|
| 17 |
+
0.6774904182553295,
|
| 18 |
+
0.6598676902055738
|
| 19 |
+
],
|
| 20 |
+
"std": [
|
| 21 |
+
0.39622931737429123,
|
| 22 |
+
1.003590449339515,
|
| 23 |
+
0.7811611351334437,
|
| 24 |
+
0.6707485714474849,
|
| 25 |
+
0.2686908204983921,
|
| 26 |
+
0.6492019642472782,
|
| 27 |
+
0.34733679400923234,
|
| 28 |
+
1.0176226676795115,
|
| 29 |
+
0.8039915805960646,
|
| 30 |
+
0.7067249045233707,
|
| 31 |
+
0.26259645177151686,
|
| 32 |
+
0.6808759458519371,
|
| 33 |
+
0.4482151151114418,
|
| 34 |
+
0.4539921671160285
|
| 35 |
+
],
|
| 36 |
+
"max": [
|
| 37 |
+
6.45659065246582,
|
| 38 |
+
4.179152488708496,
|
| 39 |
+
5.346591472625732,
|
| 40 |
+
1.7942548990249634,
|
| 41 |
+
1.8604620695114136,
|
| 42 |
+
5.43813943862915,
|
| 43 |
+
7.142920017242432,
|
| 44 |
+
4.157068729400635,
|
| 45 |
+
5.672450065612793,
|
| 46 |
+
1.95806884765625,
|
| 47 |
+
1.5663840770721436,
|
| 48 |
+
5.278968811035156,
|
| 49 |
+
1.0,
|
| 50 |
+
1.0
|
| 51 |
+
],
|
| 52 |
+
"min": [
|
| 53 |
+
-7.552278995513916,
|
| 54 |
+
-0.36354875564575195,
|
| 55 |
+
-0.18577136099338531,
|
| 56 |
+
-1.956291913986206,
|
| 57 |
+
-1.6801013946533203,
|
| 58 |
+
-7.678869724273682,
|
| 59 |
+
-8.539926528930664,
|
| 60 |
+
-0.6294453740119934,
|
| 61 |
+
-0.07775841653347015,
|
| 62 |
+
-2.1328067779541016,
|
| 63 |
+
-2.1285502910614014,
|
| 64 |
+
-8.228925704956055,
|
| 65 |
+
0.0,
|
| 66 |
+
0.0
|
| 67 |
+
],
|
| 68 |
+
"q01": [
|
| 69 |
+
-7.334665780067444,
|
| 70 |
+
-5.257390398583084e-07,
|
| 71 |
+
-2.8215323254698887e-05,
|
| 72 |
+
-1.841589093208313,
|
| 73 |
+
-1.2851550579071045,
|
| 74 |
+
-6.268352918624878,
|
| 75 |
+
-5.995517673492431,
|
| 76 |
+
-0.572821855545044,
|
| 77 |
+
-2.81171942333458e-05,
|
| 78 |
+
-1.8467556238174438,
|
| 79 |
+
-1.4545854091644288,
|
| 80 |
+
-5.945669202804565,
|
| 81 |
+
0.0,
|
| 82 |
+
0.0
|
| 83 |
+
],
|
| 84 |
+
"q99": [
|
| 85 |
+
5.016749763488781,
|
| 86 |
+
3.4792869091033936,
|
| 87 |
+
4.225283479690551,
|
| 88 |
+
1.7064905214309687,
|
| 89 |
+
1.4829185009002717,
|
| 90 |
+
3.7177388882636553,
|
| 91 |
+
1.258314609527588,
|
| 92 |
+
2.967935800552368,
|
| 93 |
+
3.2566452693939207,
|
| 94 |
+
1.766038613319397,
|
| 95 |
+
1.4066662788391113,
|
| 96 |
+
3.5830023288726807,
|
| 97 |
+
1.0,
|
| 98 |
+
1.0
|
| 99 |
+
],
|
| 100 |
+
"mask": [
|
| 101 |
+
true,
|
| 102 |
+
true,
|
| 103 |
+
true,
|
| 104 |
+
true,
|
| 105 |
+
true,
|
| 106 |
+
true,
|
| 107 |
+
true,
|
| 108 |
+
true,
|
| 109 |
+
true,
|
| 110 |
+
true,
|
| 111 |
+
true,
|
| 112 |
+
true,
|
| 113 |
+
false,
|
| 114 |
+
false
|
| 115 |
+
]
|
| 116 |
+
},
|
| 117 |
+
"state": {
|
| 118 |
+
"mean": [
|
| 119 |
+
-0.22281882727518673,
|
| 120 |
+
1.0668349397182466,
|
| 121 |
+
0.770632639825344,
|
| 122 |
+
-0.3252330579422412,
|
| 123 |
+
0.06118139047002841,
|
| 124 |
+
-0.06316296239456277,
|
| 125 |
+
0.22253762468695631,
|
| 126 |
+
1.1201750242710113,
|
| 127 |
+
0.8263923591375351,
|
| 128 |
+
-0.36159489695448405,
|
| 129 |
+
-0.024702461528722772,
|
| 130 |
+
0.05203246331657278,
|
| 131 |
+
0.6793262696266176,
|
| 132 |
+
0.6616973677277563
|
| 133 |
+
],
|
| 134 |
+
"std": [
|
| 135 |
+
0.39522870886731665,
|
| 136 |
+
1.004018577033247,
|
| 137 |
+
0.7809103811570285,
|
| 138 |
+
0.6688088363177986,
|
| 139 |
+
0.26775723818956776,
|
| 140 |
+
0.6463212583712935,
|
| 141 |
+
0.34648488321363663,
|
| 142 |
+
1.018232630032839,
|
| 143 |
+
0.8038798397682734,
|
| 144 |
+
0.7045612787595747,
|
| 145 |
+
0.26143414213134614,
|
| 146 |
+
0.6774619870452644,
|
| 147 |
+
0.44749220497027625,
|
| 148 |
+
0.45333916022037934
|
| 149 |
+
],
|
| 150 |
+
"max": [
|
| 151 |
+
6.45659065246582,
|
| 152 |
+
4.179152488708496,
|
| 153 |
+
5.346591472625732,
|
| 154 |
+
1.7942548990249634,
|
| 155 |
+
1.8604620695114136,
|
| 156 |
+
5.43813943862915,
|
| 157 |
+
7.142920017242432,
|
| 158 |
+
4.157068729400635,
|
| 159 |
+
5.672450065612793,
|
| 160 |
+
1.95806884765625,
|
| 161 |
+
1.5663840770721436,
|
| 162 |
+
5.278968811035156,
|
| 163 |
+
1.0,
|
| 164 |
+
1.0
|
| 165 |
+
],
|
| 166 |
+
"min": [
|
| 167 |
+
-7.552278995513916,
|
| 168 |
+
-0.36354875564575195,
|
| 169 |
+
-0.18577136099338531,
|
| 170 |
+
-1.956291913986206,
|
| 171 |
+
-1.6801013946533203,
|
| 172 |
+
-7.678869724273682,
|
| 173 |
+
-8.539926528930664,
|
| 174 |
+
-0.6294453740119934,
|
| 175 |
+
-0.07775841653347015,
|
| 176 |
+
-2.1328067779541016,
|
| 177 |
+
-2.1285502910614014,
|
| 178 |
+
-8.228925704956055,
|
| 179 |
+
0.0,
|
| 180 |
+
0.0
|
| 181 |
+
],
|
| 182 |
+
"q01": [
|
| 183 |
+
-7.334665780067444,
|
| 184 |
+
-5.257390398583084e-07,
|
| 185 |
+
-2.8215323254698887e-05,
|
| 186 |
+
-1.841589093208313,
|
| 187 |
+
-1.2851550579071045,
|
| 188 |
+
-6.268109321594238,
|
| 189 |
+
-5.912737417221069,
|
| 190 |
+
-0.572821855545044,
|
| 191 |
+
-2.81171942333458e-05,
|
| 192 |
+
-1.8467556238174438,
|
| 193 |
+
-1.440324923992157,
|
| 194 |
+
-5.8623305034637445,
|
| 195 |
+
0.0,
|
| 196 |
+
0.0
|
| 197 |
+
],
|
| 198 |
+
"q99": [
|
| 199 |
+
4.902149534225477,
|
| 200 |
+
3.4775466918945312,
|
| 201 |
+
4.224453401565551,
|
| 202 |
+
1.7064905214309687,
|
| 203 |
+
1.480757713317871,
|
| 204 |
+
3.609884884357428,
|
| 205 |
+
1.258314609527588,
|
| 206 |
+
2.967935800552368,
|
| 207 |
+
3.2566452693939207,
|
| 208 |
+
1.7433889162540437,
|
| 209 |
+
1.4066662788391113,
|
| 210 |
+
3.581243851184845,
|
| 211 |
+
1.0,
|
| 212 |
+
1.0
|
| 213 |
+
]
|
| 214 |
+
},
|
| 215 |
+
"num_transitions": 6075103,
|
| 216 |
+
"num_trajectories": 27500
|
| 217 |
+
}
|
| 218 |
+
}
|
settings/florence_robotwin/uniform/external_resume_certificate.json
ADDED
|
@@ -0,0 +1,138 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"branch_resume_identity": {
|
| 3 |
+
"config_sha256": "d40666777c1a9b5df6f5f6934614c0fe1b0d9af5d0c24e10d90c1fe4103e4c0f",
|
| 4 |
+
"identity_schema_version": 2,
|
| 5 |
+
"manifest_sha256": "dcb728a00da4b4ad7f989fa313d6d3355e13f57efefc7267e9ea092f2ffd8a50",
|
| 6 |
+
"model_structure_sha256": "2d1c8fd1dfa358c4975ad54934f42c24c8d10f6fcef9e8f7c2585b349b32d96b",
|
| 7 |
+
"proposal_alpha": null,
|
| 8 |
+
"proposal_sha256": null,
|
| 9 |
+
"seed": 484370,
|
| 10 |
+
"total_batch_size": 64
|
| 11 |
+
},
|
| 12 |
+
"completed_steps": 10000,
|
| 13 |
+
"optimizer_param_groups": [
|
| 14 |
+
{
|
| 15 |
+
"betas": [
|
| 16 |
+
0.9,
|
| 17 |
+
0.95
|
| 18 |
+
],
|
| 19 |
+
"eps": 1e-08,
|
| 20 |
+
"initial_lr": 1e-05,
|
| 21 |
+
"lr": 9.97215529648321e-06,
|
| 22 |
+
"name": "qwen_vl_interface",
|
| 23 |
+
"weight_decay": 1e-08
|
| 24 |
+
},
|
| 25 |
+
{
|
| 26 |
+
"betas": [
|
| 27 |
+
0.9,
|
| 28 |
+
0.95
|
| 29 |
+
],
|
| 30 |
+
"eps": 1e-08,
|
| 31 |
+
"initial_lr": 0.0001,
|
| 32 |
+
"lr": 9.972155296483209e-05,
|
| 33 |
+
"name": "action_model",
|
| 34 |
+
"weight_decay": 1e-08
|
| 35 |
+
}
|
| 36 |
+
],
|
| 37 |
+
"optimizer_state_entries": 4,
|
| 38 |
+
"parent_checkpoint": "/mnt/miaochengyu/starvla-infoflow-runtime/runs/florence-robotwin-official150k/florence-groot-robotwin-xvla3view-official150k-seed42-20260804/checkpoints/steps_10000_accelerate",
|
| 39 |
+
"parent_checkpoint_commit_sha256": "f6f6408ead21ba9cb0d7bbfa26c74733950a504264a26e5b795129ffadff6958",
|
| 40 |
+
"parent_checkpoint_files": [
|
| 41 |
+
{
|
| 42 |
+
"bytes": 1525,
|
| 43 |
+
"name": "custom_checkpoint_0.pkl",
|
| 44 |
+
"sha256": "031a5046639dfc967eebbf30b7184dbe2619e26675a74569827aa5edaf6cc85c"
|
| 45 |
+
},
|
| 46 |
+
{
|
| 47 |
+
"bytes": 13,
|
| 48 |
+
"name": "latest",
|
| 49 |
+
"sha256": "68d0ca762047fd48fdd6a06a5e6465209289c028bfd63f5557a110ff1ebc15ac"
|
| 50 |
+
},
|
| 51 |
+
{
|
| 52 |
+
"bytes": 5509764721,
|
| 53 |
+
"name": "pytorch_model/bf16_zero_pp_rank_0_mp_rank_00_optim_states.pt",
|
| 54 |
+
"sha256": "27d9b011b75f40c3d896c9681ee883eeb3d2277f201889faebf0b7fcf5492c52"
|
| 55 |
+
},
|
| 56 |
+
{
|
| 57 |
+
"bytes": 5509778801,
|
| 58 |
+
"name": "pytorch_model/bf16_zero_pp_rank_1_mp_rank_00_optim_states.pt",
|
| 59 |
+
"sha256": "f657f6f6fb48956b2b406b79511e16b9c2712dc95216b78ba5179818b821d18f"
|
| 60 |
+
},
|
| 61 |
+
{
|
| 62 |
+
"bytes": 1837527613,
|
| 63 |
+
"name": "pytorch_model/mp_rank_00_model_states.pt",
|
| 64 |
+
"sha256": "3d66f8b6c0e344c34f06d3b4708c3292cbb8a84ea5a25e650fb64b679a44eb3b"
|
| 65 |
+
},
|
| 66 |
+
{
|
| 67 |
+
"bytes": 2152621885,
|
| 68 |
+
"name": "pytorch_model.pt",
|
| 69 |
+
"sha256": "4b6c4f47df7b0f05b8c9287e480d88f2b5e711b4fb6cccbdfffd677ed9bff778"
|
| 70 |
+
},
|
| 71 |
+
{
|
| 72 |
+
"bytes": 15017,
|
| 73 |
+
"name": "random_states_0.pkl",
|
| 74 |
+
"sha256": "629d14d87e3cab680e32acec92c80c2578a7fbd1ebc0dee5475e774058f11d41"
|
| 75 |
+
},
|
| 76 |
+
{
|
| 77 |
+
"bytes": 15081,
|
| 78 |
+
"name": "random_states_1.pkl",
|
| 79 |
+
"sha256": "3cb98283ee7eaaedcee320e8e3b17e74f2f01a22d9e3b958fa2422a86215c652"
|
| 80 |
+
},
|
| 81 |
+
{
|
| 82 |
+
"bytes": 33270,
|
| 83 |
+
"name": "zero_to_fp32.py",
|
| 84 |
+
"sha256": "e8a2553b434ac043ec63b022db1e768277acf8d4e70f10a2842db6d238977135"
|
| 85 |
+
}
|
| 86 |
+
],
|
| 87 |
+
"parent_checkpoint_files_sha256": "1f8985d0633f16ac3b0381b128cdb4e479c4fdfd67ff15d58736633e4b942bc0",
|
| 88 |
+
"parent_checkpoint_integrity_manifest_sha256": null,
|
| 89 |
+
"parent_config_sha256": "d35ab9e43634c7c4a74c5299657df3c17c7d78407dd0fe3d37296a927e353d28",
|
| 90 |
+
"parent_resume_identity": {
|
| 91 |
+
"config_sha256": "2440c6327cf5e8c14efc93b800209171c725dae0898181921c62f3490eb8378f",
|
| 92 |
+
"identity_schema_version": 2,
|
| 93 |
+
"manifest_sha256": null,
|
| 94 |
+
"model_structure_sha256": "2d1c8fd1dfa358c4975ad54934f42c24c8d10f6fcef9e8f7c2585b349b32d96b",
|
| 95 |
+
"proposal_alpha": null,
|
| 96 |
+
"proposal_sha256": null,
|
| 97 |
+
"seed": 42,
|
| 98 |
+
"total_batch_size": 64
|
| 99 |
+
},
|
| 100 |
+
"proposal_sha256": null,
|
| 101 |
+
"rank_state_summaries": [
|
| 102 |
+
{
|
| 103 |
+
"optimizer_state_entries": 2,
|
| 104 |
+
"optimizer_state_sha256": "5a35add2c00249e41e72e1ab16261d87f205bfeaf23386c7263223e4ce168368",
|
| 105 |
+
"rank": 0,
|
| 106 |
+
"trainable_state_sha256": "667c6a6a7aa0dc49c32f7bef693f3272efee46064fda5d370d25788291da76a0"
|
| 107 |
+
},
|
| 108 |
+
{
|
| 109 |
+
"optimizer_state_entries": 2,
|
| 110 |
+
"optimizer_state_sha256": "073f849cd1942522dbfde195bdc29e8ee015e4c5d4be3070381027a13b9784cf",
|
| 111 |
+
"rank": 1,
|
| 112 |
+
"trainable_state_sha256": "667c6a6a7aa0dc49c32f7bef693f3272efee46064fda5d370d25788291da76a0"
|
| 113 |
+
}
|
| 114 |
+
],
|
| 115 |
+
"repeated_diffusion_steps": 4,
|
| 116 |
+
"rng_streams": {
|
| 117 |
+
"model_base": 721927,
|
| 118 |
+
"sampler": 484370,
|
| 119 |
+
"transform": 289726
|
| 120 |
+
},
|
| 121 |
+
"scheduler": {
|
| 122 |
+
"class": "LambdaLR",
|
| 123 |
+
"last_epoch": 10000,
|
| 124 |
+
"last_lr": [
|
| 125 |
+
9.97215529648321e-06,
|
| 126 |
+
9.972155296483209e-05
|
| 127 |
+
],
|
| 128 |
+
"step_count": 10001
|
| 129 |
+
},
|
| 130 |
+
"scheduler_resume_mode": "accelerate_checkpoint",
|
| 131 |
+
"schema_version": 1,
|
| 132 |
+
"status": "exact_external_resume_loaded",
|
| 133 |
+
"total_batch_size": 64,
|
| 134 |
+
"trainable_parameter_elements": 918283022,
|
| 135 |
+
"trainable_parameter_sha256": "2e8f3de6aeee46bca56fab5993103ad30302b45330037cf98f96ffd472f0e9ce",
|
| 136 |
+
"trainable_parameter_tensors": 1161,
|
| 137 |
+
"world_size": 2
|
| 138 |
+
}
|
settings/florence_robotwin/uniform/initialization_manifest.json
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"schema_version": 1,
|
| 3 |
+
"initialization_seed": 484370,
|
| 4 |
+
"action_model_sha256": "17ab11d253e73f3058c40637d5f31c3462a97cfbf61845148671de47ecca39eb",
|
| 5 |
+
"action_model_parameter_elements": 141815566
|
| 6 |
+
}
|
settings/florence_robotwin/uniform/valid_anchor_manifest.json
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2090f6b9d7e1bc5ef41897436690f031c187d625fa6b71ae0c54f3793660c757
|
| 3 |
+
size 107646168
|
settings/qwen_robotwin/common_start/config.full.yaml
ADDED
|
@@ -0,0 +1,242 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
run_id: qwen3vl-groot-robotwin-official150k-adamw8bit-mb5-ga4-seed42-20260810
|
| 2 |
+
run_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/qwen3vl-groot-robotwin-official150k
|
| 3 |
+
seed: 42
|
| 4 |
+
wandb_entity: null
|
| 5 |
+
wandb_project: InfoFlow-StarVLA-RoboTwin
|
| 6 |
+
is_debug: false
|
| 7 |
+
version_id: '0.21'
|
| 8 |
+
framework:
|
| 9 |
+
name: QwenGR00T
|
| 10 |
+
qwenvl:
|
| 11 |
+
base_vlm: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Pretrained_models/Qwen3-VL-4B-Instruct
|
| 12 |
+
attn_implementation: flash_attention_2
|
| 13 |
+
vl_hidden_dim: 2048
|
| 14 |
+
action_model:
|
| 15 |
+
action_model_type: DiT-B
|
| 16 |
+
action_hidden_dim: 2048
|
| 17 |
+
hidden_size: 1024
|
| 18 |
+
add_pos_embed: true
|
| 19 |
+
max_seq_len: 1024
|
| 20 |
+
action_dim: 14
|
| 21 |
+
state_dim: 0
|
| 22 |
+
action_horizon: 50
|
| 23 |
+
repeated_diffusion_steps: 8
|
| 24 |
+
noise_beta_alpha: 1.5
|
| 25 |
+
noise_beta_beta: 1.0
|
| 26 |
+
noise_s: 0.999
|
| 27 |
+
num_timestep_buckets: 1000
|
| 28 |
+
num_inference_timesteps: 4
|
| 29 |
+
num_target_vision_tokens: 32
|
| 30 |
+
diffusion_model_cfg:
|
| 31 |
+
cross_attention_dim: 2560
|
| 32 |
+
dropout: 0.2
|
| 33 |
+
final_dropout: true
|
| 34 |
+
interleave_self_attention: true
|
| 35 |
+
norm_type: ada_norm
|
| 36 |
+
num_layers: 16
|
| 37 |
+
output_dim: 1024
|
| 38 |
+
positional_embeddings: null
|
| 39 |
+
future_action_window_size: 49
|
| 40 |
+
past_action_window_size: 0
|
| 41 |
+
dino:
|
| 42 |
+
dino_backbone: dinov2_vits14
|
| 43 |
+
datasets:
|
| 44 |
+
vla_data:
|
| 45 |
+
dataset_py: lerobot_datasets
|
| 46 |
+
data_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Datasets/RoboTwin
|
| 47 |
+
data_mix: robotwin_all_50
|
| 48 |
+
action_type: abs_qpos
|
| 49 |
+
action_mode: abs
|
| 50 |
+
include_state: false
|
| 51 |
+
sequential_step_sampling: false
|
| 52 |
+
per_device_batch_size: 5
|
| 53 |
+
load_all_data_for_training: true
|
| 54 |
+
image_size:
|
| 55 |
+
- 224
|
| 56 |
+
- 224
|
| 57 |
+
video_backend: pyav
|
| 58 |
+
num_workers: 4
|
| 59 |
+
expected_world_size: 2
|
| 60 |
+
traceable_sampling: true
|
| 61 |
+
dataset_provenance_path: /home/miaochengyu/starVLA-infoflow-codex/protocols/robotwin/data_provenance.json
|
| 62 |
+
dataset_content_fingerprint_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/robotwin/content_fingerprints.json
|
| 63 |
+
valid_anchor_manifest_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/robotwin/valid_anchor_manifest.json
|
| 64 |
+
trainer:
|
| 65 |
+
use_deepspeed: true
|
| 66 |
+
max_train_steps: 150000
|
| 67 |
+
num_warmup_steps: 5000
|
| 68 |
+
save_interval: null
|
| 69 |
+
save_steps:
|
| 70 |
+
- 10000
|
| 71 |
+
- 11000
|
| 72 |
+
- 12000
|
| 73 |
+
- 13000
|
| 74 |
+
- 14000
|
| 75 |
+
- 15000
|
| 76 |
+
- 16000
|
| 77 |
+
- 17000
|
| 78 |
+
- 18000
|
| 79 |
+
- 19000
|
| 80 |
+
- 20000
|
| 81 |
+
- 21000
|
| 82 |
+
- 22000
|
| 83 |
+
- 23000
|
| 84 |
+
- 24000
|
| 85 |
+
- 25000
|
| 86 |
+
- 26000
|
| 87 |
+
- 27000
|
| 88 |
+
- 28000
|
| 89 |
+
- 29000
|
| 90 |
+
- 30000
|
| 91 |
+
- 31000
|
| 92 |
+
- 32000
|
| 93 |
+
- 33000
|
| 94 |
+
- 34000
|
| 95 |
+
- 35000
|
| 96 |
+
- 36000
|
| 97 |
+
- 37000
|
| 98 |
+
- 38000
|
| 99 |
+
- 39000
|
| 100 |
+
- 40000
|
| 101 |
+
- 41000
|
| 102 |
+
- 42000
|
| 103 |
+
- 43000
|
| 104 |
+
- 44000
|
| 105 |
+
- 45000
|
| 106 |
+
- 46000
|
| 107 |
+
- 47000
|
| 108 |
+
- 48000
|
| 109 |
+
- 49000
|
| 110 |
+
- 50000
|
| 111 |
+
- 51000
|
| 112 |
+
- 52000
|
| 113 |
+
- 53000
|
| 114 |
+
- 54000
|
| 115 |
+
- 55000
|
| 116 |
+
- 56000
|
| 117 |
+
- 57000
|
| 118 |
+
- 58000
|
| 119 |
+
- 59000
|
| 120 |
+
- 60000
|
| 121 |
+
- 61000
|
| 122 |
+
- 62000
|
| 123 |
+
- 63000
|
| 124 |
+
- 64000
|
| 125 |
+
- 65000
|
| 126 |
+
- 66000
|
| 127 |
+
- 67000
|
| 128 |
+
- 68000
|
| 129 |
+
- 69000
|
| 130 |
+
- 70000
|
| 131 |
+
- 71000
|
| 132 |
+
- 72000
|
| 133 |
+
- 73000
|
| 134 |
+
- 74000
|
| 135 |
+
- 75000
|
| 136 |
+
- 76000
|
| 137 |
+
- 77000
|
| 138 |
+
- 78000
|
| 139 |
+
- 79000
|
| 140 |
+
- 80000
|
| 141 |
+
- 81000
|
| 142 |
+
- 82000
|
| 143 |
+
- 83000
|
| 144 |
+
- 84000
|
| 145 |
+
- 85000
|
| 146 |
+
- 86000
|
| 147 |
+
- 87000
|
| 148 |
+
- 88000
|
| 149 |
+
- 89000
|
| 150 |
+
- 90000
|
| 151 |
+
- 91000
|
| 152 |
+
- 92000
|
| 153 |
+
- 93000
|
| 154 |
+
- 94000
|
| 155 |
+
- 95000
|
| 156 |
+
- 96000
|
| 157 |
+
- 97000
|
| 158 |
+
- 98000
|
| 159 |
+
- 99000
|
| 160 |
+
- 100000
|
| 161 |
+
- 101000
|
| 162 |
+
- 102000
|
| 163 |
+
- 103000
|
| 164 |
+
- 104000
|
| 165 |
+
- 105000
|
| 166 |
+
- 106000
|
| 167 |
+
- 107000
|
| 168 |
+
- 108000
|
| 169 |
+
- 109000
|
| 170 |
+
- 110000
|
| 171 |
+
- 111000
|
| 172 |
+
- 112000
|
| 173 |
+
- 113000
|
| 174 |
+
- 114000
|
| 175 |
+
- 115000
|
| 176 |
+
- 116000
|
| 177 |
+
- 117000
|
| 178 |
+
- 118000
|
| 179 |
+
- 119000
|
| 180 |
+
- 120000
|
| 181 |
+
- 121000
|
| 182 |
+
- 122000
|
| 183 |
+
- 123000
|
| 184 |
+
- 124000
|
| 185 |
+
- 125000
|
| 186 |
+
- 126000
|
| 187 |
+
- 127000
|
| 188 |
+
- 128000
|
| 189 |
+
- 129000
|
| 190 |
+
- 130000
|
| 191 |
+
- 131000
|
| 192 |
+
- 132000
|
| 193 |
+
- 133000
|
| 194 |
+
- 134000
|
| 195 |
+
- 135000
|
| 196 |
+
- 136000
|
| 197 |
+
- 137000
|
| 198 |
+
- 138000
|
| 199 |
+
- 139000
|
| 200 |
+
- 140000
|
| 201 |
+
- 141000
|
| 202 |
+
- 142000
|
| 203 |
+
- 143000
|
| 204 |
+
- 144000
|
| 205 |
+
- 145000
|
| 206 |
+
- 146000
|
| 207 |
+
- 147000
|
| 208 |
+
- 148000
|
| 209 |
+
- 149000
|
| 210 |
+
- 150000
|
| 211 |
+
evaluation_snapshot_interval: 1000
|
| 212 |
+
evaluation_snapshot_start_step: 10000
|
| 213 |
+
evaluation_snapshot_end_step: 20000
|
| 214 |
+
eval_interval: null
|
| 215 |
+
repeated_diffusion_steps: 4
|
| 216 |
+
learning_rate:
|
| 217 |
+
base: 1.0e-05
|
| 218 |
+
qwen_vl_interface: 1.0e-05
|
| 219 |
+
action_model: 0.0001
|
| 220 |
+
lr_scheduler_type: cosine_with_min_lr
|
| 221 |
+
scheduler_specific_kwargs:
|
| 222 |
+
min_lr: 5.0e-07
|
| 223 |
+
freeze_modules: ''
|
| 224 |
+
trainable_modules: ''
|
| 225 |
+
frozen_modules_eval: false
|
| 226 |
+
logging_frequency: 10
|
| 227 |
+
gradient_clipping: 1.0
|
| 228 |
+
gradient_accumulation_steps: 4
|
| 229 |
+
save_format: pt
|
| 230 |
+
save_final_checkpoint: false
|
| 231 |
+
save_consolidated_model: false
|
| 232 |
+
optimizer:
|
| 233 |
+
name: AdamW8bit
|
| 234 |
+
betas:
|
| 235 |
+
- 0.9
|
| 236 |
+
- 0.95
|
| 237 |
+
eps: 1.0e-08
|
| 238 |
+
weight_decay: 1.0e-08
|
| 239 |
+
is_resume: true
|
| 240 |
+
allow_model_only_resume: false
|
| 241 |
+
config_yaml: /home/miaochengyu/starVLA-infoflow-codex/examples/InfoFlow/train_files/server/qwen3vl_groot_robotwin_official150k_2gpu.yaml
|
| 242 |
+
output_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/qwen3vl-groot-robotwin-official150k/qwen3vl-groot-robotwin-official150k-adamw8bit-mb5-ga4-seed42-20260810
|
settings/qwen_robotwin/common_start/config.yaml
ADDED
|
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
datasets:
|
| 2 |
+
vla_data:
|
| 3 |
+
action_mode: abs
|
| 4 |
+
data_mix: robotwin_all_50
|
| 5 |
+
data_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Datasets/RoboTwin
|
| 6 |
+
dataset_content_fingerprint_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/robotwin/content_fingerprints.json
|
| 7 |
+
dataset_provenance_path: /home/miaochengyu/starVLA-infoflow-codex/protocols/robotwin/data_provenance.json
|
| 8 |
+
dataset_py: lerobot_datasets
|
| 9 |
+
expected_world_size: 2
|
| 10 |
+
num_workers: 4
|
| 11 |
+
per_device_batch_size: 5
|
| 12 |
+
traceable_sampling: true
|
| 13 |
+
valid_anchor_manifest_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/robotwin/valid_anchor_manifest.json
|
| 14 |
+
video_backend: pyav
|
| 15 |
+
framework:
|
| 16 |
+
action_model:
|
| 17 |
+
action_dim: 14
|
| 18 |
+
action_horizon: 50
|
| 19 |
+
action_model_type: DiT-B
|
| 20 |
+
add_pos_embed: true
|
| 21 |
+
diffusion_model_cfg:
|
| 22 |
+
cross_attention_dim: 2560
|
| 23 |
+
dropout: 0.2
|
| 24 |
+
final_dropout: true
|
| 25 |
+
interleave_self_attention: true
|
| 26 |
+
norm_type: ada_norm
|
| 27 |
+
num_layers: 16
|
| 28 |
+
output_dim: 1024
|
| 29 |
+
positional_embeddings: null
|
| 30 |
+
hidden_size: 1024
|
| 31 |
+
max_seq_len: 1024
|
| 32 |
+
noise_beta_alpha: 1.5
|
| 33 |
+
noise_beta_beta: 1.0
|
| 34 |
+
num_inference_timesteps: 4
|
| 35 |
+
num_target_vision_tokens: 32
|
| 36 |
+
num_timestep_buckets: 1000
|
| 37 |
+
state_dim: 0
|
| 38 |
+
name: QwenGR00T
|
| 39 |
+
qwenvl:
|
| 40 |
+
attn_implementation: flash_attention_2
|
| 41 |
+
base_vlm: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Pretrained_models/Qwen3-VL-4B-Instruct
|
| 42 |
+
output_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/qwen3vl-groot-robotwin-official150k/qwen3vl-groot-robotwin-official150k-adamw8bit-mb5-ga4-seed42-20260810
|
| 43 |
+
run_id: qwen3vl-groot-robotwin-official150k-adamw8bit-mb5-ga4-seed42-20260810
|
| 44 |
+
run_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/qwen3vl-groot-robotwin-official150k
|
| 45 |
+
seed: 42
|
| 46 |
+
trainer:
|
| 47 |
+
freeze_modules: ''
|
| 48 |
+
gradient_accumulation_steps: 4
|
| 49 |
+
learning_rate:
|
| 50 |
+
action_model: 0.0001
|
| 51 |
+
base: 1.0e-05
|
| 52 |
+
qwen_vl_interface: 1.0e-05
|
| 53 |
+
lr_scheduler_type: cosine_with_min_lr
|
| 54 |
+
max_train_steps: 150000
|
| 55 |
+
num_warmup_steps: 5000
|
| 56 |
+
optimizer:
|
| 57 |
+
betas:
|
| 58 |
+
- 0.9
|
| 59 |
+
- 0.95
|
| 60 |
+
eps: 1.0e-08
|
| 61 |
+
name: AdamW8bit
|
| 62 |
+
weight_decay: 1.0e-08
|
| 63 |
+
scheduler_specific_kwargs:
|
| 64 |
+
min_lr: 5.0e-07
|
settings/qwen_robotwin/common_start/dataset_statistics.json
ADDED
|
@@ -0,0 +1,218 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"new_embodiment": {
|
| 3 |
+
"action": {
|
| 4 |
+
"mean": [
|
| 5 |
+
-0.22423998869955536,
|
| 6 |
+
1.0729362124204633,
|
| 7 |
+
0.775079609751701,
|
| 8 |
+
-0.32701663097366684,
|
| 9 |
+
0.06159586461764771,
|
| 10 |
+
-0.06420705380151046,
|
| 11 |
+
0.22430071420967568,
|
| 12 |
+
1.1274866509437556,
|
| 13 |
+
0.831773659586906,
|
| 14 |
+
-0.3638282522559164,
|
| 15 |
+
-0.024959841337949897,
|
| 16 |
+
0.05302203213563182,
|
| 17 |
+
0.6774904182553295,
|
| 18 |
+
0.6598676902055738
|
| 19 |
+
],
|
| 20 |
+
"std": [
|
| 21 |
+
0.39622931737429123,
|
| 22 |
+
1.003590449339515,
|
| 23 |
+
0.7811611351334437,
|
| 24 |
+
0.6707485714474849,
|
| 25 |
+
0.2686908204983921,
|
| 26 |
+
0.6492019642472782,
|
| 27 |
+
0.34733679400923234,
|
| 28 |
+
1.0176226676795115,
|
| 29 |
+
0.8039915805960646,
|
| 30 |
+
0.7067249045233707,
|
| 31 |
+
0.26259645177151686,
|
| 32 |
+
0.6808759458519371,
|
| 33 |
+
0.4482151151114418,
|
| 34 |
+
0.4539921671160285
|
| 35 |
+
],
|
| 36 |
+
"max": [
|
| 37 |
+
6.45659065246582,
|
| 38 |
+
4.179152488708496,
|
| 39 |
+
5.346591472625732,
|
| 40 |
+
1.7942548990249634,
|
| 41 |
+
1.8604620695114136,
|
| 42 |
+
5.43813943862915,
|
| 43 |
+
7.142920017242432,
|
| 44 |
+
4.157068729400635,
|
| 45 |
+
5.672450065612793,
|
| 46 |
+
1.95806884765625,
|
| 47 |
+
1.5663840770721436,
|
| 48 |
+
5.278968811035156,
|
| 49 |
+
1.0,
|
| 50 |
+
1.0
|
| 51 |
+
],
|
| 52 |
+
"min": [
|
| 53 |
+
-7.552278995513916,
|
| 54 |
+
-0.36354875564575195,
|
| 55 |
+
-0.18577136099338531,
|
| 56 |
+
-1.956291913986206,
|
| 57 |
+
-1.6801013946533203,
|
| 58 |
+
-7.678869724273682,
|
| 59 |
+
-8.539926528930664,
|
| 60 |
+
-0.6294453740119934,
|
| 61 |
+
-0.07775841653347015,
|
| 62 |
+
-2.1328067779541016,
|
| 63 |
+
-2.1285502910614014,
|
| 64 |
+
-8.228925704956055,
|
| 65 |
+
0.0,
|
| 66 |
+
0.0
|
| 67 |
+
],
|
| 68 |
+
"q01": [
|
| 69 |
+
-7.334665780067444,
|
| 70 |
+
-5.257390398583084e-07,
|
| 71 |
+
-2.8215323254698887e-05,
|
| 72 |
+
-1.841589093208313,
|
| 73 |
+
-1.2851550579071045,
|
| 74 |
+
-6.268352918624878,
|
| 75 |
+
-5.995517673492431,
|
| 76 |
+
-0.572821855545044,
|
| 77 |
+
-2.81171942333458e-05,
|
| 78 |
+
-1.8467556238174438,
|
| 79 |
+
-1.4545854091644288,
|
| 80 |
+
-5.945669202804565,
|
| 81 |
+
0.0,
|
| 82 |
+
0.0
|
| 83 |
+
],
|
| 84 |
+
"q99": [
|
| 85 |
+
5.016749763488781,
|
| 86 |
+
3.4792869091033936,
|
| 87 |
+
4.225283479690551,
|
| 88 |
+
1.7064905214309687,
|
| 89 |
+
1.4829185009002717,
|
| 90 |
+
3.7177388882636553,
|
| 91 |
+
1.258314609527588,
|
| 92 |
+
2.967935800552368,
|
| 93 |
+
3.2566452693939207,
|
| 94 |
+
1.766038613319397,
|
| 95 |
+
1.4066662788391113,
|
| 96 |
+
3.5830023288726807,
|
| 97 |
+
1.0,
|
| 98 |
+
1.0
|
| 99 |
+
],
|
| 100 |
+
"mask": [
|
| 101 |
+
true,
|
| 102 |
+
true,
|
| 103 |
+
true,
|
| 104 |
+
true,
|
| 105 |
+
true,
|
| 106 |
+
true,
|
| 107 |
+
true,
|
| 108 |
+
true,
|
| 109 |
+
true,
|
| 110 |
+
true,
|
| 111 |
+
true,
|
| 112 |
+
true,
|
| 113 |
+
false,
|
| 114 |
+
false
|
| 115 |
+
]
|
| 116 |
+
},
|
| 117 |
+
"state": {
|
| 118 |
+
"mean": [
|
| 119 |
+
-0.22281882727518673,
|
| 120 |
+
1.0668349397182466,
|
| 121 |
+
0.770632639825344,
|
| 122 |
+
-0.3252330579422412,
|
| 123 |
+
0.06118139047002841,
|
| 124 |
+
-0.06316296239456277,
|
| 125 |
+
0.22253762468695631,
|
| 126 |
+
1.1201750242710113,
|
| 127 |
+
0.8263923591375351,
|
| 128 |
+
-0.36159489695448405,
|
| 129 |
+
-0.024702461528722772,
|
| 130 |
+
0.05203246331657278,
|
| 131 |
+
0.6793262696266176,
|
| 132 |
+
0.6616973677277563
|
| 133 |
+
],
|
| 134 |
+
"std": [
|
| 135 |
+
0.39522870886731665,
|
| 136 |
+
1.004018577033247,
|
| 137 |
+
0.7809103811570285,
|
| 138 |
+
0.6688088363177986,
|
| 139 |
+
0.26775723818956776,
|
| 140 |
+
0.6463212583712935,
|
| 141 |
+
0.34648488321363663,
|
| 142 |
+
1.018232630032839,
|
| 143 |
+
0.8038798397682734,
|
| 144 |
+
0.7045612787595747,
|
| 145 |
+
0.26143414213134614,
|
| 146 |
+
0.6774619870452644,
|
| 147 |
+
0.44749220497027625,
|
| 148 |
+
0.45333916022037934
|
| 149 |
+
],
|
| 150 |
+
"max": [
|
| 151 |
+
6.45659065246582,
|
| 152 |
+
4.179152488708496,
|
| 153 |
+
5.346591472625732,
|
| 154 |
+
1.7942548990249634,
|
| 155 |
+
1.8604620695114136,
|
| 156 |
+
5.43813943862915,
|
| 157 |
+
7.142920017242432,
|
| 158 |
+
4.157068729400635,
|
| 159 |
+
5.672450065612793,
|
| 160 |
+
1.95806884765625,
|
| 161 |
+
1.5663840770721436,
|
| 162 |
+
5.278968811035156,
|
| 163 |
+
1.0,
|
| 164 |
+
1.0
|
| 165 |
+
],
|
| 166 |
+
"min": [
|
| 167 |
+
-7.552278995513916,
|
| 168 |
+
-0.36354875564575195,
|
| 169 |
+
-0.18577136099338531,
|
| 170 |
+
-1.956291913986206,
|
| 171 |
+
-1.6801013946533203,
|
| 172 |
+
-7.678869724273682,
|
| 173 |
+
-8.539926528930664,
|
| 174 |
+
-0.6294453740119934,
|
| 175 |
+
-0.07775841653347015,
|
| 176 |
+
-2.1328067779541016,
|
| 177 |
+
-2.1285502910614014,
|
| 178 |
+
-8.228925704956055,
|
| 179 |
+
0.0,
|
| 180 |
+
0.0
|
| 181 |
+
],
|
| 182 |
+
"q01": [
|
| 183 |
+
-7.334665780067444,
|
| 184 |
+
-5.257390398583084e-07,
|
| 185 |
+
-2.8215323254698887e-05,
|
| 186 |
+
-1.841589093208313,
|
| 187 |
+
-1.2851550579071045,
|
| 188 |
+
-6.268109321594238,
|
| 189 |
+
-5.912737417221069,
|
| 190 |
+
-0.572821855545044,
|
| 191 |
+
-2.81171942333458e-05,
|
| 192 |
+
-1.8467556238174438,
|
| 193 |
+
-1.440324923992157,
|
| 194 |
+
-5.8623305034637445,
|
| 195 |
+
0.0,
|
| 196 |
+
0.0
|
| 197 |
+
],
|
| 198 |
+
"q99": [
|
| 199 |
+
4.902149534225477,
|
| 200 |
+
3.4775466918945312,
|
| 201 |
+
4.224453401565551,
|
| 202 |
+
1.7064905214309687,
|
| 203 |
+
1.480757713317871,
|
| 204 |
+
3.609884884357428,
|
| 205 |
+
1.258314609527588,
|
| 206 |
+
2.967935800552368,
|
| 207 |
+
3.2566452693939207,
|
| 208 |
+
1.7433889162540437,
|
| 209 |
+
1.4066662788391113,
|
| 210 |
+
3.581243851184845,
|
| 211 |
+
1.0,
|
| 212 |
+
1.0
|
| 213 |
+
]
|
| 214 |
+
},
|
| 215 |
+
"num_transitions": 6075103,
|
| 216 |
+
"num_trajectories": 27500
|
| 217 |
+
}
|
| 218 |
+
}
|
settings/qwen_robotwin/common_start/initialization_manifest.json
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"schema_version": 1,
|
| 3 |
+
"initialization_seed": 42,
|
| 4 |
+
"action_model_sha256": "f87884ee8b3f2bbc21809d0da733a118e4d95151aa3c3a68ae9f623cd5a7bc4d",
|
| 5 |
+
"action_model_parameter_elements": 160689934
|
| 6 |
+
}
|
settings/qwen_robotwin/common_start/valid_anchor_manifest.json
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2090f6b9d7e1bc5ef41897436690f031c187d625fa6b71ae0c54f3793660c757
|
| 3 |
+
size 107646168
|
settings/qwen_robotwin/residual/config.full.yaml
ADDED
|
@@ -0,0 +1,112 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
run_id: qwen3vl-robotwin-residual-resume11k-13k-seed484370-20260815
|
| 2 |
+
run_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/qwen3vl-robotwin-residual-resume11k-13k-20260815
|
| 3 |
+
seed: 484370
|
| 4 |
+
wandb_entity: null
|
| 5 |
+
wandb_project: InfoFlow-StarVLA-RoboTwin
|
| 6 |
+
is_debug: false
|
| 7 |
+
version_id: '0.21'
|
| 8 |
+
framework:
|
| 9 |
+
name: QwenGR00T
|
| 10 |
+
qwenvl:
|
| 11 |
+
base_vlm: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Pretrained_models/Qwen3-VL-4B-Instruct
|
| 12 |
+
attn_implementation: flash_attention_2
|
| 13 |
+
vl_hidden_dim: 2048
|
| 14 |
+
action_model:
|
| 15 |
+
action_model_type: DiT-B
|
| 16 |
+
action_hidden_dim: 2048
|
| 17 |
+
hidden_size: 1024
|
| 18 |
+
add_pos_embed: true
|
| 19 |
+
max_seq_len: 1024
|
| 20 |
+
action_dim: 14
|
| 21 |
+
state_dim: 0
|
| 22 |
+
action_horizon: 50
|
| 23 |
+
repeated_diffusion_steps: 8
|
| 24 |
+
noise_beta_alpha: 1.5
|
| 25 |
+
noise_beta_beta: 1.0
|
| 26 |
+
noise_s: 0.999
|
| 27 |
+
num_timestep_buckets: 1000
|
| 28 |
+
num_inference_timesteps: 4
|
| 29 |
+
num_target_vision_tokens: 32
|
| 30 |
+
diffusion_model_cfg:
|
| 31 |
+
cross_attention_dim: 2560
|
| 32 |
+
dropout: 0.2
|
| 33 |
+
final_dropout: true
|
| 34 |
+
interleave_self_attention: true
|
| 35 |
+
norm_type: ada_norm
|
| 36 |
+
num_layers: 16
|
| 37 |
+
output_dim: 1024
|
| 38 |
+
positional_embeddings: null
|
| 39 |
+
future_action_window_size: 49
|
| 40 |
+
past_action_window_size: 0
|
| 41 |
+
dino:
|
| 42 |
+
dino_backbone: dinov2_vits14
|
| 43 |
+
datasets:
|
| 44 |
+
vla_data:
|
| 45 |
+
dataset_py: lerobot_datasets
|
| 46 |
+
data_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Datasets/RoboTwin
|
| 47 |
+
data_mix: robotwin_all_50
|
| 48 |
+
action_type: abs_qpos
|
| 49 |
+
action_mode: abs
|
| 50 |
+
include_state: false
|
| 51 |
+
sequential_step_sampling: false
|
| 52 |
+
per_device_batch_size: 5
|
| 53 |
+
load_all_data_for_training: true
|
| 54 |
+
image_size:
|
| 55 |
+
- 224
|
| 56 |
+
- 224
|
| 57 |
+
video_backend: pyav
|
| 58 |
+
num_workers: 4
|
| 59 |
+
expected_world_size: 2
|
| 60 |
+
traceable_sampling: true
|
| 61 |
+
dataset_provenance_path: /home/miaochengyu/starVLA-infoflow-codex/protocols/robotwin/data_provenance.json
|
| 62 |
+
dataset_content_fingerprint_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/robotwin/content_fingerprints.json
|
| 63 |
+
valid_anchor_manifest_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/robotwin/valid_anchor_manifest.json
|
| 64 |
+
temporal_proposal_path: /mnt/miaochengyu/starvla-infoflow-runtime/runs/infoflow-proposals/qwen3vl_robotwin_11k_residual_grid8_gate42f_override_20260814/residual_temporal_proposal.json
|
| 65 |
+
transform_rng_seed: 289726
|
| 66 |
+
trainer:
|
| 67 |
+
use_deepspeed: true
|
| 68 |
+
max_train_steps: 150000
|
| 69 |
+
num_warmup_steps: 5000
|
| 70 |
+
save_interval: null
|
| 71 |
+
save_steps:
|
| 72 |
+
- 11500
|
| 73 |
+
- 12000
|
| 74 |
+
- 12500
|
| 75 |
+
- 13000
|
| 76 |
+
evaluation_snapshot_interval: 1000
|
| 77 |
+
evaluation_snapshot_start_step: 10000
|
| 78 |
+
evaluation_snapshot_end_step: 20000
|
| 79 |
+
eval_interval: null
|
| 80 |
+
repeated_diffusion_steps: 4
|
| 81 |
+
learning_rate:
|
| 82 |
+
base: 1.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: 5.0e-07
|
| 88 |
+
freeze_modules: ''
|
| 89 |
+
trainable_modules: ''
|
| 90 |
+
frozen_modules_eval: false
|
| 91 |
+
logging_frequency: 10
|
| 92 |
+
gradient_clipping: 1.0
|
| 93 |
+
gradient_accumulation_steps: 4
|
| 94 |
+
save_format: pt
|
| 95 |
+
save_final_checkpoint: false
|
| 96 |
+
save_consolidated_model: false
|
| 97 |
+
optimizer:
|
| 98 |
+
name: AdamW8bit
|
| 99 |
+
betas:
|
| 100 |
+
- 0.9
|
| 101 |
+
- 0.95
|
| 102 |
+
eps: 1.0e-08
|
| 103 |
+
weight_decay: 1.0e-08
|
| 104 |
+
stop_after_steps: 13000
|
| 105 |
+
is_resume: true
|
| 106 |
+
allow_model_only_resume: false
|
| 107 |
+
resume_accelerate_checkpoint: /mnt/miaochengyu/starvla-infoflow-runtime/runs/qwen3vl-groot-robotwin-official150k/qwen3vl-groot-robotwin-official150k-adamw8bit-mb5-ga4-seed42-20260810/checkpoints/steps_11000_accelerate
|
| 108 |
+
branch_parent_seed: 42
|
| 109 |
+
branch_rng_seed: 484370
|
| 110 |
+
model_rng_seed: 721927
|
| 111 |
+
config_yaml: /home/miaochengyu/starVLA-infoflow-codex/examples/InfoFlow/train_files/server/qwen3vl_groot_robotwin_official150k_2gpu.yaml
|
| 112 |
+
output_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/qwen3vl-robotwin-residual-resume11k-13k-20260815/qwen3vl-robotwin-residual-resume11k-13k-seed484370-20260815
|
settings/qwen_robotwin/residual/config.yaml
ADDED
|
@@ -0,0 +1,65 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
datasets:
|
| 2 |
+
vla_data:
|
| 3 |
+
action_mode: abs
|
| 4 |
+
data_mix: robotwin_all_50
|
| 5 |
+
data_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Datasets/RoboTwin
|
| 6 |
+
dataset_content_fingerprint_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/robotwin/content_fingerprints.json
|
| 7 |
+
dataset_provenance_path: /home/miaochengyu/starVLA-infoflow-codex/protocols/robotwin/data_provenance.json
|
| 8 |
+
dataset_py: lerobot_datasets
|
| 9 |
+
expected_world_size: 2
|
| 10 |
+
num_workers: 4
|
| 11 |
+
per_device_batch_size: 5
|
| 12 |
+
temporal_proposal_path: /mnt/miaochengyu/starvla-infoflow-runtime/runs/infoflow-proposals/qwen3vl_robotwin_11k_residual_grid8_gate42f_override_20260814/residual_temporal_proposal.json
|
| 13 |
+
traceable_sampling: true
|
| 14 |
+
valid_anchor_manifest_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/robotwin/valid_anchor_manifest.json
|
| 15 |
+
video_backend: pyav
|
| 16 |
+
framework:
|
| 17 |
+
action_model:
|
| 18 |
+
action_dim: 14
|
| 19 |
+
action_horizon: 50
|
| 20 |
+
action_model_type: DiT-B
|
| 21 |
+
add_pos_embed: true
|
| 22 |
+
diffusion_model_cfg:
|
| 23 |
+
cross_attention_dim: 2560
|
| 24 |
+
dropout: 0.2
|
| 25 |
+
final_dropout: true
|
| 26 |
+
interleave_self_attention: true
|
| 27 |
+
norm_type: ada_norm
|
| 28 |
+
num_layers: 16
|
| 29 |
+
output_dim: 1024
|
| 30 |
+
positional_embeddings: null
|
| 31 |
+
hidden_size: 1024
|
| 32 |
+
max_seq_len: 1024
|
| 33 |
+
noise_beta_alpha: 1.5
|
| 34 |
+
noise_beta_beta: 1.0
|
| 35 |
+
num_inference_timesteps: 4
|
| 36 |
+
num_target_vision_tokens: 32
|
| 37 |
+
num_timestep_buckets: 1000
|
| 38 |
+
state_dim: 0
|
| 39 |
+
name: QwenGR00T
|
| 40 |
+
qwenvl:
|
| 41 |
+
attn_implementation: flash_attention_2
|
| 42 |
+
base_vlm: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Pretrained_models/Qwen3-VL-4B-Instruct
|
| 43 |
+
output_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/qwen3vl-robotwin-residual-resume11k-13k-20260815/qwen3vl-robotwin-residual-resume11k-13k-seed484370-20260815
|
| 44 |
+
run_id: qwen3vl-robotwin-residual-resume11k-13k-seed484370-20260815
|
| 45 |
+
run_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/qwen3vl-robotwin-residual-resume11k-13k-20260815
|
| 46 |
+
seed: 484370
|
| 47 |
+
trainer:
|
| 48 |
+
freeze_modules: ''
|
| 49 |
+
gradient_accumulation_steps: 4
|
| 50 |
+
learning_rate:
|
| 51 |
+
action_model: 0.0001
|
| 52 |
+
base: 1.0e-05
|
| 53 |
+
qwen_vl_interface: 1.0e-05
|
| 54 |
+
lr_scheduler_type: cosine_with_min_lr
|
| 55 |
+
max_train_steps: 150000
|
| 56 |
+
num_warmup_steps: 5000
|
| 57 |
+
optimizer:
|
| 58 |
+
betas:
|
| 59 |
+
- 0.9
|
| 60 |
+
- 0.95
|
| 61 |
+
eps: 1.0e-08
|
| 62 |
+
name: AdamW8bit
|
| 63 |
+
weight_decay: 1.0e-08
|
| 64 |
+
scheduler_specific_kwargs:
|
| 65 |
+
min_lr: 5.0e-07
|
settings/qwen_robotwin/residual/dataset_statistics.json
ADDED
|
@@ -0,0 +1,218 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"new_embodiment": {
|
| 3 |
+
"action": {
|
| 4 |
+
"mean": [
|
| 5 |
+
-0.22423998869955536,
|
| 6 |
+
1.0729362124204633,
|
| 7 |
+
0.775079609751701,
|
| 8 |
+
-0.32701663097366684,
|
| 9 |
+
0.06159586461764771,
|
| 10 |
+
-0.06420705380151046,
|
| 11 |
+
0.22430071420967568,
|
| 12 |
+
1.1274866509437556,
|
| 13 |
+
0.831773659586906,
|
| 14 |
+
-0.3638282522559164,
|
| 15 |
+
-0.024959841337949897,
|
| 16 |
+
0.05302203213563182,
|
| 17 |
+
0.6774904182553295,
|
| 18 |
+
0.6598676902055738
|
| 19 |
+
],
|
| 20 |
+
"std": [
|
| 21 |
+
0.39622931737429123,
|
| 22 |
+
1.003590449339515,
|
| 23 |
+
0.7811611351334437,
|
| 24 |
+
0.6707485714474849,
|
| 25 |
+
0.2686908204983921,
|
| 26 |
+
0.6492019642472782,
|
| 27 |
+
0.34733679400923234,
|
| 28 |
+
1.0176226676795115,
|
| 29 |
+
0.8039915805960646,
|
| 30 |
+
0.7067249045233707,
|
| 31 |
+
0.26259645177151686,
|
| 32 |
+
0.6808759458519371,
|
| 33 |
+
0.4482151151114418,
|
| 34 |
+
0.4539921671160285
|
| 35 |
+
],
|
| 36 |
+
"max": [
|
| 37 |
+
6.45659065246582,
|
| 38 |
+
4.179152488708496,
|
| 39 |
+
5.346591472625732,
|
| 40 |
+
1.7942548990249634,
|
| 41 |
+
1.8604620695114136,
|
| 42 |
+
5.43813943862915,
|
| 43 |
+
7.142920017242432,
|
| 44 |
+
4.157068729400635,
|
| 45 |
+
5.672450065612793,
|
| 46 |
+
1.95806884765625,
|
| 47 |
+
1.5663840770721436,
|
| 48 |
+
5.278968811035156,
|
| 49 |
+
1.0,
|
| 50 |
+
1.0
|
| 51 |
+
],
|
| 52 |
+
"min": [
|
| 53 |
+
-7.552278995513916,
|
| 54 |
+
-0.36354875564575195,
|
| 55 |
+
-0.18577136099338531,
|
| 56 |
+
-1.956291913986206,
|
| 57 |
+
-1.6801013946533203,
|
| 58 |
+
-7.678869724273682,
|
| 59 |
+
-8.539926528930664,
|
| 60 |
+
-0.6294453740119934,
|
| 61 |
+
-0.07775841653347015,
|
| 62 |
+
-2.1328067779541016,
|
| 63 |
+
-2.1285502910614014,
|
| 64 |
+
-8.228925704956055,
|
| 65 |
+
0.0,
|
| 66 |
+
0.0
|
| 67 |
+
],
|
| 68 |
+
"q01": [
|
| 69 |
+
-7.334665780067444,
|
| 70 |
+
-5.257390398583084e-07,
|
| 71 |
+
-2.8215323254698887e-05,
|
| 72 |
+
-1.841589093208313,
|
| 73 |
+
-1.2851550579071045,
|
| 74 |
+
-6.268352918624878,
|
| 75 |
+
-5.995517673492431,
|
| 76 |
+
-0.572821855545044,
|
| 77 |
+
-2.81171942333458e-05,
|
| 78 |
+
-1.8467556238174438,
|
| 79 |
+
-1.4545854091644288,
|
| 80 |
+
-5.945669202804565,
|
| 81 |
+
0.0,
|
| 82 |
+
0.0
|
| 83 |
+
],
|
| 84 |
+
"q99": [
|
| 85 |
+
5.016749763488781,
|
| 86 |
+
3.4792869091033936,
|
| 87 |
+
4.225283479690551,
|
| 88 |
+
1.7064905214309687,
|
| 89 |
+
1.4829185009002717,
|
| 90 |
+
3.7177388882636553,
|
| 91 |
+
1.258314609527588,
|
| 92 |
+
2.967935800552368,
|
| 93 |
+
3.2566452693939207,
|
| 94 |
+
1.766038613319397,
|
| 95 |
+
1.4066662788391113,
|
| 96 |
+
3.5830023288726807,
|
| 97 |
+
1.0,
|
| 98 |
+
1.0
|
| 99 |
+
],
|
| 100 |
+
"mask": [
|
| 101 |
+
true,
|
| 102 |
+
true,
|
| 103 |
+
true,
|
| 104 |
+
true,
|
| 105 |
+
true,
|
| 106 |
+
true,
|
| 107 |
+
true,
|
| 108 |
+
true,
|
| 109 |
+
true,
|
| 110 |
+
true,
|
| 111 |
+
true,
|
| 112 |
+
true,
|
| 113 |
+
false,
|
| 114 |
+
false
|
| 115 |
+
]
|
| 116 |
+
},
|
| 117 |
+
"state": {
|
| 118 |
+
"mean": [
|
| 119 |
+
-0.22281882727518673,
|
| 120 |
+
1.0668349397182466,
|
| 121 |
+
0.770632639825344,
|
| 122 |
+
-0.3252330579422412,
|
| 123 |
+
0.06118139047002841,
|
| 124 |
+
-0.06316296239456277,
|
| 125 |
+
0.22253762468695631,
|
| 126 |
+
1.1201750242710113,
|
| 127 |
+
0.8263923591375351,
|
| 128 |
+
-0.36159489695448405,
|
| 129 |
+
-0.024702461528722772,
|
| 130 |
+
0.05203246331657278,
|
| 131 |
+
0.6793262696266176,
|
| 132 |
+
0.6616973677277563
|
| 133 |
+
],
|
| 134 |
+
"std": [
|
| 135 |
+
0.39522870886731665,
|
| 136 |
+
1.004018577033247,
|
| 137 |
+
0.7809103811570285,
|
| 138 |
+
0.6688088363177986,
|
| 139 |
+
0.26775723818956776,
|
| 140 |
+
0.6463212583712935,
|
| 141 |
+
0.34648488321363663,
|
| 142 |
+
1.018232630032839,
|
| 143 |
+
0.8038798397682734,
|
| 144 |
+
0.7045612787595747,
|
| 145 |
+
0.26143414213134614,
|
| 146 |
+
0.6774619870452644,
|
| 147 |
+
0.44749220497027625,
|
| 148 |
+
0.45333916022037934
|
| 149 |
+
],
|
| 150 |
+
"max": [
|
| 151 |
+
6.45659065246582,
|
| 152 |
+
4.179152488708496,
|
| 153 |
+
5.346591472625732,
|
| 154 |
+
1.7942548990249634,
|
| 155 |
+
1.8604620695114136,
|
| 156 |
+
5.43813943862915,
|
| 157 |
+
7.142920017242432,
|
| 158 |
+
4.157068729400635,
|
| 159 |
+
5.672450065612793,
|
| 160 |
+
1.95806884765625,
|
| 161 |
+
1.5663840770721436,
|
| 162 |
+
5.278968811035156,
|
| 163 |
+
1.0,
|
| 164 |
+
1.0
|
| 165 |
+
],
|
| 166 |
+
"min": [
|
| 167 |
+
-7.552278995513916,
|
| 168 |
+
-0.36354875564575195,
|
| 169 |
+
-0.18577136099338531,
|
| 170 |
+
-1.956291913986206,
|
| 171 |
+
-1.6801013946533203,
|
| 172 |
+
-7.678869724273682,
|
| 173 |
+
-8.539926528930664,
|
| 174 |
+
-0.6294453740119934,
|
| 175 |
+
-0.07775841653347015,
|
| 176 |
+
-2.1328067779541016,
|
| 177 |
+
-2.1285502910614014,
|
| 178 |
+
-8.228925704956055,
|
| 179 |
+
0.0,
|
| 180 |
+
0.0
|
| 181 |
+
],
|
| 182 |
+
"q01": [
|
| 183 |
+
-7.334665780067444,
|
| 184 |
+
-5.257390398583084e-07,
|
| 185 |
+
-2.8215323254698887e-05,
|
| 186 |
+
-1.841589093208313,
|
| 187 |
+
-1.2851550579071045,
|
| 188 |
+
-6.268109321594238,
|
| 189 |
+
-5.912737417221069,
|
| 190 |
+
-0.572821855545044,
|
| 191 |
+
-2.81171942333458e-05,
|
| 192 |
+
-1.8467556238174438,
|
| 193 |
+
-1.440324923992157,
|
| 194 |
+
-5.8623305034637445,
|
| 195 |
+
0.0,
|
| 196 |
+
0.0
|
| 197 |
+
],
|
| 198 |
+
"q99": [
|
| 199 |
+
4.902149534225477,
|
| 200 |
+
3.4775466918945312,
|
| 201 |
+
4.224453401565551,
|
| 202 |
+
1.7064905214309687,
|
| 203 |
+
1.480757713317871,
|
| 204 |
+
3.609884884357428,
|
| 205 |
+
1.258314609527588,
|
| 206 |
+
2.967935800552368,
|
| 207 |
+
3.2566452693939207,
|
| 208 |
+
1.7433889162540437,
|
| 209 |
+
1.4066662788391113,
|
| 210 |
+
3.581243851184845,
|
| 211 |
+
1.0,
|
| 212 |
+
1.0
|
| 213 |
+
]
|
| 214 |
+
},
|
| 215 |
+
"num_transitions": 6075103,
|
| 216 |
+
"num_trajectories": 27500
|
| 217 |
+
}
|
| 218 |
+
}
|
settings/qwen_robotwin/residual/external_resume_certificate.json
ADDED
|
@@ -0,0 +1,138 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"branch_resume_identity": {
|
| 3 |
+
"config_sha256": "91c26d27b4e350855849912d1438907387518f91f5443c41cd591c4a78787c3c",
|
| 4 |
+
"identity_schema_version": 2,
|
| 5 |
+
"manifest_sha256": "dcb728a00da4b4ad7f989fa313d6d3355e13f57efefc7267e9ea092f2ffd8a50",
|
| 6 |
+
"model_structure_sha256": "7809faa5245147f81b98f3dbd17c7a063a7cb07a21e18af588c748c5925e1407",
|
| 7 |
+
"proposal_alpha": 0.2,
|
| 8 |
+
"proposal_sha256": "4e9a9e9d66c2bbbb02ddd8ecec06ad0b514cd299bbfac57bf8c2d44daeb70237",
|
| 9 |
+
"seed": 484370,
|
| 10 |
+
"total_batch_size": 40
|
| 11 |
+
},
|
| 12 |
+
"completed_steps": 11000,
|
| 13 |
+
"optimizer_param_groups": [
|
| 14 |
+
{
|
| 15 |
+
"betas": [
|
| 16 |
+
0.9,
|
| 17 |
+
0.95
|
| 18 |
+
],
|
| 19 |
+
"eps": 1e-08,
|
| 20 |
+
"initial_lr": 1e-05,
|
| 21 |
+
"lr": 9.959920880962606e-06,
|
| 22 |
+
"name": "qwen_vl_interface",
|
| 23 |
+
"weight_decay": 1e-08
|
| 24 |
+
},
|
| 25 |
+
{
|
| 26 |
+
"betas": [
|
| 27 |
+
0.9,
|
| 28 |
+
0.95
|
| 29 |
+
],
|
| 30 |
+
"eps": 1e-08,
|
| 31 |
+
"initial_lr": 0.0001,
|
| 32 |
+
"lr": 9.959920880962605e-05,
|
| 33 |
+
"name": "action_model",
|
| 34 |
+
"weight_decay": 1e-08
|
| 35 |
+
}
|
| 36 |
+
],
|
| 37 |
+
"optimizer_state_entries": 8,
|
| 38 |
+
"parent_checkpoint": "/mnt/miaochengyu/starvla-infoflow-runtime/runs/qwen3vl-groot-robotwin-official150k/qwen3vl-groot-robotwin-official150k-adamw8bit-mb5-ga4-seed42-20260810/checkpoints/steps_11000_accelerate",
|
| 39 |
+
"parent_checkpoint_commit_sha256": "180f1ed2d2b7a9e6c158974fc9ec951182dfb629f6a2806808ac5639d491ce1b",
|
| 40 |
+
"parent_checkpoint_files": [
|
| 41 |
+
{
|
| 42 |
+
"bytes": 1525,
|
| 43 |
+
"name": "custom_checkpoint_0.pkl",
|
| 44 |
+
"sha256": "70788f8a2d7852a70fbb794ab9176d14ad37fb6d1b1d821a368572e1d8524390"
|
| 45 |
+
},
|
| 46 |
+
{
|
| 47 |
+
"bytes": 13,
|
| 48 |
+
"name": "latest",
|
| 49 |
+
"sha256": "68d0ca762047fd48fdd6a06a5e6465209289c028bfd63f5557a110ff1ebc15ac"
|
| 50 |
+
},
|
| 51 |
+
{
|
| 52 |
+
"bytes": 13867381301,
|
| 53 |
+
"name": "pytorch_model/bf16_zero_pp_rank_0_mp_rank_00_optim_states.pt",
|
| 54 |
+
"sha256": "42b6d10bb158b404056983ce88505c32738b29f3a062fe6835d24ce26c521a6a"
|
| 55 |
+
},
|
| 56 |
+
{
|
| 57 |
+
"bytes": 13867381301,
|
| 58 |
+
"name": "pytorch_model/bf16_zero_pp_rank_1_mp_rank_00_optim_states.pt",
|
| 59 |
+
"sha256": "7b0a1bbc92efffbf056bc40f40e93a94b67ddf3f375dc6ae263bcb048ba6ec63"
|
| 60 |
+
},
|
| 61 |
+
{
|
| 62 |
+
"bytes": 566626,
|
| 63 |
+
"name": "pytorch_model/zero_pp_rank_0_mp_rank_00_model_states.pt",
|
| 64 |
+
"sha256": "8a3ef88f9ab0df8197a836dca1d6e2416835f0b3d827cb10a78254b9a969ce7c"
|
| 65 |
+
},
|
| 66 |
+
{
|
| 67 |
+
"bytes": 566498,
|
| 68 |
+
"name": "pytorch_model/zero_pp_rank_1_mp_rank_00_model_states.pt",
|
| 69 |
+
"sha256": "67cd03fceb5ee4c2be6affe504f71a48ee10127d6ea3a1f00d05721895ffba53"
|
| 70 |
+
},
|
| 71 |
+
{
|
| 72 |
+
"bytes": 15017,
|
| 73 |
+
"name": "random_states_0.pkl",
|
| 74 |
+
"sha256": "636b1c5983ee2f8a5047025027453c9a478c0606389f2f6945b370f39cec9e65"
|
| 75 |
+
},
|
| 76 |
+
{
|
| 77 |
+
"bytes": 15081,
|
| 78 |
+
"name": "random_states_1.pkl",
|
| 79 |
+
"sha256": "416b3be53ff049935984cc65385d7dae3b0e5a82df66c9516f61ee9100c41961"
|
| 80 |
+
},
|
| 81 |
+
{
|
| 82 |
+
"bytes": 33270,
|
| 83 |
+
"name": "zero_to_fp32.py",
|
| 84 |
+
"sha256": "e8a2553b434ac043ec63b022db1e768277acf8d4e70f10a2842db6d238977135"
|
| 85 |
+
}
|
| 86 |
+
],
|
| 87 |
+
"parent_checkpoint_files_sha256": "73692dae620e3d4368fe513e8adff4fe54bdb25341b4293e34ee2a6286554f8d",
|
| 88 |
+
"parent_checkpoint_integrity_manifest_sha256": null,
|
| 89 |
+
"parent_config_sha256": "4882855d94076fc0ec19a0d511f14b1cbe82860c8b5df3dcc64c5c5f6c0d2321",
|
| 90 |
+
"parent_resume_identity": {
|
| 91 |
+
"config_sha256": "1a48ea51cf89ce72a7956aaa29f710bba0326c9b726e52795561b381507d26fa",
|
| 92 |
+
"identity_schema_version": 2,
|
| 93 |
+
"manifest_sha256": "dcb728a00da4b4ad7f989fa313d6d3355e13f57efefc7267e9ea092f2ffd8a50",
|
| 94 |
+
"model_structure_sha256": "7809faa5245147f81b98f3dbd17c7a063a7cb07a21e18af588c748c5925e1407",
|
| 95 |
+
"proposal_alpha": null,
|
| 96 |
+
"proposal_sha256": null,
|
| 97 |
+
"seed": 42,
|
| 98 |
+
"total_batch_size": 40
|
| 99 |
+
},
|
| 100 |
+
"proposal_sha256": "4e9a9e9d66c2bbbb02ddd8ecec06ad0b514cd299bbfac57bf8c2d44daeb70237",
|
| 101 |
+
"rank_state_summaries": [
|
| 102 |
+
{
|
| 103 |
+
"optimizer_state_entries": 4,
|
| 104 |
+
"optimizer_state_sha256": "e1eea37514fa0478835597d1594b7a43b45cea9adc4ccb79a0de8fb9e10fe9f0",
|
| 105 |
+
"rank": 0,
|
| 106 |
+
"trainable_state_sha256": "ca2a19167d59aab693d0a0a1e17ea9adbf9cfd7367dfa7a5c90b04cf2183b717"
|
| 107 |
+
},
|
| 108 |
+
{
|
| 109 |
+
"optimizer_state_entries": 4,
|
| 110 |
+
"optimizer_state_sha256": "62c4e2908a7e97fe9a1d60642a85968081b55345493a11b9e7f68e49fcdb22f3",
|
| 111 |
+
"rank": 1,
|
| 112 |
+
"trainable_state_sha256": "ca2a19167d59aab693d0a0a1e17ea9adbf9cfd7367dfa7a5c90b04cf2183b717"
|
| 113 |
+
}
|
| 114 |
+
],
|
| 115 |
+
"repeated_diffusion_steps": 4,
|
| 116 |
+
"rng_streams": {
|
| 117 |
+
"model_base": 721927,
|
| 118 |
+
"sampler": 484370,
|
| 119 |
+
"transform": 289726
|
| 120 |
+
},
|
| 121 |
+
"scheduler": {
|
| 122 |
+
"class": "LambdaLR",
|
| 123 |
+
"last_epoch": 11000,
|
| 124 |
+
"last_lr": [
|
| 125 |
+
9.959920880962606e-06,
|
| 126 |
+
9.959920880962605e-05
|
| 127 |
+
],
|
| 128 |
+
"step_count": 11001
|
| 129 |
+
},
|
| 130 |
+
"scheduler_resume_mode": "accelerate_checkpoint",
|
| 131 |
+
"schema_version": 1,
|
| 132 |
+
"status": "exact_external_resume_loaded",
|
| 133 |
+
"total_batch_size": 40,
|
| 134 |
+
"trainable_parameter_elements": 32022798,
|
| 135 |
+
"trainable_parameter_sha256": "1bc504b22e4375be40bbbd8250be5b5f380c14d2fb47ed85deba3604a84b5484",
|
| 136 |
+
"trainable_parameter_tensors": 957,
|
| 137 |
+
"world_size": 2
|
| 138 |
+
}
|
settings/qwen_robotwin/residual/initialization_manifest.json
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"schema_version": 1,
|
| 3 |
+
"initialization_seed": 484370,
|
| 4 |
+
"action_model_sha256": "1cab43a1e801a94110a20384a9ad6cd76a239fedd9fe17890662da7e22d47e62",
|
| 5 |
+
"action_model_parameter_elements": 160689934
|
| 6 |
+
}
|
settings/qwen_robotwin/residual/valid_anchor_manifest.json
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2090f6b9d7e1bc5ef41897436690f031c187d625fa6b71ae0c54f3793660c757
|
| 3 |
+
size 107646168
|
settings/qwen_robotwin/uniform/config.full.yaml
ADDED
|
@@ -0,0 +1,108 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
run_id: qwen3vl-robotwin-uniform-commonstart11k-plus500-seed484370-20260815
|
| 2 |
+
run_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/qwen3vl-robotwin-uniform-commonstart11k-plus500-20260815
|
| 3 |
+
seed: 484370
|
| 4 |
+
wandb_entity: null
|
| 5 |
+
wandb_project: InfoFlow-StarVLA-RoboTwin
|
| 6 |
+
is_debug: false
|
| 7 |
+
version_id: '0.21'
|
| 8 |
+
framework:
|
| 9 |
+
name: QwenGR00T
|
| 10 |
+
qwenvl:
|
| 11 |
+
base_vlm: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Pretrained_models/Qwen3-VL-4B-Instruct
|
| 12 |
+
attn_implementation: flash_attention_2
|
| 13 |
+
vl_hidden_dim: 2048
|
| 14 |
+
action_model:
|
| 15 |
+
action_model_type: DiT-B
|
| 16 |
+
action_hidden_dim: 2048
|
| 17 |
+
hidden_size: 1024
|
| 18 |
+
add_pos_embed: true
|
| 19 |
+
max_seq_len: 1024
|
| 20 |
+
action_dim: 14
|
| 21 |
+
state_dim: 0
|
| 22 |
+
action_horizon: 50
|
| 23 |
+
repeated_diffusion_steps: 8
|
| 24 |
+
noise_beta_alpha: 1.5
|
| 25 |
+
noise_beta_beta: 1.0
|
| 26 |
+
noise_s: 0.999
|
| 27 |
+
num_timestep_buckets: 1000
|
| 28 |
+
num_inference_timesteps: 4
|
| 29 |
+
num_target_vision_tokens: 32
|
| 30 |
+
diffusion_model_cfg:
|
| 31 |
+
cross_attention_dim: 2560
|
| 32 |
+
dropout: 0.2
|
| 33 |
+
final_dropout: true
|
| 34 |
+
interleave_self_attention: true
|
| 35 |
+
norm_type: ada_norm
|
| 36 |
+
num_layers: 16
|
| 37 |
+
output_dim: 1024
|
| 38 |
+
positional_embeddings: null
|
| 39 |
+
future_action_window_size: 49
|
| 40 |
+
past_action_window_size: 0
|
| 41 |
+
dino:
|
| 42 |
+
dino_backbone: dinov2_vits14
|
| 43 |
+
datasets:
|
| 44 |
+
vla_data:
|
| 45 |
+
dataset_py: lerobot_datasets
|
| 46 |
+
data_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Datasets/RoboTwin
|
| 47 |
+
data_mix: robotwin_all_50
|
| 48 |
+
action_type: abs_qpos
|
| 49 |
+
action_mode: abs
|
| 50 |
+
include_state: false
|
| 51 |
+
sequential_step_sampling: false
|
| 52 |
+
per_device_batch_size: 5
|
| 53 |
+
load_all_data_for_training: true
|
| 54 |
+
image_size:
|
| 55 |
+
- 224
|
| 56 |
+
- 224
|
| 57 |
+
video_backend: pyav
|
| 58 |
+
num_workers: 4
|
| 59 |
+
expected_world_size: 2
|
| 60 |
+
traceable_sampling: true
|
| 61 |
+
dataset_provenance_path: /home/miaochengyu/starVLA-infoflow-codex/protocols/robotwin/data_provenance.json
|
| 62 |
+
dataset_content_fingerprint_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/robotwin/content_fingerprints.json
|
| 63 |
+
valid_anchor_manifest_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/robotwin/valid_anchor_manifest.json
|
| 64 |
+
transform_rng_seed: 289726
|
| 65 |
+
trainer:
|
| 66 |
+
use_deepspeed: true
|
| 67 |
+
max_train_steps: 150000
|
| 68 |
+
num_warmup_steps: 5000
|
| 69 |
+
save_interval: null
|
| 70 |
+
save_steps:
|
| 71 |
+
- 11500
|
| 72 |
+
evaluation_snapshot_interval: 1000
|
| 73 |
+
evaluation_snapshot_start_step: 10000
|
| 74 |
+
evaluation_snapshot_end_step: 20000
|
| 75 |
+
eval_interval: null
|
| 76 |
+
repeated_diffusion_steps: 4
|
| 77 |
+
learning_rate:
|
| 78 |
+
base: 1.0e-05
|
| 79 |
+
qwen_vl_interface: 1.0e-05
|
| 80 |
+
action_model: 0.0001
|
| 81 |
+
lr_scheduler_type: cosine_with_min_lr
|
| 82 |
+
scheduler_specific_kwargs:
|
| 83 |
+
min_lr: 5.0e-07
|
| 84 |
+
freeze_modules: ''
|
| 85 |
+
trainable_modules: ''
|
| 86 |
+
frozen_modules_eval: false
|
| 87 |
+
logging_frequency: 10
|
| 88 |
+
gradient_clipping: 1.0
|
| 89 |
+
gradient_accumulation_steps: 4
|
| 90 |
+
save_format: pt
|
| 91 |
+
save_final_checkpoint: false
|
| 92 |
+
save_consolidated_model: false
|
| 93 |
+
optimizer:
|
| 94 |
+
name: AdamW8bit
|
| 95 |
+
betas:
|
| 96 |
+
- 0.9
|
| 97 |
+
- 0.95
|
| 98 |
+
eps: 1.0e-08
|
| 99 |
+
weight_decay: 1.0e-08
|
| 100 |
+
stop_after_steps: 11500
|
| 101 |
+
is_resume: true
|
| 102 |
+
allow_model_only_resume: false
|
| 103 |
+
resume_accelerate_checkpoint: /mnt/miaochengyu/starvla-infoflow-runtime/runs/qwen3vl-groot-robotwin-official150k/qwen3vl-groot-robotwin-official150k-adamw8bit-mb5-ga4-seed42-20260810/checkpoints/steps_11000_accelerate
|
| 104 |
+
branch_parent_seed: 42
|
| 105 |
+
branch_rng_seed: 484370
|
| 106 |
+
model_rng_seed: 721927
|
| 107 |
+
config_yaml: /home/miaochengyu/starVLA-infoflow-codex/examples/InfoFlow/train_files/server/qwen3vl_groot_robotwin_official150k_2gpu.yaml
|
| 108 |
+
output_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/qwen3vl-robotwin-uniform-commonstart11k-plus500-20260815/qwen3vl-robotwin-uniform-commonstart11k-plus500-seed484370-20260815
|
settings/qwen_robotwin/uniform/config.yaml
ADDED
|
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
datasets:
|
| 2 |
+
vla_data:
|
| 3 |
+
action_mode: abs
|
| 4 |
+
data_mix: robotwin_all_50
|
| 5 |
+
data_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Datasets/RoboTwin
|
| 6 |
+
dataset_content_fingerprint_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/robotwin/content_fingerprints.json
|
| 7 |
+
dataset_provenance_path: /home/miaochengyu/starVLA-infoflow-codex/protocols/robotwin/data_provenance.json
|
| 8 |
+
dataset_py: lerobot_datasets
|
| 9 |
+
expected_world_size: 2
|
| 10 |
+
num_workers: 4
|
| 11 |
+
per_device_batch_size: 5
|
| 12 |
+
traceable_sampling: true
|
| 13 |
+
valid_anchor_manifest_path: /mnt/miaochengyu/starvla-infoflow-runtime/manifests/robotwin/valid_anchor_manifest.json
|
| 14 |
+
video_backend: pyav
|
| 15 |
+
framework:
|
| 16 |
+
action_model:
|
| 17 |
+
action_dim: 14
|
| 18 |
+
action_horizon: 50
|
| 19 |
+
action_model_type: DiT-B
|
| 20 |
+
add_pos_embed: true
|
| 21 |
+
diffusion_model_cfg:
|
| 22 |
+
cross_attention_dim: 2560
|
| 23 |
+
dropout: 0.2
|
| 24 |
+
final_dropout: true
|
| 25 |
+
interleave_self_attention: true
|
| 26 |
+
norm_type: ada_norm
|
| 27 |
+
num_layers: 16
|
| 28 |
+
output_dim: 1024
|
| 29 |
+
positional_embeddings: null
|
| 30 |
+
hidden_size: 1024
|
| 31 |
+
max_seq_len: 1024
|
| 32 |
+
noise_beta_alpha: 1.5
|
| 33 |
+
noise_beta_beta: 1.0
|
| 34 |
+
num_inference_timesteps: 4
|
| 35 |
+
num_target_vision_tokens: 32
|
| 36 |
+
num_timestep_buckets: 1000
|
| 37 |
+
state_dim: 0
|
| 38 |
+
name: QwenGR00T
|
| 39 |
+
qwenvl:
|
| 40 |
+
attn_implementation: flash_attention_2
|
| 41 |
+
base_vlm: /mnt/miaochengyu/starvla-infoflow-runtime/playground/Pretrained_models/Qwen3-VL-4B-Instruct
|
| 42 |
+
output_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/qwen3vl-robotwin-uniform-commonstart11k-plus500-20260815/qwen3vl-robotwin-uniform-commonstart11k-plus500-seed484370-20260815
|
| 43 |
+
run_id: qwen3vl-robotwin-uniform-commonstart11k-plus500-seed484370-20260815
|
| 44 |
+
run_root_dir: /mnt/miaochengyu/starvla-infoflow-runtime/runs/qwen3vl-robotwin-uniform-commonstart11k-plus500-20260815
|
| 45 |
+
seed: 484370
|
| 46 |
+
trainer:
|
| 47 |
+
freeze_modules: ''
|
| 48 |
+
gradient_accumulation_steps: 4
|
| 49 |
+
learning_rate:
|
| 50 |
+
action_model: 0.0001
|
| 51 |
+
base: 1.0e-05
|
| 52 |
+
qwen_vl_interface: 1.0e-05
|
| 53 |
+
lr_scheduler_type: cosine_with_min_lr
|
| 54 |
+
max_train_steps: 150000
|
| 55 |
+
num_warmup_steps: 5000
|
| 56 |
+
optimizer:
|
| 57 |
+
betas:
|
| 58 |
+
- 0.9
|
| 59 |
+
- 0.95
|
| 60 |
+
eps: 1.0e-08
|
| 61 |
+
name: AdamW8bit
|
| 62 |
+
weight_decay: 1.0e-08
|
| 63 |
+
scheduler_specific_kwargs:
|
| 64 |
+
min_lr: 5.0e-07
|
settings/qwen_robotwin/uniform/dataset_statistics.json
ADDED
|
@@ -0,0 +1,218 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"new_embodiment": {
|
| 3 |
+
"action": {
|
| 4 |
+
"mean": [
|
| 5 |
+
-0.22423998869955536,
|
| 6 |
+
1.0729362124204633,
|
| 7 |
+
0.775079609751701,
|
| 8 |
+
-0.32701663097366684,
|
| 9 |
+
0.06159586461764771,
|
| 10 |
+
-0.06420705380151046,
|
| 11 |
+
0.22430071420967568,
|
| 12 |
+
1.1274866509437556,
|
| 13 |
+
0.831773659586906,
|
| 14 |
+
-0.3638282522559164,
|
| 15 |
+
-0.024959841337949897,
|
| 16 |
+
0.05302203213563182,
|
| 17 |
+
0.6774904182553295,
|
| 18 |
+
0.6598676902055738
|
| 19 |
+
],
|
| 20 |
+
"std": [
|
| 21 |
+
0.39622931737429123,
|
| 22 |
+
1.003590449339515,
|
| 23 |
+
0.7811611351334437,
|
| 24 |
+
0.6707485714474849,
|
| 25 |
+
0.2686908204983921,
|
| 26 |
+
0.6492019642472782,
|
| 27 |
+
0.34733679400923234,
|
| 28 |
+
1.0176226676795115,
|
| 29 |
+
0.8039915805960646,
|
| 30 |
+
0.7067249045233707,
|
| 31 |
+
0.26259645177151686,
|
| 32 |
+
0.6808759458519371,
|
| 33 |
+
0.4482151151114418,
|
| 34 |
+
0.4539921671160285
|
| 35 |
+
],
|
| 36 |
+
"max": [
|
| 37 |
+
6.45659065246582,
|
| 38 |
+
4.179152488708496,
|
| 39 |
+
5.346591472625732,
|
| 40 |
+
1.7942548990249634,
|
| 41 |
+
1.8604620695114136,
|
| 42 |
+
5.43813943862915,
|
| 43 |
+
7.142920017242432,
|
| 44 |
+
4.157068729400635,
|
| 45 |
+
5.672450065612793,
|
| 46 |
+
1.95806884765625,
|
| 47 |
+
1.5663840770721436,
|
| 48 |
+
5.278968811035156,
|
| 49 |
+
1.0,
|
| 50 |
+
1.0
|
| 51 |
+
],
|
| 52 |
+
"min": [
|
| 53 |
+
-7.552278995513916,
|
| 54 |
+
-0.36354875564575195,
|
| 55 |
+
-0.18577136099338531,
|
| 56 |
+
-1.956291913986206,
|
| 57 |
+
-1.6801013946533203,
|
| 58 |
+
-7.678869724273682,
|
| 59 |
+
-8.539926528930664,
|
| 60 |
+
-0.6294453740119934,
|
| 61 |
+
-0.07775841653347015,
|
| 62 |
+
-2.1328067779541016,
|
| 63 |
+
-2.1285502910614014,
|
| 64 |
+
-8.228925704956055,
|
| 65 |
+
0.0,
|
| 66 |
+
0.0
|
| 67 |
+
],
|
| 68 |
+
"q01": [
|
| 69 |
+
-7.334665780067444,
|
| 70 |
+
-5.257390398583084e-07,
|
| 71 |
+
-2.8215323254698887e-05,
|
| 72 |
+
-1.841589093208313,
|
| 73 |
+
-1.2851550579071045,
|
| 74 |
+
-6.268352918624878,
|
| 75 |
+
-5.995517673492431,
|
| 76 |
+
-0.572821855545044,
|
| 77 |
+
-2.81171942333458e-05,
|
| 78 |
+
-1.8467556238174438,
|
| 79 |
+
-1.4545854091644288,
|
| 80 |
+
-5.945669202804565,
|
| 81 |
+
0.0,
|
| 82 |
+
0.0
|
| 83 |
+
],
|
| 84 |
+
"q99": [
|
| 85 |
+
5.016749763488781,
|
| 86 |
+
3.4792869091033936,
|
| 87 |
+
4.225283479690551,
|
| 88 |
+
1.7064905214309687,
|
| 89 |
+
1.4829185009002717,
|
| 90 |
+
3.7177388882636553,
|
| 91 |
+
1.258314609527588,
|
| 92 |
+
2.967935800552368,
|
| 93 |
+
3.2566452693939207,
|
| 94 |
+
1.766038613319397,
|
| 95 |
+
1.4066662788391113,
|
| 96 |
+
3.5830023288726807,
|
| 97 |
+
1.0,
|
| 98 |
+
1.0
|
| 99 |
+
],
|
| 100 |
+
"mask": [
|
| 101 |
+
true,
|
| 102 |
+
true,
|
| 103 |
+
true,
|
| 104 |
+
true,
|
| 105 |
+
true,
|
| 106 |
+
true,
|
| 107 |
+
true,
|
| 108 |
+
true,
|
| 109 |
+
true,
|
| 110 |
+
true,
|
| 111 |
+
true,
|
| 112 |
+
true,
|
| 113 |
+
false,
|
| 114 |
+
false
|
| 115 |
+
]
|
| 116 |
+
},
|
| 117 |
+
"state": {
|
| 118 |
+
"mean": [
|
| 119 |
+
-0.22281882727518673,
|
| 120 |
+
1.0668349397182466,
|
| 121 |
+
0.770632639825344,
|
| 122 |
+
-0.3252330579422412,
|
| 123 |
+
0.06118139047002841,
|
| 124 |
+
-0.06316296239456277,
|
| 125 |
+
0.22253762468695631,
|
| 126 |
+
1.1201750242710113,
|
| 127 |
+
0.8263923591375351,
|
| 128 |
+
-0.36159489695448405,
|
| 129 |
+
-0.024702461528722772,
|
| 130 |
+
0.05203246331657278,
|
| 131 |
+
0.6793262696266176,
|
| 132 |
+
0.6616973677277563
|
| 133 |
+
],
|
| 134 |
+
"std": [
|
| 135 |
+
0.39522870886731665,
|
| 136 |
+
1.004018577033247,
|
| 137 |
+
0.7809103811570285,
|
| 138 |
+
0.6688088363177986,
|
| 139 |
+
0.26775723818956776,
|
| 140 |
+
0.6463212583712935,
|
| 141 |
+
0.34648488321363663,
|
| 142 |
+
1.018232630032839,
|
| 143 |
+
0.8038798397682734,
|
| 144 |
+
0.7045612787595747,
|
| 145 |
+
0.26143414213134614,
|
| 146 |
+
0.6774619870452644,
|
| 147 |
+
0.44749220497027625,
|
| 148 |
+
0.45333916022037934
|
| 149 |
+
],
|
| 150 |
+
"max": [
|
| 151 |
+
6.45659065246582,
|
| 152 |
+
4.179152488708496,
|
| 153 |
+
5.346591472625732,
|
| 154 |
+
1.7942548990249634,
|
| 155 |
+
1.8604620695114136,
|
| 156 |
+
5.43813943862915,
|
| 157 |
+
7.142920017242432,
|
| 158 |
+
4.157068729400635,
|
| 159 |
+
5.672450065612793,
|
| 160 |
+
1.95806884765625,
|
| 161 |
+
1.5663840770721436,
|
| 162 |
+
5.278968811035156,
|
| 163 |
+
1.0,
|
| 164 |
+
1.0
|
| 165 |
+
],
|
| 166 |
+
"min": [
|
| 167 |
+
-7.552278995513916,
|
| 168 |
+
-0.36354875564575195,
|
| 169 |
+
-0.18577136099338531,
|
| 170 |
+
-1.956291913986206,
|
| 171 |
+
-1.6801013946533203,
|
| 172 |
+
-7.678869724273682,
|
| 173 |
+
-8.539926528930664,
|
| 174 |
+
-0.6294453740119934,
|
| 175 |
+
-0.07775841653347015,
|
| 176 |
+
-2.1328067779541016,
|
| 177 |
+
-2.1285502910614014,
|
| 178 |
+
-8.228925704956055,
|
| 179 |
+
0.0,
|
| 180 |
+
0.0
|
| 181 |
+
],
|
| 182 |
+
"q01": [
|
| 183 |
+
-7.334665780067444,
|
| 184 |
+
-5.257390398583084e-07,
|
| 185 |
+
-2.8215323254698887e-05,
|
| 186 |
+
-1.841589093208313,
|
| 187 |
+
-1.2851550579071045,
|
| 188 |
+
-6.268109321594238,
|
| 189 |
+
-5.912737417221069,
|
| 190 |
+
-0.572821855545044,
|
| 191 |
+
-2.81171942333458e-05,
|
| 192 |
+
-1.8467556238174438,
|
| 193 |
+
-1.440324923992157,
|
| 194 |
+
-5.8623305034637445,
|
| 195 |
+
0.0,
|
| 196 |
+
0.0
|
| 197 |
+
],
|
| 198 |
+
"q99": [
|
| 199 |
+
4.902149534225477,
|
| 200 |
+
3.4775466918945312,
|
| 201 |
+
4.224453401565551,
|
| 202 |
+
1.7064905214309687,
|
| 203 |
+
1.480757713317871,
|
| 204 |
+
3.609884884357428,
|
| 205 |
+
1.258314609527588,
|
| 206 |
+
2.967935800552368,
|
| 207 |
+
3.2566452693939207,
|
| 208 |
+
1.7433889162540437,
|
| 209 |
+
1.4066662788391113,
|
| 210 |
+
3.581243851184845,
|
| 211 |
+
1.0,
|
| 212 |
+
1.0
|
| 213 |
+
]
|
| 214 |
+
},
|
| 215 |
+
"num_transitions": 6075103,
|
| 216 |
+
"num_trajectories": 27500
|
| 217 |
+
}
|
| 218 |
+
}
|
settings/qwen_robotwin/uniform/external_resume_certificate.json
ADDED
|
@@ -0,0 +1,138 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"branch_resume_identity": {
|
| 3 |
+
"config_sha256": "8cf4970a4c7c919c72b15872ea42164783860fd4cecbdc75e30d1766ea349dd3",
|
| 4 |
+
"identity_schema_version": 2,
|
| 5 |
+
"manifest_sha256": "dcb728a00da4b4ad7f989fa313d6d3355e13f57efefc7267e9ea092f2ffd8a50",
|
| 6 |
+
"model_structure_sha256": "7809faa5245147f81b98f3dbd17c7a063a7cb07a21e18af588c748c5925e1407",
|
| 7 |
+
"proposal_alpha": null,
|
| 8 |
+
"proposal_sha256": null,
|
| 9 |
+
"seed": 484370,
|
| 10 |
+
"total_batch_size": 40
|
| 11 |
+
},
|
| 12 |
+
"completed_steps": 11000,
|
| 13 |
+
"optimizer_param_groups": [
|
| 14 |
+
{
|
| 15 |
+
"betas": [
|
| 16 |
+
0.9,
|
| 17 |
+
0.95
|
| 18 |
+
],
|
| 19 |
+
"eps": 1e-08,
|
| 20 |
+
"initial_lr": 1e-05,
|
| 21 |
+
"lr": 9.959920880962606e-06,
|
| 22 |
+
"name": "qwen_vl_interface",
|
| 23 |
+
"weight_decay": 1e-08
|
| 24 |
+
},
|
| 25 |
+
{
|
| 26 |
+
"betas": [
|
| 27 |
+
0.9,
|
| 28 |
+
0.95
|
| 29 |
+
],
|
| 30 |
+
"eps": 1e-08,
|
| 31 |
+
"initial_lr": 0.0001,
|
| 32 |
+
"lr": 9.959920880962605e-05,
|
| 33 |
+
"name": "action_model",
|
| 34 |
+
"weight_decay": 1e-08
|
| 35 |
+
}
|
| 36 |
+
],
|
| 37 |
+
"optimizer_state_entries": 8,
|
| 38 |
+
"parent_checkpoint": "/mnt/miaochengyu/starvla-infoflow-runtime/runs/qwen3vl-groot-robotwin-official150k/qwen3vl-groot-robotwin-official150k-adamw8bit-mb5-ga4-seed42-20260810/checkpoints/steps_11000_accelerate",
|
| 39 |
+
"parent_checkpoint_commit_sha256": "180f1ed2d2b7a9e6c158974fc9ec951182dfb629f6a2806808ac5639d491ce1b",
|
| 40 |
+
"parent_checkpoint_files": [
|
| 41 |
+
{
|
| 42 |
+
"bytes": 1525,
|
| 43 |
+
"name": "custom_checkpoint_0.pkl",
|
| 44 |
+
"sha256": "70788f8a2d7852a70fbb794ab9176d14ad37fb6d1b1d821a368572e1d8524390"
|
| 45 |
+
},
|
| 46 |
+
{
|
| 47 |
+
"bytes": 13,
|
| 48 |
+
"name": "latest",
|
| 49 |
+
"sha256": "68d0ca762047fd48fdd6a06a5e6465209289c028bfd63f5557a110ff1ebc15ac"
|
| 50 |
+
},
|
| 51 |
+
{
|
| 52 |
+
"bytes": 13867381301,
|
| 53 |
+
"name": "pytorch_model/bf16_zero_pp_rank_0_mp_rank_00_optim_states.pt",
|
| 54 |
+
"sha256": "42b6d10bb158b404056983ce88505c32738b29f3a062fe6835d24ce26c521a6a"
|
| 55 |
+
},
|
| 56 |
+
{
|
| 57 |
+
"bytes": 13867381301,
|
| 58 |
+
"name": "pytorch_model/bf16_zero_pp_rank_1_mp_rank_00_optim_states.pt",
|
| 59 |
+
"sha256": "7b0a1bbc92efffbf056bc40f40e93a94b67ddf3f375dc6ae263bcb048ba6ec63"
|
| 60 |
+
},
|
| 61 |
+
{
|
| 62 |
+
"bytes": 566626,
|
| 63 |
+
"name": "pytorch_model/zero_pp_rank_0_mp_rank_00_model_states.pt",
|
| 64 |
+
"sha256": "8a3ef88f9ab0df8197a836dca1d6e2416835f0b3d827cb10a78254b9a969ce7c"
|
| 65 |
+
},
|
| 66 |
+
{
|
| 67 |
+
"bytes": 566498,
|
| 68 |
+
"name": "pytorch_model/zero_pp_rank_1_mp_rank_00_model_states.pt",
|
| 69 |
+
"sha256": "67cd03fceb5ee4c2be6affe504f71a48ee10127d6ea3a1f00d05721895ffba53"
|
| 70 |
+
},
|
| 71 |
+
{
|
| 72 |
+
"bytes": 15017,
|
| 73 |
+
"name": "random_states_0.pkl",
|
| 74 |
+
"sha256": "636b1c5983ee2f8a5047025027453c9a478c0606389f2f6945b370f39cec9e65"
|
| 75 |
+
},
|
| 76 |
+
{
|
| 77 |
+
"bytes": 15081,
|
| 78 |
+
"name": "random_states_1.pkl",
|
| 79 |
+
"sha256": "416b3be53ff049935984cc65385d7dae3b0e5a82df66c9516f61ee9100c41961"
|
| 80 |
+
},
|
| 81 |
+
{
|
| 82 |
+
"bytes": 33270,
|
| 83 |
+
"name": "zero_to_fp32.py",
|
| 84 |
+
"sha256": "e8a2553b434ac043ec63b022db1e768277acf8d4e70f10a2842db6d238977135"
|
| 85 |
+
}
|
| 86 |
+
],
|
| 87 |
+
"parent_checkpoint_files_sha256": "73692dae620e3d4368fe513e8adff4fe54bdb25341b4293e34ee2a6286554f8d",
|
| 88 |
+
"parent_checkpoint_integrity_manifest_sha256": null,
|
| 89 |
+
"parent_config_sha256": "4882855d94076fc0ec19a0d511f14b1cbe82860c8b5df3dcc64c5c5f6c0d2321",
|
| 90 |
+
"parent_resume_identity": {
|
| 91 |
+
"config_sha256": "1a48ea51cf89ce72a7956aaa29f710bba0326c9b726e52795561b381507d26fa",
|
| 92 |
+
"identity_schema_version": 2,
|
| 93 |
+
"manifest_sha256": "dcb728a00da4b4ad7f989fa313d6d3355e13f57efefc7267e9ea092f2ffd8a50",
|
| 94 |
+
"model_structure_sha256": "7809faa5245147f81b98f3dbd17c7a063a7cb07a21e18af588c748c5925e1407",
|
| 95 |
+
"proposal_alpha": null,
|
| 96 |
+
"proposal_sha256": null,
|
| 97 |
+
"seed": 42,
|
| 98 |
+
"total_batch_size": 40
|
| 99 |
+
},
|
| 100 |
+
"proposal_sha256": null,
|
| 101 |
+
"rank_state_summaries": [
|
| 102 |
+
{
|
| 103 |
+
"optimizer_state_entries": 4,
|
| 104 |
+
"optimizer_state_sha256": "e1eea37514fa0478835597d1594b7a43b45cea9adc4ccb79a0de8fb9e10fe9f0",
|
| 105 |
+
"rank": 0,
|
| 106 |
+
"trainable_state_sha256": "ca2a19167d59aab693d0a0a1e17ea9adbf9cfd7367dfa7a5c90b04cf2183b717"
|
| 107 |
+
},
|
| 108 |
+
{
|
| 109 |
+
"optimizer_state_entries": 4,
|
| 110 |
+
"optimizer_state_sha256": "62c4e2908a7e97fe9a1d60642a85968081b55345493a11b9e7f68e49fcdb22f3",
|
| 111 |
+
"rank": 1,
|
| 112 |
+
"trainable_state_sha256": "ca2a19167d59aab693d0a0a1e17ea9adbf9cfd7367dfa7a5c90b04cf2183b717"
|
| 113 |
+
}
|
| 114 |
+
],
|
| 115 |
+
"repeated_diffusion_steps": 4,
|
| 116 |
+
"rng_streams": {
|
| 117 |
+
"model_base": 721927,
|
| 118 |
+
"sampler": 484370,
|
| 119 |
+
"transform": 289726
|
| 120 |
+
},
|
| 121 |
+
"scheduler": {
|
| 122 |
+
"class": "LambdaLR",
|
| 123 |
+
"last_epoch": 11000,
|
| 124 |
+
"last_lr": [
|
| 125 |
+
9.959920880962606e-06,
|
| 126 |
+
9.959920880962605e-05
|
| 127 |
+
],
|
| 128 |
+
"step_count": 11001
|
| 129 |
+
},
|
| 130 |
+
"scheduler_resume_mode": "accelerate_checkpoint",
|
| 131 |
+
"schema_version": 1,
|
| 132 |
+
"status": "exact_external_resume_loaded",
|
| 133 |
+
"total_batch_size": 40,
|
| 134 |
+
"trainable_parameter_elements": 32022798,
|
| 135 |
+
"trainable_parameter_sha256": "1bc504b22e4375be40bbbd8250be5b5f380c14d2fb47ed85deba3604a84b5484",
|
| 136 |
+
"trainable_parameter_tensors": 957,
|
| 137 |
+
"world_size": 2
|
| 138 |
+
}
|
settings/qwen_robotwin/uniform/initialization_manifest.json
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"schema_version": 1,
|
| 3 |
+
"initialization_seed": 484370,
|
| 4 |
+
"action_model_sha256": "1cab43a1e801a94110a20384a9ad6cd76a239fedd9fe17890662da7e22d47e62",
|
| 5 |
+
"action_model_parameter_elements": 160689934
|
| 6 |
+
}
|
settings/qwen_robotwin/uniform/valid_anchor_manifest.json
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2090f6b9d7e1bc5ef41897436690f031c187d625fa6b71ae0c54f3793660c757
|
| 3 |
+
size 107646168
|