小编mtn*_*gld的帖子

Tensorflow Estimator - warm_start_from和model_dir

当使用tf.estimatorwith warm_start_from model_dir,并且warm_start_from目录和model_dir目录都包含有效检查点时,哪个检查点将实际恢复?

为了给出一些上下文,我的估算器代码看起来像

est = tf.estimator.Estimator(
    model_fn=model_fn,
    model_dir=model_dir,
    warm_start_from=warm_start_dir)

for epoch in range(num_epochs):
    est.train(input_fn=train_input_fn)
    est.evaluate(input_fn=eval_input_fn)
Run Code Online (Sandbox Code Playgroud)

(输入函数使用一次迭代器.)

因此在第一次迭代期间,当model_dir空为空时,我希望加载热启动检查点,但在下一个时期,我希望从上一次迭代中model_dir加入中间微调检查点.但至少从日志中看,它看起来warm_start_dir仍在被加载.

我可能会覆盖我的下一次迭代的估算器,但我想知道它是否应该在估算器中构建一些如何.

tensorflow tensorflow-estimator

9
推荐指数
1
解决办法
2505
查看次数

TensorFlow自定义估算器 - 在model_fn中进行小的更改后恢复模型

tf.estimator.Estimator用来开发我的模型,

我写了一个model_fn并训练了50,000次迭代,现在我想对我做一个小改动model_fn,例如添加一个新图层.

我不想从头开始训练,我想从50,000检查点恢复所有旧变量,并从这一点继续训练.当我尝试这样做时,我得到了一个NotFoundError

怎么办tf.estimator.Estimator呢?

tensorflow tensorflow-datasets tensorflow-estimator

7
推荐指数
1
解决办法
1933
查看次数