`neural_compressor.strategy.auto_mixed_precision`

The auto-mixed precision strategy.

Module Contents

Classes

AutoMixedPrecisionTuneStrategy

Tuning strategy for auto mixed precision.

class neural_compressor.strategy.auto_mixed_precision.AutoMixedPrecisionTuneStrategy(model, conf, q_dataloader=None, q_func=None, eval_func=None, eval_dataloader=None, eval_metric=None, resume=None, q_hooks=None)[source]: Tuning strategy for auto mixed precision.