From a32cb8ac6c77eabf39cf002e72da3f9756aded5c Mon Sep 17 00:00:00 2001 From: pikaliov Date: Wed, 22 Apr 2026 07:02:12 +0300 Subject: [PATCH] Fix Grad-CAM numpy() on tensor with grad (add detach) Co-Authored-By: Claude Opus 4.6 (1M context) --- src/training/gradcam.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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()