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' |