device: 'cuda'

data:
  train_dataset: ""
  aug_prob: 0.1
  scale_dim:
    d_0: 128
    d_1: 128
    d_2: 128

model:
  architecture: UNet4
  loss: DiceLoss
  filter_base: 16
  filter_expansion: 2
  weights:
  optimizer:
    name: 'Adam'
    lr: 0.000025

train:
  max_epochs: 500
  val_frequency: 1
  output_threshold: 0.5
  metric_path: "/home/tumor/out/metric.csv"
  save_frequency: 50
  model_directory: "/home/tumor/out/models/"
  batch_size: 10