TF-Keras (устаревший)

Hub

TF-Keras (устаревший)

tf-keras — это название версии Keras 2.x. Теперь она размещена в отдельном репозитории GitHub здесь. Хотя это устаревший фреймворк, на Hub по-прежнему размещено более 4,5 тыс. моделей. Эти модели можно загрузить с помощью библиотеки huggingface_hub. На вашем компьютере должен быть установлен либо tf-keras, либо keras<3.x.

Если вас интересует поддержка Keras 3.x, ознакомьтесь с этим руководством.

После установки достаточно использовать метод from_pretrained_keras, чтобы загрузить модель из Hub. Подробнее о from_pretrained_keras — здесь.

from huggingface_hub import from_pretrained_keras

model = from_pretrained_keras("keras-io/mobile-vit-xxs")
prediction = model.predict(image)
prediction = tf.squeeze(tf.round(prediction))
print(f'The image is a {classes[(np.argmax(prediction))]}!')

<CopyLLMTxtMenu containerStyle="float: inline-end; margin-inline-start: 10px; display: inline-flex; position: relative; z-index: 10;"></CopyLLMTxtMenu>

# The image is a sunflower!

Вы также можете разместить вашу модель tf-keras на Hub. Однако имейте в виду, что tf-keras — это устаревший фреймворк. Чтобы охватить максимальное число пользователей, мы рекомендуем создавать модель с помощью Keras 3.x и делиться ею в нативном формате, как описано выше. Подробнее о загрузке моделей tf-keras см. в документации push_to_hub_keras.

from huggingface_hub import push_to_hub_keras

push_to_hub_keras(model,
    "your-username/your-model-name",
    "your-tensorboard-log-directory",
    tags = ["object-detection", "some_other_tag"],
    **model_save_kwargs,
)

Дополнительные ресурсы

Репозиторий на GitHub

Блог-пост «Размещение Keras на 🤗 Hub для совместного обучения и воспроизводимости» (апрель 2022)

Update on GitHub