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