File size: 389 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 |
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'
|