data:
  aug_param:
    flip_lr: 0.5
    hsv: 0.3
    mini_mosaic: 0.5
    neg: 0.3
    size_range:
    - 0.85
    - 1.1
  augment: true
  cache: false
  imgsz: 1024
  train_img_dir:
  - ../datasets/codat_manga_v3/images/train
  - ../datasets/ComicErased/processed
  train_mask_dir: ../datasets/ComicSegV2
  val_img_dir:
  - ../datasets/codat_manga_v3/images/val
  val_mask_dir: ../datasets/ComicSegV2
logger:
  project: ''
  run_id: ''
  type: wandb
model:
  act: leaky
  weights: data/yolov5sblk.ckpt
resume:
  ckpt: ''
  resume_training: false
train:
  accumulation_steps: 4
  batch_size: 4
  epochs: 120
  eval_interval: 1
  linear_lr: false
  loss: dice
  lr0: 0.004
  lrf: 0.005
  momentum: 0.937
  optimizer: adam
  warmup_bias_lr: 0.1
  warmup_epochs: 3.0
  warmup_momentum: 0.8
  weight_decay: 2.0e-05