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 | |