解決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保存離當(dāng)前訓(xùn)練最近的模型數(shù)量，默認值為5。如果想全部保存，并且電腦內(nèi)存夠用，設(shè)成多大都可以。
keep_checkpoint_every_n_hours每隔n個小時保存一次模型，默認值為10,000（一般情況下應(yīng)該不會訓(xùn)練這么長時間，所以相當(dāng)于是不會按照時間來保存，按照設(shè)置的epoch保存節(jié)點數(shù)來保存）。
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。
相關(guān)文章
pyinstaller打包多個py文件和去除cmd黑框的方法
今天小編就為大家分享一篇pyinstaller打包多個py文件和去除cmd黑框的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-06-06
python正則表達式之re.match()與re.search()的用法及區(qū)別
本文主要介紹了python正則表達式之re.match()與re.search()的用法及區(qū)別，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2022-08-08
這篇文章主要為大家介紹了Python Numba庫編譯優(yōu)化利器深入探究,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2024-01-01
這篇文章主要介紹了如何基于pandas讀取csv后合并兩個股票,文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-09-09
python繼承threading.Thread實現(xiàn)有返回值的子類實例
這篇文章主要介紹了python繼承threading.Thread實現(xiàn)有返回值的子類實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-05-05
Qt通過QGraphicsview實現(xiàn)簡單縮放及還原效果
本文主要介紹通過QGraphicsview實現(xiàn)簡單的縮放以及縮放后還原原始大小，通過scale可以對view進行放大或縮小，具體內(nèi)容詳情跟隨小編一起看看吧2021-09-09
今天小編就為大家分享一篇對python中的*args與**kwgs的含義與作用詳解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-08-08
這篇文章主要介紹了Python爬蟲爬取網(wǎng)站圖片,此次python3主要用requests，解析圖片網(wǎng)址主要用beautiful soup，可以基本完成爬取圖片功能,需要的朋友可以參考一下2022-01-01
Centos環(huán)境部署django項目的全過程(永久復(fù)用)
Django是一款針對Python環(huán)境的WEB開發(fā)框架,能夠幫助我們構(gòu)架快捷,下面這篇文章主要給大家介紹了關(guān)于Centos環(huán)境部署django項目的相關(guān)資料,需要的朋友可以參考下2022-10-10
這篇文章主要介紹了Python CSV模塊使用實例,本文將舉幾個例子來介紹一下Python的CSV模塊的使用方法,包括reader、writer、DictReader、DictWriter.register_dialect等,需要的朋友可以參考下2015-04-04