SALT-SAM / AllinonSAM /eval /cholec8k /config_model_test.yml
pythn's picture
Upload with huggingface_hub
4a1f918 verified
raw
history blame contribute delete
500 Bytes
sam:
img_size: 256
num_classes: 13
sam_type: "base"
img_type: 'image'
arch: "Prompt Adapted SAM"
use_fdn: False
decoder_training: 'none'
mlp_transform: False
prompts:
USE_TEXT_PROMPT: False
USE_IMAGE_PROMPT: False
USE_SLICE_NUM: False
LOCATION: 'prepend'
DROPOUT: 0
NUM_TOKENS: 5
training:
optimizer: 'adamw'
lr: 1e-3
batch_size: 32
num_epochs: 1000
schedule_step: 100
schedule_step_factor: 0.5
weight_decay: 1e-2
loss: 'focal'
reg_multiplier: 0
use_lora: True