Skip to content

Latest commit

 

History

History
152 lines (100 loc) · 4.06 KB

CompressionReference.rst

File metadata and controls

152 lines (100 loc) · 4.06 KB
Raw

Model Compression API Reference

Compressors

Compressor

nni.compression.pytorch.compressor.Compressor

nni.compression.pytorch.compressor.Pruner

nni.compression.pytorch.compressor.Quantizer

Module Wrapper

nni.compression.pytorch.compressor.PrunerModuleWrapper

nni.compression.pytorch.compressor.QuantizerModuleWrapper

Weight Masker

nni.algorithms.compression.pytorch.pruning.weight_masker.WeightMasker

nni.algorithms.compression.pytorch.pruning.structured_pruning.StructuredWeightMasker

Pruners

nni.algorithms.compression.pytorch.pruning.sensitivity_pruner.SensitivityPruner

nni.algorithms.compression.pytorch.pruning.one_shot.OneshotPruner

nni.algorithms.compression.pytorch.pruning.one_shot.LevelPruner

nni.algorithms.compression.pytorch.pruning.one_shot.SlimPruner

nni.algorithms.compression.pytorch.pruning.one_shot.L1FilterPruner

nni.algorithms.compression.pytorch.pruning.one_shot.L2FilterPruner

nni.algorithms.compression.pytorch.pruning.one_shot.FPGMPruner

nni.algorithms.compression.pytorch.pruning.one_shot.TaylorFOWeightFilterPruner

nni.algorithms.compression.pytorch.pruning.one_shot.ActivationAPoZRankFilterPruner

nni.algorithms.compression.pytorch.pruning.one_shot.ActivationMeanRankFilterPruner

nni.algorithms.compression.pytorch.pruning.lottery_ticket.LotteryTicketPruner

nni.algorithms.compression.pytorch.pruning.agp.AGPPruner

nni.algorithms.compression.pytorch.pruning.admm_pruner.ADMMPruner

nni.algorithms.compression.pytorch.pruning.auto_compress_pruner.AutoCompressPruner

nni.algorithms.compression.pytorch.pruning.net_adapt_pruner.NetAdaptPruner

nni.algorithms.compression.pytorch.pruning.simulated_annealing_pruner.SimulatedAnnealingPruner

Quantizers

nni.algorithms.compression.pytorch.quantization.quantizers.NaiveQuantizer

nni.algorithms.compression.pytorch.quantization.quantizers.QAT_Quantizer

nni.algorithms.compression.pytorch.quantization.quantizers.DoReFaQuantizer

nni.algorithms.compression.pytorch.quantization.quantizers.BNNQuantizer

Model Speedup

Quantization Speedup

nni.compression.pytorch.quantization_speedup.backend.BaseModelSpeedup

nni.compression.pytorch.quantization_speedup.integrated_tensorrt.ModelSpeedupTensorRT

nni.compression.pytorch.quantization_speedup.calibrator.Calibrator

Compression Utilities

Sensitivity Utilities

nni.compression.pytorch.utils.sensitivity_analysis.SensitivityAnalysis

Topology Utilities

nni.compression.pytorch.utils.shape_dependency.ChannelDependency

nni.compression.pytorch.utils.shape_dependency.GroupDependency

nni.compression.pytorch.utils.mask_conflict.CatMaskPadding

nni.compression.pytorch.utils.mask_conflict.GroupMaskConflict

nni.compression.pytorch.utils.mask_conflict.ChannelMaskConflict

Model FLOPs/Parameters Counter

nni.compression.pytorch.utils.counter.count_flops_params