tf.keras.config.set_backend

Reload the backend (and the Keras package).

tf.keras.config.set_backend(
    backend
)

Example:

keras.config.set_backend("jax")

⚠️ WARNING ⚠️: Using this function is dangerous and should be done carefully. Changing the backend will NOT convert the type of any already-instantiated objects. Thus, any layers / tensors / etc. already created will no longer be usable without errors. It is strongly recommended not to keep around any Keras-originated objects instances created before calling set_backend().

This includes any function or class instance that uses any Keras functionality. All such code needs to be re-executed after calling set_backend().

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2024-06-07 UTC.