TFHub 和 Model Garden 有什么区别?

Kav*_*veh 5 tensorflow pre-trained-model tensorflow-hub

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

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

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

小智 4

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

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