From 305a42c32e7b68354c4c3bf54eb8691a9bf11c27 Mon Sep 17 00:00:00 2001 From: Tobias Fischer Date: Mon, 1 Nov 2021 18:32:11 +1000 Subject: [PATCH] Link to arxiv summary instead of pdf in pruning docs (#10282) --- docs/source/advanced/pruning_quantization.rst | 2 +- pytorch_lightning/callbacks/pruning.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/source/advanced/pruning_quantization.rst b/docs/source/advanced/pruning_quantization.rst index b9c8493b90..708a1ecb0d 100644 --- a/docs/source/advanced/pruning_quantization.rst +++ b/docs/source/advanced/pruning_quantization.rst @@ -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 `__, and more! +You can also perform iterative pruning, apply the `lottery ticket hypothesis `__, and more! .. code-block:: python diff --git a/pytorch_lightning/callbacks/pruning.py b/pytorch_lightning/callbacks/pruning.py index 70caaef8ed..83afaaf500 100644 --- a/pytorch_lightning/callbacks/pruning.py +++ b/pytorch_lightning/callbacks/pruning.py @@ -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 `_: + use_lottery_ticket_hypothesis: See `The lottery ticket hypothesis `_: - ``bool``. Whether to apply it or not. - ``Callable[[epoch], bool]``. For dynamic values. Will be called every epoch.