We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8b6c3d2 commit e8fbc2aCopy full SHA for e8fbc2a
captum/optim/_utils/circuits.py
@@ -30,8 +30,8 @@ def extract_expanded_weights(
30
specified for target2.
31
target2 (nn.Module): The end target layer. Must be above the layer
32
specified for target1.
33
- crop_shape (int or tuple of ints, optional): Specify the exact output size
34
- to crop out.
+ crop_shape (int or tuple of ints, optional): Specify the output weight
+ size to enter crop away padding.
35
model_input (tensor or tuple of tensors, optional): The input to use
36
with the specified model.
37
crop_func (Callable, optional): Specify a function to crop away the padding
captum/optim/_utils/models.py
0 commit comments