v2.4
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)
ActivationMeanRankFilterPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.iterative_pruner)
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)
advisor (nni.experiment.config.ExperimentConfig attribute)
AGPPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.iterative_pruner)
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)
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)
assessor (nni.experiment.config.ExperimentConfig attribute)
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)
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)
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_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)
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)
class_args (nni.experiment.config.AlgorithmConfig attribute)
(nni.experiment.config.CustomAlgorithmConfig attribute)
class_name (nni.experiment.config.CustomAlgorithmConfig attribute)
code_directory (nni.experiment.config.CustomAlgorithmConfig attribute)
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.compression.pytorch.compressor.Compressor method)
(nni.compression.pytorch.compressor.Pruner method)
Compressor (class in nni.compression.pytorch.compressor)
compute_target (nni.experiment.config.AmlConfig attribute)
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)
container_name (nni.experiment.config.AzureBlobConfig attribute)
container_storage_mount_point (nni.experiment.config.OpenpaiConfig 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)
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)
debug (nni.experiment.config.ExperimentConfig 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)
docker_image (nni.experiment.config.AmlConfig attribute)
(nni.experiment.config.OpenpaiConfig attribute)
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)
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)
experiment_name (nni.experiment.config.ExperimentConfig attribute)
experiment_working_directory (nni.experiment.config.ExperimentConfig attribute)
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.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)
exported_directory (nni.experiment.config.NfsConfig attribute)
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.algorithms.compression.pytorch.quantization.quantizers.QAT_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)
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.tuner.Tuner method)
generator (nni.nas.benchmarks.nds.NdsTrialConfig attribute)
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_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)
gpu_indices (nni.experiment.config.LocalConfig attribute)
(nni.experiment.config.RemoteMachineConfig attribute)
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)
host (nni.experiment.config.OpenpaiConfig attribute)
(nni.experiment.config.RemoteMachineConfig 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.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)
L2FilterPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.one_shot_pruner)
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)
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)
local_storage_mount_point (nni.experiment.config.OpenpaiConfig attribute)
log_level (nni.experiment.config.ExperimentConfig attribute)
LotteryTicketPruner (class in nni.algorithms.compression.pytorch.pruning)
(class in nni.algorithms.compression.pytorch.pruning.lottery_ticket)
M
machine_list (nni.experiment.config.RemoteConfig attribute)
mask_dict (nni.compression.pytorch.compressor.Pruner attribute)
max_experiment_duration (nni.experiment.config.ExperimentConfig 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)
max_trial_number (nni.experiment.config.ExperimentConfig attribute)
max_trial_number_per_gpu (nni.experiment.config.AmlConfig attribute)
(nni.experiment.config.LocalConfig attribute)
(nni.experiment.config.RemoteMachineConfig 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)
MsgDispatcherBase (class in nni.runtime.msg_dispatcher_base)
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.quantizers)
name (nni.experiment.config.AlgorithmConfig attribute)
(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)
nfs_server (nni.experiment.config.NfsConfig attribute)
nni_manager_ip (nni.experiment.config.ExperimentConfig attribute)
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
OneshotPruner (class in nni.algorithms.compression.pytorch.pruning.one_shot_pruner)
openpai_config (nni.experiment.config.OpenpaiConfig attribute)
openpai_config_file (nni.experiment.config.OpenpaiConfig attribute)
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)
password (nni.experiment.config.RemoteMachineConfig attribute)
platform (nni.experiment.config.AmlConfig attribute)
(nni.experiment.config.LocalConfig attribute)
(nni.experiment.config.OpenpaiConfig attribute)
(nni.experiment.config.RemoteConfig attribute)
PolicyBasedRL (class in nni.retiarii.strategy)
port (nni.experiment.config.RemoteMachineConfig attribute)
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_path (nni.experiment.config.RemoteMachineConfig attribute)
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)
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.tuner.Tuner method)
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)
resource_group (nni.experiment.config.AmlConfig attribute)
resource_group_name (nni.experiment.config.AzureBlobConfig attribute)
reuse_mode (nni.experiment.config.OpenpaiConfig attribute)
(nni.experiment.config.RemoteConfig attribute)
root_graph (nni.retiarii.Model attribute)
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)
(nni.experiment.config.ExperimentConfig attribute)
search_space_file (nni.experiment.config.ExperimentConfig 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)
serialize() (in module nni.retiarii)
set_wrappers_attribute() (nni.compression.pytorch.compressor.Compressor method)
shared_storage (nni.experiment.config.ExperimentConfig attribute)
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.iterative_pruner)
specialize_cell() (nni.retiarii.Node method)
ssh_key_file (nni.experiment.config.RemoteMachineConfig attribute)
ssh_passphrase (nni.experiment.config.RemoteMachineConfig attribute)
status (nni.retiarii.Model attribute)
step_with_optimizer() (nni.algorithms.compression.pytorch.quantization.quantizers.QAT_Quantizer method)
storage_account_key (nni.experiment.config.AzureBlobConfig attribute)
storage_account_name (nni.experiment.config.AzureBlobConfig attribute)
storage_config_name (nni.experiment.config.OpenpaiConfig attribute)
storage_type (nni.experiment.config.AzureBlobConfig attribute)
(nni.experiment.config.NfsConfig attribute)
StructuredWeightMasker (class in nni.algorithms.compression.pytorch.pruning.structured_pruning_masker)
subscription_id (nni.experiment.config.AmlConfig attribute)
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)
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)
token (nni.experiment.config.OpenpaiConfig attribute)
TPEStrategy (class in nni.retiarii.strategy)
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_service (nni.experiment.config.ExperimentConfig 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_code_directory (nni.experiment.config.ExperimentConfig attribute)
trial_command (nni.experiment.config.ExperimentConfig attribute)
trial_concurrency (nni.experiment.config.ExperimentConfig attribute)
trial_cpu_number (nni.experiment.config.OpenpaiConfig 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.assessor.Assessor method)
(nni.tuner.Tuner method)
trial_gpu_number (nni.experiment.config.ExperimentConfig attribute)
trial_memory_size (nni.experiment.config.OpenpaiConfig attribute)
Tuner (class in nni.tuner)
tuner (nni.experiment.config.ExperimentConfig attribute)
tuner_gpu_indices (nni.experiment.config.ExperimentConfig attribute)
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.tuner.Tuner method)
use_active_gpu (nni.experiment.config.LocalConfig attribute)
(nni.experiment.config.RemoteMachineConfig attribute)
use_annotation (nni.experiment.config.ExperimentConfig attribute)
user (nni.experiment.config.RemoteMachineConfig attribute)
username (nni.experiment.config.OpenpaiConfig attribute)
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.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_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)
workspace_name (nni.experiment.config.AmlConfig attribute)
Read the Docs
v: v2.4
Versions
latest
stable
v2.4
v2.3
v2.2
v2.1
v2.0
v1.9
v1.8
v1.7.1
v1.7
v1.6
md2rst
fix-docs-dependencies
Downloads
html
On Read the Docs
Project Home
Builds
Free document hosting provided by
Read the Docs
.