zixianma02 commited on
Commit
679ebf3
·
verified ·
1 Parent(s): 2f406c5

Add files using upload-large-folder tool

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. .gitattributes +33 -0
  2. config.yaml +629 -0
  3. model_and_optim/__11_0.distcp +3 -0
  4. model_and_optim/__11_5.distcp +3 -0
  5. model_and_optim/__12_0.distcp +3 -0
  6. model_and_optim/__13_1.distcp +3 -0
  7. model_and_optim/__13_14.distcp +3 -0
  8. model_and_optim/__14_14.distcp +3 -0
  9. model_and_optim/__14_6.distcp +3 -0
  10. model_and_optim/__15_3.distcp +3 -0
  11. model_and_optim/__15_8.distcp +3 -0
  12. model_and_optim/__1_12.distcp +3 -0
  13. model_and_optim/__1_14.distcp +3 -0
  14. model_and_optim/__2_0.distcp +3 -0
  15. model_and_optim/__2_2.distcp +3 -0
  16. model_and_optim/__2_8.distcp +3 -0
  17. model_and_optim/__3_13.distcp +3 -0
  18. model_and_optim/__3_14.distcp +3 -0
  19. model_and_optim/__3_6.distcp +3 -0
  20. model_and_optim/__4_2.distcp +3 -0
  21. model_and_optim/__4_9.distcp +3 -0
  22. model_and_optim/__5_1.distcp +3 -0
  23. model_and_optim/__5_2.distcp +3 -0
  24. model_and_optim/__5_3.distcp +3 -0
  25. model_and_optim/__6_7.distcp +3 -0
  26. model_and_optim/__6_8.distcp +3 -0
  27. model_and_optim/__7_1.distcp +3 -0
  28. model_and_optim/__7_13.distcp +3 -0
  29. model_and_optim/__7_7.distcp +3 -0
  30. model_and_optim/__8_1.distcp +3 -0
  31. model_and_optim/__8_6.distcp +3 -0
  32. model_and_optim/__9_1.distcp +3 -0
  33. model_and_optim/__9_14.distcp +3 -0
  34. model_and_optim/__9_3.distcp +3 -0
  35. model_and_optim/__9_9.distcp +3 -0
  36. train/rank0.pt +3 -0
  37. train/rank1.pt +3 -0
  38. train/rank10.pt +3 -0
  39. train/rank11.pt +3 -0
  40. train/rank12.pt +3 -0
  41. train/rank13.pt +3 -0
  42. train/rank14.pt +3 -0
  43. train/rank15.pt +3 -0
  44. train/rank2.pt +3 -0
  45. train/rank3.pt +3 -0
  46. train/rank4.pt +3 -0
  47. train/rank5.pt +3 -0
  48. train/rank6.pt +3 -0
  49. train/rank7.pt +3 -0
  50. train/rank8.pt +3 -0
