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

belka_gate_fusions: trainable version

Browse Source
This commit is contained in:
pikaliov
2026-05-18 15:48:49 +03:00
parent b311913034
commit a2729c065e
3 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/training/train.py
View File

@@ -56,8 +56,8 @@ class TrainConfig:
self,
train_query_file: str = "Index/train_query.txt",
val_query_file: str = "Index/test_query.txt",
data_root: str = "/mnt/data1tb/cvgl_datasets/UAV-GeoLoc",
output_dir: str = "out/caption_test",
data_root: str = "/media/servml/SSD_2_TB/datasets/cvgl_datasets/UAV-GeoLoc",
output_dir: str = "out/caption_test_exp_gate_SRGF",
epochs: int = 10,
batch_size: int = 128,
num_workers: int = 4,