提交 aa1bd1f0 编辑于 作者: Honglin Yuan's avatar Honglin Yuan 提交者: tensorflow-copybara
浏览文件

Remove `@tf.function` decorator for `predict_on_batch` to workaround a tf.data...

Remove `@tf.function` decorator for `predict_on_batch` to workaround a tf.data performance regression.

PiperOrigin-RevId: 390692772
上级 db28145d
......@@ -332,7 +332,6 @@ class _KerasModel(model_lib.Model):
def input_spec(self):
return self._input_spec
@tf.function
def predict_on_batch(self, x, training=True):
return self._keras_model(x, training=training)
......
支持 Markdown
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册