neural_compressor.experimental.nas.nas

Common classes for different NAS approaches.

NAS class for creating object of different NAS approaches. NASBase class defines the common methods of different NAS approaches.

Module Contents

Classes

NAS

Create object of different NAS approaches.

NASBase

Base class for defining the common methods of different NAS approaches.
class neural_compressor.experimental.nas.nas.NAS[source]

Create object of different NAS approaches.

Parameters:

conf_fname_or_obj (string or obj) – The path to the YAML configuration file or the object of NASConfig.

Returns:

An object of specified NAS approach.

class neural_compressor.experimental.nas.nas.NASBase(search_space=None, model_builder=None)[source]

Base class for defining the common methods of different NAS approaches.

Parameters:

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

  • model_builder (function obj) – A function to build model instance with the specified model architecture parameters.