docs: Fixed documentation examples (#28)

frgfm · Dec 23, 2020 · 1ffb38b · 1ffb38b
1 parent 3858ab9
commit 1ffb38b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/torchcam/cams/gradcam.py b/torchcam/cams/gradcam.py
@@ -76,7 +76,7 @@ class GradCAM(_GradCAM):
         >>> from torchcam.cams import GradCAM
         >>> model = resnet18(pretrained=True).eval()
         >>> cam = GradCAM(model, 'layer4')
-        >>> with torch.no_grad(): scores = model(input_tensor)
+        >>> scores = model(input_tensor)
         >>> cam(class_idx=100, scores=scores)
 
     Args:
@@ -127,7 +127,7 @@ class GradCAMpp(_GradCAM):
         >>> from torchcam.cams import GradCAMpp
         >>> model = resnet18(pretrained=True).eval()
         >>> cam = GradCAMpp(model, 'layer4')
-        >>> with torch.no_grad(): scores = model(input_tensor)
+        >>> scores = model(input_tensor)
         >>> cam(class_idx=100, scores=scores)
 
     Args:
@@ -192,7 +192,7 @@ class SmoothGradCAMpp(_GradCAM):
         >>> from torchcam.cams import SmoothGradCAMpp
         >>> model = resnet18(pretrained=True).eval()
         >>> cam = SmoothGradCAMpp(model, 'layer4', 'conv1')
-        >>> with torch.no_grad(): scores = model(input_tensor)
+        >>> scores = model(input_tensor)
         >>> cam(class_idx=100)
 
     Args: