lovodkin93 commited on
Commit
b1e1ed4
·
verified ·
1 Parent(s): 7b4390e

Upload config.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.yaml +40 -0
config.yaml ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ BASE_REPO_ID: google/paligemma-3b-pt-448
2
+ FINETUNED_MODEL_ID: lovodkin93/paligemma-3b-pt-448_lora_mmnt_concat_vertical-tgt-center-black_balanced
3
+ OUTPUT_DIR_PATH: /home/slobodkin/reference_vnli/train_ref_vnli_autorater/models/paligemma-3b-pt-448/lora_mmnt_concat_vertical-tgt-center-black_balanced
4
+ DEBUG: false
5
+ RESUME_FROM_CHECKPOINT: false
6
+ MAX_LENGTH: 300
7
+ CONCAT_REF_TGT_IMAGES: true
8
+ CONCAT_IMAGES_TYPE: vertical-tgt-center-black
9
+ TRAIN_PARAMS:
10
+ BALANCE_TRAIN_DATASETS: true
11
+ BATCH_SIZE: 4
12
+ FULL_FINETUNE: false
13
+ FINETUNE_LLM: true
14
+ FINETUNE_MM_PROJECTOR: false
15
+ FINETUNE_VISION_MODEL: false
16
+ GRADIENT_ACCUMULATION_STEPS: 2
17
+ EVAL_INTERVAL: 200
18
+ SAVE_INTERVAL: 200
19
+ SAVE_TOTAL_LIMIT: 20
20
+ METRIC_FOR_BEST_MODEL: Eval/accuracy
21
+ DATA:
22
+ MEMENTOS:
23
+ IMG_DIR: /home/slobodkin/reference_vnli/datasets/mementos/images
24
+ DATASETS_CSV_BASE_DIR: /home/slobodkin/reference_vnli/train_ref_vnli_autorater/data/mementos/multiclass
25
+ VERSION: 03Nov2024
26
+ TRAIN:
27
+ DATASETS:
28
+ - name: train_min_balanced_multi_classification
29
+ task: mementos
30
+ EVAL:
31
+ DATASETS:
32
+ - name: dev_2000_instances
33
+ task: mementos
34
+ TVQA:
35
+ DATASETS_CSV_BASE_DIR: /home/slobodkin/reference_vnli/train_ref_vnli_autorater/data/TVQA_plus
36
+ VERSION: 15Dec2024
37
+ EVAL:
38
+ DATASETS:
39
+ - name: val_balanced_full_2000_instances
40
+ task: TVQA