从训练有素的Tensorflow估算器创建模块

时间:2018-11-15 10:31:57

标签: tensorflow tensorflow-estimator

我写了一个自定义估算器,喜欢做以下事情:

  1. 像往常一样训练估算器

  2. 能够使用经过训练的变量进行预测和计算等,并将此功能存储到Tensorflow模块中以方便重用。

但是,我有点不知道该怎么做,更重要的是保存这种模块的预期方式。

我知道如何创建一个使用常量(但估计量图中没有变量)的模块,例如here

问题是 module_fn 在以后被调用。 现在,我可以将估算器图形表单磁盘的最新检查点加载到module_fn中(请参阅here),但我想知道是否还有更合乎逻辑的方法来执行此操作。例如, LatestModuleExporter 的目的到底是什么,或者打算与固定估算值一起使用,而不是自定义估算值?

0 个答案:

没有答案