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 |