基于tensorflow加載部分層的方法
更新時間：2018年07月26日 09:57:55 作者：jinglingli_SJTU
今天小編就為大家分享一篇基于tensorflow加載部分層的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
一般使用
saver.restore(sess, modeldir + "model.ckpt")
即可加載已經(jīng)訓(xùn)練好的網(wǎng)絡(luò)，可是有時候想值使用部分層的參數(shù)，這時候可以選擇在加載網(wǎng)絡(luò)之后重新初始化剩下的層
var_list = [weights['wd1'], weights['out'], biases['bd1'], biases['out'], global_step] initfc = tf.variables_initializer(var_list, name='init')
比如我們想從新初始化var_list中的各個層，在restore之后，再初始化即可
sess.run(init) saver.restore(sess, modeldir + "model.ckpt") print sess.run(global_step) #initialize several layer sess.run(initfc) print sess.run(global_step)
即可發(fā)現(xiàn)部分變量重新初始化了
以上這篇基于tensorflow加載部分層的方法就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了c++調(diào)用python中文亂碼問題,本文通過實例代碼給大家介紹的非常詳細，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2020-07-07
pyinstaller打包單個exe后無法執(zhí)行錯誤的解決方法
今天小編就為大家分享一篇pyinstaller打包單個exe后無法執(zhí)行錯誤的解決方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-06-06
這篇文章主要為大家介紹了Python妙開根號的三種方式，具有一定的參考價值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來幫助2022-01-01
python通過scapy獲取局域網(wǎng)所有主機mac地址示例
這篇文章主要介紹了python通過scapy獲取局域網(wǎng)所有主機mac地址示例,需要的朋友可以參考下2014-05-05
Python實現(xiàn)將字典內(nèi)容保存到Excel文件的四種方法
Python 中的字典是一個數(shù)據(jù)集合,其中每個值對應(yīng)一個鍵,它們是無序的、可變的,并且對字典中存儲的值和鍵的數(shù)據(jù)類型沒有限制,本文給大家介紹了Python實現(xiàn)將字典內(nèi)容保存到Excel文件的四種方法,需要的朋友可以參考下2024-07-07
如何使用Python生成4位數(shù)的隨機數(shù)字
本文討論了如何使用randint() 和randrange() 方法來生成一個四位數(shù)的數(shù)字,此外,我們還討論了另一種擁有隨機四位數(shù)號碼的途徑,感興趣的朋友跟隨小編一起看看吧2023-10-10
使用Pandas的ExcelWriter操作excel的方法
這篇文章主要介紹了使用Pandas的ExcelWriter操作excel的方法,ExcelWriter這個插件有個坑,就是已經(jīng)設(shè)置好的格式是無法更改的,因此,由pandas轉(zhuǎn)成excel的時候,必須將格式清除,尤其是表頭的格式需要大家多多注意,本文結(jié)合示例代碼講解的非常詳細,需要的朋友參考下吧2023-11-11
這篇文章主要介紹了Pyecharts 餅圖位置調(diào)整方式，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-11-11
這篇文章主要給大家介紹了關(guān)于Python中文文本處理利器jieba分詞庫使用的相關(guān)資料,jieba是python中一個重要的第三方中文分詞函數(shù)庫,文中通過代碼介紹的非常詳細,需要的朋友可以參考下2023-12-12
這篇文章主要介紹了python可視化分析繪制散點圖和邊界氣泡圖，python繪制散點圖，展現(xiàn)兩個變量間的關(guān)系，當(dāng)數(shù)據(jù)包含多組時，使用不同顏色和形狀區(qū)分2022-06-06