Tensorflow Hub与Tensorflow SavedModel?

时间:2018-09-22 09:56:33

标签: python tensorflow machine-learning deep-learning

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

SavedModelTensorflow Hub

前者在“保存和恢复模型”部分中进行了描述,但后者似乎被描述为“可重用的机器学习模块”的使用方式。

此外,在Tensorflow 2.0中谁将获得特权?

谢谢

2 个答案:

答案 0 :(得分:0)

目前建议在Tensorflow中保存模型的方法是什么?

保存或恢复模型与tensorflow hub是两个完全不同的主题:

  • TensorFlow Hub是一个用于发布,发现和使用机器学习模型的可重用部分的库。
  • 使用SavedModel保存和加载模型(变量,图形和图形的元数据)。

来源:您提供的链接。

答案 1 :(得分:0)

这些是不同的东西。 TF Hub为您提供了预训练的模块,您可以为新的用例重用和重新训练,可以根据需要修改它们,而不必从头开始进行重新训练。例如,考虑使用Inception进行转移学习。

SavedModel是保存TF模型的方法,无论它是使用TF Hub模块的新修改模型,还是您从头开始创建的模型。

希望能消除困惑