tensorfow estimator: WarmStartSettings

tensorflow version: v 1.13
在tf.estimator.WarmStartSettings里, 是可以设置具体restore哪些变量, 里面有两个参数,

ckpt_to_initialize_from, 就是需要restore的ckpt地址, 例如xx/model.ckpt-xx
vars_to_warm_start, 就是判断哪些变量需要被restore的. 需要说明的是, 默认值是".*",它会使得程序去resotre所有的TRAINABLE_VARIABLES变量, 而这个变量集合里是不含有bn层的moving_mean及moving_variance参数的, 按照官网的解释就是:
```
  Defaults to '.*', which warm-starts all variables in the TRAINABLE_VARIABLES collection. Note that this excludes variables such as accumulators and moving statistics from batch norm.
```
因此有两种解决方法, 1是根据你需要restore的变量scope名字, 显式的指定需要restore的变量, 如有多个名字, 可以放在一个列表里, 变量名字支持正则表达式寻找. 2 是传入参数[".*"], 让程序去从Global variable里去恢复所有的变量, 这样就可以把bn层的moving_mean及moving_variance参数成功restore.

tensorfow estimator: WarmStartSettings相关推荐

tf.estimator的用法
tf.estimator的用法利用 tf.estimator 训练模型时需要写两个重要的函数,一个用于数据输入的函数(input_fn),另一个用于模型创建的函数(model_fn).下面逐一来说明 ...
Tensorflow API 讲解——tf.estimator.Estimator
class Estimator(builtins.object) #介绍 Estimator 类,用来训练和验证 TensorFlow 模型. Estimator 对象包含了一个模型 model_fn ...
tensorflow estimator 实践
本文以mnist数据集为例.estimator通常是和tf的dataset一起使用,故先制作tfrecord文件,在使用estimator进行测试. 文章结构: 1.文件目录 2. 制作tfrecor ...
tf.estimator.Estimator讲解
tf.estimator.Estimator 简单介绍是一个class 所以需要初始化,作用是用来训练和评价 tensorflow 模型的 Estimator对象包装由一个名为model_fn函数 ...
tensorflow函数学习
1.学习率相关 1.tf.train.piecewise_constant(global_step, boundaries,values) 设置区间,使得不同的step有不同的学习率.学习率的变化是阶 ...
tf.estimator用法
estimator:估算器 tf.estimator -----一种高级TensorFlow API.估算器封装以下操作: 训练(training) 评价(evaluation) 预测(predict ...
怎样从tensorflow的generator object Estimator.predict at 0x7fb1ecefeaf0中取数据
怎样从tensorflow的<generator object Estimator.predict at 0x7fb1ecefeaf0>中取数据通过以下代码: predictions = ...
【 Notes 】Best linear unbiased estimator(BLUE) approach for time-of-arrival based localisation
目录 Abstract Introduction BLUE-based positioning BLUE-LSC algorithm BLUE-LLS algorithm Abstract A com ...
【 Notes 】WLLS Algorithm of TOA - Based Positioning (include the two - step WLS estimator)
目录 LLS review WLLS the two - step WLS estimator LLS review 前面有博文:LLS,提到了线性最小二乘算法,使用LLS去求解TOA-Based P ...

tensorfow estimator: WarmStartSettings

tensorfow estimator: WarmStartSettings相关推荐

最新文章

热门文章