Link to arxiv summary instead of pdf in pruning docs (#10282)

This commit is contained in:
Tobias Fischer 2021-11-01 18:32:11 +10:00 committed by GitHub
parent cb8a2165a3
commit 305a42c32e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions

View File

@ -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

View File

@ -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.