Index

_

__call__() (ResolutionStrategy method) __init__() (ArcfaceHead method) (Bottleneck method) (BottleNecksSequence method) (C3 method) (calibration_context method) (ConstantResolutionStrategy method) (ConvBNActivation method) (CsvAnnotationDataset method) (DefaultChannelsSelectorConstraint method) (DefaultResizeStrategy method) (distill method) (FakeQuantizedModel method) (FixedLatencySearchOptimizer method) (GlobalPruningLabelSelectorByChannels method) (LatencyPruningLabelSelector method) (MacCalculatorFvcore method) (Mapping method) (MobileNetBaseHead method) (MobileNetBaseStem method) (OpenvinoFakeQuantizedModel method) (OptimalPruningLabelSelector method) (PretrainOptimizer method) (PretrainResolutionStrategy method) (PruningCalibrator method) (QuantizationDistiller method) (QuantizationType method) (ResolutionSearcherWithFixedLatencyIterator method) (ResolutionStrategy method) (SearchableConv2d method) (SearchableFuseableSkipConv method) (SearchableMobileInvertedBottleneck method) (SearchableResNetD method) (SearchableResNetE method) (SearchableResNext method) (SearchOptimizer method) (SearchSpaceCommonCalculator method) (SearchSpaceLatencyCalculator method) (SearchSpaceLatencyContainer method) (SearchSpaceMacFvcoreCalculator method) (SearchSpaceMacPthflopsCalculator method) (SearchSpaceMacThopCalculator method) (SearchSpaceModel method) (SearchSpacePytorchCpuLatencyCalculator method) (SearchSpacePytorchCudaLatencyCalculator method) (SearchSpacePytorchLatencyCalculator method) (SearchVariantsContainer method) (SequentialDistiller method) (STMFakeQuantizedModel method) (ThresholdsAndScaleFactorsQuantizationDistiller method) (ThresholdsQuantizationDistiller method) (TrainOptimizer method) (TransformationParameters method) (TrtFakeQuantizedModel method) (UniformPruningLabelSelector method)

_get() (ChannelsSelectorConstraint method) _select() (ScorePruningLabelSelector method)

A

add() (Mapping method) add_distillation_nodes_to_onnx_converted_model() (in module enot.quantization) add_param_group() (FixedLatencySearchOptimizer method) (PretrainOptimizer method) (SearchOptimizer method) (TrainOptimizer method)

apply() (ChannelsSelectorConstraint method) ArcfaceHead (class in enot.models.mobilenet.heads) ASYMMETRIC (QuantizationStrategy attribute) available_calculators() (in module enot.latency.search_space_latency_calculator)

B

best_arch_latency() (in module enot.latency.search_space_latency_statistics) best_architecture (ResolutionSearcherWithFixedLatencyIterator property) best_resolution (ResolutionSearcherWithFixedLatencyIterator property) bn_tune_batches (FixedLatencySearchOptimizer property) (SearchOptimizer property) bn_validation_tune_batches (FixedLatencySearchOptimizer property) (SearchOptimizer property)

Bottleneck (class in enot.models.yolo.blocks) bottlenecks_count (BottleNecksSequence property) BottleNecksSequence (class in enot.models.yolo.blocks) build_frozen_mobilenet() (in module enot.models.mobilenet.mobilenet) build_frozen_simple_block_model() (in module enot.models.simple_block_model_builders) build_mobilenet() (in module enot.models.mobilenet.mobilenet) build_optimizer() (in module enot.optimize.utils) build_simple_block_model() (in module enot.models.simple_block_model_builders)

C

C3 (class in enot.models.yolo.blocks) calculate() (LatencyCalculator method) (MacCalculatorFvcore method) (MacCalculatorPthflops method) (MacCalculatorThop method) calibrate_and_prune_model() (in module enot.pruning) calibrate_and_prune_model_equal() (in module enot.pruning) calibrate_and_prune_model_optimal() (in module enot.pruning) calibrate_model_for_pruning() (in module enot.pruning) calibrate_quantized_model() (in module enot.quantization) calibration_context (class in enot.quantization) CalibrationMethod (class in enot.quantization.utils) cast_to_numpy() (in module enot.utils.common)

