Link to arxiv summary instead of pdf in pruning docs (#10282)
This commit is contained in:
parent
cb8a2165a3
commit
305a42c32e
|
@ -35,7 +35,7 @@ This callback supports multiple pruning functions: pass any `torch.nn.utils.prun
|
|||
# set the amount to be the fraction of parameters to prune
|
||||
trainer = Trainer(callbacks=[ModelPruning("l1_unstructured", amount=0.5)])
|
||||
|
||||
You can also perform iterative pruning, apply the `lottery ticket hypothesis <https://arxiv.org/pdf/1803.03635.pdf>`__, and more!
|
||||
You can also perform iterative pruning, apply the `lottery ticket hypothesis <https://arxiv.org/abs/1803.03635>`__, and more!
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
|
|
|
@ -131,7 +131,7 @@ class ModelPruning(Callback):
|
|||
make_pruning_permanent: Whether to remove all reparametrization pre-hooks and apply masks
|
||||
when training ends or the model is saved.
|
||||
|
||||
use_lottery_ticket_hypothesis: See `The lottery ticket hypothesis <https://arxiv.org/pdf/1803.03635.pdf>`_:
|
||||
use_lottery_ticket_hypothesis: See `The lottery ticket hypothesis <https://arxiv.org/abs/1803.03635>`_:
|
||||
|
||||
- ``bool``. Whether to apply it or not.
|
||||
- ``Callable[[epoch], bool]``. For dynamic values. Will be called every epoch.
|
||||
|
|
Loading…
Reference in New Issue