我写了一个自定义估算器,喜欢做以下事情:
像往常一样训练估算器
能够使用经过训练的变量进行预测和计算等,并将此功能存储到Tensorflow模块中以方便重用。
但是,我有点不知道该怎么做,更重要的是保存这种模块的预期方式。
我知道如何创建一个使用常量(但估计量图中没有变量)的模块,例如here。
问题是 module_fn 在以后被调用。 现在,我可以将估算器图形表单磁盘的最新检查点加载到module_fn中(请参阅here),但我想知道是否还有更合乎逻辑的方法来执行此操作。例如, LatestModuleExporter 的目的到底是什么,或者打算与固定估算值一起使用,而不是自定义估算值?