belka_gate_fusions: dataset path update to SSD_2_TB

conf/gtauav_balanced.gin

@@ -43,7 +43,7 @@ TrainConfigGTAUAV.dss_reembed_every = 1
 TrainConfigGTAUAV.use_mutex_sampler = True  # legacy flag, kept True unless disabling both samplers
 
 # ---- Output ----
-TrainConfigGTAUAV.output_dir = "out/gtauav/with_text"
+TrainConfigGTAUAV.output_dir = "out/gtauav/with_text_exp_gate_SRGF"
 
 # ---- Tracking ----
 TrainConfigGTAUAV.use_wandb = False

src/models/asymmetric_encoder.py

@@ -261,7 +261,7 @@ class TextFusionMLP(nn.Module):
 # ---------------------------------------------------------------------------
 
 # ResidualGateFusin experiment
-from residual_fusions import ResidualGateType, GatedFusionResidual
+from .residual_fusions import ResidualGateType, GatedFusionResidual
 
 class AsymmetricEncoder(nn.Module):
     """Dual encoder for CVGL with text fusion on both branches.

src/models/dual_encoder.py

@@ -22,7 +22,8 @@ import torch.nn as nn
 import torch.nn.functional as F
 
 # residual fusions exp
-from residual_fusions import ResidualGateType, GatedFusionResidual
+from .residual_fusions import ResidualGateType, GatedFusionResidual
+# from residual_fusions import ResidualGateType, GatedFusionResidual
 
 
 class ProjectionHead(nn.Module):

src/training/train_gtauav.py

@@ -58,8 +58,8 @@ from src.models.asymmetric_encoder import (
 LOGGER = logging.getLogger("caption_test.train_gtauav")
 
 # Default paths.
-_RGB_ROOT = "/home/servml/Документы/datasets/GTA-UAV-LR"
-_CAPTION_ROOT = "/home/servml/Документы/datasets/GTA-UAV-LR-captions"
+_RGB_ROOT = "/media/servml/SSD_2_2TB/datasets/cvgl_datasets/GTA-UAV-LR/"
+_CAPTION_ROOT = "/media/servml/SSD_2_2TB/datasets/cvgl_datasets/GTA-UAV-LR-captions/"
 _TRAIN_JSON = "meta/train_80.json"
 _TEST_JSON = "meta/test_20.json"
 
@@ -100,6 +100,7 @@ class TrainConfigGTAUAV:
     # Training.
     resume_from: str | None = None  # path to checkpoint for resuming
     # output_dir: str = "out/gtauav/with_text"
+    # ! ----------------- experimental gate outs path -----------------------
     output_dir: str = "out/gtauav/with_text_exp_gate_SRGF"
     epochs: int = 10
     batch_size: int = 8