标签: tensorflow tensorflow-serving tensorflow-estimator
我正在编码一个tf自定义估算器,我想编译训练有素的估算器并在另一个应用程序中使用它。
在函数model_fn中,我想在训练/评估/预测循环中有一个算法,而在编译.so时有另一个等效的算法(实际上,我在训练中使用的某些操作不会在xla框架中有任何目标操作,因此在完成培训后,我将其替换为等效操作)。
model_fn
.so
如何实现?