neural_compressor.strategy.exhaustive

The exhaustive tuning strategy.

Module Contents

Classes

ExhaustiveTuneStrategy

The exhaustive tuning strategy.

class neural_compressor.strategy.exhaustive.ExhaustiveTuneStrategy(model, conf, q_dataloader=None, q_func=None, eval_dataloader=None, eval_func=None, resume=None, q_hooks=None)

Bases: neural_compressor.strategy.strategy.TuneStrategy

The exhaustive tuning strategy.

next_tune_cfg()

Generate and yield the next tuning config using exhaustive search in tuning space.

It sequentially traverse all possible quantization tuning configurations in a tuning space.

Yields:

tune_config (dict) – A dict containing the tuning configuration for quantization.