v2.2
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
|
K
|
L
|
M
|
N
|
O
|
P
|
Q
|
R
|
S
|
T
|
U
|
V
|
W
|
Z
A
ActivationAPoZRankFilterPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.one_shot)
ActivationMeanRankFilterPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.one_shot)
add_child() (nni.nas.pytorch.utils.StructuredMutableTreeNode method)
add_model() (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner method)
ADMMPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.admm_pruner)
AGPPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.agp)
algorithm_name (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner attribute)
AMCPruner (class in nni.algorithms.compression.pytorch.pruning)
analysis() (nni.compression.pytorch.utils.sensitivity_analysis.SensitivityAnalysis method)
apply() (nni.retiarii.Mutator method)
apply_fixed_architecture() (in module nni.nas.pytorch.fixed)
arch (nni.nas.benchmarks.nasbench101.Nb101TrialConfig attribute)
(nni.nas.benchmarks.nasbench201.Nb201TrialConfig attribute)
ArchitectureCheckpoint (class in nni.nas.pytorch.callbacks)
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)
AutoCompressPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.auto_compress_pruner)
AverageMeter (class in nni.nas.pytorch.utils)
AverageMeterGroup (class in nni.nas.pytorch.utils)
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)
BaseMutator (class in nni.nas.pytorch.base_mutator)
BaseTrainer (class in nni.nas.pytorch.base_trainer)
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.quantizers)
bo (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner attribute)
build() (nni.nas.pytorch.callbacks.Callback method)
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.agp.AGPPruner method)
(nni.algorithms.compression.pytorch.pruning.auto_compress_pruner.AutoCompressPruner 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.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.structured_pruning.StructuredWeightMasker method)
(nni.algorithms.compression.pytorch.pruning.weight_masker.WeightMasker method)
(nni.compression.pytorch.compressor.Pruner method)
Callback (class in nni.nas.pytorch.callbacks)
CatMaskPadding (class in nni.compression.pytorch.utils.mask_conflict)
CdartsTrainer (class in nni.algorithms.nas.pytorch.cdarts)
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)
checkpoint() (nni.nas.pytorch.base_trainer.BaseTrainer method)
(nni.nas.pytorch.trainer.Trainer method)
choice() (nni.retiarii.Mutator method)
choices (nni.algorithms.nas.pytorch.darts.DartsMutator attribute)
(nni.nas.pytorch.mutables.LayerChoice attribute)
(nni.retiarii.nn.pytorch.LayerChoice attribute)
ChosenInputs (class in nni.retiarii.nn.pytorch)
ClassicMutator (class in nni.algorithms.nas.pytorch.classic_nas.mutator)
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.admm_pruner.ADMMPruner method)
(nni.algorithms.compression.pytorch.pruning.auto_compress_pruner.AutoCompressPruner 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.compression.pytorch.compressor.Compressor method)
(nni.compression.pytorch.compressor.Pruner method)
Compressor (class in nni.compression.pytorch.compressor)
compute_target_sparsity() (nni.algorithms.compression.pytorch.pruning.agp.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)
ConvBranch (class in nni.nas.pytorch.search_space_zoo.enas_ops)
count_flops_params() (in module nni.compression.pytorch.utils.counter)
create_cfg() (nni.algorithms.compression.pytorch.pruning.sensitivity_pruner.SensitivityPruner 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)
cut_choices() (nni.algorithms.nas.pytorch.cdarts.RegularizedDartsMutator method)
(nni.algorithms.nas.pytorch.cdarts.RegularizedMutatorParallel method)
D
DartsCell (class in nni.nas.pytorch.search_space_zoo)
DartsDiscreteMutator (class in nni.algorithms.nas.pytorch.cdarts)
DartsMutator (class in nni.algorithms.nas.pytorch.darts)
DartsTrainer (class in nni.algorithms.nas.pytorch.darts)
(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)
DilConv (class in nni.nas.pytorch.search_space_zoo.darts_ops)
DoReFaQuantizer (class in nni.algorithms.compression.pytorch.quantization.quantizers)
dry_run() (nni.retiarii.Mutator method)
E
Edge (class in nni.retiarii)
edges (nni.retiarii.Graph attribute)
enable_visualization() (nni.nas.pytorch.trainer.Trainer method)
ENASMacroGeneralModel (class in nni.nas.pytorch.search_space_zoo)
ENASMacroLayer (class in nni.nas.pytorch.search_space_zoo)
ENASMicroLayer (class in nni.nas.pytorch.search_space_zoo)
EnasMutator (class in nni.algorithms.nas.pytorch.enas)
EnasTrainer (class in nni.algorithms.nas.pytorch.enas)
(class in nni.retiarii.oneshot.pytorch)
enqueue_command() (nni.runtime.msg_dispatcher_base.MsgDispatcherBase method)
enter_mutable_scope() (nni.nas.pytorch.base_mutator.BaseMutator method)
evaluator (nni.retiarii.Model attribute)
EvolutionTuner (class in nni.algorithms.hpo.evolution_tuner)
exit_mutable_scope() (nni.nas.pytorch.base_mutator.BaseMutator method)
exploration_probability (nni.algorithms.hpo.metis_tuner.MetisTuner attribute)
export() (nni.algorithms.nas.pytorch.cdarts.RegularizedDartsMutator method)
(nni.algorithms.nas.pytorch.cdarts.RegularizedMutatorParallel method)
(nni.compression.pytorch.utils.sensitivity_analysis.SensitivityAnalysis method)
(nni.compression.pytorch.utils.shape_dependency.ChannelDependency method)
(nni.compression.pytorch.utils.shape_dependency.GroupDependency method)
(nni.nas.pytorch.base_mutator.BaseMutator method)
(nni.nas.pytorch.base_trainer.BaseTrainer method)
(nni.nas.pytorch.mutator.Mutator method)
(nni.nas.pytorch.trainer.Trainer method)
export_model() (nni.algorithms.compression.pytorch.pruning.auto_compress_pruner.AutoCompressPruner method)
(nni.algorithms.compression.pytorch.quantization.quantizers.BNNQuantizer method)
(nni.algorithms.compression.pytorch.quantization.quantizers.DoReFaQuantizer method)
(nni.algorithms.compression.pytorch.quantization.quantizers.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_results() (nni.algorithms.nas.pytorch.spos.SPOSEvolution 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)
fix_mask() (nni.compression.pytorch.utils.mask_conflict.ChannelMaskConflict method)
(nni.compression.pytorch.utils.mask_conflict.GroupMaskConflict method)
FixedArchitecture (class in nni.nas.pytorch.fixed)
flops (nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute)
(nni.nas.benchmarks.nds.NdsTrialStats attribute)
fork() (nni.retiarii.Graph method)
(nni.retiarii.Model method)
forward() (nni.compression.pytorch.compressor.PrunerModuleWrapper method)
(nni.compression.pytorch.compressor.QuantizerModuleWrapper method)
(nni.nas.pytorch.base_mutator.BaseMutator method)
(nni.nas.pytorch.mutables.InputChoice method)
(nni.nas.pytorch.mutables.LayerChoice method)
(nni.nas.pytorch.nasbench201.nasbench201_ops.Pooling method)
(nni.nas.pytorch.nasbench201.nasbench201_ops.ReLUConvBN method)
(nni.nas.pytorch.nasbench201.nasbench201_ops.Zero method)
(nni.nas.pytorch.nasbench201.NASBench201Cell method)
(nni.nas.pytorch.search_space_zoo.DartsCell method)
(nni.nas.pytorch.search_space_zoo.ENASMacroGeneralModel method)
(nni.nas.pytorch.search_space_zoo.ENASMacroLayer method)
(nni.nas.pytorch.search_space_zoo.ENASMicroLayer method)
(nni.retiarii.nn.pytorch.ChosenInputs method)
(nni.retiarii.nn.pytorch.InputChoice method)
(nni.retiarii.nn.pytorch.LayerChoice method)
(nni.retiarii.nn.pytorch.ValueChoice method)
FPGMPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.one_shot)
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.algorithms.nas.pytorch.spos.SPOSEvolution 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.tuner.Tuner method)
generator (nni.nas.benchmarks.nds.NdsTrialConfig attribute)
get_and_apply_next_architecture() (in module nni.algorithms.nas.pytorch.classic_nas)
get_best_model_id() (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner method)
get_channel_sum() (nni.algorithms.compression.pytorch.pruning.structured_pruning.StructuredWeightMasker method)
get_current_parameter() (in module nni)
get_experiment_id() (in module nni)
get_mask() (nni.algorithms.compression.pytorch.pruning.structured_pruning.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_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_sequence_id() (in module nni)
get_suggestion() (nni.algorithms.hpo.hyperopt_tuner.HyperoptTuner method)
get_trial_id() (in module nni)
global_mutable_counting() (in module nni.nas.pytorch.utils)
Good (nni.assessor.AssessResult attribute)
GPTuner (class in nni.algorithms.hpo.gp_tuner)
Graph (class in nni.retiarii)
graph (nni.retiarii.Edge attribute)
(nni.retiarii.Node attribute)
graph() (nni.nas.pytorch.mutator.Mutator method)
graphs (nni.retiarii.Model attribute)
GridSearch (class in nni.retiarii.strategy)
GridSearchTuner (class in nni.algorithms.hpo.gridsearch_tuner)
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.tuner.Tuner method)
incoming_edges (nni.retiarii.Node attribute)
infer_num_vertices() (in module nni.nas.benchmarks.nasbench101.graph_util)
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.nas.pytorch.mutables)
(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)
K
key() (nni.nas.pytorch.mutables.Mutable property)
L
L1FilterPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.one_shot)
L2FilterPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.one_shot)
latency (nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute)
LayerChoice (class in nni.nas.pytorch.mutables)
(class in nni.retiarii.nn.pytorch)
length (nni.nas.pytorch.mutables.LayerChoice attribute)
(nni.retiarii.nn.pytorch.LayerChoice attribute)
LevelPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.one_shot)
load_best_model() (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner 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_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)
LRSchedulerCallback (class in nni.nas.pytorch.callbacks)
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)
ModelCheckpoint (class in nni.nas.pytorch.callbacks)
MsgDispatcherBase (class in nni.runtime.msg_dispatcher_base)
Mutable (class in nni.nas.pytorch.mutables)
mutables() (nni.nas.pytorch.base_mutator.BaseMutator property)
MutableScope (class in nni.nas.pytorch.mutables)
mutate() (nni.retiarii.Mutator method)
Mutator (class in nni.nas.pytorch.mutator)
(class in nni.retiarii)
N
n_classes (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner attribute)
NaiveQuantizer (class in nni.algorithms.compression.pytorch.quantization.quantizers)
name (nni.retiarii.Graph attribute)
(nni.retiarii.Node attribute)
name() (nni.nas.pytorch.mutables.Mutable property)
names (nni.nas.pytorch.mutables.LayerChoice attribute)
(nni.retiarii.nn.pytorch.LayerChoice attribute)
NASBench201Cell (class in nni.nas.pytorch.nasbench201)
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
on_epoch_begin() (nni.nas.pytorch.callbacks.Callback method)
on_epoch_end() (nni.nas.pytorch.callbacks.ArchitectureCheckpoint method)
(nni.nas.pytorch.callbacks.Callback method)
(nni.nas.pytorch.callbacks.LRSchedulerCallback method)
(nni.nas.pytorch.callbacks.ModelCheckpoint method)
on_forward_input_choice() (nni.nas.pytorch.base_mutator.BaseMutator method)
(nni.nas.pytorch.mutator.Mutator method)
on_forward_layer_choice() (nni.nas.pytorch.base_mutator.BaseMutator method)
(nni.nas.pytorch.mutator.Mutator method)
OneshotPruner (class in nni.algorithms.compression.pytorch.pruning.one_shot)
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)
Pool (class in nni.nas.pytorch.search_space_zoo.enas_ops)
PoolBN (class in nni.nas.pytorch.search_space_zoo.darts_ops)
PoolBranch (class in nni.nas.pytorch.search_space_zoo.enas_ops)
Pooling (class in nni.nas.pytorch.nasbench201.nasbench201_ops)
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)
Q
QAT_Quantizer (class in nni.algorithms.compression.pytorch.quantization.quantizers)
quantize_input() (nni.compression.pytorch.compressor.Quantizer method)
quantize_output() (nni.algorithms.compression.pytorch.quantization.quantizers.BNNQuantizer method)
(nni.algorithms.compression.pytorch.quantization.quantizers.QAT_Quantizer method)
(nni.compression.pytorch.compressor.Quantizer method)
quantize_weight() (nni.algorithms.compression.pytorch.quantization.quantizers.BNNQuantizer method)
(nni.algorithms.compression.pytorch.quantization.quantizers.DoReFaQuantizer method)
(nni.algorithms.compression.pytorch.quantization.quantizers.NaiveQuantizer method)
(nni.algorithms.compression.pytorch.quantization.quantizers.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)
RandomMutator (class in nni.algorithms.nas.pytorch.random)
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.nas.pytorch.spos.SPOSEvolution method)
(nni.tuner.Tuner method)
RegularizedDartsMutator (class in nni.algorithms.nas.pytorch.cdarts)
RegularizedEvolution (class in nni.retiarii.strategy)
RegularizedMutatorParallel (class in nni.algorithms.nas.pytorch.cdarts)
ReLUConvBN (class in nni.nas.pytorch.nasbench201.nasbench201_ops)
replace_layer_choice() (nni.nas.pytorch.fixed.FixedArchitecture method)
report_final_result() (in module nni)
report_intermediate_result() (in module nni)
reset() (nni.algorithms.nas.pytorch.cdarts.RegularizedDartsMutator method)
(nni.compression.pytorch.compressor.Compressor method)
(nni.nas.pytorch.mutator.Mutator method)
(nni.nas.pytorch.utils.AverageMeter method)
reset_with_loss() (nni.algorithms.nas.pytorch.cdarts.RegularizedDartsMutator method)
(nni.algorithms.nas.pytorch.cdarts.RegularizedMutatorParallel method)
root_graph (nni.retiarii.Model attribute)
run() (nni.runtime.msg_dispatcher_base.MsgDispatcherBase method)
S
sample_final() (nni.algorithms.nas.pytorch.classic_nas.mutator.ClassicMutator method)
(nni.algorithms.nas.pytorch.darts.DartsMutator method)
(nni.algorithms.nas.pytorch.enas.EnasMutator method)
(nni.algorithms.nas.pytorch.random.RandomMutator method)
(nni.algorithms.nas.pytorch.spos.SPOSSupernetTrainingMutator method)
(nni.nas.pytorch.fixed.FixedArchitecture method)
(nni.nas.pytorch.mutator.Mutator method)
sample_search() (nni.algorithms.nas.pytorch.cdarts.DartsDiscreteMutator method)
(nni.algorithms.nas.pytorch.cdarts.RegularizedDartsMutator method)
(nni.algorithms.nas.pytorch.classic_nas.mutator.ClassicMutator method)
(nni.algorithms.nas.pytorch.darts.DartsMutator method)
(nni.algorithms.nas.pytorch.enas.EnasMutator method)
(nni.algorithms.nas.pytorch.random.RandomMutator method)
(nni.algorithms.nas.pytorch.spos.SPOSSupernetTrainingMutator method)
(nni.nas.pytorch.fixed.FixedArchitecture method)
(nni.nas.pytorch.mutator.Mutator method)
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)
SepConv (class in nni.nas.pytorch.search_space_zoo.darts_ops)
SepConvBN (class in nni.nas.pytorch.search_space_zoo.enas_ops)
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)
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.one_shot)
specialize_cell() (nni.retiarii.Node method)
SPOSEvolution (class in nni.algorithms.nas.pytorch.spos)
SPOSSupernetTrainer (class in nni.algorithms.nas.pytorch.spos)
SPOSSupernetTrainingMutator (class in nni.algorithms.nas.pytorch.spos)
status (nni.retiarii.Model attribute)
status() (nni.nas.pytorch.mutator.Mutator method)
step_with_optimizer() (nni.algorithms.compression.pytorch.quantization.quantizers.QAT_Quantizer method)
StructuredMutableTreeNode (class in nni.nas.pytorch.utils)
StructuredWeightMasker (class in nni.algorithms.compression.pytorch.pruning.structured_pruning)
successors (nni.retiarii.Node attribute)
summary() (nni.nas.pytorch.utils.AverageMeterGroup method)
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.one_shot)
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)
to_device() (in module nni.nas.pytorch.utils)
TPEStrategy (class in nni.retiarii.strategy)
train() (nni.nas.pytorch.base_trainer.BaseTrainer method)
(nni.nas.pytorch.trainer.Trainer method)
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)
train_one_epoch() (nni.algorithms.nas.pytorch.darts.DartsTrainer method)
(nni.algorithms.nas.pytorch.enas.EnasTrainer method)
(nni.algorithms.nas.pytorch.spos.SPOSSupernetTrainer method)
(nni.nas.pytorch.trainer.Trainer method)
Trainer (class in nni.nas.pytorch.trainer)
training_time (nni.nas.benchmarks.nasbench101.Nb101IntermediateStats attribute)
(nni.nas.benchmarks.nasbench101.Nb101TrialStats attribute)
(nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute)
traverse() (nni.nas.pytorch.utils.StructuredMutableTreeNode method)
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.nas.pytorch.spos.SPOSEvolution method)
(nni.assessor.Assessor method)
(nni.tuner.Tuner method)
Tuner (class in nni.tuner)
type (nni.retiarii.Operation attribute)
type() (nni.nas.pytorch.utils.StructuredMutableTreeNode method)
U
update() (nni.algorithms.hpo.networkmorphism_tuner.NetworkMorphismTuner method)
(nni.nas.pytorch.utils.AverageMeter method)
(nni.nas.pytorch.utils.AverageMeterGroup method)
update_already_pruned() (nni.compression.pytorch.utils.sensitivity_analysis.SensitivityAnalysis method)
update_epoch() (nni.algorithms.compression.pytorch.pruning.agp.AGPPruner method)
(nni.compression.pytorch.compressor.Compressor 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.nas.pytorch.spos.SPOSEvolution 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() (nni.nas.pytorch.base_trainer.BaseTrainer method)
(nni.nas.pytorch.trainer.Trainer method)
validate_config() (nni.algorithms.compression.pytorch.pruning.admm_pruner.ADMMPruner method)
(nni.algorithms.compression.pytorch.pruning.agp.AGPPruner method)
(nni.algorithms.compression.pytorch.pruning.auto_compress_pruner.AutoCompressPruner 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.OneshotPruner method)
(nni.algorithms.compression.pytorch.pruning.one_shot.SlimPruner method)
(nni.algorithms.compression.pytorch.pruning.sensitivity_pruner.SensitivityPruner method)
(nni.algorithms.compression.pytorch.pruning.simulated_annealing_pruner.SimulatedAnnealingPruner method)
(nni.algorithms.compression.pytorch.quantization.quantizers.BNNQuantizer method)
(nni.algorithms.compression.pytorch.quantization.quantizers.DoReFaQuantizer method)
(nni.algorithms.compression.pytorch.quantization.quantizers.NaiveQuantizer method)
(nni.algorithms.compression.pytorch.quantization.quantizers.QAT_Quantizer method)
(nni.compression.pytorch.compressor.Compressor method)
validate_one_epoch() (nni.algorithms.nas.pytorch.darts.DartsTrainer method)
(nni.algorithms.nas.pytorch.enas.EnasTrainer method)
(nni.algorithms.nas.pytorch.spos.SPOSSupernetTrainer method)
(nni.nas.pytorch.trainer.Trainer 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)
Z
Zero (class in nni.nas.pytorch.nasbench201.nasbench201_ops)
Read the Docs
v: v2.2
Versions
latest
stable
v2.2
v2.1
v2.0
v1.9
v1.8
v1.7.1
v1.7
v1.6
md2rst
fix-docs-dependencies
Downloads
On Read the Docs
Project Home
Builds
Free document hosting provided by
Read the Docs
.