neural_compressor.experimental.nas.search_algorithms

Search algorithms for NAS.

Module Contents

Classes

Searcher	Base class for defining the common methods of different search algorithms.
GridSearcher	Grid search.
RandomSearcher	Random search.
BayesianOptimizationSearcher	Bayesian Optimization.

class neural_compressor.experimental.nas.search_algorithms.Searcher(search_space)

Base class for defining the common methods of different search algorithms.

Parameters:

search_space (dict) – A dictionary for defining the search space.

class neural_compressor.experimental.nas.search_algorithms.GridSearcher(search_space)

Grid search.

Search the whole search space exhaustively.

Parameters:

search_space (dict) – A dictionary for defining the search space.

class neural_compressor.experimental.nas.search_algorithms.RandomSearcher(search_space, seed=42)

Random search.

Search the whole search space randomly.

Parameters:

search_space (dict) – A dictionary for defining the search space.

class neural_compressor.experimental.nas.search_algorithms.BayesianOptimizationSearcher(search_space, seed=42)

Bayesian Optimization.

Search the search space with Bayesian Optimization.

Parameters:

search_space (dict) – A dictionary for defining the search space.