当使用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仍在被加载.
我可能会覆盖我的下一次迭代的估算器,但我想知道它是否应该在估算器中构建一些如何.
我tf.estimator.Estimator用来开发我的模型,
我写了一个model_fn并训练了50,000次迭代,现在我想对我做一个小改动model_fn,例如添加一个新图层.
我不想从头开始训练,我想从50,000检查点恢复所有旧变量,并从这一点继续训练.当我尝试这样做时,我得到了一个NotFoundError
怎么办tf.estimator.Estimator呢?