Skip to content

Commit 60b930a

Browse files
authored
Update pruning.py
1 parent 43691d4 commit 60b930a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/lightning/pytorch/callbacks/pruning.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -129,7 +129,7 @@ def __init__(
129129
- ``bool``. Always apply it or not.
130130
- ``Callable[[epoch], bool]``. For dynamic values. Will be called every epoch.
131131
132-
make_pruning_permanent: Whether to remove all reparametrization pre-hooks and apply masks
132+
make_pruning_permanent: Whether to remove all reparameterization pre-hooks and apply masks
133133
when training ends or the model is saved.
134134
135135
use_lottery_ticket_hypothesis: See `The lottery ticket hypothesis <https://arxiv.org/abs/1803.03635>`_:

0 commit comments

Comments
 (0)