Установите оптимизацию модели TensorFlow

Перед установкой рекомендуется создать виртуальную среду Python. Дополнительные сведения см. в руководстве по установке TensorFlow.

Стабильные сборки

Чтобы установить последнюю версию, выполните следующее:

# Installing with the `--upgrade` flag ensures you'll get the latest version.
pip install --user --upgrade tensorflow-model-optimization

Подробности о выпуске см. в наших примечаниях к выпуску .

Необходимую версию TensorFlow и другую информацию о совместимости см. в разделе «Матрица совместимости API» на странице «Обзор» для метода, который вы собираетесь использовать. Например, для обрезки страница Обзор находится здесь .

Поскольку TensorFlow не включен как зависимость пакета оптимизации модели TensorFlow (в setup.py ), вы должны явно установить пакет TensorFlow ( tf-nightly или tf-nightly-gpu ). Это позволяет нам поддерживать один пакет вместо отдельных пакетов для TensorFlow с поддержкой CPU и GPU.

Установка из исходников

Вы также можете установить из исходников. Для этого требуется система сборки Bazel .

# To install dependencies on Ubuntu:
# sudo apt-get install bazel git python-pip
# For other platforms, see Bazel docs above.
git clone https://github.com/tensorflow/model-optimization.git
cd model-optimization
bazel build --copt=-O3 --copt=-march=native :pip_pkg
PKGDIR=$(mktemp -d)
./bazel-bin/pip_pkg $PKGDIR
pip install --user --upgrade $PKGDIR/*.whl