淺談tf.train.Saver()與tf.train.import_meta_graph的要點
(一)、tf.train.Saver()
(1). tf.train.Saver() 是用來保存tensorflow訓練模型的，默認保存全部參數(shù)
(2). 用來加載參數(shù)，注：只加載存儲在data中的權重和偏置項等需要訓練的參數(shù)，其他一律不加載，包括meta文件中的圖也不加載
(二)、tf.train.import_meta_graph
(1). 用來加載meta文件中的圖,以及圖上定義的結點參數(shù)包括權重偏置項等需要訓練的參數(shù),也包括訓練過程生成的中間參數(shù),所有參數(shù)都是通過graph調用接口get_tensor_by_name(name="訓練時的參數(shù)名稱")來獲取
(三)、總結
(1). 保存使用tf.train.Saver()
(2). 加載可以使用tf.train.import_meta_graph(".meta文件"),直接通過訓練參數(shù)名稱就可以獲取需要的參數(shù),但需要提前知道訓練時的參數(shù)名稱才能獲取,要懂得tensorflow命名規(guī)則
(3). tf.train.Saver("./checkpoints目錄/")加載的缺點是只加載了訓練參數(shù),并且必須定義與之相同(shape,dtype要相同,tf.type要相同，如：我是placeholder,你也是tf.placeholder)方能使用,當你要獲取訓練中間參數(shù)時,需要和訓練過程一樣搭建相同的網(wǎng)絡.
補充：tf.train.import_meta_graph報KeyError
我在模型恢復時，在執(zhí)行tf.train.import_meta_graph的時候報錯
后來發(fā)現(xiàn)，我的模型是在服務器上訓練的，服務器上tensorflow版本是1.11.0,而我在本地電腦上執(zhí)行的 tf.train.import_meta_graph，我本地的tensorflow是1.5.0，我將tensorflow更新到1.11.0后，就解決了。
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
這篇文章主要介紹了Python讀寫txt文本文件的操作方法全解析,包括對文本的查找和替換等技巧的講解,需要的朋友可以參考下2016-06-06
python實現(xiàn)定時壓縮指定文件夾發(fā)送郵件
這篇文章主要為大家詳細介紹了python實現(xiàn)定時壓縮指定文件夾發(fā)送郵件，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-04-04
本文主要介紹了python httpx的具體使用，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2023-04-04
Pycharm運行時總是跳出Python?Console問題
這篇文章主要介紹了Pycharm運行時總是跳出Python?Console問題，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2023-04-04
今天小編就為大家分享一篇在PyCharm中批量查找及替換的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-01-01
解決90%的常見問題的8個python NumPy函數(shù)
這篇文章主要為大家介紹了解決90%的常見問題的8個python NumPy函數(shù)示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2023-06-06
這篇文章主要介紹了Python+OpenCV 實現(xiàn)高斯濾波的過程,關于高斯濾波的解釋，它是一種線性平滑濾波，適用于消除高斯噪聲，具體實現(xiàn)過程跟隨小編一起看看吧2021-10-10
這篇文章主要介紹了Python自動化測試selenium指定截圖文件名方法，Selenium?支持?Web?瀏覽器的自動化，它提供一套測試函數(shù)，用于支持?Web?自動化測試，下文基于python實現(xiàn)指定截圖文件名方法，需要的小伙伴可以參考一下2022-05-05
淺談哪個Python庫才最適合做數(shù)據(jù)可視化
數(shù)據(jù)可視化是任何探索性數(shù)據(jù)分析或報告的關鍵步驟，目前有許多非常好的商業(yè)智能工具，比如Tableau、googledatastudio和PowerBI等，本文就詳細的進行對比，感興趣的可以了解一下2021-06-06
在本篇文章里小編給大家整理的是一篇關于postman傳遞當前時間戳知識點相關內(nèi)容，有需要的朋友們可以學習下。2019-09-09