1 files changed, 2 insertions, 3 deletions
diff --git a/dcgan.py b/dcgan.py
index 4317994..a362f69 100644
--- a/dcgan.py
+++ b/dcgan.py
@@ -1,4 +1,5 @@
 from __future__ import print_function, division
+import tensorflow.keras as keras
 from keras.datasets import mnist
 from keras.layers import Input, Dense, Reshape, Flatten, Dropout
 from keras.layers import BatchNormalization, Activation, ZeroPadding2D
@@ -76,7 +77,7 @@ class DCGAN():
         model.add(Conv2D(self.channels, kernel_size=3, padding="same"))
         model.add(Activation("tanh"))
 
-        #model.summary()
+        model.summary()
 
         noise = Input(shape=(self.latent_dim,))
         img = model(noise)
@@ -191,8 +192,6 @@ class DCGAN():
         fig.savefig("images/mnist_%d.png" % epoch)
         plt.close()
 
-'''
 if __name__ == '__main__':
     dcgan = DCGAN()
     dcgan.train(epochs=4000, batch_size=32, save_interval=50)
-'''