| # Training configuration for CLIPSeg fine-tuning | |
| seed: 42 | |
| # Data | |
| data: | |
| split_ratios: [0.70, 0.15, 0.15] # train / val / test | |
| image_size: 352 # CLIPSeg native resolution | |
| prompt_synonyms: | |
| taping: | |
| - "segment taping area" | |
| - "segment joint tape" | |
| - "segment drywall seam" | |
| - "segment drywall joint" | |
| - "segment tape line" | |
| cracks: | |
| - "segment crack" | |
| - "segment wall crack" | |
| - "segment surface crack" | |
| - "segment drywall crack" | |
| - "segment fracture" | |
| # Model | |
| model: | |
| name: "CIDAS/clipseg-rd64-refined" | |
| freeze_backbone: true | |
| # Training | |
| training: | |
| batch_size: 8 | |
| lr: 1.0e-4 | |
| weight_decay: 1.0e-4 | |
| epochs: 30 | |
| patience: 7 # early stopping | |
| bce_weight: 0.5 | |
| dice_weight: 0.5 | |
| scheduler: "cosine" | |
| num_workers: 0 # required for MPS | |
| # Evaluation | |
| evaluation: | |
| threshold: 0.5 | |
| num_visual_examples: 4 | |