From e45522ab436ae389118364cc5d70f3af4773b705 Mon Sep 17 00:00:00 2001 From: phurwicz Date: Wed, 2 Mar 2022 23:35:16 -0800 Subject: [PATCH] Compatibility change for TF1 and TF2 --- cvae/cvae.py | 4 ++-- cvae/lib/data_reader.py | 2 +- cvae/lib/data_reader_array.py | 2 +- cvae/lib/model_iaf.py | 2 +- 4 files changed, 5 insertions(+), 5 deletions(-) diff --git a/cvae/cvae.py b/cvae/cvae.py index fe96850..9b9cbb4 100644 --- a/cvae/cvae.py +++ b/cvae/cvae.py @@ -6,9 +6,9 @@ import shutil import numpy as np -import tensorflow as tf +import tensorflow.compat.v1 as tf try: - tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR) + tf.logging.set_verbosity(tf.logging.ERROR) except: pass diff --git a/cvae/lib/data_reader.py b/cvae/lib/data_reader.py index 0924e27..2421bd5 100644 --- a/cvae/lib/data_reader.py +++ b/cvae/lib/data_reader.py @@ -1,6 +1,6 @@ import threading import random -import tensorflow as tf +import tensorflow.compat.v1 as tf import numpy as np import joblib import os diff --git a/cvae/lib/data_reader_array.py b/cvae/lib/data_reader_array.py index 9960e30..f49e460 100644 --- a/cvae/lib/data_reader_array.py +++ b/cvae/lib/data_reader_array.py @@ -1,6 +1,6 @@ import threading import random -import tensorflow as tf +import tensorflow.compat.v1 as tf import numpy as np import joblib import os diff --git a/cvae/lib/model_iaf.py b/cvae/lib/model_iaf.py index d583131..7ad234c 100644 --- a/cvae/lib/model_iaf.py +++ b/cvae/lib/model_iaf.py @@ -1,4 +1,4 @@ -import tensorflow as tf +import tensorflow.compat.v1 as tf def create_variable(name, shape, initializer_type=None):