.gitattributes CHANGED
@@ -33,3 +33,36 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ model_and_optim/__3_13.distcp filter=lfs diff=lfs merge=lfs -text
37
+ model_and_optim/__7_1.distcp filter=lfs diff=lfs merge=lfs -text
38
+ model_and_optim/__8_1.distcp filter=lfs diff=lfs merge=lfs -text
39
+ model_and_optim/__4_9.distcp filter=lfs diff=lfs merge=lfs -text
40
+ model_and_optim/__14_6.distcp filter=lfs diff=lfs merge=lfs -text
41
+ model_and_optim/__2_2.distcp filter=lfs diff=lfs merge=lfs -text
42
+ model_and_optim/__2_0.distcp filter=lfs diff=lfs merge=lfs -text
43
+ model_and_optim/__3_14.distcp filter=lfs diff=lfs merge=lfs -text
44
+ model_and_optim/__12_0.distcp filter=lfs diff=lfs merge=lfs -text
45
+ model_and_optim/__5_2.distcp filter=lfs diff=lfs merge=lfs -text
46
+ model_and_optim/__1_12.distcp filter=lfs diff=lfs merge=lfs -text
47
+ model_and_optim/__15_8.distcp filter=lfs diff=lfs merge=lfs -text
48
+ model_and_optim/__6_7.distcp filter=lfs diff=lfs merge=lfs -text
49
+ model_and_optim/__15_3.distcp filter=lfs diff=lfs merge=lfs -text
50
+ model_and_optim/__9_9.distcp filter=lfs diff=lfs merge=lfs -text
51
+ model_and_optim/__1_14.distcp filter=lfs diff=lfs merge=lfs -text
52
+ model_and_optim/__11_0.distcp filter=lfs diff=lfs merge=lfs -text
53
+ model_and_optim/__14_14.distcp filter=lfs diff=lfs merge=lfs -text
54
+ model_and_optim/__7_7.distcp filter=lfs diff=lfs merge=lfs -text
55
+ model_and_optim/__5_1.distcp filter=lfs diff=lfs merge=lfs -text
56
+ model_and_optim/__11_5.distcp filter=lfs diff=lfs merge=lfs -text
57
+ model_and_optim/__5_3.distcp filter=lfs diff=lfs merge=lfs -text
58
+ model_and_optim/__8_6.distcp filter=lfs diff=lfs merge=lfs -text
59
+ model_and_optim/__13_1.distcp filter=lfs diff=lfs merge=lfs -text
60
+ model_and_optim/__9_14.distcp filter=lfs diff=lfs merge=lfs -text
61
+ model_and_optim/__3_6.distcp filter=lfs diff=lfs merge=lfs -text
62
+ model_and_optim/__7_13.distcp filter=lfs diff=lfs merge=lfs -text
63
+ model_and_optim/__6_8.distcp filter=lfs diff=lfs merge=lfs -text
64
+ model_and_optim/__2_8.distcp filter=lfs diff=lfs merge=lfs -text
65
+ model_and_optim/__13_14.distcp filter=lfs diff=lfs merge=lfs -text
66
+ model_and_optim/__4_2.distcp filter=lfs diff=lfs merge=lfs -text
67
+ model_and_optim/__9_3.distcp filter=lfs diff=lfs merge=lfs -text
68
+ model_and_optim/__9_1.distcp filter=lfs diff=lfs merge=lfs -text
config.yaml ADDED
@@ -0,0 +1,629 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ run_name: multitask_train
2
+ model:
3
+ model_name: molmo
4
+ llm:
5
+ d_model: 2560
6
+ n_heads: 32
7
+ n_kv_heads: 8
8
+ head_dim: 128
9
+ qkv_bias: false
10
+ clip_qkv: null
11
+ n_layers: 36
12
+ mlp_ratio: 4
13
+ mlp_hidden_size: 19456
14
+ activation_type: swiglu
15
+ block_type: sequential
16
+ rope: true
17
+ rope_full_precision: true
18
+ rope_theta: 1000000.0
19
+ rope_type: default
20
+ rope_factor: null
21
+ rope_high_freq_factor: null
22
+ rope_low_freq_factor: null
23
+ rope_original_max_position_embeddings: null
24
+ attention_type: sdpa
25
+ float32_attention: true
26
+ attention_dropout: 0.0
27
+ attention_layer_norm: true
28
+ attention_layer_norm_type: qwen3
29
+ residual_dropout: 0.1
30
+ response_residual_dropout: 0.0
31
+ layer_norm_type: rms
32
+ layer_norm_with_affine: true
33
+ layer_norm_eps: 1.0e-06
34
+ attention_layer_norm_with_affine: true
35
+ max_sequence_length: 4096
36
+ max_position_embeddings: null
37
+ include_bias: false
38
+ bias_for_layer_norm: null
39
+ norm_after: false
40
+ moe_num_experts: 8
41
+ moe_top_k: 2
42
+ moe_mlp_impl: sparse
43
+ moe_log_expert_assignment: false
44
+ moe_shared_expert: false
45
+ moe_lbl_in_fp32: false
46
+ moe_interleave: false
47
+ moe_loss_weight: 0.1
48
+ moe_zloss_weight: null
49
+ moe_dropless: true
50
+ moe_capacity_factor: 1.25
51
+ embedding_dropout: 0.0
52
+ scale_logits: false
53
+ vocab_size: 151936
54
+ additional_vocab_size: 128
55
+ weight_tying: true
56
+ embedding_size: 151936
57
+ use_position_ids: true
58
+ tokenizer:
59
+ identifier: Qwen/Qwen3-4B
60
+ tokenizer_dir: null
61
+ init_path: /weka/oe-training-default/mm-olmo/pretrained_llms/qwen3-4b.pt
62
+ init_incremental: null
63
+ new_embedding_init_range: 0.02
64
+ initializer_range: 0.02
65
+ normalize_input_embeds: false
66
+ activation_checkpoint: whole_layer
67
+ compile: blocks
68
+ fix_pad_tokenizer: false
69
+ init_std: 0.02
70
+ init_fn: normal
71
+ init_cutoff_factor: null
72
+ vision_backbone:
73
+ vit:
74
+ image_model_type: siglip
75
+ image_default_input_size:
76
+ - 378
77
+ - 378
78
+ image_patch_size: 14
79
+ image_pos_patch_size: 14
80
+ image_emb_dim: 1152
81
+ image_num_heads: 16
82
+ image_num_key_value_heads: 16
83
+ image_num_layers: 27
84
+ image_head_dim: 72
85
+ image_mlp_dim: 4304
86
+ image_mlp_activations: gelu_pytorch_tanh
87
+ image_dropout_rate: 0.0
88
+ image_num_pos: 729
89
+ image_norm_eps: 1.0e-06
90
+ attention_dropout: 0.0
91
+ residual_dropout: 0.0
92
+ initializer_range: 0.02
93
+ float32_attention: true
94
+ attention_type: sdpa
95
+ activation_checkpointing: true
96
+ init_path: /weka/oe-training-default/mm-olmo/pretrained_image_encoders/siglip2-so400m-14-384.pt
97
+ resize_mode: siglip
98
+ pad_value: 0.0
99
+ normalize: siglip
100
+ image_pooling_2d: attention_meanq
101
+ pooling_attention_mask: false
102
+ image_projector: mlp
103
+ image_padding_embed: null
104
+ vit_layers:
105
+ - -3
106
+ - -9
107
+ skip_unused_layers: true
108
+ image_feature_dropout: 0.0
109
+ connector_activation_checkpointing: true
110
+ compile_vit: blocks
111
+ compile_connector: dynamic
112
+ normalize_on_gpu: false
113
+ data_formatter:
114
+ prompt_templates: uber_model
115
+ message_format: role
116
+ system_prompt: demo_or_style
117
+ always_start_with_space: false
118
+ default_inference_len: 65
119
+ select_answer: best
120
+ debug: false
121
+ image_last: false
122
+ format_message_list: null
123
+ p_one_message: 0.0
124
+ eval_system_prompt_mapping: null
125
+ timestamp_mode: 50-percent-seconds
126
+ p_choice_content_in_mc: 1.0
127
+ mm_preprocessor:
128
+ crop_mode: overlap-and-resize-c2
129
+ use_col_tokens: true
130
+ max_crops: 8
131
+ pooling_w: 2
132
+ pooling_h: 2
133
+ overlap_margins:
134
+ - 4
135
+ - 4
136
+ max_images: null
137
+ max_multi_image_crops: 4
138
+ max_answer_len: null
139
+ last_message_loss_only: false
140
+ loss_token_weighting: root_subsegments
141
+ max_text_tokens: null
142
+ image_padding_mask: false
143
+ legacy_image_mask: false
144
+ bi_directional_attn: null
145
+ seed: 6198
146
+ epoch: null
147
+ dry_run: false
148
+ ft_llm: true
149
+ ft_vit: true
150
+ ft_connector: true
151
+ ft_embedding: lm_head
152
+ optimizer:
153
+ name: adamw
154
+ learning_rate: 0.0001
155
+ weight_decay: 0.01
156
+ betas:
157
+ - 0.9
158
+ - 0.95
159
+ eps: 1.0e-05
160
+ connector_learning_rate: 5.0e-06
161
+ vit_learning_rate: 5.0e-06
162
+ llm_learning_rate: 1.0e-05
163
+ frame_selector_learning_rate: 0.0001
164
+ temporal_token_scorer_learning_rate: 0.0001
165
+ connector_weight_decay: 0.0
166
+ vit_weight_decay: 0.0
167
+ llm_weight_decay: 0.0
168
+ frame_selector_weight_decay: 0.01
169
+ temporal_token_scorer_weight_decay: 0.01
170
+ connector_betas:
171
+ - 0.9
172
+ - 0.95
173
+ vit_betas:
174
+ - 0.9
175
+ - 0.95
176
+ llm_betas:
177
+ - 0.9
178
+ - 0.95
179
+ frame_selector_betas:
180
+ - 0.9
181
+ - 0.95
182
+ temporal_token_scorer_betas:
183
+ - 0.9
184
+ - 0.95
185
+ connector_eps: 1.0e-06
186
+ vit_eps: 1.0e-06
187
+ llm_eps: 1.0e-06
188
+ frame_selector_eps: 1.0e-06
189
+ temporal_token_scorer_eps: 1.0e-06
190
+ metrics_log_interval: -1
191
+ scheduler:
192
+ name: multimodal
193
+ units: steps
194
+ t_warmup: 100
195
+ t_max: null
196
+ alpha_f: 0.1
197
+ connector_t_warmup: 200
198
+ vit_t_warmup: 200
199
+ llm_t_warmup: 200
200
+ frame_selector_t_warmup: 200
201
+ temporal_token_scorer_t_warmup: 200
202
+ grad_clip_warmup_steps: null
203
+ grad_clip_warmup_factor: null
204
+ warmup_min_lr: 0.0
205
+ data:
206
+ dataset: null
207
+ mixture: null
208
+ root_size_mixture:
209
+ - rate: 0.15
210
+ mixture:
211
+ pixmo_ask_model_anything: null
212
+ pixmo_cap: 50000.0
213
+ pixmo_cap_qa_as_user_qa: null
214
+ pixmo_pointing_explanations: null
215
+ - rate: 0.5
216
+ mixture:
217
+ coco_2014_vqa_multi: null
218
+ text_vqa: null
219
+ okvqa: null
220
+ chart_qa_weighted: null
221
+ doc_qa: null
222
+ info_qa: null
223
+ ai2_diagram_v2_mix_transparent: null
224
+ a_okvqa_mc: null
225
+ a_okvqa_da: null
226
+ android_control: null
227
+ science_qa_img: null
228
+ tabwmp_da: null
229
+ st_qa: null
230
+ tally_qa: null
231
+ pixmo_clocks: 250000.0
232
+ dv_qa: 10000.0
233
+ figure_qa: 10000.0
234
+ plot_qa: 20000.0
235
+ cosyn_chart_exp: null
236
+ cosyn_chemical_exp: null
237
+ cosyn_diagram_exp: null
238
+ cosyn_document: null
239
+ cosyn_math_exp: null
240
+ cosyn_music_exp: null
241
+ cosyn_table_exp: null
242
+ - rate: 0.35
243
+ mixture:
244
+ pixmo_points_train: null
245
+ pixmo_count_train: null
246
+ pixmo_points_high_freq_train: null
247
+ cosyn_point: null
248
+ kwargs_mixture: null
249
+ split: train
250
+ seed: 50189
251
+ pad: to_max
252
+ sequence_length: 2304
253
+ max_text_seq_len: null
254
+ shuffle: true
255
+ start_index: 0
256
+ packing: null
257
+ num_workers: 2
258
+ drop_last: true
259
+ pin_memory: true
260
+ prefetch_factor: null
261
+ persistent_workers: false
262
+ timeout: 0
263
+ restore_dataloader: true
264
+ fast_forward_batches: null
265
+ evaluators: []
266
+ eval_interval: 2000
267
+ inf_evaluators:
268
+ - label: chart_qa_exp
269
+ data:
270
+ dataset: chart_qa_exp
271
+ mixture: null
272
+ root_size_mixture: null
273
+ kwargs_mixture: null
274
+ split: validation
275
+ seed: 691203
276
+ pad: to_max
277
+ sequence_length: 1792
278
+ max_text_seq_len: null
279
+ shuffle: true
280
+ start_index: 0
281
+ packing: null
282
+ num_workers: 2
283
+ drop_last: true
284
+ pin_memory: true
285
+ prefetch_factor: null
286
+ persistent_workers: true
287
+ timeout: 0
288
+ evaluator:
289
+ n_to_log: 0
290
+ num_wandb_examples: 32
291
+ save_predictions: null
292
+ save_tokens: false
293
+ vqa_eval: relaxed_correctness,scifi_relaxed_correctness,em
294
+ pointing_eval: false
295
+ count_eval: false
296
+ point_count_eval: false
297
+ android_eval: false
298
+ clock_eval: false
299
+ clock_bench_eval: false
300
+ math_vista_eval: false
301
+ temp_compass_eval: ''
302
+ temp_compass_disable_api: false
303
+ video_mme_eval: ''
304
+ mme_videoocr_mc_eval: false
305
+ mlvu_gen_eval: false
306
+ long_video_bench_eval: false
307
+ plm_fgqa_eval: false
308
+ long_video_bench_caption_eval: false
309
+ vinoground_eval: false
310
+ vixmo_caption_eval: false
311
+ dream1k_caption_eval: false
312
+ refexp_eval: false
313
+ coco_caption_eval: false
314
+ qv_highlights_eval: false
315
+ tomato: false
316
+ temporal_bench: false
317
+ max_new_tokens: 256
318
+ device_batch_size: 4
319
+ subset_num_batches: null
320
+ max_examples: 2048
321
+ console_log_interval: 20
322
+ include_image: false
323
+ - label: info_qa
324
+ data:
325
+ dataset: info_qa
326
+ mixture: null
327
+ root_size_mixture: null
328
+ kwargs_mixture: null
329
+ split: validation
330
+ seed: 691203
331
+ pad: to_max
332
+ sequence_length: 1792
333
+ max_text_seq_len: null
334
+ shuffle: true
335
+ start_index: 0
336
+ packing: null
337
+ num_workers: 2
338
+ drop_last: true
339
+ pin_memory: true
340
+ prefetch_factor: null
341
+ persistent_workers: true
342
+ timeout: 0
343
+ evaluator:
344
+ n_to_log: 0
345
+ num_wandb_examples: 32
346
+ save_predictions: null
347
+ save_tokens: false
348
+ vqa_eval: ansl,em
349
+ pointing_eval: false
350
+ count_eval: false
351
+ point_count_eval: false
352
+ android_eval: false
353
+ clock_eval: false
354
+ clock_bench_eval: false
355
+ math_vista_eval: false
356
+ temp_compass_eval: ''
357
+ temp_compass_disable_api: false
358
+ video_mme_eval: ''
359
+ mme_videoocr_mc_eval: false
360
+ mlvu_gen_eval: false
361
+ long_video_bench_eval: false
362
+ plm_fgqa_eval: false
363
+ long_video_bench_caption_eval: false
364
+ vinoground_eval: false
365
+ vixmo_caption_eval: false
366
+ dream1k_caption_eval: false
367
+ refexp_eval: false
368
+ coco_caption_eval: false
369
+ qv_highlights_eval: false
370
+ tomato: false
371
+ temporal_bench: false
372
+ max_new_tokens: 12
373
+ device_batch_size: 4
374
+ subset_num_batches: null
375
+ max_examples: 2048
376
+ console_log_interval: 20
377
+ include_image: false
378
+ - label: doc_qa
379
+ data:
380
+ dataset: doc_qa
381
+ mixture: null
382
+ root_size_mixture: null
383
+ kwargs_mixture: null
384
+ split: validation
385
+ seed: 691203
386
+ pad: to_max
387
+ sequence_length: 1792
388
+ max_text_seq_len: null
389
+ shuffle: true
390
+ start_index: 0
391
+ packing: null
392
+ num_workers: 2
393
+ drop_last: true
394
+ pin_memory: true
395
+ prefetch_factor: null
396
+ persistent_workers: true
397
+ timeout: 0
398
+ evaluator:
399
+ n_to_log: 0
400
+ num_wandb_examples: 32
401
+ save_predictions: null
402
+ save_tokens: false
403
+ vqa_eval: ansl,em
404
+ pointing_eval: false
405
+ count_eval: false
406
+ point_count_eval: false
407
+ android_eval: false
408
+ clock_eval: false
409
+ clock_bench_eval: false
410
+ math_vista_eval: false
411
+ temp_compass_eval: ''
412
+ temp_compass_disable_api: false
413
+ video_mme_eval: ''
414
+ mme_videoocr_mc_eval: false
415
+ mlvu_gen_eval: false
416
+ long_video_bench_eval: false
417
+ plm_fgqa_eval: false
418
+ long_video_bench_caption_eval: false
419
+ vinoground_eval: false
420
+ vixmo_caption_eval: false
421
+ dream1k_caption_eval: false
422
+ refexp_eval: false
423
+ coco_caption_eval: false
424
+ qv_highlights_eval: false
425
+ tomato: false
426
+ temporal_bench: false
427
+ max_new_tokens: 12
428
+ device_batch_size: 4
429
+ subset_num_batches: null
430
+ max_examples: 2048
431
+ console_log_interval: 20
432
+ include_image: false
433
+ - label: ai2_diagram
434
+ data:
435
+ dataset: ai2_diagram_v2_mix_transparent
436
+ mixture: null
437
+ root_size_mixture: null
438
+ kwargs_mixture: null
439
+ split: validation
440
+ seed: 691203
441
+ pad: to_max
442
+ sequence_length: 1792
443
+ max_text_seq_len: null
444
+ shuffle: true
445
+ start_index: 0
446
+ packing: null
447
+ num_workers: 2
448
+ drop_last: true
449
+ pin_memory: true
450
+ prefetch_factor: null
451
+ persistent_workers: true
452
+ timeout: 0
453
+ evaluator:
454
+ n_to_log: 0
455
+ num_wandb_examples: 32
456
+ save_predictions: null
457
+ save_tokens: false
458
+ vqa_eval: mc_ai2d_opaque,mc_ai2d_transparent
459
+ pointing_eval: false
460
+ count_eval: false
461
+ point_count_eval: false
462
+ android_eval: false
463
+ clock_eval: false
464
+ clock_bench_eval: false
465
+ math_vista_eval: false
466
+ temp_compass_eval: ''
467
+ temp_compass_disable_api: false
468
+ video_mme_eval: ''
469
+ mme_videoocr_mc_eval: false
470
+ mlvu_gen_eval: false
471
+ long_video_bench_eval: false
472
+ plm_fgqa_eval: false
473
+ long_video_bench_caption_eval: false
474
+ vinoground_eval: false
475
+ vixmo_caption_eval: false
476
+ dream1k_caption_eval: false
477
+ refexp_eval: false
478
+ coco_caption_eval: false
479
+ qv_highlights_eval: false
480
+ tomato: false
481
+ temporal_bench: false
482
+ max_new_tokens: 32
483
+ device_batch_size: 4
484
+ subset_num_batches: null
485
+ max_examples: 2048
486
+ console_log_interval: 20
487
+ include_image: false
488
+ - label: coco_2014_vqa
489
+ data:
490
+ dataset: coco_2014_vqa
491
+ mixture: null
492
+ root_size_mixture: null
493
+ kwargs_mixture: null
494
+ split: validation
495
+ seed: 691203
496
+ pad: to_max
497
+ sequence_length: 1792
498
+ max_text_seq_len: null
499
+ shuffle: true
500
+ start_index: 0
501
+ packing: null
502
+ num_workers: 2
503
+ drop_last: true
504
+ pin_memory: true
505
+ prefetch_factor: null
506
+ persistent_workers: true
507
+ timeout: 0
508
+ evaluator:
509
+ n_to_log: 0
510
+ num_wandb_examples: 32
511
+ save_predictions: null
512
+ save_tokens: false
513
+ vqa_eval: vqa_score
514
+ pointing_eval: false
515
+ count_eval: false
516
+ point_count_eval: false
517
+ android_eval: false
518
+ clock_eval: false
519
+ clock_bench_eval: false
520
+ math_vista_eval: false
521
+ temp_compass_eval: ''
522
+ temp_compass_disable_api: false
523
+ video_mme_eval: ''
524
+ mme_videoocr_mc_eval: false
525
+ mlvu_gen_eval: false
526
+ long_video_bench_eval: false
527
+ plm_fgqa_eval: false
528
+ long_video_bench_caption_eval: false
529
+ vinoground_eval: false
530
+ vixmo_caption_eval: false
531
+ dream1k_caption_eval: false
532
+ refexp_eval: false
533
+ coco_caption_eval: false
534
+ qv_highlights_eval: false
535
+ tomato: false
536
+ temporal_bench: false
537
+ max_new_tokens: 12
538
+ device_batch_size: 4
539
+ subset_num_batches: null
540
+ max_examples: 2048
541
+ console_log_interval: 20
542
+ include_image: false
543
+ inf_eval_interval: 2000
544
+ eval_on_last_step: true
545
+ eval_on_load: false
546
+ save_folder: /weka/oe-training-default/sanghol/molmo/models/uber-v1/uber3.4-synthetic-siglip2-qwen3_4b
547
+ checkpointer_config:
548
+ save_thread_count: null
549
+ load_thread_count: null
550
+ pre_download: false
551
+ work_dir: null
552
+ throttle_uploads: false
553
+ canceled_check_interval: 50
554
+ save_interval: 1000
555
+ save_at: null
556
+ save_final_optim: true
557
+ save_num_checkpoints_to_keep: 1
558
+ save_final_unsharded_checkpoint: false
559
+ save_interval_ephemeral: null
560
+ save_overwrite: true
561
+ load_path: null
562
+ reset_optimizer_state: false
563
+ reset_trainer_state: false
564
+ initial_model_checkpoint: /weka/oe-training-default/chrisk/molmo/models/dense-cap-v1/captioner-siglip2-qwen3_4b/step22347
565
+ allow_resume: true
566
+ max_duration: 30000
567
+ global_train_batch_size: 256
568
+ device_train_microbatch_size: 4
569
+ max_grad_norm: 1.0
570
+ multi_component_grad_norm: true
571
+ batch_divisor: global_batch
572
+ max_grad_norm_ratio: null
573
+ precision: amp_bf16
574
+ wandb:
575
+ project: molmo2-dev
576
+ entity: prior-ai2
577
+ group: uber-v1
578
+ name: uber3.4-synthetic-siglip2-qwen3_4b
579
+ tags:
580
+ - watching
581
+ log_artifacts: false
582
+ rank_zero_only: true
583
+ log_interval: 20
584
+ allow_resume: false
585
+ beaker_log_interval: 50
586
+ speed_monitor:
587
+ window_size: 20
588
+ gpu_flops_available: null
589
+ console_log_interval: 20
590
+ gen1_gc_interval: 1
591
+ compile:
592
+ mode: default
593
+ fullgraph: false
594
+ dynamic: false
595
+ backend: inductor
596
+ activation_checkpointing: true
597
+ fsdp:
598
+ fsdp2: true
599
+ precision: float
600
+ use_orig_params: true
601
+ wrapping_strategy: by_block_and_size
602
+ sharding_strategy: FULL_SHARD
603
+ hybrid_sharding_num_model_replicas: null
604
+ softmax_auxiliary_loss: true
605
+ softmax_auxiliary_loss_scale: 0.0001
606
+ saliency_score_loss_wt: null
607
+ frame_score_loss_wt: null
608
+ frame_score_loss_type: mse
609
+ frame_score_loss_target: 0.7
610
+ time_limit: null
611
+ extra_steps_after_cancel: 10
612
+ python_profiling: false
613
+ torch_profiling: false
614
+ stop_at: 30000
615
+ stop_after: null
616
+ fused_loss: false
617
+ compile_loss: true
618
+ runtime_data:
619
+ args: /gantry-runtime/launch_scripts/train_multitask_model.py 3.4-synthetic /weka/oe-training-default/chrisk/molmo/models/dense-cap-v1/captioner-siglip2-qwen3_4b
620
+ --save_overwrite --save_interval=1000 --wandb.group=uber-v1 --wandb.name=uber3.4-synthetic-siglip2-qwen3_4b
621
+ --save_folder=/weka/oe-training-default/sanghol/molmo/models/uber-v1/uber3.4-synthetic-siglip2-qwen3_4b
622
+ hostname: jupiter-cs-aus-112.reviz.ai2.in
623
+ date: 09/17/2025, 10:56
624
+ world_size: 16
625
+ resuming_from: /weka/oe-training-default/sanghol/molmo/models/uber-v1/uber3.4-synthetic-siglip2-qwen3_4b/step27000
626
+ beaker_experiment_id: 01K567APCW88M30XRRXSACWXPR
627
+ beaker_experiment_url: https://beaker.org/ex/01K567APCW88M30XRRXSACWXPR
628
+ wandb_id: 5307t37m
629
+ wandb_url: https://wandb.ai/prior-ai2/molmo2-dev/runs/5307t37m
model_and_optim/__11_0.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:036d3530bc0f84e4b39bb554e3f5366508408a07682acc89ce4c6b6c25ff25d8
3
+ size 197127554
model_and_optim/__11_5.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b4d89b473d945cd8aa4afda5ac21e4c174ce99fb6dc2d28b5ed59d26f354fa2e
3
+ size 197309771
model_and_optim/__12_0.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3867052c128477976fa5fce2163827afa499d476da8e75f5cad7ab330de1733d
3
+ size 197127554
model_and_optim/__13_1.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7b6cfa136d98a316a592271f724b5e92cc85e791164f1972f0b65e89c15604fb
3
+ size 197124856
model_and_optim/__13_14.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:544981e85717915dccd2cd32953a82ba670de251c7e51ba1c905d36bd8f4d618
3
+ size 197319955
model_and_optim/__14_14.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:18e0b83932a8388f8f75c991e177747007d8d22ee3f4721cc3bb1855befe8356
3
+ size 197319955
model_and_optim/__14_6.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:44e3fdfb3854204248eeaaf824c438c02cd2940942e7196329598dbe982c7527
3
+ size 197308194
model_and_optim/__15_3.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:67e1f8771208100081d129c3312f300125293afeb5555b9ea2530019f6b8ce00
3
+ size 197124856
model_and_optim/__15_8.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:93d9a3f125531fabe7c86738b0109c19e2359935a5a22b14ac81268a0352ff4c
3
+ size 197195331
model_and_optim/__1_12.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d0ce2d6636e08b88487346486793577ea5759c1fa02e5a2dc0238af77d447a7b
3
+ size 197145891
model_and_optim/__1_14.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:10db0f47254174128fabd4f06b9f325d0b76b95da216c16fbd9044f55419bd1f
3
+ size 197319955
model_and_optim/__2_0.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d1822b1376bd567b8a5eeebf12b54f4961ad33c20a5f61825a0c931cb0c44626
3
+ size 197127554
model_and_optim/__2_2.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8b5c9561b5b294105cc426c204ba94aa969af192ea31541317dc0f522978fd03
3
+ size 197124856
model_and_optim/__2_8.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:511fe5311412e0c60695113c12fdb49de202320f988d281bd5b11abcb0eacb92
3
+ size 197195331
model_and_optim/__3_13.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5e8eea91aa896c0c0929dd40bd6af91d37364ffdc39d0b1153355dd012626dba
3
+ size 197319955
model_and_optim/__3_14.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fcd757baf9d4dc3823fb5684477d6cc9c95d845086fd366e3d2b48a6439ef7f7
3
+ size 197319955
model_and_optim/__3_6.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:18bd08d0b52978d27846f8ba3c73ca54faa6ba35ac85b0105c3af4f47cde0dfa
3
+ size 197308194
model_and_optim/__4_2.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aa030e27ea0e133e2a904a68107813863ed72b8534f9dda9b30a2387f4fa8b4d
3
+ size 197124856
model_and_optim/__4_9.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5039f31d76b1530e3db4ae1d283e297e592d75d869244647e07f05acb584fd4a
3
+ size 197182971
model_and_optim/__5_1.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0714aaff801c1f87f68c2e865a8c92806864c716f59e3c2e387c66debb9bb020
3
+ size 197124856
model_and_optim/__5_2.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0942f55880e603dfcfd2dde9eb9246bd19b03d977cb5799a530cd5a3f3880eaa
3
+ size 197124856
model_and_optim/__5_3.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:235fa65eda6e32eee2534977603d237baf9726bd3555e3d228b3882c777e4a00
3
+ size 197124856
model_and_optim/__6_7.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:42f2bfca1f88be1bfdbe97c9cc96cc8baed131fc024b578214f857a2da140fc5
3
+ size 197182971
model_and_optim/__6_8.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6f17b8747158fada7b747868de492d9698885baa2beedbefe2b0d21990013c1d
3
+ size 197195331
model_and_optim/__7_1.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0d4eb9bf8637ea2bb7c39dc4720e39dbd3ef5109742c8c4a59f8e2c99271a861
3
+ size 197124856
model_and_optim/__7_13.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:36a1d8c511e834e147bba36295cec622a4b76ec7ac8f893511ec03d66d46de61
3
+ size 197319955
model_and_optim/__7_7.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:26405a974e58bf62834ec8b429b584633892ff41f47f14d07c14b278b76a39e4
3
+ size 197182971
model_and_optim/__8_1.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f4bb5740b18cf7c1574195df8f44a2aab2d5731eadb900a7a04e86fa37756eb5
3
+ size 197124856
model_and_optim/__8_6.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:97ef9e41bd3eeff3b2856205661404f9c6aae7ec3bd53ad1cceffa6a1935d188
3
+ size 197308194
model_and_optim/__9_1.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9984326b5354a57ca982982b56426d1046c24a2e68f02d3e7597587438383e8e
3
+ size 197124856
model_and_optim/__9_14.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f766bc24548057646be5e3e26380b50ab67d5cc7f67c1ee03a869c80d5aa681f
3
+ size 197319955
model_and_optim/__9_3.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:23942e76b2fcc80ba1c6d7153c7cd5f14c481b4a109ba1983f2151576265b825
3
+ size 197124856
model_and_optim/__9_9.distcp ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6efa678dd4a9c00f71f39a9029b0bbf46743630e6c29a1c6c7bfa82d9beee0df
3
+ size 197182971
train/rank0.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:790482fab49a0c3d2c64a9984b8fdc8b0115b669feac925eb8a2c2d51bdf85fc
3
+ size 14997
train/rank1.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:565ca8a17981b2bec78be27fbc6a592fabe408801a1b870c8b28ecda63e2ef99
3
+ size 15061
train/rank10.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:26290f02251c772bfafa80337265b84771ace54a31ca80d81e4b043e7b3b6738
3
+ size 15069
train/rank11.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6ebdc2c5fb99cd077b786716c1ff9774704307c37d9e82eb8108976052b54e62
3
+ size 15069
train/rank12.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:53bd2bdf4e4829c6af91be63447a01eeb27b6e7909405f1de9e089cb9b88c240
3
+ size 15069
train/rank13.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:106b7bf9c85649c99541d2bd79eddee0109332b21094df28f18a0d62e69aa622
3
+ size 15069
train/rank14.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:14fb7452113d272c5674e2630da1e79ae9f8bc87ac85234f8024c17e1b287991
3
+ size 15069
train/rank15.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d533fd5701673ab6de194178693854038b5f6b570c5fcc56555f68365103217b
3
+ size 15069
train/rank2.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3f7076eb17dffd55535f64c004c2cd1a0af952f1d92ece76edc645d41fe7fde9
3
+ size 15061
train/rank3.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6257eb07a70516fae3192d00df2c9b9a1b5515edab68e25bab7703446a8a2f78
3
+ size 15061
train/rank4.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:74dc3674cfa577eaa5d0e9968ab31069e33b3eca5f003cc41af41970880d8661
3
+ size 15061
train/rank5.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a66aebfe1843cbdb2ec33d9db1d8ff39815fb59da3074fd26b2f4ad1fa8d36bc
3
+ size 15061
train/rank6.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:220671d73c8066bb3775b27ca91e81897d22c081f7532875a7d7dd0463b39ddb
3
+ size 15061
train/rank7.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:baaac80ef69d25c6e82f184cc612948bc4fac1ecbe31ddc187c594eea901be59
3
+ size 15061
train/rank8.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a28511be4e382e3117f3f6951da76b80c10fbe282321987c50251bfa14b7ef20
3
+ size 15061