我们可以在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吗？

感谢您提供的任何指导！