From b3119130345100fd6e8ac41ac815c34fdcdd17a9 Mon Sep 17 00:00:00 2001 From: pikaliov Date: Mon, 18 May 2026 14:52:41 +0300 Subject: [PATCH] belka_gate_fusions: dataset path update to SSD_2_TB --- conf/gtauav_balanced.gin | 2 +- src/models/asymmetric_encoder.py | 2 +- src/models/dual_encoder.py | 3 ++- src/training/train_gtauav.py | 5 +++-- 4 files changed, 7 insertions(+), 5 deletions(-) diff --git a/conf/gtauav_balanced.gin b/conf/gtauav_balanced.gin index f96518a..12c613e 100644 --- a/conf/gtauav_balanced.gin +++ b/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 diff --git a/src/models/asymmetric_encoder.py b/src/models/asymmetric_encoder.py index da78a1c..1001c30 100644 --- a/src/models/asymmetric_encoder.py +++ b/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. diff --git a/src/models/dual_encoder.py b/src/models/dual_encoder.py index 89d435e..8d6f743 100644 --- a/src/models/dual_encoder.py +++ b/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): diff --git a/src/training/train_gtauav.py b/src/training/train_gtauav.py index 7e12c3e..e17705a 100644 --- a/src/training/train_gtauav.py +++ b/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