Tensorflow Hub 与 Tensorflow SavedModel 的区别?

我想知道目前在 Tensorflow 中保存模型的推荐方法是什么?

SavedModel 还是 Tensorflow Hub

前者在“保存和恢复模型”部分有描述,而后者似乎被描述为“可重用的机器学习模块”的方法。

此外,哪一个在 Tensorflow 2.0 中更为优先?

谢谢,


回答:

它们是不同的东西。TF Hub 为您提供预训练的模块,您可以重用并为新的用例重新训练,可以根据需要进行修改,而不必从头开始训练。可以将其视为使用 Inception 进行迁移学习的例子。

SavedModel 是您保存 TF 模型的方式,无论是使用 TF Hub 模块修改的新模型,还是您从头创建的东西。

希望这有助于消除困惑

Related Posts

梯度下降法会卡在非极小点吗?我们如何证明其正确性?

对于卡住的例子,假设我们的成本函数为 J(x,y) =…

TypeError: 不支持的操作数类型:’*’运算符不能用于’builtin_function_or_method’和’int’

我有一段手动实现感知机的代码。不幸的是,我遇到了一个错…

ValueError: 惩罚项必须为正

当我使用逻辑回归拟合模型时,显示了一个类似于 Valu…

在Keras中的有状态LSTM中,一个批次内的样本状态是否会在样本之间保留?

假设我想对时间序列进行分类,每个序列有33个时间步。我…

如何在有限数据集下使用Keras创建区分自己和他人的面部分类器?

在过去的两个月里,我一直尝试使用Keras创建一个能够…

R h2o: 如何为GBM实现自定义的stopping_metric

我发现从h2o v3.22.1.1版本开始,可以使用s…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注