Skip to content

Commit 07c9e60

Browse files
authored
Add missing InceptionV1 InceptionModule docs
1 parent 3c6c24d commit 07c9e60

File tree

1 file changed

+14
-8
lines changed

1 file changed

+14
-8
lines changed

captum/optim/models/_image/inception_v1.py

Lines changed: 14 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -291,14 +291,20 @@ def __init__(
291291
"""
292292
Args:
293293
294-
in_channels (int, optional): The number of input channels to use for the
295-
inception module.
296-
c1x1 (int, optional):
297-
c3x3reduce (int, optional):
298-
c3x3 (int, optional):
299-
c5x5reduce (int, optional):
300-
c5x5 (int, optional):
301-
pool_proj (int, optional):
294+
in_channels (int): The number of input channels to use for the first
295+
layers of the inception module branches.
296+
c1x1 (int): The number of output channels to use for the first layer in
297+
the c1x1 branch.
298+
c3x3reduce (int): The number of output channels to use for the first layer
299+
in the c3x3 branch.
300+
c3x3 (int): The number of output channels to use for the second layer in
301+
the c3x3 branch.
302+
c5x5reduce (int): The number of output channels to use for the first layer
303+
in the c5x5 branch.
304+
c5x5 (int): The number of output channels to use for the second layer in
305+
the c5x5 branch.
306+
pool_proj (int): The number of output channels to use for the second layer
307+
in the pool branch.
302308
activ (type of nn.Module, optional): The nn.Module class type to use for
303309
activation layers.
304310
Default: ``nn.ReLU``

0 commit comments

Comments
 (0)