Index

_

__call__() (ResolutionStrategy method) __init__() (ArcfaceHead method) (Bottleneck method) (BottleNecksSequence method) (C3 method) (calibration_context method) (ConstantResolutionStrategy method) (ConvBNActivation method) (CsvAnnotationDataset method) (DefaultResizeStrategy method) (EnotFixedLatencySearchOptimizer method) (EnotPretrainOptimizer method) (EnotPruningCalibrator method) (EnotSearchOptimizer method) (EnotTrainOptimizer method) (FakeQuantizedModel method) (MobileNetBaseHead method) (MobileNetBaseStem method) (OpenvinoFakeQuantizedModel method) (OptimalPruningLabelSelector method) (PretrainResolutionStrategy method) (PruningLabelSelector method) (QuantizationDistiller method) (QuantizationType method) (ResolutionSearcherWithFixedLatencyIterator method) (ResolutionStrategy method) (SearchableConv2d method) (SearchableFuseableSkipConv method) (SearchableMobileInvertedBottleneck method) (SearchableResNetD method) (SearchableResNetE method) (SearchableResNext method) (SearchSpaceCommonCalculator method) (SearchSpaceLatencyCalculator method) (SearchSpaceLatencyContainer method) (SearchSpaceMacCalculator method) (SearchSpaceMacFvcoreCalculator method) (SearchSpaceMacPthflopsCalculator method) (SearchSpaceMacThopCalculator method) (SearchSpaceModel method) (SearchSpacePytorchCpuLatencyCalculator method) (SearchSpacePytorchCudaLatencyCalculator method) (SearchSpacePytorchLatencyCalculator method) (SearchVariantsContainer method) (SequentialDistiller method) (ThresholdsAndScaleFactorsQuantizationDistiller method) (ThresholdsQuantizationDistiller method) (TopKPruningLabelSelector method) (TransformationParameters method) (TrtFakeQuantizedModel method) (UniformPruningLabelSelector method)

A

add_distillation_nodes_to_onnx_converted_model() (in module enot.quantization) add_param_group() (EnotFixedLatencySearchOptimizer method) (EnotPretrainOptimizer method) (EnotSearchOptimizer method) (EnotTrainOptimizer method)

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

B

baseline_latency (OptimalPruningLabelSelector property) best_arch_latency() (in module enot.latency.search_space_latency_statistics) best_architecture (ResolutionSearcherWithFixedLatencyIterator property) best_resolution (ResolutionSearcherWithFixedLatencyIterator property) bn_tune_batches (EnotFixedLatencySearchOptimizer property) (EnotSearchOptimizer property) bn_validation_tune_batches (EnotFixedLatencySearchOptimizer property) (EnotSearchOptimizer property)

Bottleneck (class in enot.models.yolo.blocks) bottlenecks_count (BottleNecksSequence property) BottleNecksSequence (class in enot.models.yolo.blocks) build_enot_optimizer() (in module enot.optimize.utils) 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_simple_block_model() (in module enot.models.simple_block_model_builders)

C

C3 (class in enot.models.yolo.blocks) calculate() (LatencyCalculator method) (MacCalculator 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)

CHANNELWISE (QuantizationGranularity attribute) check_pruning_config() (in module enot.pruning) compute() (SearchSpaceCommonCalculator method) (SearchSpaceLatencyCalculator method) (SearchSpacePytorchLatencyCalculator method) constant_latency (SearchSpaceLatencyContainer property) (SearchSpaceModel property) ConstantResolutionStrategy (class in enot.experimental.resolution_search.resolution_strategy) conv_mac_count() (in module enot.latency.mac_calculator) conv_out_spatial_size() (in module enot.latency.mac_calculator) 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_channel_search_step() (in module enot.pruning) default_min_channels_to_leave() (in module enot.pruning) default_sample_to_model_inputs() (in module enot.utils.dataloader2model) default_sample_to_n_samples() (in module enot.utils.dataloader2model) 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() (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)

enot_init_process_group() (in module enot.distributed) EnotFixedLatencySearchOptimizer (class in enot.optimize.search) EnotPretrainOptimizer (class in enot.optimize.pretrain) EnotPruningCalibrator (class in enot.pruning) EnotSearchOptimizer (class in enot.optimize.search) EnotTrainOptimizer (class in enot.optimize.utils)

F

FakeQuantizedModel (class in enot.quantization) 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) forward_latency() (LatencyMixin method)

G

generate_pruned_search_variants_model() (in module enot.autogeneration) get_config_for_pruning() (OptimalPruningLabelSelector method) (TopKPruningLabelSelector method) (UniformPruningLabelSelector method) get_criteria_label_dict() (in module enot.pruning) get_default_train_transform() (in module enot.utils.data.dataloaders) get_default_validation_transform() (in module enot.utils.data.dataloaders) get_labels_by_config() (OptimalPruningLabelSelector static method) (TopKPruningLabelSelector static method)

