微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

TFHub和模型花园有什么区别?

如何解决TFHub和模型花园有什么区别?

TensorFlow Hub 是预训练模型的存储库。 Model Garden 还保留了 SOTA 模型,并提供了下载和利用其模型(如 TfHub)的工具,它们都是由 TensorFlow 创建的。

为什么 Tensorflow 为模型存储库制定了两个概念?

什么时候应该使用 TfHub 检索知名模型,什么时候应该使用 Model Garden 下载模型?它们之间有什么区别?

解决方法

TF Hub 以 SavedModel、TFLite 或 TF.js 格式提供训练模型。这些工件可用于推理,有些可用于代码中进行微调。 TF Hub 不提供建模库代码来从头开始训练您自己的模型。

Model Garden 是一个建模库,用于训练 BERT、图像分类模型等。 Model Garden 提供了用于从头开始训练您自己的模型的代码以及一些检查点。

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。