TensorFlow Saver:保存和讀取模型參數(shù).ckpt實(shí)例
在使用TensorFlow的過(guò)程中，保存模型參數(shù)變量是很重要的一個(gè)環(huán)節(jié)，既可以保證訓(xùn)練過(guò)程信息不丟失，也可以幫助我們?cè)谛枰焖倩謴?fù)或使用一個(gè)模型的時(shí)候，利用之前保存好的參數(shù)之間導(dǎo)入，可以節(jié)省大量的訓(xùn)練時(shí)間。本文通過(guò)最簡(jiǎn)單的例程教大家如何保存和讀取.ckpt文件。
一、保存到文件
首先是導(dǎo)入必要的東西：
import tensorflow as tf import numpy as np
隨便寫幾個(gè)變量：
# Save to file # remember to define the same dtype and shape when restore W = tf.Variable([[1,2,3],[3,4,5]], dtype=tf.float32, name='weights') b = tf.Variable([[1,2,3]], dtype=tf.float32, name='biases') init= tf.initialize_all_variables()
定義一個(gè)saver，來(lái)存儲(chǔ)我們的各種變量：
saver = tf.train.Saver()
保存的文件用.ckpt后綴：
with tf.Session() as sess:
sess.run(init)
save_path = saver.save(sess, "my_net/save_net.ckpt")
print("Save to path: ", save_path)
上面我們就完成了保存操作。
接下來(lái)我們要把之前保存過(guò)的變量取出來(lái)。
二、取出之前保存的變量
這里要注意，取出時(shí)要先開(kāi)辟一個(gè)容器來(lái)裝，shape和type要和我們之前保存的.ckpt一樣。
# restore variables # redefine the same shape and same type for your variables W = tf.Variable(np.arange(6).reshape((2, 3)), dtype=tf.float32, name="weights") b = tf.Variable(np.arange(3).reshape((1, 3)), dtype=tf.float32, name="biases")
restore時(shí)，不需要進(jìn)行init= tf.initialize_all_variables()操作。
利用saver提取文件：
saver = tf.train.Saver()
with tf.Session() as sess:
saver.restore(sess, "my_net/save_net.ckpt")
print("weights:", sess.run(W))
print("biases:", sess.run(b))
結(jié)果：
以上這篇TensorFlow Saver:保存和讀取模型參數(shù).ckpt實(shí)例就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
python flask基于cookie和session來(lái)實(shí)現(xiàn)會(huì)話控制的實(shí)戰(zhàn)代碼
所謂的會(huì)話（session）,就是客戶端瀏覽器和服務(wù)端網(wǎng)站之間一次完整的交互過(guò)程,本文介紹falsk通過(guò)cookie和session來(lái)控制http會(huì)話的全部解析,通常我們可以用cookie和session來(lái)保持用戶登錄等,感興趣的朋友一起看看吧2024-03-03
Python基于socket模塊實(shí)現(xiàn)UDP通信功能示例
這篇文章主要介紹了Python基于socket模塊實(shí)現(xiàn)UDP通信功能,結(jié)合實(shí)例形式分析了Python使用socket模塊實(shí)現(xiàn)IPV4協(xié)議下的UDP通信客戶端與服務(wù)器端相關(guān)操作技巧,需要的朋友可以參考下2018-04-04
使用Python在PowerPoint演示文稿之間復(fù)制樣式
在專業(yè)演示文稿設(shè)計(jì)與制作領(lǐng)域,多場(chǎng)演示間保持一致性至關(guān)重要,在PowerPoint演示文稿之間復(fù)制幻燈片母版成為了一項(xiàng)關(guān)鍵技巧,本文中,我們將探討如何使用Python在不同的PowerPoint演示文稿之間復(fù)制幻燈片母版,提升演示文稿創(chuàng)作流程的效率與美觀度,需要的朋友可以參考下2024-05-05
使用Python和XPath解析動(dòng)態(tài)JSON數(shù)據(jù)的操作指南
JSON動(dòng)態(tài)數(shù)據(jù)在Python中扮演著重要的角色,為開(kāi)發(fā)者提供了處理實(shí)時(shí)和靈活數(shù)據(jù)的能力,動(dòng)態(tài)JSON數(shù)據(jù)的獲取可能涉及到網(wǎng)絡(luò)請(qǐng)求和API調(diào)用,可以使用Python和XPath來(lái)解析動(dòng)態(tài)JSON數(shù)據(jù),接下來(lái)小編就給大家介紹一下操作步驟2023-09-09
今天小編就為大家分享一篇關(guān)于Python形參打包與解包小技巧分享，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-08-08
Python中按鈕(BUTTON)樣式屬性及說(shuō)明
文章介紹了Python中tkinter庫(kù)中的Button組件,用于在GUI中添加按鈕,按鈕可以包含文本或圖像,并且可以通過(guò)點(diǎn)擊執(zhí)行特定函數(shù),文章詳細(xì)說(shuō)明了Button組件的構(gòu)造語(yǔ)法和常用參數(shù),并提供了一個(gè)代碼示例2025-01-01
5個(gè)很好的Python面試題問(wèn)題答案及分析
這篇文章主要介紹了5個(gè)很好的Python面試題問(wèn)題答案及分析，具有一定借鑒價(jià)值,需要的朋友可以參考下2018-01-01
Python?Haul利器簡(jiǎn)化數(shù)據(jù)爬取任務(wù)提高開(kāi)發(fā)效率
Haul?是一個(gè)專門為數(shù)據(jù)爬取任務(wù)而設(shè)計(jì)的?Python?庫(kù),它提供了一系列的工具和功能,幫助我們輕松處理數(shù)據(jù)爬取中的重復(fù)工作和復(fù)雜問(wèn)題2024-01-01
使用pandas實(shí)現(xiàn)csv/excel sheet互相轉(zhuǎn)換的方法
今天小編就為大家分享一篇使用pandas實(shí)現(xiàn)csv/excel sheet互相轉(zhuǎn)換的方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-12-12
Python將string轉(zhuǎn)換到float的實(shí)例方法
在本篇文章中小編給大家分享的是關(guān)于Python將string轉(zhuǎn)換到float的實(shí)例方法以及相關(guān)知識(shí)點(diǎn)，需要的朋友們參考下。2019-07-07