diff --git a/conf/gtauav_balanced.gin b/conf/gtauav_balanced.gin
index ce3075e..b67a2e9 100644
--- a/conf/gtauav_balanced.gin
+++ b/conf/gtauav_balanced.gin
@@ -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
diff --git a/src/training/train_gtauav.py b/src/training/train_gtauav.py
index 5a1c55e..82fd3eb 100644
--- a/src/training/train_gtauav.py
+++ b/src/training/train_gtauav.py
@@ -55,7 +55,7 @@ _DINO_SAT = "nn_models/DINO_SAT/model.safetensors"
 _LRSCLIP = "nn_models/LRSCLIP/DGTRS-CLIP-ViT-L-14.pt"
 
 
-@gin.configurable
+@gin.configurable(module="src.training.train_gtauav")
 @dataclass
 class TrainConfigGTAUAV:
     """Training configuration for GTA-UAV experiment."""