我想知道目前在 Tensorflow 中保存模型的推荐方法是什么?
前者在“保存和恢复模型”部分有描述,而后者似乎被描述为“可重用的机器学习模块”的方法。
此外,哪一个在 Tensorflow 2.0 中更为优先?
谢谢,
回答:
它们是不同的东西。TF Hub 为您提供预训练的模块,您可以重用并为新的用例重新训练,可以根据需要进行修改,而不必从头开始训练。可以将其视为使用 Inception 进行迁移学习的例子。
SavedModel 是您保存 TF 模型的方式,无论是使用 TF Hub 模块修改的新模型,还是您从头创建的东西。
希望这有助于消除困惑