张量流的估计器如何加载模型的特定步骤而不是最新模型？

Question

我们可以使用Estimator和ObjectMapper保存许多模型检查点。

RunConfig默认情况下将使用最新的步骤classifier，  我可以加载200 吗？

ckpt-1

Answer 1

tf.estimator.Estimator.evaluate和tf.estimator.Estimator.predict都有一个checkpoint_path参数。您应该能够在此处提供到model.ckpt-1的路径，以使用此检查点进行评估。

请注意，此参数是在最近的TF更新中添加的（可能为1.7或1.8，不确定），因此，如果您使用的是过时的版本，则可能没有此参数。有一种骇人听闻的选择：在model_dir中应该有一个名为checkpoint的文件。该文件的第一行应该是

model_checkpoint_path: "model.ckpt-xxxxxx"

其中xxxxxx是最新检查点的步数（在您的情况下为200）。您可以手动将此行更改为希望Estimator加载的任何检查点。但是，您可能之后需要将其改回，否则，如果您想继续训练模型，可能会遇到问题。