model_prompttuning.yml

sam:
  img_size: 1024
  num_classes: 5

img_type: 'images'
arch: "Prompt Adapted SAM"
use_fdn: False

prompts:
  USE_TEXT_PROMPT: True
  USE_IMAGE_PROMPT: True
  LOCATION: 'prepend'
  DROPOUT: 0.1
  NUM_TOKENS: 5

training:
  optimizer: 'adamw'
  lr: 1e-3
  batch_size: 8
  num_epochs: 1000
  schedule_step: 200
  schedule_step_factor: 0.5
  weight_decay: 1e-7
  loss: 'weighted CE+dice'