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 43691d4 commit 60b930aCopy full SHA for 60b930a
src/lightning/pytorch/callbacks/pruning.py
@@ -129,7 +129,7 @@ def __init__(
129
- ``bool``. Always apply it or not.
130
- ``Callable[[epoch], bool]``. For dynamic values. Will be called every epoch.
131
132
- make_pruning_permanent: Whether to remove all reparametrization pre-hooks and apply masks
+ make_pruning_permanent: Whether to remove all reparameterization pre-hooks and apply masks
133
when training ends or the model is saved.
134
135
use_lottery_ticket_hypothesis: See `The lottery ticket hypothesis <https://arxiv.org/abs/1803.03635>`_:
0 commit comments