neural_compressor.strategy.auto_mixed_precision

The auto-mixed precision strategy.

Classes

AutoMixedPrecisionTuneStrategy

Tuning strategy for auto mixed precision.

Module Contents

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.