Fix bug where examples fail with "doesn't work when executing eagerly".

Fixes: #29
#31
PiperOrigin-RevId: 322138125
Change-Id: I8569e5d5907cd2e32189729d850cec8db671378c

tensorflow_gan/examples/cifar/eval.py

@@ -22,6 +22,7 @@
 from absl import app
 from absl import flags
 
+import tensorflow.compat.v1 as tf
 from tensorflow_gan.examples.cifar import eval_lib
 
 FLAGS = flags.FLAGS
@@ -66,4 +67,5 @@ def main(_):
 
 
 if __name__ == '__main__':
+  tf.disable_v2_behavior()
   app.run(main)

tensorflow_gan/examples/cifar/train.py

@@ -22,6 +22,7 @@
 from absl import flags
 from absl import logging
 
+import tensorflow.compat.v1 as tf
 from tensorflow_gan.examples.cifar import train_lib
 
 # ML Hparams.
@@ -58,4 +59,5 @@ def main(_):
 
 if __name__ == '__main__':
   logging.set_verbosity(logging.INFO)
+  tf.disable_v2_behavior()
   app.run(main)

tensorflow_gan/examples/cyclegan/train.py

@@ -22,6 +22,7 @@
 from absl import app
 from absl import flags
 
+import tensorflow.compat.v1 as tf
 from tensorflow_gan.examples.cyclegan import train_lib
 
 flags.DEFINE_string('image_set_x_file_pattern', None,
@@ -63,4 +64,5 @@ def main(_):
 
 
 if __name__ == '__main__':
+  tf.disable_v2_behavior()
   app.run(main)

tensorflow_gan/examples/mnist/conditional_eval.py

@@ -22,6 +22,7 @@
 from absl import app
 from absl import flags
 
+import tensorflow.compat.v1 as tf
 from tensorflow_gan.examples.mnist import conditional_eval_lib
 
 flags.DEFINE_string('checkpoint_dir', '/tmp/mnist/',
@@ -56,4 +57,5 @@ def main(_):
 
 
 if __name__ == '__main__':
+  tf.disable_v2_behavior()
   app.run(main)

tensorflow_gan/examples/mnist/eval.py

@@ -22,6 +22,7 @@
 from absl import app
 from absl import flags
 
+import tensorflow.compat.v1 as tf
 from tensorflow_gan.examples.mnist import eval_lib
 
 FLAGS = flags.FLAGS
@@ -61,4 +62,5 @@ def main(_):
 
 
 if __name__ == '__main__':
+  tf.disable_v2_behavior()
   app.run(main)

tensorflow_gan/examples/mnist/infogan_eval.py

@@ -29,6 +29,7 @@
 from absl import app
 from absl import flags
 
+import tensorflow.compat.v1 as tf
 from tensorflow_gan.examples.mnist import infogan_eval_lib
 
 flags.DEFINE_string('checkpoint_dir', '/tmp/mnist/',
@@ -68,4 +69,5 @@ def main(_):
 
 
 if __name__ == '__main__':
+  tf.disable_v2_behavior()
   app.run(main)

tensorflow_gan/examples/mnist/train.py

@@ -23,6 +23,7 @@
 from absl import flags
 from absl import logging
 
+import tensorflow.compat.v1 as tf
 from tensorflow_gan.examples.mnist import train_lib
 
 
@@ -54,4 +55,5 @@ def main(_):
 
 if __name__ == '__main__':
   logging.set_verbosity(logging.INFO)
+  tf.disable_v2_behavior()
   app.run(main)

tensorflow_gan/examples/mnist_estimator/train.py

@@ -22,6 +22,7 @@
 from absl import app
 from absl import flags
 
+import tensorflow.compat.v1 as tf
 from tensorflow_gan.examples.mnist_estimator import train_lib
 
 flags.DEFINE_integer('batch_size', 32,
@@ -46,4 +47,5 @@ def main(_):
 
 
 if __name__ == '__main__':
+  tf.disable_v2_behavior()
   app.run(main)

tensorflow_gan/examples/mnist_estimator/train_experiment.py

@@ -22,6 +22,7 @@
 from absl import app
 from absl import flags
 
+import tensorflow.compat.v1 as tf
 from tensorflow_gan.examples.mnist_estimator import train_experiment_lib
 
 # ML Hparams.
@@ -62,4 +63,5 @@ def main(_):
 
 
 if __name__ == '__main__':
+  tf.disable_v2_behavior()
   app.run(main)

tensorflow_gan/examples/progressive_gan/train_main.py

@@ -169,4 +169,5 @@ def main(_):
 
 
 if __name__ == '__main__':
+  tf.disable_v2_behavior()
   tf.app.run()

tensorflow_gan/examples/self_attention_estimator/train_experiment_main.py

@@ -27,6 +27,8 @@
 
 from absl import app
 from absl import flags
+
+import tensorflow.compat.v1 as tf
 from tensorflow_gan.examples.self_attention_estimator import train_experiment
 
 
@@ -140,4 +142,5 @@ def main(_):
 
 
 if __name__ == '__main__':
+  tf.disable_v2_behavior()
   app.run(main)

tensorflow_gan/examples/stargan/train.py

@@ -22,6 +22,7 @@
 from absl import app
 from absl import flags
 
+import tensorflow.compat.v1 as tf
 from tensorflow_gan.examples.stargan import train_lib
 
 # FLAGS for data.
@@ -66,4 +67,5 @@ def main(_):
 
 
 if __name__ == '__main__':
+  tf.disable_v2_behavior()
   app.run(main)

tensorflow_gan/examples/stargan_estimator/train.py

@@ -22,6 +22,7 @@
 from absl import app
 from absl import flags
 
+import tensorflow.compat.v1 as tf
 from tensorflow_gan.examples.stargan_estimator import train_lib
 
 # FLAGS for data.
@@ -69,4 +70,5 @@ def main(_):
 
 
 if __name__ == '__main__':
+  tf.disable_v2_behavior()
   app.run(main)