Chengpengaaa commited on
Commit
4cfba01
·
verified ·
1 Parent(s): d6bec81

Upload folder using huggingface_hub

Browse files
qwenvl-eep0108/checkpoints/steps_5000_pytorch_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fb5525ac67a9e861feaa2d405abd626df367ac79ad4c8994690426e9ebdb76fb
3
+ size 6867598474
qwenvl-eep0108/config.json ADDED
@@ -0,0 +1,87 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "run_id": "QwenAdapter-0108-eep",
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": "qwenadapter-eep-0108",
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": 20,
23
+ "state_dim": 20,
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-0108-eep"
31
+ },
32
+ "datasets": {
33
+ "vla_data": {
34
+ "dataset_py": "lerobot_datasets",
35
+ "data_root_dir": "/data/LumosVLA/LumosVLA/data/0106_data_10_in_1",
36
+ "data_mix": "train_0107eep",
37
+ "per_device_batch_size": 32,
38
+ "num_workers": 4,
39
+ "action_type": "abs_ee",
40
+ "image_size": 224
41
+ }
42
+ },
43
+ "trainer": {
44
+ "epochs": 100,
45
+ "max_train_steps": 15000,
46
+ "num_warmup_steps": 1000,
47
+ "save_interval": 2500,
48
+ "eval_interval": 100000,
49
+ "learning_rate": {
50
+ "base": 5e-05,
51
+ "qwen_vl_interface": 1e-05,
52
+ "action_model": 0.0002
53
+ },
54
+ "lr_scheduler_type": "polynomial",
55
+ "scheduler_specific_kwargs": {
56
+ "lr_end": 1e-06,
57
+ "power": 1.0
58
+ },
59
+ "freeze_modules": "qwen_vl_interface",
60
+ "loss_scale": {
61
+ "vla": 1.0,
62
+ "vlm": 0.05
63
+ },
64
+ "max_grad_norm": 1.0,
65
+ "warmup_ratio": 0.1,
66
+ "weight_decay": 0.0,
67
+ "logging_frequency": 10,
68
+ "gradient_clipping": 1.0,
69
+ "gradient_accumulation_steps": 1,
70
+ "optimizer": {
71
+ "name": "AdamW",
72
+ "betas": [
73
+ 0.9,
74
+ 0.95
75
+ ],
76
+ "eps": 1e-08,
77
+ "weight_decay": 0
78
+ }
79
+ },
80
+ "is_resume": false,
81
+ "resume_epoch": null,
82
+ "resume_step": null,
83
+ "enable_gradient_checkpointing": true,
84
+ "enable_mixed_precision_training": true,
85
+ "wandb_run_name": "v1",
86
+ "output_dir": "/data/LumosVLA/LumosVLA/results/QwenAdapter-0108-eep"
87
+ }
qwenvl-eep0108/config.yaml ADDED
@@ -0,0 +1,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ run_id: QwenAdapter-0108-eep
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: qwenadapter-eep-0108
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: 20
20
+ state_dim: 20
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-0108-eep
27
+ datasets:
28
+ vla_data:
29
+ dataset_py: lerobot_datasets
30
+ data_root_dir: /data/LumosVLA/LumosVLA/data/0106_data_10_in_1
31
+ data_mix: train_0107eep
32
+ per_device_batch_size: 32
33
+ num_workers: 4
34
+ action_type: abs_ee
35
+ image_size: 224
36
+ trainer:
37
+ epochs: 100
38
+ max_train_steps: 15000
39
+ num_warmup_steps: 1000
40
+ save_interval: 2500
41
+ eval_interval: 100000
42
+ learning_rate:
43
+ base: 5.0e-05
44
+ qwen_vl_interface: 1.0e-05
45
+ action_model: 0.0002
46
+ lr_scheduler_type: polynomial
47
+ scheduler_specific_kwargs:
48
+ lr_end: 1.0e-06
49
+ power: 1.0
50
+ freeze_modules: qwen_vl_interface
51
+ loss_scale:
52
+ vla: 1.0
53
+ vlm: 0.05
54
+ max_grad_norm: 1.0
55
+ warmup_ratio: 0.1
56
+ weight_decay: 0.0
57
+ logging_frequency: 10
58
+ gradient_clipping: 1.0
59
+ gradient_accumulation_steps: 1
60
+ optimizer:
61
+ name: AdamW
62
+ betas:
63
+ - 0.9
64
+ - 0.95
65
+ eps: 1.0e-08
66
+ weight_decay: 0
67
+ is_resume: false
68
+ resume_epoch: null
69
+ resume_step: null
70
+ enable_gradient_checkpointing: true
71
+ enable_mixed_precision_training: true
72
+ wandb_run_name: v1
73
+ output_dir: /data/LumosVLA/LumosVLA/results/QwenAdapter-0108-eep
qwenvl-eep0108/dataset_statistics.json ADDED
@@ -0,0 +1,218 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "action": {
4
+ "mean": [
5
+ 0.23282504230737688,
6
+ -0.05165965789929033,
7
+ 0.21386985182762147,
8
+ -91.47970352172852,
9
+ 20.121428585052488,
10
+ 42.44750967025757,
11
+ 0.23724629282951357,
12
+ 0.08887637909501792,
13
+ 0.21939275562763214,
14
+ 45.390340900421144,
15
+ 29.264903831481938,
16
+ -119.05479431152344,
17
+ 0.03987310770899058,
18
+ 0.03885028753429652
19
+ ],
20
+ "std": [
21
+ 0.09953932969670672,
22
+ 0.1263747541518203,
23
+ 0.07284480907274637,
24
+ 133.6138685971796,
25
+ 18.383623797067543,
26
+ 155.94565103777856,
27
+ 0.10225355728400369,
28
+ 0.08295887480529306,
29
+ 0.06472772608178973,
30
+ 158.29364115016972,
31
+ 16.00381071103632,
32
+ 97.27639744542212,
33
+ 0.026943868199832993,
34
+ 0.02694874948904217
35
+ ],
36
+ "max": [
37
+ 0.5771363973617554,
38
+ 0.5499705076217651,
39
+ 1.1769417524337769,
40
+ 179.99998474121094,
41
+ 89.99874114990234,
42
+ 179.9999237060547,
43
+ 0.5673798322677612,
44
+ 0.5647112131118774,
45
+ 0.6844663023948669,
46
+ 179.99998474121094,
47
+ 89.99887084960938,
48
+ 179.99996948242188,
49
+ 0.05999999865889549,
50
+ 0.05999999865889549
51
+ ],
52
+ "min": [
53
+ -0.15116086602210999,
54
+ -0.5200218558311462,
55
+ 0.050602782517671585,
56
+ -180.0,
57
+ -20.672367095947266,
58
+ -179.99993896484375,
59
+ -0.0248020701110363,
60
+ -0.29858818650245667,
61
+ 0.06846695393323898,
62
+ -179.99998474121094,
63
+ -13.50222110748291,
64
+ -179.9997100830078,
65
+ 0.0,
66
+ 0.0
67
+ ],
68
+ "q01": [
69
+ 0.05501473695039749,
70
+ -0.3686788082122803,
71
+ 0.0868777185678482,
72
+ -179.78305053710938,
73
+ -9.416550636291504,
74
+ -179.69680786132812,
75
+ 0.056286852806806564,
76
+ -0.18694551289081573,
77
+ 0.10547168552875519,
78
+ -179.8378448486328,
79
+ -0.19658219814300537,
80
+ -179.51170349121094,
81
+ 0.0,
82
+ 0.0
83
+ ],
84
+ "q99": [
85
+ 0.47519516944885254,
86
+ 0.4694986939430237,
87
+ 0.46994128823280334,
88
+ 179.755126953125,
89
+ 89.57238006591797,
90
+ 179.736328125,
91
+ 0.48799216747283936,
92
+ 0.3558615446090698,
93
+ 0.43878117203712463,
94
+ 179.83042907714844,
95
+ 89.67033386230469,
96
+ 179.49917602539062,
97
+ 0.05999999865889549,
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
+ false,
114
+ false
115
+ ]
116
+ },
117
+ "state": {
118
+ "mean": [
119
+ 0.24175260961055758,
120
+ -0.049470676161581656,
121
+ 0.21608835756778721,
122
+ -1.413518315553665,
123
+ 0.428703111410141,
124
+ 0.7577013611793519,
125
+ 0.24074497371912004,
126
+ 0.08981470651924611,
127
+ 0.21937729418277743,
128
+ 0.8831917621195315,
129
+ 0.6019748389720918,
130
+ -2.0126270771026613,
131
+ 0.04148903861641884,
132
+ 0.03994188271462917
133
+ ],
134
+ "std": [
135
+ 0.09280708088090048,
136
+ 0.12742504567945281,
137
+ 0.0744079447762679,
138
+ 2.5498722585501863,
139
+ 0.3793943873796246,
140
+ 2.808145353959299,
141
+ 0.09863927576384764,
142
+ 0.0815411369762193,
143
+ 0.06492154445492224,
144
+ 2.805201274470533,
145
+ 0.31938597693868265,
146
+ 1.884995382965943,
147
+ 0.025358022526895105,
148
+ 0.026140522839142637
149
+ ],
150
+ "max": [
151
+ 0.5393329858779907,
152
+ 0.48442599177360535,
153
+ 0.6967939734458923,
154
+ 3.1415927410125732,
155
+ 1.5707963705062866,
156
+ 3.1415927410125732,
157
+ 0.5489550232887268,
158
+ 0.44545501470565796,
159
+ 0.6456500291824341,
160
+ 3.1415927410125732,
161
+ 1.5707963705062866,
162
+ 3.1415927410125732,
163
+ 0.06159999966621399,
164
+ 0.05998999997973442
165
+ ],
166
+ "min": [
167
+ -0.10788500308990479,
168
+ -0.5191159844398499,
169
+ 0.05315700173377991,
170
+ -3.1410515308380127,
171
+ -1.5302174091339111,
172
+ -3.1410515308380127,
173
+ -0.02319999970495701,
174
+ -0.26581600308418274,
175
+ 0.06903599947690964,
176
+ -3.1410515308380127,
177
+ -0.23581144213676453,
178
+ -3.1410515308380127,
179
+ 0.00014000000373926014,
180
+ 7.000000186963007e-05
181
+ ],
182
+ "q01": [
183
+ 0.056127000600099564,
184
+ -0.36912801861763,
185
+ 0.08751751482486725,
186
+ -3.137159585952759,
187
+ -0.16739033162593842,
188
+ -3.1360411643981934,
189
+ 0.056008998304605484,
190
+ -0.15914200246334076,
191
+ 0.10651859641075134,
192
+ -3.138237953186035,
193
+ -0.005713686812669039,
194
+ -3.131770372390747,
195
+ 0.0004199999966658652,
196
+ 7.000000186963007e-05
197
+ ],
198
+ "q99": [
199
+ 0.4720889925956726,
200
+ 0.4121397137641907,
201
+ 0.4779260456562042,
202
+ 3.1415927410125732,
203
+ 1.5365502834320068,
204
+ 3.1415927410125732,
205
+ 0.4841437041759491,
206
+ 0.35178446769714355,
207
+ 0.42531776428222656,
208
+ 3.1415927410125732,
209
+ 1.5435858964920044,
210
+ 3.1415927410125732,
211
+ 0.059780001640319824,
212
+ 0.05992000177502632
213
+ ]
214
+ },
215
+ "num_transitions": 885334,
216
+ "num_trajectories": 881
217
+ }
218
+ }