训练,保存和加载Tensorflow模型

时间:2018-10-31 10:11:37

标签: python-3.x tensorflow

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

2 个答案:

答案 0 :(得分:0)

您是否参考过this指南?它非常清楚地说明了如何以所有可能的格式加载和保存张量流模型。

答案 1 :(得分:0)

如果您不熟悉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')

加载模型并进行预测