tensorflow 只恢復部分模型參數(shù)的實例
更新時間：2020年01月06日 09:11:38 作者：平和之心
今天小編就為大家分享一篇tensorflow 只恢復部分模型參數(shù)的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
我就廢話不多說了，直接上代碼吧！
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 只恢復部分模型參數(shù)的實例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
Python生成可執(zhí)行文件.exe操作完整流程記錄
.exe是文件擴展名,帶有.exe擴展名的文件名按下Enter鍵就可運行,這篇文章主要給大家介紹了關于Python生成可執(zhí)行文件.exe操作完整流程的相關資料,文中通過代碼介紹的非常詳細,需要的朋友可以參考下2024-05-05
Flask是一個微型的Python開發(fā)的Web框架，基于Werkzeug WSGI工具箱和Jinja2 模板引擎。Flask使用BSD授權。Flask也被稱為“microframework”，因為它使用簡單的核心，用extension增加其他功能。本文主要介紹了它的異常處理機制2021-06-06
python bluetooth藍牙信息獲取藍牙設備類型的方法
這篇文章主要介紹了python bluetooth藍牙信息獲取藍牙設備類型的方法，具體轉(zhuǎn)化方法文中給大家介紹的非常詳細，非常不錯，具有一定的參考借鑒價值,需要的朋友可以參考下2019-11-11
這篇文章主要介紹了Python實現(xiàn)密碼薄文件讀寫操作，本文給大家介紹的非常詳細，具有一定的參考借鑒價值,需要的朋友可以參考下2019-12-12
這篇文章主要介紹了python使用turtle庫來畫一朵花，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2019-03-03
本文主要介紹了教你pycharm運行Django第一個項目的實現(xiàn)，文中通過圖文示例介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2021-08-08
今天小編就為大家分享一篇Pycharm新建模板默認添加個人信息的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-07-07
Django ForeignKey與數(shù)據(jù)庫的FOREIGN KEY約束詳解
這篇文章主要介紹了Django ForeignKey與數(shù)據(jù)庫的FOREIGN KEY約束詳解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-05-05
這篇文章主要為大家介紹了Python中列表的幾個常用方法總結(jié)，文中的示例代碼講解詳細，對我們學習Python列表有一定幫助，需要的可以參考一下2022-03-03
python3中類的重點與難點:類屬性和實例屬性的區(qū)別說明
這篇文章主要介紹了python3中類的重點與難點:類屬性和實例屬性的區(qū)別說明，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2021-06-06