Pikaliov/caption-test
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

belka_gate_fusions: dataset path update to SSD_2_TB

Browse Source
This commit is contained in:
pikaliov
2026-05-18 14:52:41 +03:00
parent b7542b27de
commit b311913034
4 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/training/train_gtauav.py
View File

@@ -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