lovodkin93 commited on
Commit
7df87fb
1 Parent(s): a3c94af

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-ft-nlvr2-448
2
+ FINETUNED_MODEL_ID: lovodkin93/paligemma-3b-ft-nlvr2-448_ref_vnli_lora_only_SUTI
3
+ OUTPUT_DIR_PATH: /home/slobodkin/reference_vnli/train_ref_vnli_autorater/models/paligemma-3b-ft-nlvr2-448/lora_only_SUTI_min_balanced_multiclass
4
+ DEBUG: false
5
+ RESUME_FROM_CHECKPOINT: false
6
+ MAX_LENGTH: 100
7
+ CONCAT_REF_TGT_IMAGES: false
8
+ TRAIN_PARAMS:
9
+ BALANCE_TRAIN_DATASETS: false
10
+ BATCH_SIZE: 2
11
+ FULL_FINETUNE: false
12
+ FINETUNE_LLM: true
13
+ FINETUNE_MM_PROJECTOR: false
14
+ FINETUNE_VISION_MODEL: false
15
+ GRADIENT_ACCUMULATION_STEPS: 2
16
+ EVAL_INTERVAL: 100
17
+ SAVE_INTERVAL: 100
18
+ SAVE_TOTAL_LIMIT: 2
19
+ METRIC_FOR_BEST_MODEL: Eval/accuracy
20
+ DATA:
21
+ MEMENTOS:
22
+ IMG_DIR: /home/slobodkin/reference_vnli/datasets/mementos/images
23
+ DATASETS_CSV_BASE_DIR: /home/slobodkin/reference_vnli/train_ref_vnli_autorater/data/mementos/multiclass
24
+ VERSION: 03Nov2024
25
+ EVAL:
26
+ DATASETS:
27
+ - name: dev_2000_instances
28
+ task: mementos
29
+ SUTI:
30
+ IMG_DIR: /home/slobodkin/reference_vnli/datasets/SUTI/images
31
+ DATASETS_CSV_BASE_DIR: /home/slobodkin/reference_vnli/train_ref_vnli_autorater/data/SUTI/multiclass
32
+ VERSION: 08Nov2024
33
+ TRAIN:
34
+ DATASETS:
35
+ - name: train_min_balanced_multi_classification
36
+ task: SUTI
37
+ EVAL:
38
+ DATASETS:
39
+ - name: dev_2000_instances
40
+ task: SUTI