Update docs: training improvements (learnable temp, augmentations, warmup)

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
pikaliov
2026-04-21 18:08:06 +03:00
parent 998d52cb57
commit 3014a5def8
2 changed files with 12 additions and 3 deletions

View File

@@ -149,9 +149,17 @@ Meta-файл `meta/seg_filter.json`: исключение изображени
## Конфигурации
### V3 (GTA-UAV)
Параметры: 10 epochs, batch 64, lr=1e-4, AMP, cosine LR, image 256x256.
Train: 15,693 pairs (cross-area), Test: 18,015 pairs.
Eval на cross-area split (primary).
Параметры:
- 10 epochs, batch 64, AMP, image 256x256
- **Optimizer:** AdamW, per-group LR: proj=1e-4, text=1e-5 (10x lower)
- **Scheduler:** linear warmup (2 epochs) + cosine annealing (per-step)
- **Loss:** InfoNCE с learnable temperature (CLIP logit_scale), init=0.07, clamp [0.01, 0.5]
- **Augmentations:**
- Drone: RandomResizedCrop(0.7-1.0), HFlip, Rotation(15°), ColorJitter, Grayscale(5%), GaussianBlur
- Satellite: RandomResizedCrop(0.7-1.0), HFlip, ColorJitter, Grayscale(5%)
- Eval: Resize+CenterCrop (clean, no augmentation)
- Train: 15,693 pairs → 13,622 after seg filter (cross-area)
- Test: 18,015 pairs
### V2 (UAV-GeoLoc, gin)
| Конфиг | Gate init | Описание |