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.

有人知道如何解决这个问题吗？

- Bert Hanz

---

你可以在colab中切换到TensorFlow 1.x。 - Dr. Snoopy

Answer 1

尝试使用来自tensorflow路径的keras后端。您的代码给我报错，但这个方法对我有效。

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)

- Nick