File size: 588 Bytes
4a1f918 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
sam:
img_size: 512
num_classes: 2
sam_type: "base"
img_type: 'image'
arch: "Prompt Adapted SAM"
use_fdn: False
decoder_training: 'none'
mlp_transform: False
prompts:
USE_TEXT_PROMPT: True
NUM_TEXT_REPEAT: 1
USE_IMAGE_PROMPT: False
USE_SLICE_NUM: False
LOCATION: 'prepend'
DROPOUT: 0
NUM_TOKENS: 5
decoder_training: full
prompt_encoder: full
sam_encoder: full
training:
optimizer: 'adamw'
lr: 1e-4
batch_size: 6
num_epochs: 500
schedule_step: 200
schedule_step_factor: 0.2
weight_decay: 1e-2
loss: 'focal+dice'
reg_multiplier: 0
# use_lora: False |