tensorflow獲取預訓練模型某層參數(shù)并賦值到當前網(wǎng)絡指定層方式
更新時間：2020年01月24日 10:39:23 作者：曲草
今天小編就為大家分享一篇tensorflow獲取預訓練模型某層參數(shù)并賦值到當前網(wǎng)絡指定層方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
已經(jīng)有了一個預訓練的模型，我需要從其中取出某一層，把該層的weights和biases賦值到新的網(wǎng)絡結構中，可以使用tensorflow中的pywrap_tensorflow(用來讀取預訓練模型的參數(shù)值)結合Session.assign()進行操作。
這種需求即預訓練模型可能為單分支網(wǎng)絡，當前網(wǎng)絡為多分支，我需要把單分支A復用到到多個分支去（B，C，D）。
先導入對應的工具包
from tensorflow.python import pywrap_tensorflow
接下來的操作在一個tf.Session中進行
reader = pywrap_tensorflow.NewCheckpointReader(pre_train_model_path)
# 獲取當前圖可訓練變量
trainable_variables = tf.trainable_variables()
# 需要賦值的當前網(wǎng)絡層變量，這里只是隨便起的名字。
restore_v_target_name = "fc_target"
# 需要的預訓練模型中的某層的名字
restore_v_source_name = "fc_source"
for v in trainable_variables:
if restore_v_target_name == v.name:
# 回復weights和biases
sess.run(
tf.assign(v, reader.get_tensor(restore_v_source_name + "/weights"))) if "weights" in v.name else sess.run(
tf.assign(v, reader.get_tensor(restore_v_source_name + "/biases")))
以上這篇tensorflow獲取預訓練模型某層參數(shù)并賦值到當前網(wǎng)絡指定層方式就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
今天小編就為大家分享一篇python中使用print輸出中文的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-07-07
這篇文章主要介紹了10 分鐘快速入門 Python3的教程，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2019-01-01
這篇文章主要介紹了Django 頁面緩存的cache_key是如何生成的，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2021-03-03
使用python將圖片格式轉(zhuǎn)換為ico格式的示例
今天小編就為大家分享一篇使用python將圖片格式轉(zhuǎn)換為ico格式的示例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-10-10
這篇文章主要介紹了如何在python中使用selenium的示例，selenium提供了一個通用的接口,可模擬用戶來操作瀏覽器，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2017-12-12
這篇文章主要介紹了pygame實現(xiàn)成語填空游戲，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-10-10
PyTorch搭建雙向LSTM實現(xiàn)時間序列負荷預測
這篇文章主要為大家介紹了PyTorch搭建雙向LSTM實現(xiàn)時間序列負荷預測，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2022-05-05
本文將介紹處理大量數(shù)據(jù)時非常方便的工具，例如tqdm與 multiprocessing?imap??一起使用、并行處理檔案、繪制和處理3D數(shù)據(jù)等，感興趣的小伙伴可以了解一下2021-12-12
開始你的Python，這里有靠譜的Pycharm漢化安裝詳細教程，本文圖文并茂給大家介紹的非常詳細，具有一定的參考借鑒價值,需要的朋友可以參考下2020-01-01
這篇文章主要介紹了Pycharm github配置實現(xiàn)過程圖解,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-10-10