belka_gate_fusions: inject GateFusionResidual variants to asym and dual encoder

2026-05-18 11:24:58 +03:00
parent e2fd16e66c
commit b7542b27de
4 changed files with 167 additions and 121 deletions
--- a/src/training/train_gtauav.py
+++ b/src/training/train_gtauav.py
@@ -99,7 +99,8 @@ class TrainConfigGTAUAV:

    # Training.
    resume_from: str | None = None  # path to checkpoint for resuming
-    output_dir: str = "out/gtauav/with_text"
+    # output_dir: str = "out/gtauav/with_text"
+    output_dir: str = "out/gtauav/with_text_exp_gate_SRGF"
    epochs: int = 10
    batch_size: int = 8
    num_workers: int = 4