img_size: 256
num_classes: 10
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'