v2.6
Table of Contents
Overview
Installation
QuickStart
Auto (Hyper-parameter) Tuning
Neural Architecture Search
Model Compression
Feature Engineering
References
Use Cases and Solutions
Research and Publications
FAQ
How to Contribute
Change Log
NNI
Docs
»
Index
Edit on GitHub
Index
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
L
|
M
|
N
|
O
|
P
|
Q
|
R
|
S
|
T
|
U
|
V
|
W
A
ActivationAPoZRankFilterPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.iterative_pruner)
ActivationAPoZRankPruner (class in nni.algorithms.compression.v2.pytorch.pruning)
ActivationMeanRankFilterPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.iterative_pruner)
ActivationMeanRankPruner (class in nni.algorithms.compression.v2.pytorch.pruning)
add_model() (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner method)
ADMMPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.iterative_pruner)
(class in nni.algorithms.compression.v2.pytorch.pruning)
AGPPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.iterative_pruner)
(class in nni.algorithms.compression.v2.pytorch.pruning)
algorithm_name (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner attribute)
AMCPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.v2.pytorch.pruning)
analysis() (nni.compression.pytorch.utils.sensitivity_analysis.SensitivityAnalysis method)
apply() (nni.retiarii.Mutator method)
arch (nni.nas.benchmarks.nasbench101.Nb101TrialConfig attribute)
(nni.nas.benchmarks.nasbench201.Nb201TrialConfig attribute)
arguments (nni.retiarii.evaluator.FunctionalEvaluator attribute)
assess_trial() (nni.algorithms.hpo.curvefitting_assessor.CurvefittingAssessor method)
(nni.algorithms.hpo.medianstop_assessor.MedianstopAssessor method)
(nni.assessor.Assessor method)
Assessor (class in nni.assessor)
AssessResult (class in nni.assessor)
AutoActivation (class in nni.retiarii.nn.pytorch)
AutoCompressPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.auto_compress_pruner)
(class in nni.algorithms.compression.v2.pytorch.pruning)
AVG_POOL_3X3 (nni.nas.benchmarks.nasbench201 attribute)
(nni.nas.benchmarks.nds.constants attribute)
B
Bad (nni.assessor.AssessResult attribute)
base_lr (nni.nas.benchmarks.nds.NdsTrialConfig attribute)
BaseModelSpeedup (class in nni.compression.pytorch.quantization_speedup.backend)
basic_unit() (in module nni.retiarii)
BatchTuner (class in nni.algorithms.hpo.batch_tuner)
best_test_acc (nni.nas.benchmarks.nds.NdsTrialStats attribute)
best_train_acc (nni.nas.benchmarks.nds.NdsTrialStats attribute)
best_train_loss (nni.nas.benchmarks.nds.NdsTrialStats attribute)
beta (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner attribute)
bind_sampler() (nni.retiarii.Mutator method)
BNNQuantizer (class in nni.algorithms.compression.pytorch.quantization)
bo (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner attribute)
build_dependency() (nni.compression.pytorch.utils.shape_dependency.ChannelDependency method)
(nni.compression.pytorch.utils.shape_dependency.GroupDependency method)
C
calc_mask() (nni.algorithms.compression.pytorch.pruning.auto_compress_pruner.AutoCompressPruner method)
(nni.algorithms.compression.pytorch.pruning.iterative_pruner.AGPPruner method)
(nni.algorithms.compression.pytorch.pruning.lottery_ticket.LotteryTicketPruner method)
(nni.algorithms.compression.pytorch.pruning.net_adapt_pruner.NetAdaptPruner method)
(nni.algorithms.compression.pytorch.pruning.sensitivity_pruner.SensitivityPruner method)
(nni.algorithms.compression.pytorch.pruning.simulated_annealing_pruner.SimulatedAnnealingPruner method)
(nni.algorithms.compression.pytorch.pruning.structured_pruning_masker.StructuredWeightMasker method)
(nni.algorithms.compression.pytorch.pruning.transformer_pruner.TransformerHeadPruner method)
(nni.algorithms.compression.pytorch.pruning.weight_masker.WeightMasker method)
(nni.compression.pytorch.compressor.Pruner method)
Calibrator (class in nni.compression.pytorch.quantization_speedup.calibrator)
Cell (class in nni.retiarii.nn.pytorch)
cell (nni.retiarii.Node attribute)
cell_spec (nni.nas.benchmarks.nds.NdsTrialConfig attribute)
ChannelDependency (class in nni.compression.pytorch.utils.shape_dependency)
ChannelMaskConflict (class in nni.compression.pytorch.utils.mask_conflict)
choice() (nni.retiarii.Mutator method)
choices (nni.retiarii.nn.pytorch.LayerChoice attribute)
ChosenInputs (class in nni.retiarii.nn.pytorch)
Classification (class in nni.retiarii.evaluator.pytorch.lightning)
Classification() (in module nni.retiarii.evaluator.pytorch.cgo.evaluator)
cold_start_num (nni.algorithms.hpo.metis_tuner.MetisTuner attribute)
command_queue_worker() (nni.runtime.msg_dispatcher_base.MsgDispatcherBase method)
compress() (nni.algorithms.compression.pytorch.pruning.auto_compress_pruner.AutoCompressPruner method)
(nni.algorithms.compression.pytorch.pruning.iterative_pruner.ADMMPruner method)
(nni.algorithms.compression.pytorch.pruning.iterative_pruner.AGPPruner method)
(nni.algorithms.compression.pytorch.pruning.iterative_pruner.IterativePruner method)
(nni.algorithms.compression.pytorch.pruning.net_adapt_pruner.NetAdaptPruner method)
(nni.algorithms.compression.pytorch.pruning.sensitivity_pruner.SensitivityPruner method)
(nni.algorithms.compression.pytorch.pruning.simulated_annealing_pruner.SimulatedAnnealingPruner method)
(nni.algorithms.compression.pytorch.pruning.transformer_pruner.TransformerHeadPruner method)
(nni.algorithms.compression.pytorch.quantization.ObserverQuantizer method)
(nni.compression.pytorch.compressor.Compressor method)
(nni.compression.pytorch.compressor.Pruner method)
(nni.compression.pytorch.quantization_speedup.backend.BaseModelSpeedup method)
(nni.compression.pytorch.quantization_speedup.integrated_tensorrt.ModelSpeedupTensorRT method)
Compressor (class in nni.compression.pytorch.compressor)
compute_target_sparsity() (nni.algorithms.compression.pytorch.pruning.iterative_pruner.AGPPruner method)
config (nni.nas.benchmarks.nasbench101.Nb101TrialStats attribute)
(nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute)
(nni.nas.benchmarks.nds.NdsTrialStats attribute)
CONV1X1_BN_RELU (nni.nas.benchmarks.nasbench101 attribute)
CONV3X3_BN_RELU (nni.nas.benchmarks.nasbench101 attribute)
CONV_1X1 (nni.nas.benchmarks.nasbench201 attribute)
(nni.nas.benchmarks.nds.constants attribute)
CONV_3X1_1X3 (nni.nas.benchmarks.nds.constants attribute)
CONV_3X3 (nni.nas.benchmarks.nasbench201 attribute)
(nni.nas.benchmarks.nds.constants attribute)
CONV_7X1_1X7 (nni.nas.benchmarks.nds.constants attribute)
count_flops_params() (in module nni.compression.pytorch.utils.counter)
create_cfg() (nni.algorithms.compression.pytorch.pruning.sensitivity_pruner.SensitivityPruner method)
create_fixed_module() (nni.retiarii.nn.pytorch.InputChoice class method)
(nni.retiarii.nn.pytorch.LayerChoice class method)
(nni.retiarii.nn.pytorch.Repeat class method)
(nni.retiarii.nn.pytorch.ValueChoice class method)
current_epoch (nni.nas.benchmarks.nasbench101.Nb101IntermediateStats attribute)
(nni.nas.benchmarks.nasbench201.Nb201IntermediateStats attribute)
(nni.nas.benchmarks.nds.NdsIntermediateStats attribute)
current_sparsity() (nni.algorithms.compression.pytorch.pruning.sensitivity_pruner.SensitivityPruner method)
CurvefittingAssessor (class in nni.algorithms.hpo.curvefitting_assessor)
D
DartsTrainer (class in nni.retiarii.oneshot.pytorch)
dataset (nni.nas.benchmarks.nasbench201.Nb201TrialConfig attribute)
(nni.nas.benchmarks.nds.NdsTrialConfig attribute)
default_model_len (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner attribute)
default_model_width (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner attribute)
dependency_sets (nni.compression.pytorch.utils.shape_dependency.ChannelDependency property)
DIL_CONV_3X3 (nni.nas.benchmarks.nds.constants attribute)
DIL_CONV_5X5 (nni.nas.benchmarks.nds.constants attribute)
DIL_SEP_CONV_3X3 (nni.nas.benchmarks.nds.constants attribute)
DoReFaQuantizer (class in nni.algorithms.compression.pytorch.quantization)
dry_run() (nni.retiarii.Mutator method)
dump() (in module nni)
E
Edge (class in nni.retiarii)
edges (nni.retiarii.Graph attribute)
EnasTrainer (class in nni.retiarii.oneshot.pytorch)
enqueue_command() (nni.runtime.msg_dispatcher_base.MsgDispatcherBase method)
evaluator (nni.retiarii.Model attribute)
EvolutionTuner (class in nni.algorithms.hpo.evolution_tuner)
exploration_probability (nni.algorithms.hpo.metis_tuner.MetisTuner attribute)
export() (nni.compression.pytorch.utils.sensitivity_analysis.SensitivityAnalysis method)
(nni.compression.pytorch.utils.shape_dependency.ChannelDependency method)
(nni.compression.pytorch.utils.shape_dependency.GroupDependency method)
export_model() (nni.algorithms.compression.pytorch.pruning.auto_compress_pruner.AutoCompressPruner method)
(nni.algorithms.compression.pytorch.quantization.BNNQuantizer method)
(nni.algorithms.compression.pytorch.quantization.DoReFaQuantizer method)
(nni.algorithms.compression.pytorch.quantization.LsqQuantizer method)
(nni.algorithms.compression.pytorch.quantization.ObserverQuantizer method)
(nni.algorithms.compression.pytorch.quantization.QAT_Quantizer method)
(nni.compression.pytorch.compressor.Pruner method)
(nni.compression.pytorch.compressor.Quantizer method)
export_model_save() (nni.compression.pytorch.compressor.Quantizer method)
export_quantized_model() (nni.compression.pytorch.quantization_speedup.backend.BaseModelSpeedup method)
(nni.compression.pytorch.quantization_speedup.integrated_tensorrt.ModelSpeedupTensorRT method)
F
final_test_acc (nni.nas.benchmarks.nds.NdsTrialStats attribute)
final_train_acc (nni.nas.benchmarks.nds.NdsTrialStats attribute)
final_train_loss (nni.nas.benchmarks.nds.NdsTrialStats attribute)
find_conv_bn_patterns() (nni.compression.pytorch.compressor.Quantizer method)
fix_mask() (nni.compression.pytorch.utils.mask_conflict.ChannelMaskConflict method)
(nni.compression.pytorch.utils.mask_conflict.GroupMaskConflict method)
fixed_arch() (in module nni.retiarii)
flops (nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute)
(nni.nas.benchmarks.nds.NdsTrialStats attribute)
fold_bn() (nni.compression.pytorch.compressor.Quantizer method)
fork() (nni.retiarii.Graph method)
(nni.retiarii.Model method)
forward() (nni.compression.pytorch.compressor.PrunerModuleWrapper method)
(nni.compression.pytorch.compressor.QuantizerModuleWrapper method)
(nni.retiarii.nn.pytorch.AutoActivation method)
(nni.retiarii.nn.pytorch.Cell method)
(nni.retiarii.nn.pytorch.ChosenInputs method)
(nni.retiarii.nn.pytorch.InputChoice method)
(nni.retiarii.nn.pytorch.LayerChoice method)
(nni.retiarii.nn.pytorch.Repeat method)
(nni.retiarii.nn.pytorch.ValueChoice method)
FPGMPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.one_shot_pruner)
(class in nni.algorithms.compression.v2.pytorch.pruning)
function (nni.retiarii.evaluator.FunctionalEvaluator attribute)
FunctionalEvaluator (class in nni.retiarii.evaluator)
G
generate() (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner method)
generate_multiple_parameters() (nni.algorithms.hpo.evolution_tuner.EvolutionTuner method)
(nni.tuner.Tuner method)
generate_parameters() (nni.algorithms.hpo.batch_tuner.BatchTuner method)
(nni.algorithms.hpo.evolution_tuner.EvolutionTuner method)
(nni.algorithms.hpo.gp_tuner.GPTuner method)
(nni.algorithms.hpo.gridsearch_tuner.GridSearchTuner method)
(nni.algorithms.hpo.hyperopt_tuner.HyperoptTuner method)
(nni.algorithms.hpo.metis_tuner.MetisTuner method)
(nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner method)
(nni.algorithms.hpo.random_tuner.RandomTuner method)
(nni.algorithms.hpo.tpe_tuner.TpeTuner method)
(nni.tuner.Tuner method)
generator (nni.nas.benchmarks.nds.NdsTrialConfig attribute)
get_batch() (nni.compression.pytorch.quantization_speedup.calibrator.Calibrator method)
get_best_model_id() (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner method)
get_channel_sum() (nni.algorithms.compression.pytorch.pruning.structured_pruning_masker.StructuredWeightMasker method)
get_current_parameter() (in module nni)
get_experiment_id() (in module nni)
get_mask() (nni.algorithms.compression.pytorch.pruning.structured_pruning_masker.StructuredWeightMasker method)
get_metric_value_by_id() (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner method)
get_modules_to_compress() (nni.compression.pytorch.compressor.Compressor method)
get_modules_wrapper() (nni.compression.pytorch.compressor.Compressor method)
get_next_parameter() (in module nni)
get_node_by_id() (nni.retiarii.Graph method)
get_node_by_name() (nni.retiarii.Graph method)
(nni.retiarii.Model method)
get_node_by_python_name() (nni.retiarii.Graph method)
(nni.retiarii.Model method)
get_nodes() (nni.retiarii.Model method)
get_nodes_by_label() (nni.retiarii.Model method)
get_nodes_by_type() (nni.retiarii.Graph method)
(nni.retiarii.Model method)
get_prune_iterations() (nni.algorithms.compression.pytorch.pruning.lottery_ticket.LotteryTicketPruner method)
get_pruned_weights() (nni.compression.pytorch.compressor.Pruner method)
get_sequence_id() (in module nni)
get_suggestion() (nni.algorithms.hpo.hyperopt_tuner.HyperoptTuner method)
get_trial_id() (in module nni)
Good (nni.assessor.AssessResult attribute)
GPTuner (class in nni.algorithms.hpo.gp_tuner)
grad_scale() (nni.algorithms.compression.pytorch.quantization.LsqQuantizer static method)
Graph (class in nni.retiarii)
graph (nni.retiarii.Edge attribute)
(nni.retiarii.Node attribute)
graphs (nni.retiarii.Model attribute)
GridSearch (class in nni.retiarii.strategy)
GridSearchTuner (class in nni.algorithms.hpo.gridsearch_tuner)
group_weight_names_by_graph() (nni.algorithms.compression.pytorch.pruning.transformer_pruner.TransformerHeadPruner method)
group_weights_by_name() (nni.algorithms.compression.pytorch.pruning.transformer_pruner.TransformerHeadPruner method)
GroupDependency (class in nni.compression.pytorch.utils.shape_dependency)
GroupMaskConflict (class in nni.compression.pytorch.utils.mask_conflict)
H
handle_add_customized_trial() (nni.algorithms.hpo.hyperband_advisor.Hyperband method)
(nni.runtime.msg_dispatcher_base.MsgDispatcherBase method)
handle_import_data() (nni.algorithms.hpo.hyperband_advisor.Hyperband method)
(nni.runtime.msg_dispatcher_base.MsgDispatcherBase method)
handle_initialize() (nni.algorithms.hpo.hyperband_advisor.Hyperband method)
(nni.runtime.msg_dispatcher_base.MsgDispatcherBase method)
handle_report_metric_data() (nni.algorithms.hpo.hyperband_advisor.Hyperband method)
(nni.runtime.msg_dispatcher_base.MsgDispatcherBase method)
handle_request_trial_jobs() (nni.algorithms.hpo.hyperband_advisor.Hyperband method)
(nni.runtime.msg_dispatcher_base.MsgDispatcherBase method)
handle_trial_end() (nni.algorithms.hpo.hyperband_advisor.Hyperband method)
(nni.runtime.msg_dispatcher_base.MsgDispatcherBase method)
handle_update_search_space() (nni.algorithms.hpo.hyperband_advisor.Hyperband method)
(nni.runtime.msg_dispatcher_base.MsgDispatcherBase method)
hash (nni.nas.benchmarks.nasbench101.Nb101TrialConfig attribute)
hash_module() (in module nni.nas.benchmarks.nasbench101.graph_util)
head (nni.retiarii.Edge attribute)
head_slot (nni.retiarii.Edge attribute)
hidden_nodes (nni.retiarii.Graph attribute)
history (nni.retiarii.Model attribute)
Hyperband (class in nni.algorithms.hpo.hyperband_advisor)
HyperoptTuner (class in nni.algorithms.hpo.hyperopt_tuner)
I
id (nni.retiarii.Graph attribute)
(nni.retiarii.Node attribute)
import_data() (nni.algorithms.hpo.batch_tuner.BatchTuner method)
(nni.algorithms.hpo.evolution_tuner.EvolutionTuner method)
(nni.algorithms.hpo.gp_tuner.GPTuner method)
(nni.algorithms.hpo.gridsearch_tuner.GridSearchTuner method)
(nni.algorithms.hpo.hyperopt_tuner.HyperoptTuner method)
(nni.algorithms.hpo.metis_tuner.MetisTuner method)
(nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner method)
(nni.algorithms.hpo.tpe_tuner.TpeTuner method)
(nni.tuner.Tuner method)
incoming_edges (nni.retiarii.Node attribute)
infer_num_vertices() (in module nni.nas.benchmarks.nasbench101.graph_util)
inference() (nni.compression.pytorch.quantization_speedup.backend.BaseModelSpeedup method)
(nni.compression.pytorch.quantization_speedup.integrated_tensorrt.ModelSpeedupTensorRT method)
init_search() (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner method)
INPUT (nni.nas.benchmarks.nasbench101 attribute)
input_names (nni.retiarii.Graph attribute)
input_node (nni.retiarii.Graph attribute)
input_shape (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner attribute)
InputChoice (class in nni.retiarii.nn.pytorch)
intermediate_metrics (nni.retiarii.Model attribute)
is_valid() (nni.algorithms.hpo.batch_tuner.BatchTuner method)
iter_time (nni.nas.benchmarks.nds.NdsTrialStats attribute)
IterativePruner (class in nni.algorithms.compression.pytorch.pruning.iterative_pruner)
L
L1FilterPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.one_shot_pruner)
L1NormPruner (class in nni.algorithms.compression.v2.pytorch.pruning)
L2FilterPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.one_shot_pruner)
L2NormPruner (class in nni.algorithms.compression.v2.pytorch.pruning)
label (nni.retiarii.Node attribute)
latency (nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute)
LayerChoice (class in nni.retiarii.nn.pytorch)
length (nni.retiarii.nn.pytorch.LayerChoice attribute)
LevelPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.one_shot_pruner)
(class in nni.algorithms.compression.v2.pytorch.pruning)
LightningModule (class in nni.retiarii.evaluator.pytorch.lightning)
LinearPruner (class in nni.algorithms.compression.v2.pytorch.pruning)
load() (in module nni)
load_best_model() (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner method)
load_calibration_config() (nni.algorithms.compression.pytorch.quantization.LsqQuantizer method)
(nni.algorithms.compression.pytorch.quantization.QAT_Quantizer method)
(nni.compression.pytorch.compressor.Quantizer method)
load_checkpoint() (nni.assessor.Assessor method)
(nni.tuner.Tuner method)
load_model_by_id() (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner method)
load_model_state_dict() (nni.compression.pytorch.compressor.Pruner method)
load_quantized_model() (nni.compression.pytorch.quantization_speedup.integrated_tensorrt.ModelSpeedupTensorRT method)
load_sensitivity() (nni.algorithms.compression.pytorch.pruning.sensitivity_pruner.SensitivityPruner method)
load_state_dict() (nni.compression.pytorch.utils.sensitivity_analysis.SensitivityAnalysis method)
LotteryTicketPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.lottery_ticket)
(class in nni.algorithms.compression.v2.pytorch.pruning)
LsqQuantizer (class in nni.algorithms.compression.pytorch.quantization)
M
mask_dict (nni.compression.pytorch.compressor.Pruner attribute)
max_model_size (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner attribute)
MAX_POOL_3X3 (nni.nas.benchmarks.nds.constants attribute)
MAX_POOL_5X5 (nni.nas.benchmarks.nds.constants attribute)
MAX_POOL_7X7 (nni.nas.benchmarks.nds.constants attribute)
MAXPOOL3X3 (nni.nas.benchmarks.nasbench101 attribute)
MedianstopAssessor (class in nni.algorithms.hpo.medianstop_assessor)
merge_parameter() (in module nni.utils)
MetisTuner (class in nni.algorithms.hpo.metis_tuner)
metric (nni.retiarii.Model attribute)
miscs_update_idxs_vals() (nni.algorithms.hpo.hyperopt_tuner.HyperoptTuner method)
Model (class in nni.retiarii)
model (nni.retiarii.Graph attribute)
model_family (nni.nas.benchmarks.nds.NdsTrialConfig attribute)
model_spec (nni.nas.benchmarks.nds.NdsTrialConfig attribute)
model_wrapper() (in module nni.retiarii)
ModelSpeedup (class in nni.compression.pytorch)
ModelSpeedupTensorRT (class in nni.compression.pytorch.quantization_speedup.integrated_tensorrt)
MovementPruner (class in nni.algorithms.compression.v2.pytorch.pruning)
MsgDispatcherBase (class in nni.runtime.msg_dispatcher_base)
MultiModelSupervisedLearningModule() (in module nni.retiarii.evaluator.pytorch.cgo.evaluator)
mutate() (nni.retiarii.Mutator method)
Mutator (class in nni.retiarii)
N
n_classes (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner attribute)
NaiveQuantizer (class in nni.algorithms.compression.pytorch.quantization)
name (nni.retiarii.Graph attribute)
(nni.retiarii.Node attribute)
names (nni.retiarii.nn.pytorch.LayerChoice attribute)
nasbench_format_to_architecture_repr() (in module nni.nas.benchmarks.nasbench101.graph_util)
Nb101IntermediateStats (class in nni.nas.benchmarks.nasbench101)
Nb101TrialConfig (class in nni.nas.benchmarks.nasbench101)
Nb101TrialStats (class in nni.nas.benchmarks.nasbench101)
Nb201IntermediateStats (class in nni.nas.benchmarks.nasbench201)
Nb201TrialConfig (class in nni.nas.benchmarks.nasbench201)
Nb201TrialStats (class in nni.nas.benchmarks.nasbench201)
NdsIntermediateStats (class in nni.nas.benchmarks.nds)
NdsTrialConfig (class in nni.nas.benchmarks.nds)
NdsTrialStats (class in nni.nas.benchmarks.nds)
NetAdaptPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.net_adapt_pruner)
NetworkMorphismTuner (class in nni.algorithms.hpo.networkmorphism_tuner)
no_candidates (nni.algorithms.hpo.metis_tuner.MetisTuner attribute)
no_resampling (nni.algorithms.hpo.metis_tuner.MetisTuner attribute)
Node (class in nni.retiarii)
nodes (nni.retiarii.Graph attribute)
NONE (nni.nas.benchmarks.nasbench201 attribute)
(nni.nas.benchmarks.nds.constants attribute)
normalize() (nni.algorithms.compression.pytorch.pruning.sensitivity_pruner.SensitivityPruner method)
num_cells (nni.nas.benchmarks.nasbench201.Nb201TrialConfig attribute)
num_channels (nni.nas.benchmarks.nasbench201.Nb201TrialConfig attribute)
num_epochs (nni.nas.benchmarks.nasbench101.Nb101TrialConfig attribute)
(nni.nas.benchmarks.nasbench201.Nb201TrialConfig attribute)
(nni.nas.benchmarks.nds.NdsTrialConfig attribute)
num_vertices (nni.nas.benchmarks.nasbench101.Nb101TrialConfig attribute)
O
ObserverQuantizer (class in nni.algorithms.compression.pytorch.quantization)
OneshotPruner (class in nni.algorithms.compression.pytorch.pruning.one_shot_pruner)
Operation (class in nni.retiarii)
operation (nni.retiarii.Node attribute)
optimize_mode (nni.algorithms.hpo.metis_tuner.MetisTuner attribute)
(nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner attribute)
ori_test_acc (nni.nas.benchmarks.nasbench201.Nb201IntermediateStats attribute)
(nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute)
ori_test_evaluation_time (nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute)
ori_test_loss (nni.nas.benchmarks.nasbench201.Nb201IntermediateStats attribute)
(nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute)
outgoing_edges (nni.retiarii.Node attribute)
OUTPUT (nni.nas.benchmarks.nasbench101 attribute)
output_names (nni.retiarii.Graph attribute)
output_node (nni.retiarii.Graph attribute)
P
parameters (nni.nas.benchmarks.nasbench101.Nb101TrialStats attribute)
(nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute)
(nni.nas.benchmarks.nds.NdsTrialStats attribute)
(nni.retiarii.Operation attribute)
PolicyBasedRL (class in nni.retiarii.strategy)
predecessors (nni.retiarii.Node attribute)
process_command_thread() (nni.runtime.msg_dispatcher_base.MsgDispatcherBase method)
proposer (nni.nas.benchmarks.nds.NdsTrialConfig attribute)
ProxylessTrainer (class in nni.retiarii.oneshot.pytorch)
prune_iteration_start() (nni.algorithms.compression.pytorch.pruning.lottery_ticket.LotteryTicketPruner method)
Pruner (class in nni.compression.pytorch.compressor)
PrunerModuleWrapper (class in nni.compression.pytorch.compressor)
python_class (nni.retiarii.Model attribute)
python_init_params (nni.retiarii.Model attribute)
python_name (nni.retiarii.Graph attribute)
(nni.retiarii.Node attribute)
Q
QAT_Quantizer (class in nni.algorithms.compression.pytorch.quantization)
quantize_input() (nni.algorithms.compression.pytorch.quantization.LsqQuantizer method)
(nni.algorithms.compression.pytorch.quantization.ObserverQuantizer method)
(nni.algorithms.compression.pytorch.quantization.QAT_Quantizer method)
(nni.compression.pytorch.compressor.Quantizer method)
quantize_output() (nni.algorithms.compression.pytorch.quantization.BNNQuantizer method)
(nni.algorithms.compression.pytorch.quantization.LsqQuantizer method)
(nni.algorithms.compression.pytorch.quantization.ObserverQuantizer method)
(nni.algorithms.compression.pytorch.quantization.QAT_Quantizer method)
(nni.compression.pytorch.compressor.Quantizer method)
quantize_weight() (nni.algorithms.compression.pytorch.quantization.BNNQuantizer method)
(nni.algorithms.compression.pytorch.quantization.DoReFaQuantizer method)
(nni.algorithms.compression.pytorch.quantization.LsqQuantizer method)
(nni.algorithms.compression.pytorch.quantization.NaiveQuantizer method)
(nni.algorithms.compression.pytorch.quantization.ObserverQuantizer method)
(nni.algorithms.compression.pytorch.quantization.QAT_Quantizer method)
(nni.compression.pytorch.compressor.Quantizer method)
Quantizer (class in nni.compression.pytorch.compressor)
QuantizerModuleWrapper (class in nni.compression.pytorch.compressor)
query_nb101_trial_stats() (in module nni.nas.benchmarks.nasbench101)
query_nb201_trial_stats() (in module nni.nas.benchmarks.nasbench201)
query_nds_trial_stats() (in module nni.nas.benchmarks.nds)
R
Random (class in nni.retiarii.strategy)
RandomTuner (class in nni.algorithms.hpo.random_tuner)
read_calibration_cache() (nni.compression.pytorch.quantization_speedup.calibrator.Calibrator method)
receive_trial_result() (nni.algorithms.hpo.batch_tuner.BatchTuner method)
(nni.algorithms.hpo.evolution_tuner.EvolutionTuner method)
(nni.algorithms.hpo.gp_tuner.GPTuner method)
(nni.algorithms.hpo.gridsearch_tuner.GridSearchTuner method)
(nni.algorithms.hpo.hyperopt_tuner.HyperoptTuner method)
(nni.algorithms.hpo.metis_tuner.MetisTuner method)
(nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner method)
(nni.algorithms.hpo.random_tuner.RandomTuner method)
(nni.algorithms.hpo.tpe_tuner.TpeTuner method)
(nni.tuner.Tuner method)
record_shape() (nni.compression.pytorch.compressor.Quantizer method)
Regression (class in nni.retiarii.evaluator.pytorch.lightning)
Regression() (in module nni.retiarii.evaluator.pytorch.cgo.evaluator)
RegularizedEvolution (class in nni.retiarii.strategy)
remove_ungrouped_modules() (nni.algorithms.compression.pytorch.pruning.transformer_pruner.TransformerHeadPruner method)
Repeat (class in nni.retiarii.nn.pytorch)
report_final_result() (in module nni)
report_intermediate_result() (in module nni)
reset() (nni.compression.pytorch.compressor.Compressor method)
root_graph (nni.retiarii.Model attribute)
round_pass() (nni.algorithms.compression.pytorch.quantization.LsqQuantizer static method)
run() (nni.runtime.msg_dispatcher_base.MsgDispatcherBase method)
S
save_checkpoint() (nni.assessor.Assessor method)
(nni.tuner.Tuner method)
search_space (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner attribute)
seed (nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute)
(nni.nas.benchmarks.nds.NdsTrialStats attribute)
select_config() (nni.compression.pytorch.compressor.Compressor method)
selection_num_starting_points (nni.algorithms.hpo.metis_tuner.MetisTuner attribute)
SensitivityAnalysis (class in nni.compression.pytorch.utils.sensitivity_analysis)
SensitivityPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.sensitivity_pruner)
SEP_CONV_3X3 (nni.nas.benchmarks.nds.constants attribute)
SEP_CONV_5X5 (nni.nas.benchmarks.nds.constants attribute)
SEP_CONV_7X7 (nni.nas.benchmarks.nds.constants attribute)
set_wrappers_attribute() (nni.compression.pytorch.compressor.Compressor method)
SimulatedAnnealingPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.simulated_annealing_pruner)
(class in nni.algorithms.compression.v2.pytorch.pruning)
SinglePathTrainer (class in nni.retiarii.oneshot.pytorch)
SKIP_CONNECT (nni.nas.benchmarks.nasbench201 attribute)
(nni.nas.benchmarks.nds.constants attribute)
SlimPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.iterative_pruner)
(class in nni.algorithms.compression.v2.pytorch.pruning)
specialize_cell() (nni.retiarii.Node method)
status (nni.retiarii.Model attribute)
step_with_optimizer() (nni.algorithms.compression.pytorch.quantization.LsqQuantizer method)
(nni.algorithms.compression.pytorch.quantization.QAT_Quantizer method)
StructuredWeightMasker (class in nni.algorithms.compression.pytorch.pruning.structured_pruning_masker)
successors (nni.retiarii.Node attribute)
T
t_min (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner attribute)
tail (nni.retiarii.Edge attribute)
tail_slot (nni.retiarii.Edge attribute)
TaylorFOWeightFilterPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.iterative_pruner)
TaylorFOWeightPruner (class in nni.algorithms.compression.v2.pytorch.pruning)
test_acc (nni.nas.benchmarks.nasbench101.Nb101IntermediateStats attribute)
(nni.nas.benchmarks.nasbench101.Nb101TrialStats attribute)
(nni.nas.benchmarks.nasbench201.Nb201IntermediateStats attribute)
(nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute)
(nni.nas.benchmarks.nds.NdsIntermediateStats attribute)
test_evaluation_time (nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute)
test_loss (nni.nas.benchmarks.nasbench201.Nb201IntermediateStats attribute)
(nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute)
TPEStrategy (class in nni.retiarii.strategy)
TpeTuner (class in nni.algorithms.hpo.tpe_tuner)
trace() (in module nni)
train_acc (nni.nas.benchmarks.nasbench101.Nb101IntermediateStats attribute)
(nni.nas.benchmarks.nasbench101.Nb101TrialStats attribute)
(nni.nas.benchmarks.nasbench201.Nb201IntermediateStats attribute)
(nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute)
(nni.nas.benchmarks.nds.NdsIntermediateStats attribute)
train_loss (nni.nas.benchmarks.nasbench201.Nb201IntermediateStats attribute)
(nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute)
(nni.nas.benchmarks.nds.NdsIntermediateStats attribute)
training_time (nni.nas.benchmarks.nasbench101.Nb101IntermediateStats attribute)
(nni.nas.benchmarks.nasbench101.Nb101TrialStats attribute)
(nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute)
TransformerHeadPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.transformer_pruner)
trial (nni.nas.benchmarks.nasbench101.Nb101IntermediateStats attribute)
(nni.nas.benchmarks.nasbench201.Nb201IntermediateStats attribute)
(nni.nas.benchmarks.nds.NdsIntermediateStats attribute)
trial_end() (nni.algorithms.hpo.curvefitting_assessor.CurvefittingAssessor method)
(nni.algorithms.hpo.evolution_tuner.EvolutionTuner method)
(nni.algorithms.hpo.medianstop_assessor.MedianstopAssessor method)
(nni.algorithms.hpo.tpe_tuner.TpeTuner method)
(nni.assessor.Assessor method)
(nni.tuner.Tuner method)
Tuner (class in nni.tuner)
type (nni.retiarii.Operation attribute)
U
update() (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner method)
update_already_pruned() (nni.compression.pytorch.utils.sensitivity_analysis.SensitivityAnalysis method)
update_epoch() (nni.algorithms.compression.pytorch.pruning.iterative_pruner.AGPPruner method)
(nni.compression.pytorch.compressor.Compressor method)
update_mask() (nni.algorithms.compression.pytorch.pruning.transformer_pruner.TransformerHeadPruner method)
update_search_space() (nni.algorithms.hpo.batch_tuner.BatchTuner method)
(nni.algorithms.hpo.evolution_tuner.EvolutionTuner method)
(nni.algorithms.hpo.gp_tuner.GPTuner method)
(nni.algorithms.hpo.gridsearch_tuner.GridSearchTuner method)
(nni.algorithms.hpo.hyperopt_tuner.HyperoptTuner method)
(nni.algorithms.hpo.metis_tuner.MetisTuner method)
(nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner method)
(nni.algorithms.hpo.random_tuner.RandomTuner method)
(nni.algorithms.hpo.tpe_tuner.TpeTuner method)
(nni.tuner.Tuner method)
V
valid_acc (nni.nas.benchmarks.nasbench101.Nb101IntermediateStats attribute)
(nni.nas.benchmarks.nasbench101.Nb101TrialStats attribute)
(nni.nas.benchmarks.nasbench201.Nb201IntermediateStats attribute)
(nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute)
valid_evaluation_time (nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute)
valid_loss (nni.nas.benchmarks.nasbench201.Nb201IntermediateStats attribute)
(nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute)
validate_config() (nni.algorithms.compression.pytorch.pruning.auto_compress_pruner.AutoCompressPruner method)
(nni.algorithms.compression.pytorch.pruning.iterative_pruner.ADMMPruner method)
(nni.algorithms.compression.pytorch.pruning.iterative_pruner.AGPPruner method)
(nni.algorithms.compression.pytorch.pruning.iterative_pruner.SlimPruner method)
(nni.algorithms.compression.pytorch.pruning.lottery_ticket.LotteryTicketPruner method)
(nni.algorithms.compression.pytorch.pruning.net_adapt_pruner.NetAdaptPruner method)
(nni.algorithms.compression.pytorch.pruning.one_shot_pruner.OneshotPruner method)
(nni.algorithms.compression.pytorch.pruning.sensitivity_pruner.SensitivityPruner method)
(nni.algorithms.compression.pytorch.pruning.simulated_annealing_pruner.SimulatedAnnealingPruner method)
(nni.algorithms.compression.pytorch.pruning.transformer_pruner.TransformerHeadPruner method)
(nni.algorithms.compression.pytorch.quantization.BNNQuantizer method)
(nni.algorithms.compression.pytorch.quantization.DoReFaQuantizer method)
(nni.algorithms.compression.pytorch.quantization.NaiveQuantizer method)
(nni.algorithms.compression.pytorch.quantization.ObserverQuantizer method)
(nni.algorithms.compression.pytorch.quantization.QAT_Quantizer method)
(nni.compression.pytorch.compressor.Compressor method)
validate_weight_groups() (nni.algorithms.compression.pytorch.pruning.transformer_pruner.TransformerHeadPruner method)
ValueChoice (class in nni.retiarii.nn.pytorch)
verbose (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner attribute)
W
weight_decay (nni.nas.benchmarks.nds.NdsTrialConfig attribute)
WeightMasker (class in nni.algorithms.compression.pytorch.pruning.weight_masker)
write_calibration_cache() (nni.compression.pytorch.quantization_speedup.calibrator.Calibrator method)
Read the Docs
v: v2.6
Versions
latest
stable
v2.6
v2.5
v2.4
v2.3
v2.2
v2.1
v2.0
v1.9
v1.8
v1.7.1
v1.7
v1.6
Downloads
html
On Read the Docs
Project Home
Builds
Free document hosting provided by
Read the Docs
.