溫馨提示×

您好，登錄后才能下訂單哦！

密碼登錄×
登錄注冊(cè)×
其他方式登錄
點(diǎn)擊 登錄注冊(cè) 即表示同意《億速云用戶(hù)服務(wù)條款》

如何修改TensorFlow保存數(shù)量的限制

發(fā)布時(shí)間：2021-03-03 15:54:47 來(lái)源：億速云 閱讀：228 作者：Leah 欄目：開(kāi)發(fā)技術(shù)

本篇文章為大家展示了如何修改TensorFlow保存數(shù)量的限制，內(nèi)容簡(jiǎn)明扼要并且容易理解，絕對(duì)能使你眼前一亮，通過(guò)這篇文章的詳細(xì)介紹希望你能有所收獲。

設(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)保存）。

上述內(nèi)容就是如何修改TensorFlow保存數(shù)量的限制，你們學(xué)到知識(shí)或技能了嗎？如果還想學(xué)到更多技能或者豐富自己的知識(shí)儲(chǔ)備，歡迎關(guān)注億速云行業(yè)資訊頻道。

向AI問(wèn)一下細(xì)節(jié)

免責(zé)聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng)，如果涉及侵權(quán)請(qǐng)聯(lián)系站長(zhǎng)郵箱：is@yisu.com進(jìn)行舉報(bào)，并提供相關(guān)證據(jù)，一經(jīng)查實(shí)，將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI
助
手