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. # GTA-UAV Balanced: GatedFusion with L1/L2/L3 captions on both branches.
# query = sigma(alpha) * drone + (1-sigma(alpha)) * text -> InfoNCE vs gallery # 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.losses.multi_infonce
import src.training.train_gtauav
# ---- Training ---- # ---- Training ----
TrainConfigGTAUAV.epochs = 10 TrainConfigGTAUAV.epochs = 10

2
src/training/train_gtauav.py
View File

@@ -55,7 +55,7 @@ _DINO_SAT = "nn_models/DINO_SAT/model.safetensors"
_LRSCLIP = "nn_models/LRSCLIP/DGTRS-CLIP-ViT-L-14.pt" _LRSCLIP = "nn_models/LRSCLIP/DGTRS-CLIP-ViT-L-14.pt"
@gin.configurable @gin.configurable(module="src.training.train_gtauav")
@dataclass @dataclass
class TrainConfigGTAUAV: class TrainConfigGTAUAV:
"""Training configuration for GTA-UAV experiment.""" """Training configuration for GTA-UAV experiment."""