diff --git a/pytorch_grad_cam/utils/model_targets.py b/pytorch_grad_cam/utils/model_targets.py index 489dd1987..343fbd398 100644 --- 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: