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