实现tensorflow高级api时出错

我试图实现tensorflows提供的高级api，特别是基线分类器。然而，当我尝试训练模型时，我得到了以下结果

错误：

NotFoundError (see above for traceback): Key baseline/bias not found in checkpoint
     [[Node: save/RestoreV2 = RestoreV2[dtypes=[DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_INT64], _device="/job:localhost/replica:0/task:0/device:CPU:0"](_arg_save/Const_0_0, save/RestoreV2/tensor_names, save/RestoreV2/shape_and_slices)]]

代码：

---

似乎您在

model\u dir=“./checkpoints\u tutorial17-1/”

中有一个检查点，它来自另一个模型，而不是基线分类器。具体来说，该文件夹中有一个检查点文件和model.ckpt-*文件

正如tensorflow所记录的：

• model_dir：用于保存模型参数、图形等的目录。这也可用于将检查点从目录加载到估计器中，以继续训练先前保存的模型。如果是PathLike对象，则将解析路径。如果没有，则如果设置，将使用配置中的model_dir。如果两者都已设置，则它们必须相同。如果两者都为“无”，则将使用临时目录

这里，

BaselineClassifier

将首先构建一个使用

baseline/bias

的图。然后它发现在

model\u dir

中有一个以前的检查点。它将尝试加载此检查点，您应该会看到一条信息（如果您已经执行了

tf.logging.set\u verbosity（tf.logging.info）

）如下所示

"INFO:tensorflow:Restoring parameters from .../checkpoints_tutorial17-1\model.ckpt-..."

由于

model_dir

中的此检查点不是来自

基线分类器

，因此它不会有

基线/偏差

<代码>基线分类器找不到它，因此将抛出错误

"INFO:tensorflow:Restoring parameters from .../checkpoints_tutorial17-1\model.ckpt-..."