Keras 'set_session' недоступен для Tensorflow 2.0

Question

Попытка сделать следующее:

import tensorflow as tf
from keras.models import load_model, Model
from keras import backend as K

sess = tf.compat.v1.Session()
K.set_session(sess)

Когда я запускаю это в Google Colab, я получаю:

RuntimeError: `set_session` is not available when using TensorFlow 2.0.

Кто-нибудь знает, как это исправить?

Answer 1

попробуйте использовать бэкэнд keras из пути tensorflow. Ваш код выдает ошибку, но у меня это работает.

import tensorflow as tf
from tensorflow.keras.models import load_model, Model
from tensorflow.python.keras import backend as K

sess = tf.compat.v1.Session()
K.set_session(sess)