Index

_

__call__() (ResolutionStrategy method) __init__() (ArcfaceHead method) (Bottleneck method) (BottleNecksSequence method) (C3 method) (calibration_context method) (ConstantResolutionStrategy method) (ConvBNActivation method) (CsvAnnotationDataset method) (DefaultResizeStrategy method) (distill method) (FakeQuantizedModel method) (FixedLatencySearchOptimizer method) (Mapping method) (MobileNetBaseHead method) (MobileNetBaseStem method) (OpenvinoFakeQuantizedModel method) (OptimalPruningLabelSelector method) (PretrainOptimizer method) (PretrainResolutionStrategy method) (PruningCalibrator method) (PruningLabelSelector 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) (ThresholdsAndScaleFactorsQuantizationDistiller method) (ThresholdsQuantizationDistiller method) (TopKPruningLabelSelector method) (TrainOptimizer method) (TransformationParameters method) (TrtFakeQuantizedModel method) (UniformPruningLabelSelector 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)

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 (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)

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) 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_architecture_optimization_strategy() (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 (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)

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_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_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_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) 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) 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) logging_config() (in module enot.logging)

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) minimal_network_latency (OptimalPruningLabelSelector property)

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)

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

payload() (Mapping method) plot_latency_heatmap() (in module enot.visualization) plot_probability_heatmap() (in module enot.visualization) prepare_log() (in module enot.logging) 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_cfg (OptimalPruningLabelSelector property) (TopKPruningLabelSelector property) 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) 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) 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 (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() (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() (FixedLatencySearchOptimizer method) (PretrainOptimizer method) (SearchOptimizer method) (TrainOptimizer method) step() (FixedLatencySearchOptimizer method) (PretrainOptimizer method) (SearchOptimizer method) (TrainOptimizer 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)

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)