tf.keras.config.set_backend  |  TensorFlow v2.16.1 (original) (raw)

tf.keras.config.set_backend

Stay organized with collections Save and categorize content based on your preferences.

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 notto 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 callingset_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.