在tensorflow中設(shè)置保存checkpoint的最大數(shù)量實例
更新時間：2020年01月21日 09:56:53 作者：橘的皮瓜著披。
今天小編就為大家分享一篇在tensorflow中設(shè)置保存checkpoint的最大數(shù)量實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
1、我就廢話不多說了，直接上代碼吧！
# Set up a RunConfig to only save checkpoints once per training cycle.
run_config = tf.estimator.RunConfig(save_checkpoints_secs=1e9,keep_checkpoint_max = 10)
model = tf.estimator.Estimator(
model_fn=deeplab_model_focal_class_imbalance_loss_adaptive.deeplabv3_plus_model_fn,
model_dir=FLAGS.model_dir,
config=run_config,
params={
'output_stride': FLAGS.output_stride,
'batch_size': FLAGS.batch_size,
'base_architecture': FLAGS.base_architecture,
'pre_trained_model': FLAGS.pre_trained_model,
'batch_norm_decay': _BATCH_NORM_DECAY,
'num_classes': _NUM_CLASSES,
'tensorboard_images_max_outputs': FLAGS.tensorboard_images_max_outputs,
'weight_decay': FLAGS.weight_decay,
'learning_rate_policy': FLAGS.learning_rate_policy,
'num_train': _NUM_IMAGES['train'],
'initial_learning_rate': FLAGS.initial_learning_rate,
'max_iter': FLAGS.max_iter,
'end_learning_rate': FLAGS.end_learning_rate,
'power': _POWER,
'momentum': _MOMENTUM,
'freeze_batch_norm': FLAGS.freeze_batch_norm,
'initial_global_step': FLAGS.initial_global_step
})
以上這篇在tensorflow中設(shè)置保存checkpoint的最大數(shù)量實例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要為大家介紹了python虛擬機解釋器及運行過程的介紹，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2022-06-06
在本篇內(nèi)容里小編給大家整理的是一篇關(guān)于python爬蟲篩選工作實例講解內(nèi)容，有需要的朋友們可以參考下。2020-11-11
Python中創(chuàng)建包和增添包的路徑(sys.path.append()）
本文主要介紹了Python中創(chuàng)建包和增添包的路徑(sys.path.append()）,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2025-01-01
計算機二級python學(xué)習(xí)教程（2） python語言基本語法元素
這篇文章主要為大家詳細(xì)介紹了計算機二級python學(xué)習(xí)教程的第2篇，Python語言基本語法元素，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-05-05
Python實現(xiàn)讀取Properties配置文件的方法
這篇文章主要介紹了Python實現(xiàn)讀取Properties配置文件的方法,結(jié)合實例形式分析了Python讀取Properties配置文件類的定義與使用相關(guān)操作技巧,需要的朋友可以參考下2018-03-03
這篇文章主要介紹了pycharm引入其他目錄的包報錯,import報錯的解決,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-08-08
關(guān)鍵字是預(yù)先保留的標(biāo)識符,每個關(guān)鍵字都有特殊的含義,下面這篇文章主要給大家介紹了關(guān)于Python新手入門之常用關(guān)鍵字的簡單示例,文中通過代碼介紹的非常詳細(xì),需要的朋友可以參考下2024-03-03
這篇文章主要為大家詳細(xì)介紹了pygame游戲之旅，教大家如何創(chuàng)建游戲窗口界面，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-11-11
今天小編就為大家分享一篇python中in在list和dict中查找效率的對比分析，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-05-05
這篇文章主要介紹了Python聚類算法之DBSACN,結(jié)合實例形式詳細(xì)分析了DBSACN算法的原理與具體實現(xiàn)技巧,具有一定參考借鑒價值,需要的朋友可以參考下2015-11-11