Skip to content

Latest commit

 

History

History
155 lines (102 loc) · 4.25 KB

CompressionReference.rst

File metadata and controls

155 lines (102 loc) · 4.25 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_masker.StructuredWeightMasker

Pruners

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

nni.algorithms.compression.pytorch.pruning.one_shot_pruner.OneshotPruner

nni.algorithms.compression.pytorch.pruning.one_shot_pruner.LevelPruner

nni.algorithms.compression.pytorch.pruning.one_shot_pruner.L1FilterPruner

nni.algorithms.compression.pytorch.pruning.one_shot_pruner.L2FilterPruner

nni.algorithms.compression.pytorch.pruning.one_shot_pruner.FPGMPruner

nni.algorithms.compression.pytorch.pruning.iterative_pruner.IterativePruner

nni.algorithms.compression.pytorch.pruning.iterative_pruner.SlimPruner

nni.algorithms.compression.pytorch.pruning.iterative_pruner.TaylorFOWeightFilterPruner

nni.algorithms.compression.pytorch.pruning.iterative_pruner.ActivationAPoZRankFilterPruner

nni.algorithms.compression.pytorch.pruning.iterative_pruner.ActivationMeanRankFilterPruner

nni.algorithms.compression.pytorch.pruning.iterative_pruner.AGPPruner

nni.algorithms.compression.pytorch.pruning.iterative_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

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

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