fix: resolve TensorFlow keras lazy loader recursion error #3126
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Set TF_USE_LEGACY_KERAS=1 to force TensorFlow 2.19.0 to use
tf_keras instead of Keras 3.x if the environment has Keras 3.x
installed. This work around prevents TensorFlow's lazy loader
from infinitely recursing when trying to determine which
Keras version to use. See:
https://keras.io/getting_started/#tensorflow--keras-2-backwards-compatibility.
This fixes generate_test_models.py which was failing with:
"RecursionError: maximum recursion depth exceeded in comparison"
at line checking "if self._tfll_keras_version == 'keras_3'"
BUG=tensorflow#3125
|
Oops, must ensure that tf_keras is installed. Have I mentioned how much I dislike our non-hermetic Python setup in Bazel? I might fix that soon. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Set TF_USE_LEGACY_KERAS=1 to force TensorFlow 2.19.0 to use tf_keras
instead of Keras 3.x if the build environment has both keras 3.10.0 and
tf_keras 2.19.0 installed. This prevents TensorFlow's lazy loader from
entering infinite recursion when trying to determine which Keras version
to use for tf.keras.models.Sequential().
This fixes runtime_test which was failing with:
"RecursionError: maximum recursion depth exceeded in comparison"
at line checking "if self._tfll_keras_version == 'keras_3'"
BUG=#3125