framework.rst 1.33 KB
Newer Older
J-shang's avatar
J-shang committed
1
2
Framework Related
=================
3
4
5
6

Pruner
------

J-shang's avatar
J-shang committed
7
.. autoclass:: nni.algorithms.compression.v2.pytorch.base.Pruner
8
9
    :members:

J-shang's avatar
J-shang committed
10
11
PrunerModuleWrapper
-------------------
12

J-shang's avatar
J-shang committed
13
.. autoclass:: nni.algorithms.compression.v2.pytorch.base.PrunerModuleWrapper
14

J-shang's avatar
J-shang committed
15
16
BasicPruner
-----------
17

J-shang's avatar
J-shang committed
18
.. autoclass:: nni.algorithms.compression.v2.pytorch.pruning.basic_pruner.BasicPruner
19
20
    :members:

J-shang's avatar
J-shang committed
21
22
DataCollector
-------------
23

J-shang's avatar
J-shang committed
24
.. autoclass:: nni.algorithms.compression.v2.pytorch.pruning.tools.DataCollector
25
26
    :members:

J-shang's avatar
J-shang committed
27
MetricsCalculator
28
29
30
31
32
-----------------

.. autoclass:: nni.algorithms.compression.v2.pytorch.pruning.tools.MetricsCalculator
    :members:

J-shang's avatar
J-shang committed
33
34
35
SparsityAllocator
-----------------

36
37
38
.. autoclass:: nni.algorithms.compression.v2.pytorch.pruning.tools.SparsityAllocator
    :members:

J-shang's avatar
J-shang committed
39
40
41
BasePruningScheduler
--------------------

42
43
44
.. autoclass:: nni.algorithms.compression.v2.pytorch.base.BasePruningScheduler
    :members:

J-shang's avatar
J-shang committed
45
46
47
TaskGenerator
-------------

48
49
50
.. autoclass:: nni.algorithms.compression.v2.pytorch.pruning.tools.TaskGenerator
    :members:

J-shang's avatar
J-shang committed
51
52
53
Quantizer
---------

54
55
56
.. autoclass:: nni.compression.pytorch.compressor.Quantizer
    :members:

J-shang's avatar
J-shang committed
57
58
59
QuantizerModuleWrapper
----------------------

60
61
62
.. autoclass:: nni.compression.pytorch.compressor.QuantizerModuleWrapper
    :members:

J-shang's avatar
J-shang committed
63
64
65
QuantGrad
---------

66
67
.. autoclass:: nni.compression.pytorch.compressor.QuantGrad
    :members: