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

Question

我们可以使用 Estimator 和 RunConfig 保存模型的许多检查点.
classifier eval 将使用最新的步骤 200默认情况下，
我可以加载 ckpt-1 ?

my_checkpointing_config = tf.estimator.RunConfig(
    save_checkpoints_secs = 20*60,  # Save checkpoints every 20 minutes.
    keep_checkpoint_max = 10,       # Retain the 10 most recent checkpoints.
)
classifier = tf.estimator.DNNClassifier(
    feature_columns=my_feature_columns,
    hidden_units=[10, 10],
    n_classes=3,
    model_dir='models/iris',
    config=my_checkpointing_config)



$ ls -1 models/iris
checkpoint
events.out.tfevents.timestamp.hostname
graph.pbtxt
model.ckpt-1.data-00000-of-00001
model.ckpt-1.index
model.ckpt-1.meta
model.ckpt-200.data-00000-of-00001
model.ckpt-200.index
model.ckpt-200.meta

Answer 1

两者 tf.estimator.Estimator.evaluate和 tf.estimator.Estimator.predict有一个 checkpoint_path争论。您应该能够提供到 model.ckpt-1 的路径此处使用此检查点进行评估。

请注意，此参数是在最近的 TF 更新(可能是 1.7 或 1.8，不确定)中添加的，因此如果您使用的是过时的版本，则可能没有此参数可用。有一个 hacky 替代方案:在 model_dir 中应该有一个名为 checkpoint 的文件.这个文件的第一行应该是

model_checkpoint_path: "model.ckpt-xxxxxx"

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