tensorflow/model-optimization

A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.

What it solves

它提供了一套工具来优化机器学习模型，使其在部署和执行时更加高效。

How it works

该工具包使用量化和稀疏权重的剪枝等技术来减小模型大小并提高性能。它包含稳定的 Python API 以及针对 Keras 的特定集成。

Highlights

  • 量化支持
  • 稀疏权重的剪枝
  • 针对 Keras 的专用 API
  • 稳定的 Python API