Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[TensorFlow] Upgrade to keras v3 #1461

Open
felixdittrich92 opened this issue Feb 9, 2024 · 0 comments
Open

[TensorFlow] Upgrade to keras v3 #1461

felixdittrich92 opened this issue Feb 9, 2024 · 0 comments
Assignees
Labels
framework: tensorflow Related to TensorFlow backend module: models Related to doctr.models type: breaking change Introducing a breaking change
Milestone

Comments

@felixdittrich92
Copy link
Contributor

felixdittrich92 commented Feb 9, 2024

🚀 The feature

TensorFlow >= 2.16.0 switches to Keras v3 (keras-core) which will especially break the saving and loading logic from models in our case

  • pin to >=TF 2.16.0 (requires py3.9 min)

Keras: https://keras.io/api/models/model_saving_apis/weights_saving_and_loading/#save_weights-method
tf2onnx: https://github.com/onnx/tensorflow-onnx?tab=readme-ov-file#from_keras-tf-20-and-newer
hf hub: https://huggingface.co/docs/huggingface_hub/guides/upload

(Optional) Provide a script to convert old models to new format

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
framework: tensorflow Related to TensorFlow backend module: models Related to doctr.models type: breaking change Introducing a breaking change
Projects
None yet
Development

No branches or pull requests

1 participant