File tree Expand file tree Collapse file tree 1 file changed +14
-8
lines changed
captum/optim/models/_image Expand file tree Collapse file tree 1 file changed +14
-8
lines changed Original file line number Diff line number Diff line change @@ -291,14 +291,20 @@ def __init__(
291
291
"""
292
292
Args:
293
293
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.
302
308
activ (type of nn.Module, optional): The nn.Module class type to use for
303
309
activation layers.
304
310
Default: ``nn.ReLU``
You can’t perform that action at this time.
0 commit comments