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

Init refactor branch

Browse Source
This commit is contained in:
pikaliov
2026-04-27 11:57:29 +03:00
parent c6fcd2222c
commit a27b5a7357
3 changed files with 593 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
conf/gtauav_balanced.gin
View File

@@ -34,10 +34,10 @@ TrainConfigGTAUAV.label_smoothing = 0.1
TrainConfigGTAUAV.learnable_temperature = True
TrainConfigGTAUAV.weight_q2g = 0.6
TrainConfigGTAUAV.weight_g2q = 0.4
TrainConfigGTAUAV.neg_bank_size = 4096
TrainConfigGTAUAV.neg_bank_size = 0 # 4096
# ---- Sampling ----
TrainConfigGTAUAV.sampler_type = "dss" # "dss" or "mutex"
TrainConfigGTAUAV.sampler_type = "mutex" # "dss" or "mutex"
TrainConfigGTAUAV.dss_warmup_epochs = 1 # first N epochs use mutex-only (untrained embeds not useful)
TrainConfigGTAUAV.dss_reembed_every = 1
TrainConfigGTAUAV.use_mutex_sampler = True # legacy flag, kept True unless disabling both samplers
@@ -48,7 +48,7 @@ TrainConfigGTAUAV.output_dir = "out/gtauav/with_text"
# ---- Tracking ----
TrainConfigGTAUAV.use_wandb = False
TrainConfigGTAUAV.use_tb = True
TrainConfigGTAUAV.use_gradcam = True
TrainConfigGTAUAV.use_gradcam = False
TrainConfigGTAUAV.gradcam_every = 5
TrainConfigGTAUAV.use_profiler = False
TrainConfigGTAUAV.log_grad_norms = True
@@ -61,4 +61,4 @@ InfoNCELoss.weight_q2g = 0.6
InfoNCELoss.weight_g2q = 0.4
InfoNCELoss.tau_min = 0.01
InfoNCELoss.tau_max = 0.1
InfoNCELoss.hard_mining_k = 512 # 0 = use whole queue (disable mining)
InfoNCELoss.hard_mining_k = 0 # 512 # 0 = use whole queue (disable mining)