From bfac53be8a44ee1e648838cd25f920af3dcd3b5d Mon Sep 17 00:00:00 2001 From: unknown Date: Thu, 20 Jul 2023 21:45:55 +0700 Subject: [PATCH] add required gad when init basecam class --- pytorch_grad_cam/base_cam.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pytorch_grad_cam/base_cam.py b/pytorch_grad_cam/base_cam.py index 7ee192971..590dc0489 100644 --- a/pytorch_grad_cam/base_cam.py +++ b/pytorch_grad_cam/base_cam.py @@ -16,6 +16,8 @@ def __init__(self, reshape_transform: Callable = None, compute_input_gradient: bool = False, uses_gradients: bool = True) -> None: + for params in model.parameters(): + params.requires_grad = True self.model = model.eval() self.target_layers = target_layers self.cuda = use_cuda