Chengpengaaa commited on
Commit
b9be06d
·
verified ·
1 Parent(s): a24e24d

Upload folder using huggingface_hub

Browse files
qwen-all-v1/checkpoints/steps_15000_pytorch_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:946a5f11bc230fb47461305bd701bf4db372ba750d03f3659b026a753d0c1464
3
+ size 6817169669
qwen-all-v1/config.json ADDED
@@ -0,0 +1,88 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "run_id": "QwenAdapter-0106",
3
+ "run_root_dir": "/data/LumosVLA/LumosVLA/results",
4
+ "seed": 42,
5
+ "trackers": [
6
+ "jsonl",
7
+ "wandb"
8
+ ],
9
+ "wandb_entity": "xspadex-nanyang-technological-university-singapore",
10
+ "wandb_project": "starvla-groot-2.5vl-3B",
11
+ "is_debug": false,
12
+ "framework": {
13
+ "name": "QwenAdapter",
14
+ "qwenvl": {
15
+ "base_vlm": "/data/LumosVLA/LumosVLA/playground/Pretrained_models/hub/models--Qwen--Qwen3-VL-2B-Instruct/snapshots/89644892e4d85e24eaac8bacfd4f463576704203",
16
+ "attn_implementation": "flash_attention_2",
17
+ "vl_hidden_dim": 2048
18
+ },
19
+ "action_model": {
20
+ "action_model_type": "VLA-Adapter",
21
+ "hidden_dim": 2048,
22
+ "action_dim": 14,
23
+ "state_dim": 14,
24
+ "action_query_num": 64,
25
+ "use_pro_version": true,
26
+ "use_proprio": true,
27
+ "phase": "Training",
28
+ "num_actions_chunk": 30
29
+ },
30
+ "framework_py": "QwenAdapter-0106"
31
+ },
32
+ "datasets": {
33
+ "vla_data": {
34
+ "dataset_py": "lerobot_datasets",
35
+ "data_root_dir": "/data/LumosVLA/LumosVLA/data/1230_data_7_in_1",
36
+ "data_mix": "train_1230",
37
+ "per_device_batch_size": 16,
38
+ "num_workers": 8,
39
+ "action_type": "abs_ee",
40
+ "image_size": [
41
+ 224,
42
+ 224
43
+ ]
44
+ }
45
+ },
46
+ "trainer": {
47
+ "epochs": 100,
48
+ "max_train_steps": 50000,
49
+ "num_warmup_steps": 1000,
50
+ "save_interval": 5000,
51
+ "eval_interval": 10000,
52
+ "learning_rate": {
53
+ "base": 2.5e-05,
54
+ "qwen_vl_interface": 1e-05,
55
+ "action_model": 0.0001
56
+ },
57
+ "lr_scheduler_type": "cosine_with_min_lr",
58
+ "scheduler_specific_kwargs": {
59
+ "min_lr": 1e-06
60
+ },
61
+ "freeze_modules": "qwen_vl_interface",
62
+ "loss_scale": {
63
+ "vla": 1.0,
64
+ "vlm": 0.05
65
+ },
66
+ "max_grad_norm": 1.0,
67
+ "warmup_ratio": 0.1,
68
+ "weight_decay": 0.0,
69
+ "logging_frequency": 10,
70
+ "gradient_clipping": 1.0,
71
+ "gradient_accumulation_steps": 1,
72
+ "optimizer": {
73
+ "name": "AdamW",
74
+ "betas": [
75
+ 0.9,
76
+ 0.95
77
+ ],
78
+ "eps": 1e-08,
79
+ "weight_decay": 1e-08
80
+ },
81
+ "is_resume": false,
82
+ "resume_epoch": null,
83
+ "resume_step": null,
84
+ "enable_gradient_checkpointing": true,
85
+ "enable_mixed_precision_training": true
86
+ },
87
+ "output_dir": "/data/LumosVLA/LumosVLA/results/QwenAdapter-0106"
88
+ }
qwen-all-v1/config.yaml ADDED
@@ -0,0 +1,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ run_id: QwenAdapter-0106
2
+ run_root_dir: /data/LumosVLA/LumosVLA/results
3
+ seed: 42
4
+ trackers:
5
+ - jsonl
6
+ - wandb
7
+ wandb_entity: xspadex-nanyang-technological-university-singapore
8
+ wandb_project: starvla-groot-2.5vl-3B
9
+ is_debug: false
10
+ framework:
11
+ name: QwenAdapter
12
+ qwenvl:
13
+ base_vlm: /data/LumosVLA/LumosVLA/playground/Pretrained_models/hub/models--Qwen--Qwen3-VL-2B-Instruct/snapshots/89644892e4d85e24eaac8bacfd4f463576704203
14
+ attn_implementation: flash_attention_2
15
+ vl_hidden_dim: 2048
16
+ action_model:
17
+ action_model_type: VLA-Adapter
18
+ hidden_dim: 2048
19
+ action_dim: 14
20
+ state_dim: 14
21
+ action_query_num: 64
22
+ use_pro_version: true
23
+ use_proprio: true
24
+ phase: Training
25
+ num_actions_chunk: 30
26
+ framework_py: QwenAdapter-0106
27
+ datasets:
28
+ vla_data:
29
+ dataset_py: lerobot_datasets
30
+ data_root_dir: /data/LumosVLA/LumosVLA/data/1230_data_7_in_1
31
+ data_mix: train_1230
32
+ per_device_batch_size: 16
33
+ num_workers: 8
34
+ action_type: abs_ee
35
+ image_size:
36
+ - 224
37
+ - 224
38
+ trainer:
39
+ epochs: 100
40
+ max_train_steps: 50000
41
+ num_warmup_steps: 1000
42
+ save_interval: 5000
43
+ eval_interval: 10000
44
+ learning_rate:
45
+ base: 2.5e-05
46
+ qwen_vl_interface: 1.0e-05
47
+ action_model: 0.0001
48
+ lr_scheduler_type: cosine_with_min_lr
49
+ scheduler_specific_kwargs:
50
+ min_lr: 1.0e-06
51
+ freeze_modules: qwen_vl_interface
52
+ loss_scale:
53
+ vla: 1.0
54
+ vlm: 0.05
55
+ max_grad_norm: 1.0
56
+ warmup_ratio: 0.1
57
+ weight_decay: 0.0
58
+ logging_frequency: 10
59
+ gradient_clipping: 1.0
60
+ gradient_accumulation_steps: 1
61
+ optimizer:
62
+ name: AdamW
63
+ betas:
64
+ - 0.9
65
+ - 0.95
66
+ eps: 1.0e-08
67
+ weight_decay: 1.0e-08
68
+ is_resume: false
69
+ resume_epoch: null
70
+ resume_step: null
71
+ enable_gradient_checkpointing: true
72
+ enable_mixed_precision_training: true
73
+ output_dir: /data/LumosVLA/LumosVLA/results/QwenAdapter-0106
qwen-all-v1/dataset_statistics.json ADDED
@@ -0,0 +1,218 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ -0.16321200245459166,
6
+ 1.4521609033857077,
7
+ -0.9421584350722177,
8
+ -0.17235713984285084,
9
+ 0.6999681762286597,
10
+ 0.019915780850819187,
11
+ 0.03984901175967285,
12
+ 0.24289310191358843,
13
+ 1.3446637051446098,
14
+ -0.8462631191526142,
15
+ 0.14655291927712305,
16
+ 0.6068966857024602,
17
+ -0.1327239607593843,
18
+ 0.03956502383308752
19
+ ],
20
+ "std": [
21
+ 0.4805919514279753,
22
+ 0.5138003856380448,
23
+ 0.41102662911826815,
24
+ 0.3577144366234307,
25
+ 0.3016522331624895,
26
+ 0.5490499483479746,
27
+ 0.02682619089760716,
28
+ 0.3654577319930143,
29
+ 0.5116360367226789,
30
+ 0.3962255526627343,
31
+ 0.4488282767408623,
32
+ 0.2948004042563088,
33
+ 0.6312928416021599,
34
+ 0.026576370264582067
35
+ ],
36
+ "max": [
37
+ 1.5166736841201782,
38
+ 2.873859167098999,
39
+ 0.03595378249883652,
40
+ 1.7434443235397339,
41
+ 1.2570908069610596,
42
+ 3.012822151184082,
43
+ 0.05999999865889549,
44
+ 1.8831404447555542,
45
+ 2.8448691368103027,
46
+ 0.01151917316019535,
47
+ 1.744596242904663,
48
+ 1.2279962301254272,
49
+ 2.137557029724121,
50
+ 0.05999999865889549
51
+ ],
52
+ "min": [
53
+ -1.8999130725860596,
54
+ -0.005323254037648439,
55
+ -2.8321456909179688,
56
+ -1.747144341468811,
57
+ -1.2225507497787476,
58
+ -2.125462055206299,
59
+ 0.0,
60
+ -1.2692208290100098,
61
+ -0.0023387412074953318,
62
+ -2.831395387649536,
63
+ -1.7527817487716675,
64
+ -1.0821739435195923,
65
+ -2.1614506244659424,
66
+ 0.0
67
+ ],
68
+ "q01": [
69
+ -1.1235190629959106,
70
+ 0.0,
71
+ -2.0828940868377686,
72
+ -1.2669345140457153,
73
+ -0.5976105332374573,
74
+ -1.2526793479919434,
75
+ 0.0,
76
+ -0.9868741035461426,
77
+ -0.0012915435945615172,
78
+ -2.087623357772827,
79
+ -1.3179678916931152,
80
+ 0.0,
81
+ -1.8024719953536987,
82
+ 0.0
83
+ ],
84
+ "q99": [
85
+ 1.2632519006729126,
86
+ 2.4042139053344727,
87
+ 0.0,
88
+ 1.0156922340393066,
89
+ 1.2199327945709229,
90
+ 2.035996437072754,
91
+ 0.05999999865889549,
92
+ 1.1853927373886108,
93
+ 2.40316104888916,
94
+ 0.00062831852119416,
95
+ 1.2153974771499634,
96
+ 1.21909499168396,
97
+ 1.6403651237487793,
98
+ 0.05999999865889549
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
+ true,
114
+ true
115
+ ]
116
+ },
117
+ "state": {
118
+ "mean": [
119
+ -0.16321200245459166,
120
+ 1.4521609033857077,
121
+ -0.9421584350722177,
122
+ -0.17235713984285084,
123
+ 0.6999681762286597,
124
+ 0.019915780850819187,
125
+ 0.041679546769176214,
126
+ 0.24289310191358843,
127
+ 1.3446637051446098,
128
+ -0.8462631191526142,
129
+ 0.14655291927712305,
130
+ 0.6068966857024602,
131
+ -0.1327239607593843,
132
+ 0.04096942687673229
133
+ ],
134
+ "std": [
135
+ 0.4805919514279753,
136
+ 0.5138003856380448,
137
+ 0.41102662911826815,
138
+ 0.3577144366234307,
139
+ 0.3016522331624895,
140
+ 0.5490499483479746,
141
+ 0.025123654324128476,
142
+ 0.3654577319930143,
143
+ 0.5116360367226789,
144
+ 0.3962255526627343,
145
+ 0.4488282767408623,
146
+ 0.2948004042563088,
147
+ 0.6312928416021599,
148
+ 0.02559506757389033
149
+ ],
150
+ "max": [
151
+ 1.5166736841201782,
152
+ 2.873859167098999,
153
+ 0.03595378249883652,
154
+ 1.7434443235397339,
155
+ 1.2570908069610596,
156
+ 3.012822151184082,
157
+ 0.06159999966621399,
158
+ 1.8831404447555542,
159
+ 2.8448691368103027,
160
+ 0.01151917316019535,
161
+ 1.744596242904663,
162
+ 1.2279962301254272,
163
+ 2.137557029724121,
164
+ 0.05998999997973442
165
+ ],
166
+ "min": [
167
+ -1.8999130725860596,
168
+ -0.005323254037648439,
169
+ -2.8321456909179688,
170
+ -1.747144341468811,
171
+ -1.2225507497787476,
172
+ -2.125462055206299,
173
+ 0.00014000000373926014,
174
+ -1.2692208290100098,
175
+ -0.0023387412074953318,
176
+ -2.831395387649536,
177
+ -1.7527817487716675,
178
+ -1.0821739435195923,
179
+ -2.1614506244659424,
180
+ 7.000000186963007e-05
181
+ ],
182
+ "q01": [
183
+ -1.1235190629959106,
184
+ 0.0,
185
+ -2.0828940868377686,
186
+ -1.2669345140457153,
187
+ -0.5976105332374573,
188
+ -1.2526793479919434,
189
+ 0.0006300000241026282,
190
+ -0.9868741035461426,
191
+ -0.0012915435945615172,
192
+ -2.087623357772827,
193
+ -1.3179678916931152,
194
+ 0.0,
195
+ -1.8024719953536987,
196
+ 7.000000186963007e-05
197
+ ],
198
+ "q99": [
199
+ 1.2632519006729126,
200
+ 2.4042139053344727,
201
+ 0.0,
202
+ 1.0156922340393066,
203
+ 1.2199327945709229,
204
+ 2.035996437072754,
205
+ 0.059780001640319824,
206
+ 1.1853927373886108,
207
+ 2.40316104888916,
208
+ 0.00062831852119416,
209
+ 1.2153974771499634,
210
+ 1.21909499168396,
211
+ 1.6403651237487793,
212
+ 0.05992000177502632
213
+ ]
214
+ },
215
+ "num_transitions": 691216,
216
+ "num_trajectories": 707
217
+ }
218
+ }