diff --git a/src/training/gradcam.py b/src/training/gradcam.py
index 409ef29..0a94bb7 100644
--- a/src/training/gradcam.py
+++ b/src/training/gradcam.py
@@ -146,7 +146,7 @@ class DINOv3GradCAM:
         # Denormalize from ImageNet stats.
         mean = torch.tensor([0.485, 0.456, 0.406]).view(3, 1, 1).to(image.device)
         std = torch.tensor([0.229, 0.224, 0.225]).view(3, 1, 1).to(image.device)
-        img = (image * std + mean).clamp(0, 1).cpu().numpy().transpose(1, 2, 0)  # [H, W, 3]
+        img = (image * std + mean).clamp(0, 1).detach().cpu().numpy().transpose(1, 2, 0)  # [H, W, 3]
 
         # Resize heatmap to image size.
         heatmap_resized = torch.tensor(heatmap).unsqueeze(0).unsqueeze(0).float()