From f9147645fc3fb6ebe9b8c54e0cfb96a5740e3fd7 Mon Sep 17 00:00:00 2001 From: felix Date: Fri, 25 Oct 2024 19:27:15 +0200 Subject: [PATCH] true not false --- doctr/models/utils/tensorflow.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/doctr/models/utils/tensorflow.py b/doctr/models/utils/tensorflow.py index b5dc04c231..792089bda4 100644 --- a/doctr/models/utils/tensorflow.py +++ b/doctr/models/utils/tensorflow.py @@ -163,8 +163,8 @@ def export_model_to_onnx( """ # get the users eager mode eager_mode = tf.executing_eagerly() - # set eager mode to false to avoid issues with tf2onnx - tf.config.run_functions_eagerly(False) + # set eager mode to true to avoid issues with tf2onnx + tf.config.run_functions_eagerly(True) large_model = kwargs.get("large_model", False) model_proto, _ = tf2onnx.convert.from_keras( model,