TF-Keras (旧版)

Hub

tf-keras 是 Keras 2.x 版本的名称。它现在托管在一个单独的 GitHub 仓库这里。虽然它是一个旧版框架，但在 Hub 上仍然托管着 4.5k+ 个模型。这些模型可以使用 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))]}!')

# The image is a sunflower!

你也可以在 Hub 上托管你的 tf-keras 模型。但是，请记住 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 年 4 月)

< > 在 GitHub 上更新