All modules for which code is available
- nni.algorithms.compression.pytorch.pruning.admm_pruner
- nni.algorithms.compression.pytorch.pruning.agp
- nni.algorithms.compression.pytorch.pruning.amc.amc_pruner
- nni.algorithms.compression.pytorch.pruning.auto_compress_pruner
- nni.algorithms.compression.pytorch.pruning.lottery_ticket
- nni.algorithms.compression.pytorch.pruning.net_adapt_pruner
- nni.algorithms.compression.pytorch.pruning.one_shot
- nni.algorithms.compression.pytorch.pruning.sensitivity_pruner
- nni.algorithms.compression.pytorch.pruning.simulated_annealing_pruner
- nni.algorithms.hpo.batch_tuner
- nni.algorithms.hpo.curvefitting_assessor.curvefitting_assessor
- nni.algorithms.hpo.evolution_tuner
- nni.algorithms.hpo.gp_tuner.gp_tuner
- nni.algorithms.hpo.gridsearch_tuner
- nni.algorithms.hpo.hyperband_advisor
- nni.algorithms.hpo.hyperopt_tuner
- nni.algorithms.hpo.medianstop_assessor
- nni.algorithms.hpo.metis_tuner.metis_tuner
- nni.algorithms.hpo.networkmorphism_tuner.networkmorphism_tuner
- nni.algorithms.nas.pytorch.cdarts.mutator
- nni.algorithms.nas.pytorch.cdarts.trainer
- nni.algorithms.nas.pytorch.classic_nas.mutator
- nni.algorithms.nas.pytorch.darts.mutator
- nni.algorithms.nas.pytorch.darts.trainer
- nni.algorithms.nas.pytorch.enas.mutator
- nni.algorithms.nas.pytorch.enas.trainer
- nni.algorithms.nas.pytorch.random.mutator
- nni.algorithms.nas.pytorch.spos.evolution
- nni.algorithms.nas.pytorch.spos.mutator
- nni.algorithms.nas.pytorch.spos.trainer
- nni.assessor
- nni.compression.pytorch.utils.counter
- nni.compression.pytorch.utils.mask_conflict
- nni.compression.pytorch.utils.sensitivity_analysis
- nni.compression.pytorch.utils.shape_dependency
- nni.experiment.experiment
- nni.nas.benchmarks.nasbench101.graph_util
- nni.nas.benchmarks.nasbench101.model
- nni.nas.benchmarks.nasbench101.query
- nni.nas.benchmarks.nasbench201.model
- nni.nas.benchmarks.nasbench201.query
- nni.nas.benchmarks.nds.model
- nni.nas.benchmarks.nds.query
- nni.nas.pytorch.base_mutator
- nni.nas.pytorch.base_trainer
- nni.nas.pytorch.callbacks
- nni.nas.pytorch.fixed
- nni.nas.pytorch.mutables
- nni.nas.pytorch.mutator
- nni.nas.pytorch.nasbench201.nasbench201
- nni.nas.pytorch.nasbench201.nasbench201_ops
- nni.nas.pytorch.search_space_zoo.darts_cell
- nni.nas.pytorch.search_space_zoo.darts_ops
- nni.nas.pytorch.search_space_zoo.enas_cell
- nni.nas.pytorch.search_space_zoo.enas_ops
- nni.nas.pytorch.trainer
- nni.nas.pytorch.utils
- nni.retiarii.experiment
- nni.retiarii.graph
- nni.retiarii.mutator
- nni.retiarii.nn.pytorch.nn
- nni.retiarii.operation
- nni.retiarii.strategies.random_strategy
- nni.retiarii.strategies.tpe_strategy
- nni.retiarii.trainer.pytorch.base
- nni.retiarii.trainer.pytorch.darts
- nni.retiarii.trainer.pytorch.enas
- nni.retiarii.trainer.pytorch.proxyless
- nni.retiarii.trainer.pytorch.random
- nni.runtime.msg_dispatcher_base
- nni.trial
- nni.tuner
- nni.utils
- peewee
- playhouse.sqlite_ext