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