CompressionReference.md 826 Bytes
Newer Older
Chi Song's avatar
Chi Song 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
# 模型压缩 Python API 参考

```eval_rst
.. contents::
```

## 灵敏度工具

```eval_rst
..  autoclass:: nni.compression.torch.utils.sensitivity_analysis.SensitivityAnalysis
    :members:

```

## 拓扑结构工具

```eval_rst
..  autoclass:: nni.compression.torch.utils.shape_dependency.ChannelDependency
    :members:

..  autoclass:: nni.compression.torch.utils.shape_dependency.GroupDependency
    :members:

..  autoclass:: nni.compression.torch.utils.mask_conflict.CatMaskPadding
    :members:

..  autoclass:: nni.compression.torch.utils.mask_conflict.GroupMaskConflict
    :members:

..  autoclass:: nni.compression.torch.utils.mask_conflict.ChannelMaskConflict
    :members:

```

## 模型 FLOPs 和参数计数器

```eval_rst
..  autofunction:: nni.compression.torch.utils.counter.count_flops_params

```