Этот раздел содержит дополнительные коллекции справочных страниц API для проектов и пакетов, отдельных от пакета tensorflow , но не имеющих выделенных страниц дочернего сайта.
Репозиторий моделей TensorFlow
Репозиторий моделей TensorFlow предоставляет реализации современных моделей (SOTA).
Каталог Official /projects содержит коллекцию моделей SOTA, использующих высокоуровневый API TensorFlow. Они должны хорошо поддерживаться, тестироваться и обновляться с помощью новейшего API TensorFlow.
Код библиотеки, используемый для построения и обучения этих моделей, доступен в виде пакета pip. Вы можете установить его с помощью:
$ pip install tensorflow-models-official # For the latest release
$ #or
$ pip install tf-models-nightly # For the nightly build
Чтобы установить пакет из исходников, обратитесь к этим инструкциям .
Пакет tensorflow-models-official содержит два модуля верхнего уровня: tensorflow_models и orbit . Вы можете импортировать их с помощью:
import tensorflow_models as tfm
import orbit
Модели тензорного потока
Модуль tensorflow_models занимается построением моделей и настройкой обучения. Функциональность для конкретного приложения доступна в tfm.vision и tfm.nlp .
Орбита
Модуль orbit определяет гибкую и легкую библиотеку для написания настраиваемого кода цикла обучения в TensorFlow. Orbit гибко подходит к типам моделей, с которыми она работает. Вы можете использовать Orbit для обучения моделей Keras (в качестве альтернативы Model.fit ), но вам вообще не нужно использовать Keras. Orbit легко интегрируется с tf.distribute и поддерживает работу на различных типах устройств (CPU, GPU и TPU).