解決tensorflow 釋放圖,刪除變量問題
問題,在一個程序內(nèi)構(gòu)建好了一個圖,運(yùn)行完之后想重新使用這個圖進(jìn)行計算,或者想同時在train完的時候做test,就會提示***變量已存在。
解決辦法:
待一個圖運(yùn)行完之后加上tf.reset_default_graph來重置默認(rèn)的圖。
補(bǔ)充知識:tensorflow 張量插入或者刪除的辦法
我就廢話不多說了,大家還是直接看代碼吧~
part1 = new_enc_in[:, 0:3 ] part2 = new_enc_in[:, 3:] ###part_new 要插入的張量 new_enc_in = tf.concat([part1, part_new, part2], 1) ##得到的張量里part_new為新插入列 ##如果要刪除,則修改part2的范圍 part1 = new_enc_in[:, 0:3] part2 = new_enc_in[:, 4:] new_enc_in = tf.concat([part1, part2], 1) ###原張量的第四列被刪除
以上這篇解決tensorflow 釋放圖,刪除變量問題就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
解讀torch.cuda.amp自動混合精度訓(xùn)練之節(jié)省顯存并加快推理速度
這篇文章主要介紹了torch.cuda.amp自動混合精度訓(xùn)練之節(jié)省顯存并加快推理速度問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-08-08淺談keras2 predict和fit_generator的坑
這篇文章主要介紹了淺談keras2 predict和fit_generator的坑,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-06-06python使用標(biāo)準(zhǔn)庫根據(jù)進(jìn)程名如何獲取進(jìn)程的pid詳解
Python有一套很有用的標(biāo)準(zhǔn)庫(standard library)。標(biāo)準(zhǔn)庫會隨著Python解釋器,一起安裝在你的電腦中的,所以下面這篇文章主要給大家介紹了關(guān)于python使用標(biāo)準(zhǔn)庫根據(jù)進(jìn)程名如何獲取進(jìn)程pid的相關(guān)資料,需要的朋友可以參考下。2017-10-10