tensorflow estimator:训练和服务中model_fn的不同路径

时间:2019-03-09 14:52:19

标签: tensorflow tensorflow-serving tensorflow-estimator

我正在编码一个tf自定义估算器,我想编译训练有素的估算器并在另一个应用程序中使用它。

在函数model_fn中,我想在训练/评估/预测循环中有一个算法,而在编译.so时有另一个等效的算法(实际上,我在训练中使用的某些操作不会在xla框架中有任何目标操作,因此在完成培训后,我将其替换为等效操作)。

如何实现?

0 个答案:

没有答案