Skip to content

Commit 675e9e4

Browse files
author
Svetlana Karslioglu
committed
Update
1 parent 3aaa3d2 commit 675e9e4

File tree

4 files changed

+22
-14
lines changed

4 files changed

+22
-14
lines changed

.pyspelling.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,8 @@ matrix:
2222
#- intermediate_source/nvfuser_intro_tutorial.py
2323
#- intermediate_source/parametrizations.py
2424
#- intermediate_source/per_sample_grads.py
25-
- intermediate_source/pipeline_tutorial.py
25+
#- intermediate_source/pipeline_tutorial.py
26+
- intermediate_source/pruning_tutorial.py
2627
dictionary:
2728
wordlists:
2829
- en-wordlist.txt

en-wordlist.txt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,9 @@
1+
subnetworks
2+
sparsify
3+
LeCun
4+
prepruned
5+
dimensionality
6+
unpruned
17
RPC
28
multihead
39
GPU's

intermediate_source/pipeline_tutorial.py

Lines changed: 12 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@
3535
# As a result, our focus is on ``nn.TransformerEncoder`` and we split the model
3636
# such that half of the ``nn.TransformerEncoderLayer`` are on one GPU and the
3737
# other half are on another. To do this, we pull out the ``Encoder`` and
38-
# ``Decoder`` sections into seperate modules and then build an ``nn.Sequential``
38+
# ``Decoder`` sections into separate modules and then build an ``nn.Sequential``
3939
# representing the original Transformer module.
4040

4141
import sys
@@ -134,16 +134,17 @@ def forward(self, x):
134134
# length 6:
135135
#
136136
# .. math::
137-
# \begin{bmatrix}
138-
# \text{A} & \text{B} & \text{C} & \ldots & \text{X} & \text{Y} & \text{Z}
139-
# \end{bmatrix}
140-
# \Rightarrow
141-
# \begin{bmatrix}
142-
# \begin{bmatrix}\text{A} \\ \text{B} \\ \text{C} \\ \text{D} \\ \text{E} \\ \text{F}\end{bmatrix} &
143-
# \begin{bmatrix}\text{G} \\ \text{H} \\ \text{I} \\ \text{J} \\ \text{K} \\ \text{L}\end{bmatrix} &
144-
# \begin{bmatrix}\text{M} \\ \text{N} \\ \text{O} \\ \text{P} \\ \text{Q} \\ \text{R}\end{bmatrix} &
145-
# \begin{bmatrix}\text{S} \\ \text{T} \\ \text{U} \\ \text{V} \\ \text{W} \\ \text{X}\end{bmatrix}
146-
# \end{bmatrix}
137+
#
138+
# \begin{bmatrix}
139+
# \text{A} & \text{B} & \text{C} & \ldots & \text{X} & \text{Y} & \text{Z}
140+
# \end{bmatrix}
141+
# \Rightarrow
142+
# \begin{bmatrix}
143+
# \begin{bmatrix}\text{A} \\ \text{B} \\ \text{C} \\ \text{D} \\ \text{E} \\ \text{F}\end{bmatrix} &
144+
# \begin{bmatrix}\text{G} \\ \text{H} \\ \text{I} \\ \text{J} \\ \text{K} \\ \text{L}\end{bmatrix} &
145+
# \begin{bmatrix}\text{M} \\ \text{N} \\ \text{O} \\ \text{P} \\ \text{Q} \\ \text{R}\end{bmatrix} &
146+
# \begin{bmatrix}\text{S} \\ \text{T} \\ \text{U} \\ \text{V} \\ \text{W} \\ \text{X}\end{bmatrix}
147+
# \end{bmatrix}
147148
#
148149
# These columns are treated as independent by the model, which means that
149150
# the dependence of ``G`` and ``F`` can not be learned, but allows more

intermediate_source/pruning_tutorial.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -339,8 +339,8 @@ def forward(self, x):
339339
# pruning this technique implements (supported options are ``global``,
340340
# ``structured``, and ``unstructured``). This is needed to determine
341341
# how to combine masks in the case in which pruning is applied
342-
# iteratively. In other words, when pruning a pre-pruned parameter,
343-
# the current prunining techique is expected to act on the unpruned
342+
# iteratively. In other words, when pruning a prepruned parameter,
343+
# the current pruning technique is expected to act on the unpruned
344344
# portion of the parameter. Specifying the ``PRUNING_TYPE`` will
345345
# enable the ``PruningContainer`` (which handles the iterative
346346
# application of pruning masks) to correctly identify the slice of the

0 commit comments

Comments
 (0)