From 416b5dd7e27af299bc5fed54669c24ee1f2bcc15 Mon Sep 17 00:00:00 2001 From: Robert Sachunsky Date: Thu, 10 Jun 2021 21:09:11 +0200 Subject: [PATCH] adapt to TF1 deprecation --- sbb_binarize/sbb_binarize.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sbb_binarize/sbb_binarize.py b/sbb_binarize/sbb_binarize.py index 557ed29..8d3a741 100644 --- a/sbb_binarize/sbb_binarize.py +++ b/sbb_binarize/sbb_binarize.py @@ -39,10 +39,10 @@ class SbbBinarizer: self.models.append(self.load_model(model_file)) def start_new_session(self): - config = tf.ConfigProto() + config = tf.compat.v1.ConfigProto() config.gpu_options.allow_growth = True - self.session = tf.Session(config=config) # tf.InteractiveSession() + self.session = tf.compat.v1.Session(config=config) # tf.InteractiveSession() tensorflow_backend.set_session(self.session) def end_session(self):