Skip to content

Commit 3c6c24d

Browse files
authored
Improve docs (#554)
* Update common.py * Update inception_v1.py
1 parent 57ea951 commit 3c6c24d

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

captum/optim/_utils/image/common.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ def make_grid_image(
7070

7171
def show(
7272
x: torch.Tensor,
73-
figsize: Optional[Tuple[int, int]] = (8, 8),
73+
figsize: Optional[Tuple[int, int]] = None,
7474
scale: float = 255.0,
7575
images_per_row: Optional[int] = None,
7676
padding: int = 2,
@@ -84,7 +84,7 @@ def show(
8484
x (torch.Tensor): The tensor you want to display as an image.
8585
figsize (Tuple[int, int], optional): height & width to use
8686
for displaying the image figure.
87-
Default: ``(8, 8)``
87+
Default: ``None``
8888
scale (float, optional): Value to multiply the input tensor by so that
8989
it's value range is [0-255] for display.
9090
Default: ``255.0``

captum/optim/models/_image/inception_v1.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ def googlenet(
5353
Default: ``False``
5454
5555
Returns:
56-
**model** (InceptionV1): An Inception5h model instance.
56+
model (InceptionV1): An Inception5h model instance.
5757
"""
5858

5959
if pretrained:

0 commit comments

Comments
 (0)