Skip to content

Commit 336c62f

Browse files
authored
Spelling fixes and add top channels to tutorial
1 parent 6892f3c commit 336c62f

File tree

2 files changed

+92
-18
lines changed

2 files changed

+92
-18
lines changed

captum/optim/_utils/circuits.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,8 +65,8 @@ def get_expanded_weights(
6565

6666
def max2avg_pool2d(model, value: Optional[Any] = float("-inf")) -> None:
6767
"""
68-
Replace all non-linear MaxPool2d layers with their linear AvgPool2d equivalents.
69-
This allows us to ignore non-linear values when calculating expanded weights.
68+
Replace all nonlinear MaxPool2d layers with their linear AvgPool2d equivalents.
69+
This allows us to ignore nonlinear values when calculating expanded weights.
7070
7171
Args:
7272
model (nn.Module): A PyTorch model instance.

tutorials/optimviz/WeightVisualization_OptimViz.ipynb

Lines changed: 90 additions & 16 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)