Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
pruner.rst 2.03 KB
Newer Older
Yuge Zhang's avatar
Resolve conflicts for #4760 (#4762)  
Yuge Zhang committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123 
Pruner
======

Basic Pruner
------------

.. _level-pruner:

Level Pruner
^^^^^^^^^^^^

..  autoclass:: nni.compression.pytorch.pruning.LevelPruner

.. _l1-norm-pruner:

L1 Norm Pruner
^^^^^^^^^^^^^^

.. autoclass:: nni.compression.pytorch.pruning.L1NormPruner

.. _l2-norm-pruner:

L2 Norm Pruner
^^^^^^^^^^^^^^

.. autoclass:: nni.compression.pytorch.pruning.L2NormPruner

.. _fpgm-pruner:

FPGM Pruner
^^^^^^^^^^^

.. autoclass:: nni.compression.pytorch.pruning.FPGMPruner

.. _slim-pruner:

Slim Pruner
^^^^^^^^^^^

.. autoclass:: nni.compression.pytorch.pruning.SlimPruner

.. _activation-apoz-rank-pruner:

Activation APoZ Rank Pruner
^^^^^^^^^^^^^^^^^^^^^^^^^^^

.. autoclass:: nni.compression.pytorch.pruning.ActivationAPoZRankPruner

.. _activation-mean-rank-pruner:

Activation Mean Rank Pruner
^^^^^^^^^^^^^^^^^^^^^^^^^^^

.. autoclass:: nni.compression.pytorch.pruning.ActivationMeanRankPruner

.. _taylor-fo-weight-pruner:

Taylor FO Weight Pruner
^^^^^^^^^^^^^^^^^^^^^^^

.. autoclass:: nni.compression.pytorch.pruning.TaylorFOWeightPruner

.. _admm-pruner:

ADMM Pruner
^^^^^^^^^^^

.. autoclass:: nni.compression.pytorch.pruning.ADMMPruner

Scheduled Pruners
-----------------

.. _linear-pruner:

Linear Pruner
^^^^^^^^^^^^^

.. autoclass:: nni.compression.pytorch.pruning.LinearPruner

.. _agp-pruner:

AGP Pruner
^^^^^^^^^^

.. autoclass:: nni.compression.pytorch.pruning.AGPPruner

.. _lottery-ticket-pruner:

Lottery Ticket Pruner
^^^^^^^^^^^^^^^^^^^^^

.. autoclass:: nni.compression.pytorch.pruning.LotteryTicketPruner

.. _simulated-annealing-pruner:

Simulated Annealing Pruner
^^^^^^^^^^^^^^^^^^^^^^^^^^

.. autoclass:: nni.compression.pytorch.pruning.SimulatedAnnealingPruner

.. _auto-compress-pruner:

Auto Compress Pruner
^^^^^^^^^^^^^^^^^^^^

.. autoclass:: nni.compression.pytorch.pruning.AutoCompressPruner

.. _amc-pruner:

AMC Pruner
^^^^^^^^^^

..  autoclass:: nni.compression.pytorch.pruning.AMCPruner

Other Pruner
------------

.. _movement-pruner:

Movement Pruner
^^^^^^^^^^^^^^^

.. autoclass:: nni.compression.pytorch.pruning.MovementPruner