使用Estimator export_saved_model时出错(未找到:在检查点中未找到关键global_step)

时间:2018-11-20 13:51:25

标签: tensorflow google-cloud-ml

我正在尝试为Google Cloud ML准备一个预先训练的模型。我正在尝试使用估计量导出模型。在估算器加载检查点期间,出现以下错误:

<p>
  {{stepper.selectedIndex}} step works!
</p>
<button mat-button (click)="previous()">Back</button>
<button mat-button (click)="next()">Next</button>
  • TensorFlow版本(使用下面的命令):1.12.0
  • Python版本:3.6

这是我正在使用的代码:

var button= FindViewById<Button>(Resource.Id.the_button);
var gradientDrawable = button.Background.Current as GradientDrawable;
gradientDrawable.SetColor(your_color);

我已经搜索了很多这个问题。 tensorflow的较旧版本有一个错误报告(我认为是1.2.0,但现在不确定)。我可以使用tf.saved_model.simple_save加载和保存此模型,并且在对它运行预测时可以使用。

我不确定这是一个错误还是缺少一些非常简单的东西。我已经在tensorflow github repo上发布了同样的东西,但是目前还没有任何回应。

2 个答案:

答案 0 :(得分:0)

在代码中,您可以使用任何优化程序将损失降到最低

train_op = optimizer.minimize( loss , global_step=tf.train.get_global_step())

您可能没有提到global_step=,因此您会收到错误消息。

答案 1 :(得分:0)

我终于设法通过手动添加“ global_step”变量使其工作,并将其导出为新的检查点并加载它。 加载当前检查点后,我运行了以下代码:

b = tf.Variable(load_step, name="global_step", dtype=tf.int64)
sess.run(b.initializer)
saver = tf.train.Saver()
saver.save(sess,'UpdatedModel/model.ckpt', global_step=load_step)

然后在我之前的代码中,我使用了新文件夹MODEL_DIR,它可以正常工作。