true not false

mindee · Oct 25, 2024 · f914764 · f914764
1 parent 8a12442
commit f914764
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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,