相关疑难解决方法(0)

从 TensorFlow Estimator 模型 (2.0) 保存、加载和预测

是否有Estimator在 TF2 中序列化和恢复模型的指南?文档非常参差不齐,其中大部分都没有更新到 TF2。我还没有在任何地方看到一个清晰而完整的例子,任何地方都Estimator被保存,从磁盘加载并用于从新输入进行预测。

TBH,我对这看起来有多复杂感到有些困惑。Estimators 被宣传为拟合标准模型的简单、相对高级的方法,但在生产中使用它们的过程似乎非常神秘。例如,当我通过tf.saved_model.load(export_path)获取一个AutoTrackable对象从磁盘加载模型时:

<tensorflow.python.training.tracking.tracking.AutoTrackable at 0x7fc42e779f60>

不清楚为什么我不Estimator回来。看起来曾经有一个听起来很有用的函数tf.contrib.predictor.from_saved_model,但自从contrib消失了,它似乎不再起作用了(除了它出现在 TFLite 中)。

任何指针都会非常有帮助。如你所见,我有点失落。

tensorflow

5
推荐指数
1
解决办法
1555
查看次数

标签 统计

tensorflow ×1