belka_gate_fusions: dataset path update to SSD_2_TB

2026-05-18 14:52:41 +03:00
parent b7542b27de
commit b311913034
4 changed files with 7 additions and 5 deletions
--- 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.