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