neural_compressor.strategy.strategy

The base class for tuning strategy.

Module Contents

Classes

TuneStrategyMeta

Tuning strategy metaclass.

TuneStrategy

Basic class for tuning strategy.

Functions

strategy_registry(cls)

Class decorator used to register all TuneStrategy subclasses.
neural_compressor.strategy.strategy.strategy_registry(cls)[source]

Class decorator used to register all TuneStrategy subclasses.

Parameters:

cls (class) – The class of register.

Returns:

The class of register.

Return type:

cls

class neural_compressor.strategy.strategy.TuneStrategyMeta[source]

Tuning strategy metaclass.

class neural_compressor.strategy.strategy.TuneStrategy(model, conf, q_dataloader=None, q_func=None, eval_func=None, eval_dataloader=None, eval_metric=None, resume=None, q_hooks=None)[source]

Basic class for tuning strategy.