如何在Tensorflow中使用TPU Estimator进行乐观还原?

时间:2018-09-06 13:00:17

标签: tensorflow tensorflow-estimator

我正在尝试通过tf.estimator(特别是TPUEstimator)实现GAN的逐步增长。我指定了一个模型目录供估算器使用。经过几步后,模型在model_fn中添加了更多操作,估计器希望仅从检查点还原先前的变量(优化还原)。但是,在添加新操作后,它抱怨该变量在检查点文件中不存在。

到目前为止,我已经尝试了热启动设置,但由于在检查点错误中未找到相同的设置而失败。我已经看到了用于乐观还原的代码,但是我不确定如何覆盖tf.estimator中的原始还原机制。

有人做过还是应该请求功能?

0 个答案:

没有答案