img_size: 512
num_classes: 1
in_channels: 3
img_type: 'image'
arch: "TransUNet"
use_fdn: False


training:
  optimizer: 'adamw'
  lr: 1e-4
  batch_size: 32
  num_epochs: 500
  schedule_step: 2100
  schedule_step_factor: 0.5
  weight_decay: 1e-2
  loss: 'dice+CE'