Image-Text-to-Text
English
weizhiwang commited on
Commit
f4836d5
·
verified ·
1 Parent(s): 9430e34

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. checkpoints/latest-checkpoint.pt +3 -0
  2. config.json +73 -0
  3. config.yaml +66 -0
checkpoints/latest-checkpoint.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a4f68b9169d7419850d408c6eb75bb606ff66301540c32670503026965943496
3
+ size 7125006107
config.json ADDED
@@ -0,0 +1,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": {
3
+ "align_stage_components": [
4
+ "download/llava-laion-cc-sbu-558k/chat.json",
5
+ "download/llava-laion-cc-sbu-558k"
6
+ ],
7
+ "dataset_id": "llava-v15",
8
+ "dataset_resampled": true,
9
+ "dataset_root_dir": "data",
10
+ "finetune_stage_components": [
11
+ "/local/home/weizhiwang/data/MAmmoTH-VL-Instruct-12M/mammoth_si_10M_simple.jsonl",
12
+ "/share/edc/home/weizhiwang/data/MAmmoTH-VL-Instruct-12M/single_image_data"
13
+ ],
14
+ "max_num_images": 6,
15
+ "min_num_images": 1,
16
+ "train_num_samples": 200000,
17
+ "type": "llava-v15",
18
+ "workers": 4
19
+ },
20
+ "model": {
21
+ "align_epochs": 1,
22
+ "align_global_batch_size": 256,
23
+ "align_learning_rate": 0.001,
24
+ "align_lr_scheduler_type": "linear-warmup+cosine-decay",
25
+ "align_max_grad_norm": 1.0,
26
+ "align_max_steps": null,
27
+ "align_per_device_batch_size": 16,
28
+ "align_train_strategy": "fsdp-shard-grad-op",
29
+ "align_warmup_ratio": 0.03,
30
+ "align_weight_decay": 0.0,
31
+ "arch_specifier": "full-align+729-avgpool",
32
+ "enable_gradient_checkpointing": true,
33
+ "enable_mixed_precision_training": true,
34
+ "finetune_epochs": 1,
35
+ "finetune_global_batch_size": 128,
36
+ "finetune_learning_rate": 2e-05,
37
+ "finetune_lr_scheduler_type": "linear-warmup+cosine-decay",
38
+ "finetune_max_grad_norm": 1.0,
39
+ "finetune_max_steps": null,
40
+ "finetune_per_device_batch_size": 2,
41
+ "finetune_train_strategy": "fsdp-full-shard",
42
+ "finetune_warmup_ratio": 0.03,
43
+ "finetune_weight_decay": 0.1,
44
+ "image_resize_strategy": "resize-naive",
45
+ "llm_backbone_id": "qwen2.5-1.5b-instruct",
46
+ "llm_max_length": 4096,
47
+ "model_id": "qwen2.5-1.5b-instruct-continue-training-ccs-datacomp-mlm-filter-mammoth-10m",
48
+ "pretrain_epochs": 1,
49
+ "pretrain_global_batch_size": 512,
50
+ "pretrain_learning_rate": 5e-05,
51
+ "pretrain_lr_scheduler_type": "linear-warmup+cosine-decay",
52
+ "pretrain_max_grad_norm": 1.0,
53
+ "pretrain_max_steps": null,
54
+ "pretrain_per_device_batch_size": 16,
55
+ "pretrain_train_strategy": "fsdp-full-shard",
56
+ "pretrain_warmup_ratio": 0.03,
57
+ "pretrain_weight_decay": 0.01,
58
+ "reduce_in_full_precision": false,
59
+ "type": "one-stage+7b",
60
+ "vision_backbone_id": "siglip-vit-so400m-384px"
61
+ },
62
+ "mount_path": "Qwen",
63
+ "pretrained_checkpoint": "/local/home/weizhiwang/checkpoints/obelics+qwen2.5-1.5b-instruct-continue-training-ccs-datacomp-mlm-filter+stage-pretrain+x7/checkpoints/latest-checkpoint.pt",
64
+ "run_id": "qwen2.5-1.5b-instruct-continue-training-ccs-datacomp-mlm-filter-mammoth-10m+stage-finetune+x7",
65
+ "run_root_dir": "/share/edc/home/weizhiwang/checkpoints",
66
+ "seed": 7,
67
+ "stage": "finetune",
68
+ "trackers": [
69
+ "jsonl"
70
+ ],
71
+ "wandb_entity": null,
72
+ "wandb_project": "mmpretrain"
73
+ }
config.yaml ADDED
@@ -0,0 +1,66 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset:
2
+ align_stage_components:
3
+ - download/llava-laion-cc-sbu-558k/chat.json
4
+ - download/llava-laion-cc-sbu-558k
5
+ dataset_id: llava-v15
6
+ dataset_resampled: true
7
+ dataset_root_dir: data
8
+ finetune_stage_components:
9
+ - /local/home/weizhiwang/data/MAmmoTH-VL-Instruct-12M/mammoth_si_10M_simple.jsonl
10
+ - /share/edc/home/weizhiwang/data/MAmmoTH-VL-Instruct-12M/single_image_data
11
+ max_num_images: 6
12
+ min_num_images: 1
13
+ train_num_samples: 200000
14
+ type: llava-v15
15
+ workers: 4
16
+ model:
17
+ align_epochs: 1
18
+ align_global_batch_size: 256
19
+ align_learning_rate: 0.001
20
+ align_lr_scheduler_type: linear-warmup+cosine-decay
21
+ align_max_grad_norm: 1.0
22
+ align_max_steps: null
23
+ align_per_device_batch_size: 16
24
+ align_train_strategy: fsdp-shard-grad-op
25
+ align_warmup_ratio: 0.03
26
+ align_weight_decay: 0.0
27
+ arch_specifier: full-align+729-avgpool
28
+ enable_gradient_checkpointing: true
29
+ enable_mixed_precision_training: true
30
+ finetune_epochs: 1
31
+ finetune_global_batch_size: 128
32
+ finetune_learning_rate: 2.0e-05
33
+ finetune_lr_scheduler_type: linear-warmup+cosine-decay
34
+ finetune_max_grad_norm: 1.0
35
+ finetune_max_steps: null
36
+ finetune_per_device_batch_size: 2
37
+ finetune_train_strategy: fsdp-full-shard
38
+ finetune_warmup_ratio: 0.03
39
+ finetune_weight_decay: 0.1
40
+ image_resize_strategy: resize-naive
41
+ llm_backbone_id: qwen2.5-1.5b-instruct
42
+ llm_max_length: 4096
43
+ model_id: qwen2.5-1.5b-instruct-continue-training-ccs-datacomp-mlm-filter-mammoth-10m
44
+ pretrain_epochs: 1
45
+ pretrain_global_batch_size: 512
46
+ pretrain_learning_rate: 5.0e-05
47
+ pretrain_lr_scheduler_type: linear-warmup+cosine-decay
48
+ pretrain_max_grad_norm: 1.0
49
+ pretrain_max_steps: null
50
+ pretrain_per_device_batch_size: 16
51
+ pretrain_train_strategy: fsdp-full-shard
52
+ pretrain_warmup_ratio: 0.03
53
+ pretrain_weight_decay: 0.01
54
+ reduce_in_full_precision: false
55
+ type: one-stage+7b
56
+ vision_backbone_id: siglip-vit-so400m-384px
57
+ mount_path: Qwen
58
+ pretrained_checkpoint: /local/home/weizhiwang/checkpoints/obelics+qwen2.5-1.5b-instruct-continue-training-ccs-datacomp-mlm-filter+stage-pretrain+x7/checkpoints/latest-checkpoint.pt
59
+ run_id: qwen2.5-1.5b-instruct-continue-training-ccs-datacomp-mlm-filter-mammoth-10m+stage-finetune+x7
60
+ run_root_dir: /share/edc/home/weizhiwang/checkpoints
61
+ seed: 7
62
+ stage: finetune
63
+ trackers:
64
+ - jsonl
65
+ wandb_entity: null
66
+ wandb_project: mmpretrain