Я пытаюсь перенести код из 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?
Спасибо за любые советы, которые вы можете дать!