TensorFlow Saver:保存和讀取模型參數(shù).ckpt實(shí)例
在使用TensorFlow的過程中，保存模型參數(shù)變量是很重要的一個環(huán)節(jié)，既可以保證訓(xùn)練過程信息不丟失，也可以幫助我們在需要快速恢復(fù)或使用一個模型的時(shí)候，利用之前保存好的參數(shù)之間導(dǎo)入，可以節(jié)省大量的訓(xùn)練時(shí)間。本文通過最簡單的例程教大家如何保存和讀取.ckpt文件。
一、保存到文件
首先是導(dǎo)入必要的東西：
import tensorflow as tf import numpy as np
隨便寫幾個變量：
# 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()
定義一個saver，來存儲我們的各種變量：
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)
上面我們就完成了保存操作。
接下來我們要把之前保存過的變量取出來。
二、取出之前保存的變量
這里要注意，取出時(shí)要先開辟一個容器來裝，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í)例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
python flask基于cookie和session來實(shí)現(xiàn)會話控制的實(shí)戰(zhàn)代碼
所謂的會話（session）,就是客戶端瀏覽器和服務(wù)端網(wǎng)站之間一次完整的交互過程,本文介紹falsk通過cookie和session來控制http會話的全部解析,通常我們可以用cookie和session來保持用戶登錄等,感興趣的朋友一起看看吧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)域,多場演示間保持一致性至關(guān)重要,在PowerPoint演示文稿之間復(fù)制幻燈片母版成為了一項(xiàng)關(guān)鍵技巧,本文中,我們將探討如何使用Python在不同的PowerPoint演示文稿之間復(fù)制幻燈片母版,提升演示文稿創(chuàng)作流程的效率與美觀度,需要的朋友可以參考下2024-05-05
使用Python和XPath解析動態(tài)JSON數(shù)據(jù)的操作指南
JSON動態(tài)數(shù)據(jù)在Python中扮演著重要的角色,為開發(fā)者提供了處理實(shí)時(shí)和靈活數(shù)據(jù)的能力,動態(tài)JSON數(shù)據(jù)的獲取可能涉及到網(wǎng)絡(luò)請求和API調(diào)用,可以使用Python和XPath來解析動態(tài)JSON數(shù)據(jù),接下來小編就給大家介紹一下操作步驟2023-09-09
今天小編就為大家分享一篇關(guān)于Python形參打包與解包小技巧分享，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-08-08
文章介紹了Python中tkinter庫中的Button組件,用于在GUI中添加按鈕,按鈕可以包含文本或圖像,并且可以通過點(diǎn)擊執(zhí)行特定函數(shù),文章詳細(xì)說明了Button組件的構(gòu)造語法和常用參數(shù),并提供了一個代碼示例2025-01-01
這篇文章主要介紹了5個很好的Python面試題問題答案及分析，具有一定借鑒價(jià)值,需要的朋友可以參考下2018-01-01
Python?Haul利器簡化數(shù)據(jù)爬取任務(wù)提高開發(fā)效率
Haul?是一個專門為數(shù)據(jù)爬取任務(wù)而設(shè)計(jì)的?Python?庫,它提供了一系列的工具和功能,幫助我們輕松處理數(shù)據(jù)爬取中的重復(fù)工作和復(fù)雜問題2024-01-01
使用pandas實(shí)現(xiàn)csv/excel sheet互相轉(zhuǎn)換的方法
今天小編就為大家分享一篇使用pandas實(shí)現(xiàn)csv/excel sheet互相轉(zhuǎn)換的方法，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-12-12
Python將string轉(zhuǎn)換到float的實(shí)例方法
在本篇文章中小編給大家分享的是關(guān)于Python將string轉(zhuǎn)換到float的實(shí)例方法以及相關(guān)知識點(diǎn)，需要的朋友們參考下。2019-07-07