在tensorflow中设置保存checkpoint的最大数量实例


Posted in Python onJanuary 21, 2020

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中设置保存checkpoint的最大数量实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python查看FTP是否能连接成功的方法
Jul 30 Python
Python tkinter实现的图片移动碰撞动画效果【附源码下载】
Jan 04 Python
python3+PyQt5+Qt Designer实现扩展对话框
Apr 20 Python
Python 爬虫之Beautiful Soup模块使用指南
Jul 05 Python
深入浅析Python中的迭代器
Jun 04 Python
python 将字符串完成特定的向右移动方法
Jun 11 Python
Python实现代码统计工具
Sep 19 Python
python实现小世界网络生成
Nov 21 Python
python ftplib模块使用代码实例
Dec 31 Python
Pytorch模型转onnx模型实例
Jan 15 Python
浅谈tensorflow之内存暴涨问题
Feb 05 Python
在keras中实现查看其训练loss值
Jun 16 Python
TensorFlow——Checkpoint为模型添加检查点的实例
Jan 21 #Python
tensorflow estimator 使用hook实现finetune方式
Jan 21 #Python
Python实现FLV视频拼接功能
Jan 21 #Python
TFRecord格式存储数据与队列读取实例
Jan 21 #Python
TensorFlow dataset.shuffle、batch、repeat的使用详解
Jan 21 #Python
使用 tf.nn.dynamic_rnn 展开时间维度方式
Jan 21 #Python
python爬取本站电子书信息并入库的实现代码
Jan 20 #Python
You might like
PHP ? EasyUI DataGrid 资料取的方式介绍
2012/11/07 PHP
PHP获取表单所有复选框的值的方法
2014/08/28 PHP
ThinkPHP框架整合微信支付之Native 扫码支付模式二图文详解
2019/04/09 PHP
Laravel框架控制器,视图及模型操作图文详解
2019/12/04 PHP
PHP常量DIRECTORY_SEPARATOR原理及用法解析
2020/11/10 PHP
jquery+ajax每秒向后台发送请求数据然后返回页面的代码
2011/01/17 Javascript
用JavaScript实现类似于ListBox功能示例代码
2014/03/09 Javascript
悬浮广告方法日常收集整理
2016/03/18 Javascript
基于JavaScript实现下拉列表左右移动代码
2017/02/07 Javascript
jQuery文字轮播特效
2017/02/12 Javascript
微信小程序 开发之全局配置
2017/05/05 Javascript
echart简介_动力节点Java学院整理
2017/08/11 Javascript
详解webpack3如何正确引用并使用jQuery库
2017/08/26 jQuery
Vue中使用vee-validate表单验证的方法
2018/05/09 Javascript
详解vue中axios请求的封装
2019/04/08 Javascript
vuejs移动端实现div拖拽移动
2019/07/25 Javascript
关于layui 弹出层一闪而过就消失的解决方法
2019/09/09 Javascript
使用Vue Composition API写出清晰、可扩展的表单实现
2020/06/10 Javascript
[10:14]2018DOTA2国际邀请赛寻真——paiN Gaming不仅为自己而战
2018/08/14 DOTA
Python数组条件过滤filter函数使用示例
2014/07/22 Python
Python中实现三目运算的方法
2015/06/21 Python
python中requests小技巧
2017/05/10 Python
Python实现读取TXT文件数据并存进内置数据库SQLite3的方法
2017/08/08 Python
python提取包含关键字的整行数据方法
2018/12/11 Python
Python爬虫之UserAgent的使用实例
2019/02/21 Python
pyqt弹出新对话框,以及关闭对话框获取数据的实例
2019/06/18 Python
Anaconda之conda常用命令介绍(安装、更新、删除)
2019/10/06 Python
keras打印loss对权重的导数方式
2020/06/10 Python
Python 处理日期时间的Arrow库使用
2020/08/18 Python
用HTML5 实现橡皮擦的涂抹效果的教程
2015/05/11 HTML / CSS
浅析HTML5 Landmark
2020/09/11 HTML / CSS
青年创业培训欢迎词
2014/01/08 职场文书
作弊检讨书1000字
2014/02/01 职场文书
职称评定个人总结
2015/03/05 职场文书
安全责任协议书范本
2016/03/23 职场文书
SQL Server使用T-SQL语句批处理
2022/05/20 SQL Server