Keras 'set_session' 不适用于 Tensorflow 2.0答案

【问题标题】：Keras 'set_session' not available for Tensorflow 2.0Keras 'set_session' 不适用于 Tensorflow 2.0
【发布时间】：2020-10-31 12:04:08
【问题描述】：

尝试执行以下操作：

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.

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

【问题讨论】：

你可以在 colab 中切换到 TensorFlow 1.x

标签： python-3.x machine-learning keras tensorflow2.0

【解决方案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)

【讨论】：