如何使用张量流模型检查点以.h5格式保存张量流模型？

Question

我正在研究tensorflow项目。我已经使用python flask成功地训练，测试并做出了预测。但是为了每次进行预测，我都必须再次使用检查点加载完整模型。如果保存模型.h5格式，则无需加载数据集即可预测数据集。我不知道如何使用检查点以.h5格式保存tensorflow模型。如果有人知道该怎么做，请帮助我或向我转发任何链接。

谢谢。

Answer 1

您可以使用tf.train.Saver类保存和恢复张量流模型。 尽管这不会以.h5格式存储模型。

您可以参考以下部分以进一步了解：
Save
Restore

希望这会有所帮助！