ChannelsSelectorConstraint (class in enot.pruning) CHANNELWISE (QuantizationGranularity attribute) compute() (SearchSpaceCommonCalculator method) (SearchSpaceLatencyCalculator method) (SearchSpacePytorchLatencyCalculator method) constant_latency (SearchSpaceLatencyContainer property) (SearchSpaceModel property) ConstantResolutionStrategy (class in enot.experimental.resolution_search.resolution_strategy) conv2d_adapter() (in module enot.distillation.adapter) ConvBNActivation (class in enot.models.yolo.blocks) create_data_loader() (in module enot.utils.data.dataloaders) create_data_loader_from_csv_annotation() (in module enot.utils.data.dataloaders) CsvAnnotationDataset (class in enot.utils.data.csv_annotation_dataset) current_latency() (in module enot.latency.search_space_latency_statistics)

D

default_sample_to_model_inputs() (in module enot.utils.dataloader2model) default_sample_to_n_samples() (in module enot.utils.dataloader2model) DefaultChannelsSelectorConstraint (class in enot.pruning) DefaultQuantizationDistiller (in module enot.quantization) DefaultResizeStrategy (class in enot.experimental.resolution_search.resolution_strategy) depth_mult (TransformationParameters property) dist_only() (in module enot.distributed)

distill (class in enot.distillation) distill() (QuantizationDistiller method) (SequentialDistiller method) (ThresholdsAndScaleFactorsQuantizationDistiller method) distillation_context (class in enot.quantization) DistillationLayer (class in enot.distillation) DistillationLayerSelectionStrategy (class in enot.quantization) DistillerInterface (class in enot.quantization)

E

enable_calibration_mode() (FakeQuantizedModel method) (OpenvinoFakeQuantizedModel method) (TrtFakeQuantizedModel method)

enable_quantization_mode() (FakeQuantizedModel method) (OpenvinoFakeQuantizedModel method) (TrtFakeQuantizedModel method) extract_labels_info() (PruningLabelSelector static method)

F

FakeQuantizedModel (class in enot.quantization) FixedLatencySearchOptimizer (class in enot.optimize.search) float_model_from_quantized_model() (in module enot.quantization) forward() (ArcfaceHead method) (Bottleneck method) (BottleNecksSequence method) (C3 method) (ConvBNActivation method) (MobileNetBaseHead method) (MobileNetBaseStem method) (SearchableConv2d method) (SearchableMobileInvertedBottleneck method) (SearchableResNetD method) (SearchableResNetE method) (SearchableResNext method) (SearchSpaceModel method) (SearchVariantsContainer method)

forward_latency (SearchSpaceModel property)

G

generate_pruned_search_variants_model() (in module enot.autogeneration) get() (ChannelsSelectorConstraint method) get_default_train_transform() (in module enot.utils.data.dataloaders) get_default_validation_transform() (in module enot.utils.data.dataloaders) get_label_filters() (in module enot.pruning.tools) get_local_rank() (in module enot.distributed)

get_module_name_by_tracing_node() (in module enot.tensor_trace.tools) get_most_important_labels() (PruningLabelSelector static method) get_network_by_indexes() (SearchSpaceModel method) get_network_with_best_arch() (SearchSpaceModel method) get_tracing_node_by_module_name() (in module enot.tensor_trace.tools) get_world_size() (in module enot.distributed) GlobalPruningLabelSelectorByChannels (class in enot.pruning)

H

HALF_TO_EVEN (RoundingFunction attribute)

HALF_UP (RoundingFunction attribute)

I

init_convnet_params() (in module enot.utils.common) init_process_group() (in module enot.distributed) init_torch() (in module enot.distributed) initialize_latency() (in module enot.latency.search_space_latency_calculator) initialize_output_distribution_optimization() (SearchSpaceModel method) inverted_conv2d_adapter() (in module enot.distillation.adapter)

