brian-gordon commited on
Commit
f8e9e23
1 Parent(s): 035de6d

Upload config.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.yaml +42 -0
config.yaml ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ BASE_REPO_ID: google/paligemma-3b-pt-448
2
+ FINETUNED_MODEL_ID: brian-gordon/paligemma-3b-pt-448_DOCCI_CorruptionsV2-ALL
3
+ OUTPUT_DIR_PATH: /home/gordonbrian/ImageUnderstandingSprint/experiments
4
+ MAX_LENGTH: 1024
5
+ TRAIN_PARAMS:
6
+ BATCH_SIZE: 2
7
+ USE_IMAGES: true
8
+ FULL_FINETUNE: false
9
+ FINETUNE_LLM: false
10
+ FINETUNE_MM_PROJECTOR: false
11
+ FINETUNE_VISION_MODEL: false
12
+ GRADIENT_ACCUMULATION_STEPS: 2
13
+ DATA:
14
+ DOCCI:
15
+ IMG_DIR: /home/gordonbrian/ImageUnderstandingSprint/images/docci_images
16
+ TRAIN:
17
+ CORRUPTION_PROB: null
18
+ N_CORRUPTIONS: 1
19
+ CORRUPTION_TYPES:
20
+ - SWAP
21
+ - ADD
22
+ DATASETS:
23
+ - name: NewDOCCIDataset
24
+ csv_path: /home/gordonbrian/ImageUnderstandingSprint/data/DOCCI_V2_Corruptions_MultiTurn_With_Consistency_01Dec2024--train.csv
25
+ tasks:
26
+ - Annotation
27
+ - Text
28
+ - SxS_Paragraph
29
+ - SxS_Sentence
30
+ - VNLI_With_Context
31
+ - VNLI_Claims
32
+ EVAL:
33
+ CORRUPTION_PROB: null
34
+ N_CORRUPTIONS: 1
35
+ CORRUPTION_TYPES:
36
+ - SWAP
37
+ - ADD
38
+ DATASETS:
39
+ - name: NewDOCCIDataset
40
+ csv_path: /home/gordonbrian/ImageUnderstandingSprint/data/DOCCI_V2_Corruptions_MultiTurn_With_Consistency_01Dec2024--test.csv
41
+ tasks:
42
+ - VNLI_Claims