updated generator.py

generator.py

@@ -1,10 +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_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)