SALT-SAM / AllinonSAM /eval /cholec8k /model_baseline.yml
pythn's picture
Upload with huggingface_hub
4a1f918 verified
img_size: 256
num_classes: 12
in_channels: 3
img_type: 'image'
arch: "UNet"
use_fdn: False
training:
optimizer: 'adamw'
lr: 1e-4
batch_size: 16
num_epochs: 500
schedule_step: 2100
schedule_step_factor: 0.5
weight_decay: 1e-2
loss: 'focal'