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