# GTA-UAV Balanced (asymmetric, full MONA): WEB drone encoder + SAT satellite encoder.
# MONA injected into all 24 ViT blocks of each encoder.
# Same loss/sampling/optimizer as gtauav_balanced.gin; differs only in model arch.
#
# Trainable: ~17.6M (MONA 2× × 24 blocks + LoRA + TextFusionMLP + gates + tau)
# Total params: ~733M (2× DINOv3-L + DGTRS-CLIP)
# VRAM target (RTX 4090, 24 GB): ~16-20 GB at batch=8 with gradient checkpointing.

include 'conf/gtauav_balanced.gin'

# ---- Model overrides: asymmetric + full MONA ----
TrainConfigGTAUAV.shared_encoder = False    # WEB for drone, SAT for satellite
TrainConfigGTAUAV.mona_last_n_blocks = 24   # MONA in all 24 ViT blocks (was 12)

# ---- Output ----
TrainConfigGTAUAV.output_dir = "out/gtauav/balanced_asym"