seed: 42

paths:
  root_dir: D:/AI/ImageAI
  master_dir: D:/AI/ImageAI/MasterPNG
  sparse_dir: D:/AI/ImageAI/SparsePNG
  splits_dir: D:/AI/ImageAI/splits
  runs_dir: D:/AI/ImageAI/runs
  checkpoints_dir: D:/AI/ImageAI/checkpoints
  outputs_dir: D:/AI/ImageAI/outputs

data:
  train_split: train_native_even.txt
  val_split: val_native_even.txt
  test_split: test_native_even.txt
  sample_mode: native_image
  patch_size: null
  eval_patch_size: null
  full_frame_size: null
  full_frame_pad_mode: edge
  train_batch_size: 1
  val_batch_size: 1
  num_workers: 12
  pin_memory: true
  persistent_workers: true
  prefetch_factor: 2
  train_derive_sparse_from_gt: false
  val_derive_sparse_from_gt: false
  train_horizontal_flip: false
  train_vertical_flip: false

model:
  name: sparse_unet
  in_channels: 16
  out_channels: 12
  base_channels: 96

loss:
  missing_weight: 1.0
  known_weight: 0.05

training:
  device: cuda
  amp: true
  max_steps: 1000000
  warmup_steps: 1000
  validate_every: 1000
  save_every: 1000
  log_every: 50
  learning_rate: 0.0002
  min_learning_rate: 0.000001
  weight_decay: 0.01
  betas: [0.9, 0.99]
  grad_clip_norm: 1.0
  run_name: sparse_unet_native_bc96
  resume: null

inference:
  tile_size: 512
  tile_overlap: 64