【问题标题】:What is the difference between TFHub and Model Garden?TFHub 和模型花园有什么区别?
【发布时间】:2021-08-16 15:53:12
【问题描述】:

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

为什么Tensorflow 为模型存储库提出两个概念?

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

【问题讨论】:

  • 我们主要期望 TF Hub 的用户将模型视为具有明确定义的输入和输出的黑盒。如果模型满足您的需求(例如准确性、速度、大小、预训练类别),那么就使用它!如果您想深入了解模型结构的内部细节,那么您可能对代码优先资源更感兴趣,例如 Model Garden 或 Keras 应用程序。这些通常带有预训练的检查点,可以使用提供的代码加载。

标签: tensorflow pre-trained-model tensorflow-hub


【解决方案1】:

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

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

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 2010-12-05
    • 1970-01-01
    • 2023-03-22
    • 1970-01-01
    • 2012-09-15
    • 1970-01-01
    • 2012-02-03
    • 2021-04-16
    相关资源
    最近更新 更多