trainer:
  name: supervised_segmentation

train:
  seed: 42
  epochs: 200
  batch_size: 4
  val_batch_size: 4
  amp: true
  num_workers: 4
  pin_memory: true
  persistent_workers: true
  prefetch_factor: 2
  device: cuda
  grad_clip:
    enabled: true
    max_norm: 1.0
    norm_type: 2.0
  auto_batch_size:
    enabled: false
    target_memory_fraction: 0.75
    reference_gpu_gb: 8.0
    reference_batch_size: 4
    min_batch_size: 1
    max_batch_size: 8

metrics:
  task_mode: binary
  metrics:
    - name: dice
    - name: iou

validation:
  enabled: true
  interval: 1
  threshold: 0.5
  early_stopping: true
  early_stopping_patience: 40
  early_stopping_min_delta: 0.0
  metrics:
    task_mode: binary
    metrics:
      - name: dice
      - name: iou

dataset:
  name: ultrasound_sup_seg
  dataset_name: BUSI
  root: data/BUSI
  split: train
  split_file: null
  val_split: val
  val_split_file: null
  image_size: [256, 256]
  in_channels: 3
  num_classes: 1

model:
  model_name: swinv2_tiny_patch4_window8_256
  load_weights: false
  decoder_channels: [384, 192, 96, 96]
  fwta_wavelet: haar
  fwta_level: 1
  fwta_sigma_ratio: 0.35
  fwta_tau_fourier: 0.15
  fwta_gate_temperature: 1.0
  use_multiscale_features: true
  include_patch_embed: true

optimizer:
  name: adamw
  lr: 1.0e-4
  weight_decay: 0.05

scheduler:
  name: cosine
  warmup:
    name: linear
    params:
      start_factor: 0.1
      total_iters: 10
  params:
    T_max: 190
    eta_min: 1.0e-6

checkpoint:
  dir: outputs/supervised_segmentation/train_sup_us_template
  save: true
  save_last: true
  monitor: dice
  monitor_mode: max
  resume: null
  resume_strict: true
  resume_training: true

logging:
  log_interval: 10
  print_training_setup: true
  use_swanlab: true
  project: X_SSL_Net
  experiment_name: supervised_segmentation
  swanlab_mode: null