首页 > 解决方案 > Tensorflow Hub 与 Tensorflow SavedModel 对比?

问题描述

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

SavedModel还是Tensorflow Hub

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

此外,哪一个会在 Tensorflow 2.0 中获得特权?

谢谢,

标签: pythontensorflowmachine-learningdeep-learning

解决方案


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

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

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

来源:您提供的链接。


推荐阅读