neural_compressor.experimental.nas.basic_nas

Basic NAS approach class.

Module Contents

Classes

BasicNAS

Basic NAS approach.
class neural_compressor.experimental.nas.basic_nas.BasicNAS(conf_fname_or_obj, search_space=None, model_builder=None)[source]

Basic NAS approach.

Defining the pipeline for basic NAS approach.

Parameters:

  • conf_fname (string) – The path to the YAML configuration file.

  • 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.