add sys

AirShower_WGAN/ShowerGAN.py

@@ -9,6 +9,7 @@ from keras.utils import plot_model
 import keras.backend.tensorflow_backend as KTF
 import utils
 import tensorflow as tf
+import sys
 
 
 KTF.set_session(utils.get_session())  # Allows 2 jobs per GPU, Please do not change this during the tutorial

MNIST_GAN/MNIST_GAN.py

@@ -6,7 +6,7 @@ import mnist
 import matplotlib.pyplot as plt
 import keras.backend.tensorflow_backend as KTF
 from gan import build_generator, build_discriminator, plot_images, make_trainable, get_session
-
+import sys
 
 log_dir="."
 KTF.set_session(get_session())  # Allows 2 jobs per GPU, Please do not change this during the tutorial