neural_compressor.strategy.mse_v2

The MSE_V2 tuning strategy.

Classes

MSE_V2TuneStrategy

The mse_v2 tuning strategy.

Module Contents

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

The mse_v2 tuning strategy.

MSE_v2 is a strategy with a two stages fallback and revert fallback. Note that, only tensorflow framework and pytorch FX backend is currently supported for mse_v2 tuning strategy.