我們都知道tensorflow最后生成的模型文件含：

?

checkpoint

xxxxx.meta

xxxxx.ckpt.data-xxx

xxxxx.index

?

學習和使用tensorflow的小伙伴肯定都會進行這個過程，我們來看一下怎么操作

上代碼：

import tensorflow as tfmodel_name = 'xx/xxx/xxx'#啟動一個會話，意味著開始訓練 with tf.Session() as sess:init = tf.global_variables_initializer()sess.run(init)#聲明一個saversaver = tf.train.Saver()for epoch in range(max_epochs):training() #訓練部分 #保存模型if (epoch%10) == 0 or (epoch + 1) == max_epochs:saver.save(sess,model_name)

值得注意的點是，saver的聲明要在會話里，不然會報錯類似于：使用了未初始化(uninitialized)的變量。

這是最基本的用法，我們不用去考慮saver.save函數的參數問題。我們下一篇文章介紹進階用法。

總結

以上是生活随笔為你收集整理的Tensorflow详解保存模型（基础版）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。