在 Keras 中将 TensorFlow 设置为会话的问题

Question

执行以下操作：

from keras import backend as K
sess = tf.Session()
K.set_session(sess)

即使我正确导入了 Keras 和 TensorFlow，我也得到以下信息：

module 'keras.backend' has no attribute 'set_session'

任何帮助，将不胜感激！

Answer 1

.compat.v1.之后需要tf

例子：

tf.compat.v1.keras.backend.set_session(tf.compat.v1.Session(config=config));

（在 Anaconda、Python 3.7、tensorflow 2.0.0 中测试）

Answer 2

我想你需要这个：

from keras.backend.tensorflow_backend import set_session

像这样使用它：

from keras import backend as K
K.tensorflow_backend.set_session(sess)

对我来说效果很好