Binary model target

jacobgil · Jun 15, 2023 · 58a565a · 58a565a
1 parent 2183a9c
commit 58a565a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_grad_cam/utils/model_targets.py b/pytorch_grad_cam/utils/model_targets.py
@@ -32,7 +32,7 @@ def __call__(self, model_output):
             sign = 1
         else:
             sign = -1
-        return model_output * sign
+        return torch.abs(model_output) * sign
 
 
 class SoftmaxOutputTarget: