Pikaliov/caption-test
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix gin-config ambiguity: pin module name, remove redundant import

Browse Source 
- Use gin.configurable(module=...) to prevent __main__ vs module name clash
- Remove `import src.training.train_gtauav` from gin files (already loaded)
- Use short selector names (TrainConfigGTAUAV) in all gin configs

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
pikaliov
2026-04-21 20:45:32 +03:00
parent 29a09349e7
commit 69857e0ade
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
conf/gtauav_balanced.gin
View File

@@ -1,9 +1,11 @@
# GTA-UAV Balanced: GatedFusion with L1/L2/L3 captions on both branches.
# query = sigma(alpha) * drone + (1-sigma(alpha)) * text -> InfoNCE vs gallery
# 20 epochs, DINOv3 + DGTRS-CLIP, MONA + LoRA adapters.
# 10 epochs, DINOv3 + DGTRS-CLIP, MONA + LoRA adapters.
#
# NOTE: TrainConfigGTAUAV is registered by train_gtauav.py before gin parsing.
# InfoNCELoss is registered via import below.
import src.losses.multi_infonce
import src.training.train_gtauav
# ---- Training ----
TrainConfigGTAUAV.epochs = 10