is_bn() (in module enot.utils.batch_norm) is_dist() (in module enot.distributed) is_floating_tensor() (in module enot.utils.common) is_local_master() (in module enot.distributed) is_master() (in module enot.distributed) iterate_by_submodules() (in module enot.utils.common)

K

kernel_size (TransformationParameters property)

L

latency_type (SearchSpaceLatencyContainer property) (SearchSpaceModel property) LatencyCalculator (class in enot.latency.latency_calculator) LatencyPruningLabelSelector (class in enot.pruning) LAYERWISE (QuantizationGranularity attribute) load() (in module enot.utils.common)

load_from_bytes() (SearchSpaceLatencyContainer class method) load_from_file() (SearchSpaceLatencyContainer class method) load_state_dict() (FixedLatencySearchOptimizer method) (PretrainOptimizer method) (SearchOptimizer method) (TrainOptimizer method)

M

MacCalculatorFvcore (class in enot.latency.latency_calculator) MacCalculatorPthflops (class in enot.latency.latency_calculator) MacCalculatorThop (class in enot.latency.latency_calculator) Mapping (class in enot.distillation) master_only() (in module enot.distributed) max_latency() (in module enot.latency.search_space_latency_statistics) max_latency_value (FixedLatencySearchOptimizer property) mean_latency() (in module enot.latency.search_space_latency_statistics) median_latency() (in module enot.latency.search_space_latency_statistics) min_latency() (in module enot.latency.search_space_latency_statistics) MIN_MAX (CalibrationMethod attribute) MobileNetBaseHead (class in enot.models.mobilenet.heads)

MobileNetBaseStem (class in enot.models.mobilenet.stems) model (FixedLatencySearchOptimizer property) (PretrainOptimizer property) (SearchOptimizer property) (TrainOptimizer property) model_reset_bn() (in module enot.utils.batch_norm) model_set_bn_momentum() (in module enot.utils.batch_norm) model_step() (FixedLatencySearchOptimizer method) (PretrainOptimizer method) (SearchOptimizer method) (TrainOptimizer method) ModelPruningInfo (class in enot.pruning)

N

n_prunable_groups (ModelPruningInfo property)

O

OpenvinoFakeQuantizedModel (class in enot.quantization) operations_latencies (SearchSpaceLatencyContainer property)

optimal_quantization_scheme() (in module enot.quantization.utils) OptimalPruningLabelSelector (class in enot.pruning) output_distribution_optimization_enabled (SearchSpaceModel property)

P

payload() (Mapping method) plot_latency_heatmap() (in module enot.visualization) plot_probability_heatmap() (in module enot.visualization) prepare_validation_model() (FixedLatencySearchOptimizer method) (SearchOptimizer method) PretrainOptimizer (class in enot.optimize.pretrain)

PretrainResolutionStrategy (class in enot.experimental.resolution_search.resolution_strategy) profile_speed() (in module enot.utils.common) prune_model() (in module enot.pruning) pruning_info (PruningCalibrator property) PruningCalibrator (class in enot.pruning) PruningLabelSelector (class in enot.pruning)

Q

QuantDistillationModule (class in enot.quantization) quantization_parameters() (FakeQuantizedModel method) (OpenvinoFakeQuantizedModel method) (TrtFakeQuantizedModel method)

QuantizationDistiller (class in enot.quantization) QuantizationGranularity (class in enot.quantization.utils) QuantizationStrategy (class in enot.quantization.utils) QuantizationType (class in enot.quantization.utils)

R

regular_parameters() (FakeQuantizedModel method) (OpenvinoFakeQuantizedModel method) (TrtFakeQuantizedModel method) replace_ops() (in module enot.utils.common) replace_prunable_modules() (in module enot.utils.module_replacement) reset_bn() (in module enot.utils.batch_norm)

