Index _ | A | B | C | D | E | F | G | H | I | L | M | N | O | P | Q | R | S | T | U | V | W _ _sampled (nni.retiarii.oneshot.pytorch.supermodule.sampling.PathSamplingInput attribute) (nni.retiarii.oneshot.pytorch.supermodule.sampling.PathSamplingLayer attribute) (nni.retiarii.oneshot.pytorch.supermodule.sampling.PathSamplingRepeat attribute) A ActivationAPoZRankPruner (class in nni.compression.pytorch.pruning) ActivationMeanRankPruner (class in nni.compression.pytorch.pruning) ADMMPruner (class in nni.compression.pytorch.pruning) advance_lr_schedulers() (nni.retiarii.oneshot.pytorch.base_lightning.BaseOneShotLightningModule method) advance_optimization() (nni.retiarii.oneshot.pytorch.base_lightning.BaseOneShotLightningModule method) AGPPruner (class in nni.compression.pytorch.pruning) all_options() (nni.retiarii.nn.pytorch.ValueChoice method) AMCPruner (class in nni.compression.pytorch.pruning) AmoebaNet (class in nni.retiarii.hub.pytorch) analysis() (nni.compression.pytorch.utils.SensitivityAnalysis method) apply() (nni.retiarii.Mutator method) arch (nni.nas.benchmarks.nasbench101.Nb101TrialConfig attribute) (nni.nas.benchmarks.nasbench201.Nb201TrialConfig attribute) architecture_optimizers() (nni.retiarii.oneshot.pytorch.base_lightning.BaseOneShotLightningModule method) arguments (nni.retiarii.evaluator.FunctionalEvaluator attribute) assess_trial() (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.compression.pytorch.pruning) AutoformerSpace (class in nni.retiarii.hub.pytorch) B Bad (nni.assessor.AssessResult attribute) base_lr (nni.nas.benchmarks.nds.NdsTrialConfig attribute) BaseOneShotLightningModule (class in nni.retiarii.oneshot.pytorch.base_lightning) BaseOneShotTrainer (class in nni.retiarii.oneshot) BasePruningScheduler (class in nni.algorithms.compression.v2.pytorch.base) BaseStrategy (class in nni.retiarii.strategy) BaseSuperNetModule (class in nni.retiarii.oneshot.pytorch.base_lightning) basic_unit() (in module nni.retiarii) BasicPruner (class in nni.algorithms.compression.v2.pytorch.pruning.basic_pruner) 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) bind_sampler() (nni.retiarii.Mutator method) BNNQuantizer (class in nni.algorithms.compression.pytorch.quantization) BOHB (class in nni.algorithms.hpo.bohb_advisor) bound_type (nni.retiarii.oneshot.pytorch.supermodule.operation.MixedBatchNorm2d attribute) (nni.retiarii.oneshot.pytorch.supermodule.operation.MixedConv2d attribute) (nni.retiarii.oneshot.pytorch.supermodule.operation.MixedLayerNorm attribute) (nni.retiarii.oneshot.pytorch.supermodule.operation.MixedLinear attribute) (nni.retiarii.oneshot.pytorch.supermodule.operation.MixedMultiHeadAttention attribute) budget_exhausted() (in module nni.retiarii.execution) build_dependency() (nni.compression.pytorch.utils.ChannelDependency method) (nni.compression.pytorch.utils.GroupDependency method) C calculate_metrics() (nni.algorithms.compression.v2.pytorch.pruning.tools.MetricsCalculator 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) ChannelMaskConflict (class in nni.compression.pytorch.utils) choice() (nni.retiarii.Mutator method) choices (nni.retiarii.nn.pytorch.LayerChoice attribute) chosen (nni.retiarii.nn.pytorch.ChosenInputs attribute) ChosenInputs (class in nni.retiarii.nn.pytorch) Classification (class in nni.retiarii.evaluator.pytorch) (class in nni.retiarii.evaluator.pytorch.cgo.evaluator) ClassificationModule (class in nni.retiarii.evaluator.pytorch) collect() (nni.algorithms.compression.v2.pytorch.pruning.tools.DataCollector method) common_target_masks_generation() (nni.algorithms.compression.v2.pytorch.pruning.tools.SparsityAllocator method) compress() (nni.algorithms.compression.v2.pytorch.base.BasePruningScheduler method) (nni.algorithms.compression.v2.pytorch.base.Pruner method) (nni.algorithms.compression.v2.pytorch.pruning.basic_pruner.BasicPruner method) (nni.compression.pytorch.quantization_speedup.ModelSpeedupTensorRT method) compute_sparsity() (in module nni.algorithms.compression.v2.pytorch.utils.pruning) ConcatLoader (class in nni.retiarii.oneshot.pytorch.dataloader) condition() (nni.retiarii.nn.pytorch.ValueChoice static method) config (nni.experiment.Experiment attribute) (nni.nas.benchmarks.nasbench101.Nb101TrialStats attribute) (nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute) (nni.nas.benchmarks.nds.NdsTrialStats attribute) configure_architecture_optimizers() (nni.retiarii.oneshot.pytorch.base_lightning.BaseOneShotLightningModule method) configure_optimizers() (nni.retiarii.oneshot.pytorch.base_lightning.BaseOneShotLightningModule method) connect() (nni.experiment.Experiment class method) ContextStack (class in nni.retiarii.utils) count_flops_params() (in module nni.compression.pytorch.utils) create_fixed_module() (nni.retiarii.nn.pytorch.mutation_utils.Mutable class method) current_context() (nni.retiarii.utils.ModelNamespace static method) current_epoch (nni.nas.benchmarks.nasbench101.Nb101IntermediateStats attribute) (nni.nas.benchmarks.nasbench201.Nb201IntermediateStats attribute) (nni.nas.benchmarks.nds.NdsIntermediateStats attribute) CurvefittingAssessor (class in nni.algorithms.hpo.curvefitting_assessor) D DARTS (class in nni.retiarii.hub.pytorch) (class in nni.retiarii.strategy) DartsTrainer (class in nni.retiarii.oneshot.pytorch) DataCollector (class in nni.algorithms.compression.v2.pytorch.pruning.tools) DataLoader (class in nni.retiarii.evaluator.pytorch) dataset (nni.nas.benchmarks.nasbench201.Nb201TrialConfig attribute) (nni.nas.benchmarks.nds.NdsTrialConfig attribute) default_mutation_hooks() (nni.retiarii.oneshot.pytorch.base_lightning.BaseOneShotLightningModule method) dependency_sets (nni.compression.pytorch.utils.ChannelDependency property) DifferentiableMixedCell (class in nni.retiarii.oneshot.pytorch.supermodule.differentiable) DifferentiableMixedInput (class in nni.retiarii.oneshot.pytorch.supermodule.differentiable) DifferentiableMixedLayer (class in nni.retiarii.oneshot.pytorch.supermodule.differentiable) DifferentiableMixedRepeat (class in nni.retiarii.oneshot.pytorch.supermodule.differentiable) DNGOTuner (class in nni.algorithms.hpo.dngo_tuner) DoReFaQuantizer (class in nni.algorithms.compression.pytorch.quantization) downsampling (nni.retiarii.hub.pytorch.nasnet.NDSStage attribute) dry_run() (nni.retiarii.Mutator method) dump() (in module nni.common.serializer) E Edge (class in nni.retiarii) edges (nni.retiarii.Graph attribute) ENAS (class in nni.retiarii.hub.pytorch) (class in nni.retiarii.strategy) EnasTrainer (class in nni.retiarii.oneshot.pytorch) estimated_out_channels (nni.retiarii.hub.pytorch.nasnet.NDSStage attribute) estimated_out_channels_prev (nni.retiarii.hub.pytorch.nasnet.NDSStage attribute) evaluate() (nni.retiarii.Evaluator method) (nni.retiarii.nn.pytorch.ValueChoice method) Evaluator (class in nni.retiarii) evaluator (nni.retiarii.Model attribute) EvolutionTuner (class in nni.algorithms.hpo.evolution_tuner) Experiment (class in nni.experiment) export() (nni.compression.pytorch.utils.ChannelDependency method) (nni.compression.pytorch.utils.GroupDependency method) (nni.compression.pytorch.utils.SensitivityAnalysis method) (nni.retiarii.oneshot.pytorch.base_lightning.BaseOneShotLightningModule method) (nni.retiarii.oneshot.pytorch.base_lightning.BaseSuperNetModule method) (nni.retiarii.oneshot.pytorch.supermodule.differentiable.DifferentiableMixedCell method) (nni.retiarii.oneshot.pytorch.supermodule.differentiable.DifferentiableMixedInput method) (nni.retiarii.oneshot.pytorch.supermodule.differentiable.DifferentiableMixedLayer method) (nni.retiarii.oneshot.pytorch.supermodule.differentiable.DifferentiableMixedRepeat method) (nni.retiarii.oneshot.pytorch.supermodule.differentiable.MixedOpDifferentiablePolicy method) (nni.retiarii.oneshot.pytorch.supermodule.operation.MixedOperation method) (nni.retiarii.oneshot.pytorch.supermodule.operation.MixedOperationSamplingPolicy method) (nni.retiarii.oneshot.pytorch.supermodule.sampling.MixedOpPathSamplingPolicy method) (nni.retiarii.oneshot.pytorch.supermodule.sampling.PathSamplingCell method) (nni.retiarii.oneshot.pytorch.supermodule.sampling.PathSamplingInput method) (nni.retiarii.oneshot.pytorch.supermodule.sampling.PathSamplingLayer method) (nni.retiarii.oneshot.pytorch.supermodule.sampling.PathSamplingRepeat method) export_data() (nni.experiment.Experiment method) export_model() (nni.algorithms.compression.v2.pytorch.base.Pruner method) (nni.compression.pytorch.compressor.Quantizer method) export_model_save() (nni.compression.pytorch.compressor.Quantizer method) export_probs() (nni.retiarii.oneshot.pytorch.base_lightning.BaseOneShotLightningModule method) (nni.retiarii.oneshot.pytorch.base_lightning.BaseSuperNetModule method) (nni.retiarii.oneshot.pytorch.supermodule.differentiable.DifferentiableMixedCell method) (nni.retiarii.oneshot.pytorch.supermodule.differentiable.DifferentiableMixedRepeat method) (nni.retiarii.oneshot.pytorch.supermodule.differentiable.MixedOpDifferentiablePolicy method) (nni.retiarii.oneshot.pytorch.supermodule.operation.MixedOperation method) (nni.retiarii.oneshot.pytorch.supermodule.operation.MixedOperationSamplingPolicy method) export_quantized_model() (nni.compression.pytorch.quantization_speedup.ModelSpeedupTensorRT method) export_top_models() (nni.retiarii.experiment.pytorch.RetiariiExperiment 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) FIRST() (nni.retiarii.nn.pytorch.ModelParameterChoice static method) first_cell_transformation_factory() (nni.retiarii.hub.pytorch.nasnet.NDSStage method) fix_mask() (nni.compression.pytorch.utils.ChannelMaskConflict method) (nni.compression.pytorch.utils.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.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.NasBench101Cell method) (nni.retiarii.nn.pytorch.NasBench201Cell method) (nni.retiarii.nn.pytorch.ValueChoice method) (nni.retiarii.oneshot.pytorch.supermodule.differentiable.DifferentiableMixedInput method) (nni.retiarii.oneshot.pytorch.supermodule.differentiable.DifferentiableMixedLayer method) (nni.retiarii.oneshot.pytorch.supermodule.operation.MixedOperation method) (nni.retiarii.oneshot.pytorch.supermodule.proxyless.ProxylessMixedInput method) (nni.retiarii.oneshot.pytorch.supermodule.proxyless.ProxylessMixedLayer method) forward_argument() (nni.retiarii.oneshot.pytorch.supermodule.operation.MixedOperation method) (nni.retiarii.oneshot.pytorch.supermodule.operation.MixedOperationSamplingPolicy method) forward_with_args() (nni.retiarii.oneshot.pytorch.supermodule.operation.MixedOperation method) FPGMPruner (class in nni.compression.pytorch.pruning) function (nni.retiarii.evaluator.FunctionalEvaluator attribute) FunctionalEvaluator (class in nni.retiarii.evaluator) G generate_multiple_parameters() (nni.tuner.Tuner method) generate_parameters() (nni.tuner.Tuner method) generate_sparsity() (nni.algorithms.compression.v2.pytorch.pruning.tools.SparsityAllocator method) generate_task() (nni.algorithms.compression.v2.pytorch.base.BasePruningScheduler method) generator (nni.nas.benchmarks.nds.NdsTrialConfig attribute) get() (nni.common.serializer.Traceable method) get_all_experiments_metadata() (nni.experiment.Experiment method) get_and_register_default_listener() (in module nni.retiarii.execution) get_best_result() (nni.algorithms.compression.v2.pytorch.base.BasePruningScheduler method) (nni.algorithms.compression.v2.pytorch.pruning.tools.TaskGenerator method) get_bits_length() (nni.compression.pytorch.compressor.QuantGrad class method) get_default_framework() (in module nni.common.framework) get_execution_engine() (in module nni.retiarii.execution) get_experiment_id() (in module nni) get_experiment_metadata() (nni.experiment.Experiment method) get_experiment_profile() (nni.experiment.Experiment method) get_job_metrics() (nni.experiment.Experiment method) get_job_statistics() (nni.experiment.Experiment method) get_modules_wrapper() (nni.algorithms.compression.v2.pytorch.base.Pruner 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_origin2wrapped_parameter_name_map() (nni.algorithms.compression.v2.pytorch.base.Pruner method) get_sequence_id() (in module nni) get_status() (nni.experiment.Experiment method) get_trial_id() (in module nni) get_trial_job() (nni.experiment.Experiment method) 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) 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) GroupMaskConflict (class in nni.compression.pytorch.utils) GumbelDARTS (class in nni.retiarii.strategy) GumbelSoftmax (class in nni.retiarii.oneshot.pytorch.supermodule.differentiable) H hash (nni.nas.benchmarks.nasbench101.Nb101TrialConfig attribute) 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.experiment.Experiment attribute) (nni.retiarii.Graph attribute) (nni.retiarii.Node attribute) import_data() (nni.tuner.Tuner method) incoming_edges (nni.retiarii.Node attribute) infer_modules_masks() (nni.compression.pytorch.speedup.ModelSpeedup method) inference() (nni.compression.pytorch.quantization_speedup.ModelSpeedupTensorRT method) init_execution_engine() (in module nni.retiarii.execution) initialize_speedup() (nni.compression.pytorch.speedup.ModelSpeedup method) input_names (nni.retiarii.Graph attribute) input_node (nni.retiarii.Graph attribute) InputChoice (class in nni.retiarii.nn.pytorch) intermediate_metrics (nni.retiarii.Model attribute) InvalidMutation (class in nni.retiarii) InvertedResidual (class in nni.retiarii.hub.pytorch.proxylessnas) is_stopped_exec() (in module nni.retiarii.execution) is_traceable() (in module nni.common.serializer) is_wrapped_with_trace() (in module nni.common.serializer) iter_time (nni.nas.benchmarks.nds.NdsTrialStats attribute) L L1NormPruner (class in nni.compression.pytorch.pruning) L2NormPruner (class in nni.compression.pytorch.pruning) label (nni.retiarii.Node attribute) (nni.retiarii.oneshot.pytorch.supermodule.differentiable.DifferentiableMixedInput attribute) (nni.retiarii.oneshot.pytorch.supermodule.differentiable.DifferentiableMixedLayer attribute) (nni.retiarii.oneshot.pytorch.supermodule.sampling.PathSamplingLayer attribute) LAST() (nni.retiarii.nn.pytorch.ModelParameterChoice static method) 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.compression.pytorch.pruning) Lightning (class in nni.retiarii.evaluator.pytorch) LightningEvaluator (class in nni.compression.pytorch) LightningModule (class in nni.retiarii.evaluator.pytorch) LinearPruner (class in nni.compression.pytorch.pruning) list_models() (in module nni.retiarii.execution) list_trial_jobs() (nni.experiment.Experiment method) load() (in module nni.common.serializer) load_calibration_config() (nni.compression.pytorch.compressor.Quantizer method) load_checkpoint() (nni.assessor.Assessor method) (nni.tuner.Tuner method) load_masks() (nni.algorithms.compression.v2.pytorch.base.Pruner method) load_quantized_model() (nni.compression.pytorch.quantization_speedup.ModelSpeedupTensorRT method) load_state_dict() (nni.compression.pytorch.utils.SensitivityAnalysis method) LotteryTicketPruner (class in nni.compression.pytorch.pruning) LsqQuantizer (class in nni.algorithms.compression.pytorch.quantization) M max() (nni.retiarii.nn.pytorch.ValueChoice static method) MedianstopAssessor (class in nni.algorithms.hpo.medianstop_assessor) MetisTuner (class in nni.algorithms.hpo.metis_tuner) metric (nni.retiarii.Model attribute) MetricsCalculator (class in nni.algorithms.compression.v2.pytorch.pruning.tools) min() (nni.retiarii.nn.pytorch.ValueChoice static method) MixedBatchNorm2d (class in nni.retiarii.oneshot.pytorch.supermodule.operation) MixedConv2d (class in nni.retiarii.oneshot.pytorch.supermodule.operation) MixedLayerNorm (class in nni.retiarii.oneshot.pytorch.supermodule.operation) MixedLinear (class in nni.retiarii.oneshot.pytorch.supermodule.operation) MixedMultiHeadAttention (class in nni.retiarii.oneshot.pytorch.supermodule.operation) MixedOpDifferentiablePolicy (class in nni.retiarii.oneshot.pytorch.supermodule.differentiable) MixedOperation (class in nni.retiarii.oneshot.pytorch.supermodule.operation) MixedOperationSamplingPolicy (class in nni.retiarii.oneshot.pytorch.supermodule.operation) MixedOpPathSamplingPolicy (class in nni.retiarii.oneshot.pytorch.supermodule.sampling) MobileNetV3Space (class in nni.retiarii.hub.pytorch) Model (class in nni.retiarii) model (nni.retiarii.Graph attribute) (nni.retiarii.oneshot.pytorch.base_lightning.BaseOneShotLightningModule attribute) model_family (nni.nas.benchmarks.nds.NdsTrialConfig attribute) model_spec (nni.nas.benchmarks.nds.NdsTrialConfig attribute) model_wrapper() (in module nni.retiarii) ModelNamespace (class in nni.retiarii.utils) ModelParameterChoice (class in nni.retiarii.nn.pytorch) ModelSpeedup (class in nni.compression.pytorch.speedup) ModelSpeedupTensorRT (class in nni.compression.pytorch.quantization_speedup) module nni.common.framework nni.common.serializer nni.nas.benchmarks.nasbench101 nni.nas.benchmarks.nasbench201 nni.nas.benchmarks.nds nni.retiarii.execution nni.retiarii.nn.pytorch.mutation_utils nni.retiarii.oneshot.pytorch.base_lightning nni.retiarii.oneshot.pytorch.dataloader nni.retiarii.oneshot.pytorch.supermodule.differentiable nni.retiarii.oneshot.pytorch.supermodule.operation nni.retiarii.oneshot.pytorch.supermodule.proxyless nni.retiarii.oneshot.pytorch.supermodule.sampling nni.retiarii.utils nni.typehint MovementPruner (class in nni.compression.pytorch.pruning) MultiModelSupervisedLearningModule (class in nni.retiarii.evaluator.pytorch.cgo.evaluator) Mutable (class in nni.retiarii.nn.pytorch.mutation_utils) mutate() (nni.retiarii.Mutator method) (nni.retiarii.oneshot.pytorch.base_lightning.BaseSuperNetModule class method) (nni.retiarii.oneshot.pytorch.supermodule.operation.MixedOperation class method) (nni.retiarii.oneshot.pytorch.supermodule.sampling.PathSamplingCell class method) mutate_kwargs() (nni.retiarii.oneshot.pytorch.base_lightning.BaseOneShotLightningModule method) Mutator (class in nni.retiarii) N NaiveQuantizer (class in nni.algorithms.compression.pytorch.quantization) name (nni.retiarii.Graph attribute) (nni.retiarii.Node attribute) named_parameters() (nni.retiarii.oneshot.pytorch.supermodule.differentiable.DifferentiableMixedInput method) (nni.retiarii.oneshot.pytorch.supermodule.differentiable.DifferentiableMixedLayer method) names (nni.retiarii.nn.pytorch.LayerChoice attribute) nas_modules (nni.retiarii.oneshot.pytorch.base_lightning.BaseOneShotLightningModule attribute) NasBench101 (class in nni.retiarii.hub.pytorch) NasBench101Cell (class in nni.retiarii.nn.pytorch) NasBench201 (class in nni.retiarii.hub.pytorch) NasBench201Cell (class in nni.retiarii.nn.pytorch) NASNet (class in nni.retiarii.hub.pytorch) 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) NDS (class in nni.retiarii.hub.pytorch.nasnet) NdsIntermediateStats (class in nni.nas.benchmarks.nds) NDSStage (class in nni.retiarii.hub.pytorch.nasnet) NDSStageDifferentiable (class in nni.retiarii.hub.pytorch.nasnet) NDSStagePathSampling (class in nni.retiarii.hub.pytorch.nasnet) NdsTrialConfig (class in nni.nas.benchmarks.nds) NdsTrialStats (class in nni.nas.benchmarks.nds) next() (nni.algorithms.compression.v2.pytorch.pruning.tools.TaskGenerator method) next_label() (nni.retiarii.utils.ModelNamespace static method) nni.common.framework module nni.common.serializer module nni.nas.benchmarks.nasbench101 module nni.nas.benchmarks.nasbench201 module nni.nas.benchmarks.nds module nni.retiarii.execution module nni.retiarii.nn.pytorch.mutation_utils module nni.retiarii.oneshot.pytorch.base_lightning module nni.retiarii.oneshot.pytorch.dataloader module nni.retiarii.oneshot.pytorch.supermodule.differentiable module nni.retiarii.oneshot.pytorch.supermodule.operation module nni.retiarii.oneshot.pytorch.supermodule.proxyless module nni.retiarii.oneshot.pytorch.supermodule.sampling module nni.retiarii.utils module nni.typehint module no_default_hook() (in module nni.retiarii.oneshot.pytorch.base_lightning) NoContextError Node (class in nni.retiarii) nodes (nni.retiarii.Graph attribute) 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) op_candidates_factory (nni.retiarii.nn.pytorch.Cell attribute) op_names (nni.retiarii.oneshot.pytorch.supermodule.differentiable.DifferentiableMixedLayer attribute) Operation (class in nni.retiarii) operation (nni.retiarii.Node 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) original_state_dict_hooks() (in module nni.retiarii.utils) outgoing_edges (nni.retiarii.Node attribute) output_names (nni.retiarii.Graph attribute) output_node (nni.retiarii.Graph attribute) output_node_indices (nni.retiarii.nn.pytorch.Cell attribute) P Parameters (in module nni.typehint) parameters (nni.nas.benchmarks.nasbench101.Nb101TrialStats attribute) (nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute) (nni.nas.benchmarks.nds.NdsTrialStats attribute) (nni.retiarii.Operation attribute) parameters() (nni.retiarii.oneshot.pytorch.supermodule.differentiable.DifferentiableMixedInput method) (nni.retiarii.oneshot.pytorch.supermodule.differentiable.DifferentiableMixedLayer method) PathSamplingCell (class in nni.retiarii.oneshot.pytorch.supermodule.sampling) PathSamplingInput (class in nni.retiarii.oneshot.pytorch.supermodule.sampling) PathSamplingLayer (class in nni.retiarii.oneshot.pytorch.supermodule.sampling) PathSamplingRepeat (class in nni.retiarii.oneshot.pytorch.supermodule.sampling) PBTTuner (class in nni.algorithms.hpo.pbt_tuner) Placeholder (class in nni.retiarii.nn.pytorch) PNAS (class in nni.retiarii.hub.pytorch) PolicyBasedRL (class in nni.retiarii.strategy) port (nni.experiment.Experiment attribute) PPOTuner (class in nni.algorithms.hpo.ppo_tuner) predecessors (nni.retiarii.Node attribute) proposer (nni.nas.benchmarks.nds.NdsTrialConfig attribute) Proxyless (class in nni.retiarii.strategy) ProxylessMixedInput (class in nni.retiarii.oneshot.pytorch.supermodule.proxyless) ProxylessMixedLayer (class in nni.retiarii.oneshot.pytorch.supermodule.proxyless) ProxylessNAS (class in nni.retiarii.hub.pytorch) ProxylessTrainer (class in nni.retiarii.oneshot.pytorch) Pruner (class in nni.algorithms.compression.v2.pytorch.base) PrunerModuleWrapper (class in nni.algorithms.compression.v2.pytorch.base) pruning_one_step() (nni.algorithms.compression.v2.pytorch.base.BasePruningScheduler method) python_class (nni.retiarii.Model attribute) python_init_params (nni.retiarii.Model attribute) python_name (nni.retiarii.Graph attribute) (nni.retiarii.Node attribute) python_object (nni.retiarii.Model attribute) Q QAT_Quantizer (class in nni.algorithms.compression.pytorch.quantization) quant_backward() (nni.compression.pytorch.compressor.QuantGrad static method) QuantGrad (class in nni.compression.pytorch.compressor) quantize_input() (nni.compression.pytorch.compressor.Quantizer method) quantize_output() (nni.compression.pytorch.compressor.Quantizer method) quantize_weight() (nni.compression.pytorch.compressor.Quantizer method) Quantizer (class in nni.compression.pytorch.compressor) QuantizerModuleWrapper (class in nni.compression.pytorch.compressor) query_available_resources() (in module nni.retiarii.execution) 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) RandomOneShot (class in nni.retiarii.strategy) RandomTuner (class in nni.algorithms.hpo.random_tuner) receive_task_result() (nni.algorithms.compression.v2.pytorch.pruning.tools.TaskGenerator method) receive_trial_result() (nni.tuner.Tuner method) record_shape() (nni.compression.pytorch.compressor.Quantizer method) record_task_result() (nni.algorithms.compression.v2.pytorch.base.BasePruningScheduler method) reduction (nni.retiarii.nn.pytorch.ChosenInputs attribute) reduction() (nni.retiarii.oneshot.pytorch.supermodule.differentiable.DifferentiableMixedInput method) (nni.retiarii.oneshot.pytorch.supermodule.differentiable.DifferentiableMixedLayer method) (nni.retiarii.oneshot.pytorch.supermodule.differentiable.DifferentiableMixedRepeat method) (nni.retiarii.oneshot.pytorch.supermodule.sampling.PathSamplingInput method) (nni.retiarii.oneshot.pytorch.supermodule.sampling.PathSamplingLayer method) (nni.retiarii.oneshot.pytorch.supermodule.sampling.PathSamplingRepeat method) Regression (class in nni.retiarii.evaluator.pytorch) (class in nni.retiarii.evaluator.pytorch.cgo.evaluator) RegressionModule (class in nni.retiarii.evaluator.pytorch) RegularizedEvolution (class in nni.retiarii.strategy) ReinforceController (class in nni.retiarii.oneshot.pytorch.enas) Repeat (class in nni.retiarii.nn.pytorch) replace_compressed_modules() (nni.compression.pytorch.speedup.ModelSpeedup method) replace_input_choice() (in module nni.retiarii.oneshot.pytorch.utils) replace_layer_choice() (in module nni.retiarii.oneshot.pytorch.utils) replace_submodule() (nni.compression.pytorch.speedup.ModelSpeedup method) report_final_result() (in module nni) report_intermediate_result() (in module nni) resample() (nni.retiarii.oneshot.pytorch.base_lightning.BaseOneShotLightningModule method) (nni.retiarii.oneshot.pytorch.base_lightning.BaseSuperNetModule method) (nni.retiarii.oneshot.pytorch.supermodule.differentiable.DifferentiableMixedCell method) (nni.retiarii.oneshot.pytorch.supermodule.differentiable.DifferentiableMixedInput method) (nni.retiarii.oneshot.pytorch.supermodule.differentiable.DifferentiableMixedLayer method) (nni.retiarii.oneshot.pytorch.supermodule.differentiable.DifferentiableMixedRepeat method) (nni.retiarii.oneshot.pytorch.supermodule.differentiable.MixedOpDifferentiablePolicy method) (nni.retiarii.oneshot.pytorch.supermodule.operation.MixedOperation method) (nni.retiarii.oneshot.pytorch.supermodule.operation.MixedOperationSamplingPolicy method) (nni.retiarii.oneshot.pytorch.supermodule.proxyless.ProxylessMixedInput method) (nni.retiarii.oneshot.pytorch.supermodule.proxyless.ProxylessMixedLayer method) (nni.retiarii.oneshot.pytorch.supermodule.sampling.MixedOpPathSamplingPolicy method) (nni.retiarii.oneshot.pytorch.supermodule.sampling.PathSamplingCell method) (nni.retiarii.oneshot.pytorch.supermodule.sampling.PathSamplingInput method) (nni.retiarii.oneshot.pytorch.supermodule.sampling.PathSamplingLayer method) (nni.retiarii.oneshot.pytorch.supermodule.sampling.PathSamplingRepeat method) reset() (nni.algorithms.compression.v2.pytorch.pruning.tools.DataCollector method) reset_tools() (nni.algorithms.compression.v2.pytorch.pruning.basic_pruner.BasicPruner method) resume() (nni.experiment.Experiment static method) (nni.retiarii.experiment.pytorch.RetiariiExperiment static method) RetiariiExeConfig (class in nni.retiarii.experiment.pytorch) RetiariiExperiment (class in nni.retiarii.experiment.pytorch) root_graph (nni.retiarii.Model attribute) run() (nni.experiment.Experiment method) (nni.retiarii.experiment.pytorch.RetiariiExperiment method) running_mode (nni.retiarii.evaluator.pytorch.LightningModule attribute) S Sampler (class in nni.retiarii) save_checkpoint() (nni.assessor.Assessor method) (nni.tuner.Tuner method) search_space_spec() (nni.retiarii.oneshot.pytorch.base_lightning.BaseOneShotLightningModule method) (nni.retiarii.oneshot.pytorch.base_lightning.BaseSuperNetModule method) SearchSpace (in module nni.typehint) seed (nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute) (nni.nas.benchmarks.nds.NdsTrialStats attribute) SensitivityAnalysis (class in nni.compression.pytorch.utils) set_default_framework() (in module nni.common.framework) set_drop_path_prob() (nni.retiarii.hub.pytorch.nasnet.NDS method) set_execution_engine() (in module nni.retiarii.execution) set_model() (nni.retiarii.evaluator.pytorch.LightningModule method) shortcut_framework() (in module nni.common.framework) shortcut_module() (in module nni.common.framework) show_pruned_weights() (nni.algorithms.compression.v2.pytorch.base.Pruner method) ShuffleNetSpace (class in nni.retiarii.hub.pytorch) SimulatedAnnealingPruner (class in nni.compression.pytorch.pruning) SinglePathTrainer (class in nni.retiarii.oneshot.pytorch) slice_param() (nni.retiarii.oneshot.pytorch.supermodule.operation.MixedOperation method) SlimPruner (class in nni.compression.pytorch.pruning) SMACTuner (class in nni.algorithms.hpo.smac_tuner) SparsityAllocator (class in nni.algorithms.compression.v2.pytorch.pruning.tools) special_target_masks_generation() (nni.algorithms.compression.v2.pytorch.pruning.tools.SparsityAllocator method) specialize_cell() (nni.retiarii.Node method) speedup_model() (nni.compression.pytorch.speedup.ModelSpeedup method) start() (nni.experiment.Experiment method) (nni.retiarii.experiment.pytorch.RetiariiExperiment method) status (nni.retiarii.Model attribute) stop() (nni.experiment.Experiment method) (nni.retiarii.experiment.pytorch.RetiariiExperiment method) sub_state_dict() (nni.retiarii.strategy.RandomOneShot method) submit_models() (in module nni.retiarii.execution) successors (nni.retiarii.Node attribute) super_init_argument() (nni.retiarii.oneshot.pytorch.supermodule.operation.MixedOperation method) T tail (nni.retiarii.Edge attribute) tail_slot (nni.retiarii.Edge attribute) TaskGenerator (class in nni.algorithms.compression.v2.pytorch.pruning.tools) TaylorFOWeightPruner (class in nni.compression.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) to_float() (nni.retiarii.nn.pytorch.ValueChoice static method) to_int() (nni.retiarii.nn.pytorch.ValueChoice static method) TorchEvaluator (class in nni.compression.pytorch) TPE (class in nni.retiarii.strategy) TpeArguments (class in nni.algorithms.hpo.tpe_tuner) TpeTuner (class in nni.algorithms.hpo.tpe_tuner) trace() (in module nni.common.serializer) trace_args (nni.common.serializer.Traceable property) trace_copy() (nni.common.serializer.Traceable method) trace_kwargs (nni.common.serializer.Traceable property) trace_symbol (nni.common.serializer.Traceable property) Traceable (class in nni.common.serializer) 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) Trainer (class in nni.retiarii.evaluator.pytorch) training_time (nni.nas.benchmarks.nasbench101.Nb101IntermediateStats attribute) (nni.nas.benchmarks.nasbench101.Nb101TrialStats attribute) (nni.nas.benchmarks.nasbench201.Nb201TrialStats attribute) TransformersEvaluator (class in nni.compression.pytorch) Translatable (class in nni.common.serializer) traverse_and_mutate_submodules() (in module nni.retiarii.oneshot.pytorch.base_lightning) trial (nni.nas.benchmarks.nasbench101.Nb101IntermediateStats attribute) (nni.nas.benchmarks.nasbench201.Nb201IntermediateStats attribute) (nni.nas.benchmarks.nds.NdsIntermediateStats attribute) trial_end() (nni.assessor.Assessor method) (nni.tuner.Tuner method) TrialMetric (in module nni.typehint) Tuner (class in nni.tuner) type (nni.retiarii.Operation attribute) U update_already_pruned() (nni.compression.pytorch.utils.SensitivityAnalysis method) update_direct_sparsity() (nni.compression.pytorch.speedup.ModelSpeedup method) update_indirect_sparsity() (nni.compression.pytorch.speedup.ModelSpeedup method) update_max_experiment_duration() (nni.experiment.Experiment method) update_max_trial_number() (nni.experiment.Experiment method) update_search_space() (nni.experiment.Experiment method) (nni.tuner.Tuner method) update_trial_concurrency() (nni.experiment.Experiment 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) ValueChoice (class in nni.retiarii.nn.pytorch) view() (nni.experiment.Experiment static method) (nni.retiarii.experiment.pytorch.RetiariiExperiment static method) W wait_models() (in module nni.retiarii.execution) weight_decay (nni.nas.benchmarks.nds.NdsTrialConfig attribute)