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 提供了用于从头开始训练您自己的模型的代码以及一些检查点。