default.yaml
679 bytes
| 1 | data: |
|---|---|
| 2 | raw_dir: data/raw # iga alamkaust on üks riik |
| 3 | manifest: data/manifest.csv # clean.py väljund; kui puudub, skannitakse raw_dir |
| 4 | min_per_class: 100 # riigid, millel on vähem pilte, jäetakse välja |
| 5 | val_frac: 0.1 |
| 6 | test_frac: 0.1 |
| 7 | num_workers: 2 |
| 8 | |
| 9 | model: |
| 10 | backbone: vit_base_patch16_clip_224.openai |
| 11 | dropout: 0.2 |
| 12 | |
| 13 | train: |
| 14 | seed: 42 |
| 15 | batch_size: 64 |
| 16 | epochs_head: 3 # faas 1: ainult klassifitseerimispea |
| 17 | epochs_finetune: 8 # faas 2: viimased plokid lahti |
| 18 | unfreeze_blocks: 4 |
| 19 | lr_head: 1.0e-3 |
| 20 | lr_head_finetune: 1.0e-4 |
| 21 | lr_backbone: 2.0e-5 |
| 22 | weight_decay: 0.01 |
| 23 | label_smoothing: 0.1 |
| 24 | out_dir: runs/clip_vit_b16 |
| 25 | |