updated generator file

generator.py

@@ -1,11 +1,11 @@
-# import tensorflow.contrib as tf_contrib
+import tensorflow.contrib as tf_contrib
 import tensorflow as tf
 
 def layer_norm(x, scope='layer_norm') :
-    # return tf_contrib.layers.layer_norm(x,
-    #                                     center=True, scale=True,
-    #                                     scope=scope)
-    return tf.keras.layers.LayerNormalization(x, center=True, scale=True)
+    return tf_contrib.layers.layer_norm(x,
+                                        center=True, scale=True,
+                                        scope=scope)
+    # return tf.keras.layers.LayerNormalization(x, center=True, scale=True)
 
 def lrelu(x, alpha=0.2):
     return tf.nn.leaky_relu(x, alpha)