keras - 我们可以摆脱 Tensorflow 2.0 中的 set_session 吗？

Question

我正在尝试将代码从 Tensorflow 1.13 迁移到 Tensorflow 2.0，但我在会话声明步骤中遇到了问题。

现在，我替换tf.keras.backend.set_session为tf.compat.v1.keras.backend.set_session.

config = tf.compat.v1.ConfigProto()

config.gpu_options.allow_growth = True
if self._gpu_fraction:
    config.gpu_options.per_process_gpu_memory_fraction = self._gpu_fraction
sess = tf.compat.v1.Session(config=config)
tf.compat.v1.keras.backend.set_session(sess)

代码正在运行，但我收到警告： The name tf.keras.backend.set_session is deprecated. Please use tf.compat.v1.keras.backend.set_session instead.

我知道会话随着 Tensorflow 2 发展了很多，但我不确定我是否可以/应该摆脱这个set_session。我也不确定警告：tf.compat.v1.keras.backend.set_session打电话给旧的tf.keras.backend.set_session吗？

感谢您提供的任何指导！

标签： kerastensorflow2.0