brian-gordon's picture
Upload config.yaml with huggingface_hub
2119824 verified
raw
history blame
No virus
2.46 kB
BASE_REPO_ID: google/paligemma-3b-pt-448
FINETUNED_MODEL_ID: brian-gordon/paligemma-3b-pt-448_DOCCI-IIW_ALL-TASKS
OUTPUT_DIR_PATH: /home/dcor/briangordon/LongCaptionProject/eval_output
MAX_LENGTH: 1024
TRAIN_PARAMS:
BATCH_SIZE: 2
USE_IMAGES: true
DATA:
DOCCI:
IMG_DIR: /home/dcor/datasets/docci_data/docci_images
DATASETS_CSV_BASE_DIR: /home/dcor/datasets/docci_data/datasets
VERSION: 21Apr2024
TRAIN:
DATASETS:
- name: DOCCI_description_to_annotated
task: Annotation
- name: DOCCI_description_to_fixed_decription_E2E
task: Text
- name: DOCCI_SxS_Long_Description
task: SxS_Paragraph
- name: DOCCI_SxS_Sentence_Level
task: SxS_Sentence
- name: DOCCI_Long_Description_with_target_sentence_VNLI
task: VNLI_With_Context
- name: DOCCI_Claims_Consistency_VNLI
task: VNLI_Claims
EVAL:
DATASETS:
- name: DOCCI_description_to_annotated
task: Annotation
- name: DOCCI_description_to_fixed_decription_E2E
task: Text
- name: DOCCI_SxS_Long_Description
task: SxS_Paragraph
- name: DOCCI_SxS_Sentence_Level
task: SxS_Sentence
- name: DOCCI_Long_Description_with_target_sentence_VNLI
task: VNLI_With_Context
- name: DOCCI_Claims_Consistency_VNLI
task: VNLI_Claims
IIW:
IMG_DIR: /home/dcor/datasets/IIW/images_from_sstable
DATASETS_CSV_BASE_DIR: /home/dcor/datasets/IIW/gemini_mpp_32k
VERSION: 17_August_2024
TRAIN:
DATASETS:
- name: IIW_description_to_annotated
task: Annotation
- name: IIW_description_to_fixed_decription_E2E
task: Text
- name: IIW_SxS_Long_Description
task: SxS_Paragraph
- name: IIW_SxS_Sentence_Level
task: SxS_Sentence
- name: IIW_Long_Description_with_target_sentence_VNLI
task: VNLI_With_Context
- name: IIW_Claims_Consistency_VNLI
task: VNLI_Claims
EVAL:
VERSION: 17_August_2024
DATASETS:
- name: IIW_description_to_annotated
task: Annotation
- name: IIW_description_to_fixed_decription_E2E
task: Text
- name: IIW_SxS_Long_Description
task: SxS_Paragraph
- name: IIW_SxS_Sentence_Level
task: SxS_Sentence
- name: IIW_Long_Description_with_target_sentence_VNLI
task: VNLI_With_Context
- name: IIW_Claims_Consistency_VNLI
task: VNLI_Claims