訓練，保存和加載Tensorflow模型
[英]Train, Save and Load a Tensorflow Model
問題描述
參照此內容訓練MNIST數據集的GAN模型，我想保存一個模型並還原它以進行進一步的預測。 在了解了保存和導入Tensorflow模型后，我能夠保存和恢復一些輸入和輸出變量，但是對於這個網絡，我只能在經過某些特定的迭代之后保存模型，並且無法預測某些輸出。
2 個解決方案
解決方案1
0 2018-10-31 10:59:27
您是否參考了本指南？ 它非常清楚地說明了如何以所有可能的格式加載和保存張量流模型。
解決方案2
0 2018-10-31 11:10:25
如果您不熟悉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')
加載模型並進行預測
保存並加載自定義 Tensorflow Model（自回歸 seq2seq 多元時間序列 GRU/RNN）
[英]Save and Load Custom Tensorflow Model (Autoregressive seq2seq multivariate time series GRU/RNN)
聲明:本站的技術帖子網頁，遵循CC BY-SA 4.0協議，如果您需要轉載，請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.