run:
  out_root: artefacts
  name: null
  seed: 42
  device: auto
  num_workers: 0
  early_stopping_patience: 5
  metric_for_best: ap
data:
  source: local
  local:
    root: dataset
    splits:
      train: train
      val: val
      test: test
    classes:
      negative: non_illustration
      positive: illustration
  hf:
    dataset: null
    image_column: image
    label_column: label
    splits:
      train: train
      val: validation
      test: test
    label_names:
    - '0'
    - '1'
model:
  backend: transformers
  name: apple/mobilevitv2-1.0-imagenet1k-256
  pretrained: true
  freeze_backbone: false
preprocess:
  processor_use_fast: false
  mode: auto
  img_size: 288
  mean:
  - 0.485
  - 0.456
  - 0.406
  std:
  - 0.229
  - 0.224
  - 0.225
  augment:
    enabled: true
    hflip: true
    rotation_deg: 5
    jitter:
      brightness: 0.12
      contrast: 0.12
      saturation: 0.08
      hue: 0.02
train:
  epochs: 30
  batch_size: 32
  lr: 5e-5
export:
  opset: 18