get_labels_for_uniform_pruning() (in module enot.pruning) get_latency_container() (SearchSpaceModel method) get_least_important_labels_by_config() (in module enot.pruning) get_local_rank() (in module enot.distributed) get_network_by_indexes() (SearchSpaceModel method) get_network_with_best_arch() (SearchSpaceModel method) get_world_size() (in module enot.distributed)

H

HALF_TO_EVEN (RoundingFunction attribute)

HALF_UP (RoundingFunction attribute)

I

init_convnet_params() (in module enot.utils.common) init_torch() (in module enot.distributed) initialize_latency() (in module enot.latency.search_space_latency_calculator) initialize_output_distribution_optimization() (SearchSpaceModel method) 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) iterate_over_gate_criteria() (in module enot.pruning)

K

kernel_size (TransformationParameters property)

L

labels_for_pruning (OptimalPruningLabelSelector property) (PruningLabelSelector property) latency_type (SearchSpaceLatencyContainer property) (SearchSpaceModel property) LatencyCalculator (class in enot.latency.latency_calculator) LatencyMixin (class in enot.latency.latency_mixin) LAYERWISE (QuantizationGranularity attribute)

load() (in module enot.utils.common) load_from_bytes() (SearchSpaceLatencyContainer class method) load_from_file() (SearchSpaceLatencyContainer class method) load_state_dict() (EnotFixedLatencySearchOptimizer method) (EnotPretrainOptimizer method) (EnotSearchOptimizer method) (EnotTrainOptimizer method) logging_config() (in module enot.logging)

M

MacCalculator (class in enot.latency.latency_calculator) MacCalculatorFvcore (class in enot.latency.latency_calculator) MacCalculatorPthflops (class in enot.latency.latency_calculator) MacCalculatorThop (class in enot.latency.latency_calculator) master_only() (in module enot.distributed) max_latency() (in module enot.latency.search_space_latency_statistics) max_latency_value (EnotFixedLatencySearchOptimizer property) mean_latency() (in module enot.latency.search_space_latency_statistics) median_latency() (in module enot.latency.search_space_latency_statistics) mib_mac_count() (in module enot.latency.mac_calculator) min_latency() (in module enot.latency.search_space_latency_statistics) MIN_MAX (CalibrationMethod attribute) minimal_network_latency (OptimalPruningLabelSelector property)

MobileNetBaseHead (class in enot.models.mobilenet.heads) MobileNetBaseStem (class in enot.models.mobilenet.stems) model (EnotFixedLatencySearchOptimizer property) (EnotPretrainOptimizer property) (EnotSearchOptimizer property) (EnotTrainOptimizer property) model_reset_bn() (in module enot.utils.batch_norm) model_set_bn_momentum() (in module enot.utils.batch_norm) model_step() (EnotFixedLatencySearchOptimizer method) (EnotPretrainOptimizer method) (EnotSearchOptimizer method) (EnotTrainOptimizer method) ModelPruningInfo (class in enot.pruning)

N

n_prunable_groups (ModelPruningInfo property)

n_search_steps (OptimalPruningLabelSelector 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

plot_latency_heatmap() (in module enot.visualization) plot_probability_heatmap() (in module enot.visualization) prepare_log() (in module enot.logging) prepare_validation_model() (EnotFixedLatencySearchOptimizer method) (EnotSearchOptimizer method) PretrainResolutionStrategy (class in enot.experimental.resolution_search.resolution_strategy)

profile_speed() (in module enot.utils.common) prune_model() (in module enot.pruning) pruning_cfg (OptimalPruningLabelSelector property) (TopKPruningLabelSelector property) pruning_info (EnotPruningCalibrator property) 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) 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) RMSELoss (class in enot.quantization) rn_mac_count() (in module enot.latency.mac_calculator) 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) search_space (EnotFixedLatencySearchOptimizer property) (EnotPretrainOptimizer property) (EnotSearchOptimizer 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) 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) SearchSpaceMacCalculator (class in enot.latency.search_space_latency_calculator) 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() (OptimalPruningLabelSelector method) (PruningLabelSelector method) (TopKPruningLabelSelector method) SequentialDistiller (class in enot.quantization) set_resolution_target_metric() (ResolutionSearcherWithFixedLatencyIterator method) split_into_groups() (in module enot.utils.train) state_dict() (EnotFixedLatencySearchOptimizer method) (EnotPretrainOptimizer method) (EnotSearchOptimizer method) (EnotTrainOptimizer method) step() (EnotFixedLatencySearchOptimizer method) (EnotPretrainOptimizer method) (EnotSearchOptimizer method) (EnotTrainOptimizer method) summary() (ModelPruningInfo method) SYMMETRIC (QuantizationStrategy attribute) sync_model() (in module enot.distributed)

T

ThresholdsAndScaleFactorsQuantizationDistiller (class in enot.quantization) ThresholdsQuantizationDistiller (class in enot.quantization) TopKPruningLabelSelector (class in enot.pruning)

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() (EnotFixedLatencySearchOptimizer method) (EnotPretrainOptimizer method) (EnotSearchOptimizer method) (EnotTrainOptimizer method)