tf.keras.config.set_backend

Перезагружает бэкенд (и пакет Keras).

tf.keras.config.set_backend(
    backend
)

Пример:

keras.config.set_backend("jax")

⚠️ ПРЕДУПРЕЖДЕНИЕ ⚠️: Использовать эту функцию опасно, поэтому делать это нужно с осторожностью. Смена бэкенда НЕ преобразует тип уже созданных экземпляров объектов. Поэтому любые уже созданные слои / тензоры / и т. д. станут непригодны для использования без ошибок. Настоятельно рекомендуется не сохранять никакие экземпляры объектов Keras, созданные до вызова set_backend().

Это относится к любой функции или экземпляру класса, использующему функциональность Keras. Весь такой код необходимо выполнить повторно после вызова set_backend().