训练、保存和加载 TensorFlow 模型

【问题标题】：Train, Save and Load a Tensorflow Model训练、保存和加载 TensorFlow 模型
【发布时间】：2018-10-31 19:10:25
【问题描述】：

请参阅this 为 MNIST 数据集训练 GAN 模型，我想保存模型并恢复它以进行进一步预测。在对Saving and Importing a Tensorflow Model 有所了解之后，我能够保存和恢复输入和输出的一些变量，但是对于this 网络，我只能在一些特定的迭代之后保存模型并且无法预测一些输出。

【问题讨论】：

你有什么问题？
保存上述模型后，我以后如何预测结果，因为该模型不使用任何变量进行预测，而是使用函数。

标签： python-3.x tensorflow

【解决方案1】：

您是否参考了this 指南？它非常清楚地解释了如何以所有可能的格式加载和保存 tensorflow 模型。

【讨论】：

不，我没有参考本指南
对你有帮助吗？

【解决方案2】：

如果您是 ML 新手，我建议您先尝试一下 Keras，它更易于使用。见https://keras.io/getting-started/faq/#how-can-i-save-a-keras-model，几乎可以用：

model.save('my_model.h5')

将模型保存到磁盘。

model = load_model('my_model.h5')

加载您的模型并进行预测

【讨论】：

我习惯了 keras 这么简单，但我是 TF 新手，所以我问了这个问题