ApiReference.rst

Retiarii API Reference

Inline Mutation APIs

nni.retiarii.nn.pytorch.LayerChoice

nni.retiarii.nn.pytorch.InputChoice

nni.retiarii.nn.pytorch.ValueChoice

nni.retiarii.nn.pytorch.ChosenInputs

nni.retiarii.nn.pytorch.Repeat

nni.retiarii.nn.pytorch.Cell

Graph Mutation APIs

nni.retiarii.Mutator

nni.retiarii.Model

nni.retiarii.Graph

nni.retiarii.Node

nni.retiarii.Edge

nni.retiarii.Operation

Evaluators

nni.retiarii.evaluator.FunctionalEvaluator

nni.retiarii.evaluator.pytorch.lightning.LightningModule

nni.retiarii.evaluator.pytorch.lightning.Classification

nni.retiarii.evaluator.pytorch.lightning.Regression

Oneshot Trainers

nni.retiarii.oneshot.pytorch.DartsTrainer

nni.retiarii.oneshot.pytorch.EnasTrainer

nni.retiarii.oneshot.pytorch.ProxylessTrainer

nni.retiarii.oneshot.pytorch.SinglePathTrainer

Exploration Strategies

nni.retiarii.strategy.Random

nni.retiarii.strategy.GridSearch

nni.retiarii.strategy.RegularizedEvolution

nni.retiarii.strategy.TPEStrategy

nni.retiarii.strategy.PolicyBasedRL

Retiarii Experiments

nni.retiarii.experiment.pytorch.RetiariiExperiment

nni.retiarii.experiment.pytorch.RetiariiExeConfig

Utilities

nni.retiarii.serialize

nni.retiarii.fixed_arch