adapt to TF1 deprecation

4 years ago · 416b5dd7e2
parent ad50ed287d
commit 416b5dd7e2
1 changed files with 2 additions and 2 deletions
--- 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):