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

尝试使用 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)