解決TensorFlow訓(xùn)練模型及保存數(shù)量限制的問題
更新時間：2021年03月03日 08:36:35 作者：Destiny_Ren
這篇文章主要介紹了解決TensorFlow訓(xùn)練模型及保存數(shù)量限制的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
每次卷積神經(jīng)網(wǎng)絡(luò)訓(xùn)練的結(jié)果都只保存了最后一部分，查閱了一下相關(guān)資料，發(fā)現(xiàn)是定義saver時采用的默認值，這里進行如下設(shè)置：
saver = tf.train.Saver( max_to_keep = 100 , keep_checkpoint_every_n_hours = 1 )
補充：解決TensorFlow只能保存5個模型的問題
直奔主題
在訓(xùn)練模型的代碼中找到這句代碼：tf.train.Saver（），
改成：
tf.train.Saver(max_to_keep = m) # m為你想保存的模型數(shù)量
擴展
Saver類中的可選參數(shù)
tf.train.Saver(max_to_keep = m, keep_checkpoint_every_n_hours = n)
max_to_keep保存離當前訓(xùn)練最近的模型數(shù)量，默認值為5。如果想全部保存，并且電腦內(nèi)存夠用，設(shè)成多大都可以。
keep_checkpoint_every_n_hours每隔n個小時保存一次模型，默認值為10,000（一般情況下應(yīng)該不會訓(xùn)練這么長時間，所以相當于是不會按照時間來保存，按照設(shè)置的epoch保存節(jié)點數(shù)來保存）。
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。
相關(guān)文章
在本篇文章里小編給大家分享的是一篇關(guān)于Python編寫萬花尺圖案實例的內(nèi)容，有興趣的朋友們可以參考下。2021-01-01
這篇文章主要為大家詳細介紹了python實現(xiàn)圖像增強算法，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2022-08-08
Pytorch evaluation每次運行結(jié)果不同的解決
今天小編就為大家分享一篇Pytorch evaluation每次運行結(jié)果不同的解決，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-01-01
這篇文章主要介紹了Python3.5面向?qū)ο缶幊?結(jié)合圖文與實例形式詳細分析了Python面向?qū)ο缶幊滔嚓P(guān)的概念、類定義、實例化、實例變量、類變量、析構(gòu)函數(shù)等相關(guān)原理及使用技巧,需要的朋友可以參考下2019-04-04
Python 如何優(yōu)雅的將數(shù)字轉(zhuǎn)化為時間格式的方法
這篇文章主要介紹了Python 如何優(yōu)雅的將數(shù)字轉(zhuǎn)化為時間格式的方法，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習或者工作具有一定的參考學(xué)習價值，需要的朋友們下面隨著小編來一起學(xué)習學(xué)習吧2019-09-09
這篇文章主要介紹了Python OS模塊,結(jié)合實例形式總結(jié)分析了Python使用OS解析文件路徑、判斷文件、目錄等相關(guān)操作技巧,需要的朋友可以參考下2019-04-04
箭頭向量圖十分常見，比如天氣預(yù)報在顯示風場的時候，就會貼心地用箭頭指明風的方向，下面就跟隨小編一起學(xué)習一下如何利用Python繪制箭頭向量圖吧2023-08-08
Python最強大的功能之一是能夠十分方便地使用性能優(yōu)異的第三方庫,這些庫涵蓋了大量的不同領(lǐng)域,下面這篇文章主要給大家介紹了關(guān)于利用pip安裝python第三方庫的4種方法,需要的朋友可以參考下2023-02-02
這篇文章主要為大家介紹了python的循環(huán)，具有一定的參考價值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來幫助2021-12-12
Django框架會話技術(shù)實例分析【Cookie與Session】
這篇文章主要介紹了Django框架會話技術(shù),結(jié)合實例形式分析了Django框架Cookie與Session相關(guān)使用技巧與注意事項,需要的朋友可以參考下2019-05-05