revert back the binary model output

jacobgil · Jul 23, 2023 · af2d53a · af2d53a
1 parent c01a950
commit af2d53a
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 torch.abs(model_output) * sign
+        return model_output * sign
 
 
 class SoftmaxOutputTarget: