#7 · 主分类: 模型动物园与中心

hub

embeddings image-classification machine-learning ml python tensorflow transfer-learning

通过重用TensorFlow模型部分进行迁移学习的库。

项目最后更新:01/17/25

GitHub Stars

3.5K

Forks数量

1.6K

贡献者数量

108

许可证

Apache-2.0

收录理由

如今多数团队不再从零训练图像或文本编码器,TensorFlow Hub 正是让这种复用变得顺手的那块拼图。你只要给出一个已发布模型的句柄,它就能把权重加载成可直接使用的 Keras 层,想在现有编码器上加个微调头,几行代码就能搞定,根本不用跑一次完整的训练流程。这省去了自己托管或维护检查点的麻烦。需要留意的是,从 2023 年底起,模型托管迁到了 Kaggle Models,新句柄都指向那里;而旧 tfhub.dev 链接通过加载器仍然能用。如果你正用预训练组件拼装 TensorFlow 流水线,从这里起步是最顺理成章的。

相关文章

暂无针对该项目的文章。

如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。

同类热门项目