tyleryzhu commited on
Commit
6d1891f
1 Parent(s): 03baaf4

Upload merv-base/config.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. merv-base/config.yaml +62 -0
merv-base/config.yaml ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset:
2
+ align_stage_components:
3
+ - download/videollava/valley_llavaimage.json
4
+ - download/videollava
5
+ dataset_root_dir: data
6
+ finetune_stage_components:
7
+ - download/videollava/videochatgpt_llavaimage_tune.json
8
+ - download/videollava
9
+ type: videollava
10
+ hf_token: .hf_token
11
+ model:
12
+ align_epochs: 1
13
+ align_global_batch_size: 256
14
+ align_learning_rate: 0.001
15
+ align_lr_scheduler_type: linear-warmup+cosine-decay
16
+ align_max_grad_norm: 1.0
17
+ align_max_steps: null
18
+ align_per_device_batch_size: 16
19
+ align_train_strategy: fsdp-shard-grad-op
20
+ align_warmup_ratio: 0.03
21
+ align_weight_decay: 0.0
22
+ arch_specifier: no-align+3davg+linear
23
+ enable_gradient_checkpointing: true
24
+ enable_mixed_precision_training: true
25
+ feature_fusion: cross_attention_avg_lq
26
+ finetune_epochs: 1
27
+ finetune_global_batch_size: 128
28
+ finetune_learning_rate: 2.0e-05
29
+ finetune_lr_scheduler_type: linear-warmup+cosine-decay
30
+ finetune_max_grad_norm: 1.0
31
+ finetune_max_steps: null
32
+ finetune_per_device_batch_size: 8
33
+ finetune_train_strategy: fsdp-full-shard
34
+ finetune_warmup_ratio: 0.03
35
+ finetune_weight_decay: 0.1
36
+ image_resize_strategy: resize-naive
37
+ llm_backbone_id: llama2-7b-pure
38
+ llm_max_length: 2048
39
+ model_id: merv-base
40
+ num_frames:
41
+ - 16
42
+ - 16
43
+ - 32
44
+ - 16
45
+ projector_token_length: 64
46
+ reduce_in_full_precision: false
47
+ type: merv-base
48
+ video_backbone_ids:
49
+ - languagebind-video-noclass
50
+ - dinov2-video-all-tokens
51
+ - vivit-google-b-all-no-cls-16frames
52
+ - siglip-vit-b16-224px-all-no-cls
53
+ visual_feature_length: 1024
54
+ pretrained_checkpoint: null
55
+ run_id: merv-base
56
+ run_root_dir: runs
57
+ seed: 7
58
+ stage: finetune
59
+ trackers:
60
+ - jsonl
61
+ - wandb
62
+ type: merv-base