diff --git a/src/training/gradcam.py b/src/training/gradcam.py
index c9d2e91..409ef29 100644
--- a/src/training/gradcam.py
+++ b/src/training/gradcam.py
@@ -200,8 +200,11 @@ def generate_gradcam_samples(
     save_dir.mkdir(parents=True, exist_ok=True)
 
     model.eval()
-    cam_drone = DINOv3GradCAM(model.drone_encoder, target_layer_idx=-1)
-    cam_sat = DINOv3GradCAM(model.sat_encoder, target_layer_idx=-1)
+    # Shared encoder mode uses single image_encoder for both branches.
+    drone_enc = model.image_encoder if model.shared_encoder else model.drone_encoder
+    sat_enc = model.image_encoder if model.shared_encoder else model.sat_encoder
+    cam_drone = DINOv3GradCAM(drone_enc, target_layer_idx=-1)
+    cam_sat = DINOv3GradCAM(sat_enc, target_layer_idx=-1)
 
     overlays = []
     batch = next(iter(dataloader))