revert back the binary model output

jacobgil · jacobgil · commit af2d53ad6c80 · 2023-07-23T14:23:13.000+03:00
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 torch.abs(model_output) * sign
+        return model_output * sign
 
 
 class SoftmaxOutputTarget:
