基于tensorflow加載部分層的方法
更新時(shí)間：2018年07月26日 09:57:55 作者：jinglingli_SJTU
今天小編就為大家分享一篇基于tensorflow加載部分層的方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
一般使用
saver.restore(sess, modeldir + "model.ckpt")
即可加載已經(jīng)訓(xùn)練好的網(wǎng)絡(luò)，可是有時(shí)候想值使用部分層的參數(shù)，這時(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中的各個(gè)層，在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加載部分層的方法就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
python開(kāi)發(fā)準(zhǔn)備工作之配置虛擬環(huán)境(非常重要)
這篇文章主要介紹了python開(kāi)發(fā)準(zhǔn)備工作之配置虛擬環(huán)境(非常重要)，小編覺(jué)得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2019-02-02
wxPython是Python編程語(yǔ)言的一個(gè)GUI工具箱。他使得Python程序員能夠輕松的創(chuàng)建具有健壯、功能強(qiáng)大的圖形用戶界面的程序。2009-02-02
Python中venv虛擬環(huán)境超詳細(xì)講解
虛擬環(huán)境是一個(gè)獨(dú)立的Python環(huán)境,它與系統(tǒng)的全局Python環(huán)境隔離,這篇文章主要介紹了Python中venv虛擬環(huán)境的相關(guān)資料,文中通過(guò)代碼介紹的非常詳細(xì),需要的朋友可以參考下2025-04-04
python使用pygame創(chuàng)建精靈Sprite
這篇文章主要介紹了使用Pygame創(chuàng)建精靈Sprite，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2021-04-04
Python操作遠(yuǎn)程服務(wù)器 paramiko模塊詳細(xì)介紹
這篇文章主要介紹了Python操作遠(yuǎn)程服務(wù)器 paramiko模塊詳細(xì)介紹,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-08-08
這篇文章主要介紹了python將文本中的空格替換為換行的方法,需要的朋友可以參考下2018-03-03
keras 獲取某層輸出 獲取復(fù)用層的多次輸出實(shí)例
這篇文章主要介紹了keras 獲取某層輸出 獲取復(fù)用層的多次輸出實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-05-05
使用python將多個(gè)excel文件合并到同一個(gè)文件的方法
這篇文章主要介紹了使用python將多個(gè)excel文件合并到同一個(gè)文件的方法，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2019-07-07
python實(shí)現(xiàn)通過(guò)pil模塊對(duì)圖片格式進(jìn)行轉(zhuǎn)換的方法
這篇文章主要介紹了python實(shí)現(xiàn)通過(guò)pil模塊對(duì)圖片格式進(jìn)行轉(zhuǎn)換的方法,涉及Python中pil模塊的使用技巧,具有一定參考借鑒價(jià)值,需要的朋友可以參考下2015-03-03
Python中實(shí)現(xiàn)繪制動(dòng)態(tài)旋轉(zhuǎn)的星形
結(jié)合matplotlib和numpy創(chuàng)建動(dòng)態(tài)五角星動(dòng)畫,步驟包括安裝庫(kù)、編寫代碼和運(yùn)行腳本,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2024-11-11