Tensorflow 自定義loss的情況下初始化部分變量方式
一般情況下，tensorflow里面變量初始化過(guò)程為：
#variables ........... #..................... init = tf.initialize_all_variables() sess.run(init)
這里 tf.initialize_all_variables() 會(huì)初始化所有的變量。
實(shí)際過(guò)程中，假設(shè)有a, b, c三個(gè)變量，其中a已經(jīng)被初始化了，只想單獨(dú)初始化b，c，那么：
#variables ... ... init = tf.variables_initializer([b,c]) sess.run(init)
此外，如果自行修改了optimizer，如下代碼就會(huì)報(bào)錯(cuò)：
#definition of variables a, b, c ... .... my_optimizer = tf.train.RMSProp(learning_rate = 0.1).minimize(my_cost) init = tf.variables_initializer([b,c]) sess.run(init)
這是因?yàn)樽约憾x的optimizer會(huì)生成新的variables，但是在init里面并沒有初始化，所以無(wú)法訪問(wèn)，會(huì)報(bào)錯(cuò)。解決方法如下：
a = tf.Variables(...) #line N temp = set(tf.all_variables()) b = tf.Variables(...) c = tf.Variables(...) #definition of my optimizer optimizer = tf.train....... init = tf.variables_initializer(set(tf.all_varialbles())-temp) # line M sess.run(init)
首先，temp = set(tf.all_variables()) 將該行(line N)代碼之前的所有變量保存在temp中，接下來(lái)定義變量b, c，以及自定義的optimizer，然后 set(tf.all_varialbles()存儲(chǔ)了改行(line M)之前的所有變量（包括optimizer生成的變量以及temp中所含的變量），set(tf.all_varialbles())-temp相減得到line N~M這幾行定義的變量。
以上這篇Tensorflow 自定義loss的情況下初始化部分變量方式就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了python文件選擇對(duì)話框的操作方法，本文通過(guò)實(shí)例代碼給大家介紹的非常詳細(xì)，具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2019-06-06
Python代碼統(tǒng)計(jì)耗時(shí)的方法詳解
在現(xiàn)代軟件開發(fā)中,性能優(yōu)化是一個(gè)至關(guān)重要的環(huán)節(jié),無(wú)論是開發(fā)大型系統(tǒng)還是小型工具,開發(fā)者都需要對(duì)代碼的執(zhí)行時(shí)間進(jìn)行精確測(cè)量,以便找出瓶頸并優(yōu)化性能,本文給大家介紹了Python代碼統(tǒng)計(jì)耗時(shí)的方法,需要的朋友可以參考下2025-02-02
tensorflow基本操作小白快速構(gòu)建線性回歸和分類模型
這篇文章主要介紹了tensorflow基本操作,快速構(gòu)建線性回歸和分類模型,圖文代碼示例非常詳細(xì)，有需要的朋友可以借鑒參考下，希望可以對(duì)大家有所幫助2021-08-08
python實(shí)現(xiàn)簡(jiǎn)單倒計(jì)時(shí)功能
這篇文章主要為大家詳細(xì)介紹了python實(shí)現(xiàn)簡(jiǎn)單倒計(jì)時(shí)功能，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2021-04-04
Python實(shí)現(xiàn)訪問(wèn)者模式詳情
這篇文章主要介紹了Python實(shí)現(xiàn)訪問(wèn)者模式詳情，訪問(wèn)者模式，指作用于一個(gè)對(duì)象結(jié)構(gòu)體上的元素的操作。訪問(wèn)者可以使用戶在不改變?cè)摻Y(jié)構(gòu)體中的類的基礎(chǔ)上定義一個(gè)新的操作，下文更多相關(guān)資料,需要的朋友可以參考下2022-03-03
Sentry的安裝、配置、使用教程(Sentry日志手機(jī)系統(tǒng))
Sentry?是一個(gè)實(shí)時(shí)事件日志記錄和聚合平臺(tái)，由于ExceptionLess官方提供的客戶端只有.Net/.NetCore平臺(tái)和js的，本文繼續(xù)介紹另一個(gè)日志收集系統(tǒng)Sentry，感興趣的朋友一起看看吧2022-07-07
Python-pip配置國(guó)內(nèi)鏡像源快速下載包的方法詳解
pip如果不配置國(guó)內(nèi)鏡像源的話,下載包的速度非常慢,畢竟默認(rèn)的源在國(guó)外呢,這篇文章主要介紹了Python-pip配置國(guó)內(nèi)鏡像源快速下載包的方法詳解,需要的朋友可以參考下2024-01-01
pip安裝python庫(kù)時(shí)報(bào)錯(cuò)的問(wèn)題解決
本文主要介紹了在Windows系統(tǒng)上解決pip命令找不到的問(wèn)題的兩種方法,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2025-03-03
隊(duì)列（queue）是一種先進(jìn)先出的（First In First Out）的線性表，簡(jiǎn)稱FIFO。隊(duì)列允許在一端進(jìn)行插入操作，而在另一端進(jìn)行刪除操作。允許插入的一端為隊(duì)尾，允許刪除的一端為隊(duì)頭。隊(duì)列不允許在中間部位進(jìn)行操作2021-10-10
python函數(shù)存儲(chǔ)在模塊的優(yōu)點(diǎn)及用法總結(jié)
在本篇文章里小編給大家整理了一篇關(guān)于python函數(shù)存儲(chǔ)在模塊的優(yōu)點(diǎn)及用法相關(guān)內(nèi)容，有興趣的朋友們可以跟著學(xué)習(xí)下。2021-10-10