tensorflow 只恢復(fù)部分模型參數(shù)的實(shí)例
更新時(shí)間：2020年01月06日 09:11:38 作者：平和之心
今天小編就為大家分享一篇tensorflow 只恢復(fù)部分模型參數(shù)的實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
我就廢話不多說(shuō)了，直接上代碼吧！
import tensorflow as tf
def model_1():
with tf.variable_scope("var_a"):
a = tf.Variable(initial_value=[1, 2, 3], name="a")
vars = [var for var in tf.trainable_variables() if var.name.startswith("var_a")]
print(len(vars))
return vars
def model_2():
vars1 = model_1()
with tf.variable_scope("var_b"):
a = tf.Variable(initial_value=[1, 2, 3], name="a")
vars2 = [var for var in tf.trainable_variables() if var.name.startswith("var")]
print(len(vars2))
return vars1
def pretrain_model1():
print("-------- model 1 ------")
vars = model_1()
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
saver = tf.train.Saver()
saver.save(sess, "./model.ckpt")
def train_model2():
print("-------- model 2 ------")
model1_vars = model_2()
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
saver = tf.train.Saver(var_list=model1_vars)
saver.restore(sess, "./model.ckpt")
vars = sess.run([model1_vars])
for var in vars:
print(var)
step = 2
if step == 1:
pretrain_model1()
else:
train_model2()
以上這篇tensorflow 只恢復(fù)部分模型參數(shù)的實(shí)例就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
如何給windows設(shè)置定時(shí)任務(wù)并運(yùn)行python腳本
這篇文章主要介紹了如何給windows設(shè)置定時(shí)任務(wù)并運(yùn)行python腳本，本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2021-08-08
python正則表達(dá)式抓取成語(yǔ)網(wǎng)站
做NLPproject時(shí)需要一個(gè)成語(yǔ)庫(kù)，我需要的是純成語(yǔ)，網(wǎng)上找的都是有詳細(xì)解釋的。于是自己寫了一個(gè)爬成語(yǔ)的python程序2013-11-11
這篇文章主要介紹了如何利用python畫畫，幫助大家更好的理解和使用python的turtle庫(kù)，感興趣的朋友可以了解下2020-12-12
PyQt5結(jié)合matplotlib繪圖的實(shí)現(xiàn)示例
這篇文章主要介紹了PyQt5結(jié)合matplotlib繪圖的實(shí)現(xiàn)示例，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-09-09
這篇文章主要介紹了pytorch模型保存方式,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2024-09-09
這篇文章主要介紹了Python正則re模塊使用步驟及原理解析,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-08-08
Python 讀取 YUV(NV12) 視頻文件實(shí)例
今天小編就為大家分享一篇Python 讀取 YUV(NV12) 視頻文件實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-12-12
PyCharm無(wú)法識(shí)別PyQt5的2種解決方法，ModuleNotFoundError: No module named
這篇文章主要介紹了PyCharm無(wú)法識(shí)別PyQt5的兩種解決辦法，ModuleNotFoundError: No module named 'pyqt5',需要的朋友可以參考下2020-02-02
在本篇文章里小編給大家分享了關(guān)于python自省的相關(guān)知識(shí)點(diǎn)內(nèi)容，需要的朋友們可以參考學(xué)習(xí)下。2020-06-06
對(duì)python產(chǎn)生隨機(jī)的二維數(shù)組實(shí)例詳解
今天小編就為大家分享一篇對(duì)python產(chǎn)生隨機(jī)的二維數(shù)組實(shí)例詳解，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-12-12