contrib module not found with tensorflow 2.5.3

GiorgiaAuroraAdorni / gansformer-reproducibility-challenge

Replication of the novel Generative Adversarial Transformer.

MIT License

3 stars 3 forks source link

contrib module not found with tensorflow 2.5.3 #6

Closed karen-jin closed 2 years ago

karen-jin commented 2 years ago

File "/content/gansformer-reproducibility-challenge/src/dnnlib/tflib/tfutil.py", line 16, in import tensorflow.contrib # requires TensorFlow 1.x! ModuleNotFoundError: No module named 'tensorflow.contrib'

karen-jin commented 2 years ago

added to be back compatible.

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()