seed: 1337
output_dir: outputs/default
amp: true
compile: false
channels_last_3d: false
num_workers: 4
pin_memory: true
wandb:
  enabled: true
  project: SACFlow-FM
  entity: null
  mode: online
  tags: []
  log_images_every: 250
  watch_model: false

distributed:
  backend: nccl
  find_unused_parameters: false

data:
  manifest: data_json/mmwhs_ct2mr.json
  spacing: null
  patch_size: [96, 96, 96]
  batch_size: 1
  val_batch_size: 1
  cache_rate: 0.0
  in_channels: 1
  num_classes: 5
  foreground_labels: null
  intensity:
    clip: null
    normalize: zscore_nonzero
  augmentation:
    random_flip: true
    random_intensity_shift: 0.10
    random_intensity_scale: 0.10

model:
  name: unet3d
  in_channels: 1
  num_classes: 5
  base_channels: 32
  levels: 4
  norm: instance
  dropout: 0.0
  deep_supervision: false
  adapter:
    enabled: false
    bottleneck_ratio: 0.25
    locations: [prelogit]

optim:
  optimizer: adamw
  lr: 2.0e-4
  weight_decay: 1.0e-4
  betas: [0.9, 0.999]
  scheduler: cosine
  warmup_steps: 500
  max_steps: 30000
  grad_clip_norm: 12.0

train:
  mode: source_train
  epochs: 100
  steps_per_epoch: 250
  val_every: 1
  save_every: 5
  log_every: 20
  resume: null
  source_checkpoint: null
  memory_path: null
  ema_decay: 0.99
  pseudo_conf_threshold: 0.70
  class_balanced_thresholds: false
  loss:
    ce: 1.0
    dice: 1.0
    entropy: 0.0
    consistency: 0.0

eval:
  checkpoint: null
  split: target_test
  save_predictions: false
  sliding_window: true
  roi_size: [96, 96, 96]
  sw_batch_size: 1
  overlap: 0.5