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) (PretrainResolutionStrategy 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) (TransformationParameters method) (TrtFakeQuantizedModel 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

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_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) 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_sample_to_model_inputs() (in module enot.utils.dataloader2model) DefaultQuantizationDistiller (class in enot.quantization) DefaultResizeStrategy (class in enot.experimental.resolution_search.resolution_strategy) depth_mult (TransformationParameters property)

dist_only() (in module enot.distributed) 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_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_for_equal_pruning() (in module enot.pruning)

get_latency_container() (SearchSpaceModel method) 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

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)

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)

O

OpenvinoFakeQuantizedModel (class in enot.quantization)

operations_latencies (SearchSpaceLatencyContainer property) 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_info (EnotPruningCalibrator property)

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) 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) 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) SYMMETRIC (QuantizationStrategy attribute) sync_model() (in module enot.distributed)

T

TransformationParameters (class in enot.autogeneration)

TrtFakeQuantizedModel (class in enot.quantization) tune_bn_stats() (in module enot.utils.batch_norm)

W

width_expansion (BottleNecksSequence property)

width_mult (TransformationParameters property)

Z

zero_grad() (EnotFixedLatencySearchOptimizer method) (EnotPretrainOptimizer method) (EnotSearchOptimizer method) (EnotTrainOptimizer method)