| 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 |
|
|
