TensorFlow利用saver保存和提取參數(shù)的實例
更新時間：2018年07月26日 09:44:21 作者：winycg
今天小編就為大家分享一篇TensorFlow利用saver保存和提取參數(shù)的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
在訓(xùn)練循環(huán)中，定期調(diào)用 saver.save() 方法，向文件夾中寫入包含了當(dāng)前模型中所有可訓(xùn)練變量的 checkpoint 文件。
saver.save(sess, FLAGS.train_dir, global_step=step)
global_step是訓(xùn)練的第幾步
保存參數(shù)：
import tensorflow as tf W = tf.Variable([[1, 2, 3]], dtype=tf.float32) b = tf.Variable([[1]], dtype=tf.float32) saver = tf.train.Saver() sess = tf.InteractiveSession() tf.global_variables_initializer().run() # 必須要指定文件夾，保存到ckpt文件 save_path = saver.save(sess, "winycg/1.ckpt") print(save_path)
一次 saver.save() 后可以在文件夾中看到新增的四個文件，實際上每調(diào)用一次保存操作會創(chuàng)建后3個數(shù)據(jù)文件并創(chuàng)建一個檢查點（checkpoint）文件，簡單理解就是權(quán)重等參數(shù)被保存到 .chkp.data 文件中，以字典的形式；圖和元數(shù)據(jù)被保存到 .chkp.meta 文件中，可以被 tf.train.import_meta_graph 加載到當(dāng)前默認(rèn)的圖。
讀取參數(shù)：
import tensorflow as tf
import numpy as np
W = tf.Variable(np.arange(3).reshape(1, 3), dtype=tf.float32)
b = tf.Variable(np.arange(1).reshape(1, 1), dtype=tf.float32)
saver = tf.train.Saver()
sess = tf.InteractiveSession()
# 讀取參數(shù)時不需要global_variables_initializer()
save_path = saver.restore(sess, "parameter/1.ckpt")
print("weights:", sess.run(W))
print("bias:", sess.run(b))
weights: [[ 1. 2. 3.]]
bias: [[ 1.]]
以上這篇TensorFlow利用saver保存和提取參數(shù)的實例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
Tornado協(xié)程在python2.7如何返回值(實現(xiàn)方法)
下面小編就為大家?guī)硪黄猅ornado協(xié)程在python2.7如何返回值(實現(xiàn)方法)。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2017-06-06
工作中通常會遇到大量的數(shù)據(jù)集需要處理,其中的一項就是將含有某些數(shù)據(jù)的行刪除掉,下面這篇文章主要給大家介紹了關(guān)于如何利用Pandas刪除某列指定值所在的行的相關(guān)資料,需要的朋友可以參考下2022-04-04
這篇文章主要介紹了PIP和conda 更換國內(nèi)安裝源的方法步驟，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-09-09
很多人不理解“元編程”是個什么東西,關(guān)于它也沒有一個十分準(zhǔn)確的定義.這篇文章要說的是Python里的元編程,實際上也不一定就真的符合“元編程”的定義.只不過我無法找到一個更準(zhǔn)確的名字來代表這篇文章的主題,所以就借了這么一個名號,需要的朋友可以參考下2021-05-05
Python-OpenCV中的cv2.inpaint()函數(shù)的使用
大多數(shù)人會在家里放一些舊的退化照片，上面有一些黑點，一些筆畫等。你有沒有想過恢復(fù)它？本文就來介紹一下方法，感興趣的可以了解一下2021-06-06
Python一行代碼實現(xiàn)ChatGPT接入微信機器人
這篇文章主要為大家介紹了Python一行代碼實現(xiàn)ChatGPT接入微信機器人示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2023-03-03
這篇文章主要介紹了django 實現(xiàn)電子支付功能的示例代碼,小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-07-07
Python?eval()和exec()函數(shù)使用詳解
exec函數(shù)執(zhí)行的是python語句，沒有返回值，eval函數(shù)執(zhí)行的是python表達(dá)式，有返回值，exec函數(shù)和eval函數(shù)都可以傳入命名空間作為參數(shù)，本文給大家介紹下Python?eval()和exec()函數(shù)，感興趣的朋友跟隨小編一起看看吧2022-11-11
本文主要介紹了python多線程中獲取函數(shù)返回值的三種方法，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-03-03
Python實戰(zhàn)之實現(xiàn)百度智能圖片識別
這篇文章主要介紹了如何利用Python編寫一個百度智能圖片識別項目，文中的示例代碼講解詳細(xì)，對我們學(xué)習(xí)Python有一定幫助，需要的可以參考一下2022-01-01