nncf.common.pruning.statistics
#
Classes#
Contains statistics of the filter pruning algorithm. |
- class nncf.common.pruning.statistics.FilterPruningStatistics(model_statistics, current_pruning_level, target_pruning_level, prune_flops)[source]#
Bases:
nncf.api.statistics.Statistics
Contains statistics of the filter pruning algorithm.
- Parameters:
model_statistics (PrunedModelStatistics) – Statistics of the pruned model.
current_pruning_level (float) – A current level of the pruning for the algorithm for the current epoch.
target_pruning_level (float) – A target level of the pruning for the algorithm.
prune_flops (bool) – Is pruning algo sets flops pruning level or not (filter pruning level).