- Add unified experiment tracker (W&B + TensorBoard) with graceful fallback - Add gradient norm monitoring per param group (MONA, LoRA, MLP, gates, tau) - Add Grad-CAM visualization for DINOv3 drone/satellite encoders - Add PyTorch Profiler wrapper + torchinfo model summary - Add gin-config support to train_gtauav.py with CLI overrides - Add v3 gin configs: gtauav_balanced, gtauav_baseline, gtauav_text_heavy, gtauav_image_heavy - Generate metric plots every epoch (not just on eval) - Set default epochs to 10 - Update README and CLAUDE.md with new tooling and usage docs Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
10 lines
308 B
Plaintext
10 lines
308 B
Plaintext
# GTA-UAV Baseline: no text fusion (gate forced to 1.0).
|
|
# query = drone_only -> InfoNCE vs satellite
|
|
# Reference R@1 for delta computation.
|
|
|
|
include 'conf/gtauav_balanced.gin'
|
|
|
|
TrainConfigGTAUAV.baseline_mode = True
|
|
TrainConfigGTAUAV.output_dir = "out/gtauav/baseline"
|
|
TrainConfigGTAUAV.use_gradcam = False
|