lovodkin93's picture
Upload config.yaml with huggingface_hub
7df87fb verified
raw
history blame
1.34 kB
BASE_REPO_ID: google/paligemma-3b-ft-nlvr2-448
FINETUNED_MODEL_ID: lovodkin93/paligemma-3b-ft-nlvr2-448_ref_vnli_lora_only_SUTI
OUTPUT_DIR_PATH: /home/slobodkin/reference_vnli/train_ref_vnli_autorater/models/paligemma-3b-ft-nlvr2-448/lora_only_SUTI_min_balanced_multiclass
DEBUG: false
RESUME_FROM_CHECKPOINT: false
MAX_LENGTH: 100
CONCAT_REF_TGT_IMAGES: false
TRAIN_PARAMS:
BALANCE_TRAIN_DATASETS: false
BATCH_SIZE: 2
FULL_FINETUNE: false
FINETUNE_LLM: true
FINETUNE_MM_PROJECTOR: false
FINETUNE_VISION_MODEL: false
GRADIENT_ACCUMULATION_STEPS: 2
EVAL_INTERVAL: 100
SAVE_INTERVAL: 100
SAVE_TOTAL_LIMIT: 2
METRIC_FOR_BEST_MODEL: Eval/accuracy
DATA:
MEMENTOS:
IMG_DIR: /home/slobodkin/reference_vnli/datasets/mementos/images
DATASETS_CSV_BASE_DIR: /home/slobodkin/reference_vnli/train_ref_vnli_autorater/data/mementos/multiclass
VERSION: 03Nov2024
EVAL:
DATASETS:
- name: dev_2000_instances
task: mementos
SUTI:
IMG_DIR: /home/slobodkin/reference_vnli/datasets/SUTI/images
DATASETS_CSV_BASE_DIR: /home/slobodkin/reference_vnli/train_ref_vnli_autorater/data/SUTI/multiclass
VERSION: 08Nov2024
TRAIN:
DATASETS:
- name: train_min_balanced_multi_classification
task: SUTI
EVAL:
DATASETS:
- name: dev_2000_instances
task: SUTI