reset_latency() (in module enot.latency.search_space_latency_calculator) ResolutionSearcherWithFixedLatencyIterator (class in enot.experimental.resolution_search.fixed_latency_search) ResolutionStrategy (class in enot.experimental.resolution_search.resolution_strategy) revert_modules_replacement() (in module enot.utils.module_replacement) RMSELoss (class in enot.quantization) RoundingFunction (class in enot.quantization.utils)

S

sample_latencies() (in module enot.latency.search_space_latency_statistics) save() (in module enot.utils.common) (QuantizationDistiller method) save_to_bytes() (SearchSpaceLatencyContainer method) save_to_file() (SearchSpaceLatencyContainer method) ScorePruningLabelSelector (class in enot.pruning) search_space (FixedLatencySearchOptimizer property) (PretrainOptimizer property) (SearchOptimizer property) search_variants_containers (SearchSpaceModel property) SearchableConv2d (class in enot.models.operations.conv_blocks) SearchableFuseableSkipConv (class in enot.models.operations.conv_blocks) SearchableMobileInvertedBottleneck (class in enot.models.operations.mobilenet_blocks) SearchableResNetD (class in enot.models.operations.resnet_blocks) SearchableResNetE (class in enot.models.operations.resnet_blocks) SearchableResNext (class in enot.models.operations.resnext_blocks) SearchOptimizer (class in enot.optimize.search) SearchSpaceCommonCalculator (class in enot.latency.search_space_latency_calculator) SearchSpaceLatencyCalculator (class in enot.latency.search_space_latency_calculator) SearchSpaceLatencyContainer (class in enot.latency.search_space_latency_container) SearchSpaceMacFvcoreCalculator (class in enot.latency.search_space_latency_calculator) SearchSpaceMacPthflopsCalculator (class in enot.latency.search_space_latency_calculator)

SearchSpaceMacThopCalculator (class in enot.latency.search_space_latency_calculator) SearchSpaceModel (class in enot.models) SearchSpacePytorchCpuLatencyCalculator (class in enot.latency.search_space_latency_calculator) SearchSpacePytorchCudaLatencyCalculator (class in enot.latency.search_space_latency_calculator) SearchSpacePytorchLatencyCalculator (class in enot.latency.search_space_latency_calculator) SearchVariantsContainer (class in enot.models) select() (PruningLabelSelector method) (ScorePruningLabelSelector method) SequentialDistiller (class in enot.quantization) set_resolution_target_metric() (ResolutionSearcherWithFixedLatencyIterator method) split_into_groups() (in module enot.utils.train) state_dict() (FixedLatencySearchOptimizer method) (PretrainOptimizer method) (SearchOptimizer method) (TrainOptimizer method) step() (FixedLatencySearchOptimizer method) (PretrainOptimizer method) (SearchOptimizer method) (TrainOptimizer method) STMFakeQuantizedModel (class in enot.quantization) summary() (ModelPruningInfo method) SYMMETRIC (QuantizationStrategy attribute) sync_model() (in module enot.distributed)

T

tabulate_label_dependencies() (in module enot.pruning.tools) tabulate_label_filters() (in module enot.pruning.tools) tabulate_module_dependencies() (in module enot.pruning.tools) tabulate_unprunable_groups() (in module enot.pruning.tools) ThresholdsAndScaleFactorsQuantizationDistiller (class in enot.quantization)

ThresholdsQuantizationDistiller (class in enot.quantization) TrainOptimizer (class in enot.optimize.utils) TransformationParameters (class in enot.autogeneration) TrtFakeQuantizedModel (class in enot.quantization) tune_bn_stats() (in module enot.utils.batch_norm)

U

UniformPruningLabelSelector (class in enot.pruning)

W

width_expansion (BottleNecksSequence property)

width_mult (TransformationParameters property)

Z

zero_grad() (FixedLatencySearchOptimizer method) (PretrainOptimizer method) (SearchOptimizer method) (TrainOptimizer method)