解決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)文章
pyinstaller打包多個(gè)py文件和去除cmd黑框的方法
今天小編就為大家分享一篇pyinstaller打包多個(gè)py文件和去除cmd黑框的方法,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-06-06python正則表達(dá)式之re.match()與re.search()的用法及區(qū)別
本文主要介紹了python正則表達(dá)式之re.match()與re.search()的用法及區(qū)別,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2022-08-08深入探究Python Numba庫(kù)編譯優(yōu)化利器
這篇文章主要為大家介紹了Python Numba庫(kù)編譯優(yōu)化利器深入探究,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2024-01-01python繼承threading.Thread實(shí)現(xiàn)有返回值的子類(lèi)實(shí)例
這篇文章主要介紹了python繼承threading.Thread實(shí)現(xiàn)有返回值的子類(lèi)實(shí)例,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-05-05Qt通過(guò)QGraphicsview實(shí)現(xiàn)簡(jiǎn)單縮放及還原效果
本文主要介紹通過(guò)QGraphicsview實(shí)現(xiàn)簡(jiǎn)單的縮放以及縮放后還原原始大小,通過(guò)scale可以對(duì)view進(jìn)行放大或縮小,具體內(nèi)容詳情跟隨小編一起看看吧2021-09-09對(duì)python中的*args與**kwgs的含義與作用詳解
今天小編就為大家分享一篇對(duì)python中的*args與**kwgs的含義與作用詳解,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-08-08Centos環(huán)境部署django項(xiàng)目的全過(guò)程(永久復(fù)用)
Django是一款針對(duì)Python環(huán)境的WEB開(kāi)發(fā)框架,能夠幫助我們構(gòu)架快捷,下面這篇文章主要給大家介紹了關(guān)于Centos環(huán)境部署django項(xiàng)目的相關(guān)資料,需要的朋友可以參考下2022-10-10