Add import

1 files changed, 1 insertions, 1 deletions

diff --git a/cdcgan.py b/cdcgan.py
index 895def2..f2122b8 100755
--- a/cdcgan.py
+++ b/cdcgan.py
@@ -7,7 +7,7 @@ from keras.datasets import mnist
 from keras.layers import Input, Dense, Reshape, Flatten, Dropout, multiply
 from keras.layers import BatchNormalization, Embedding, Activation, ZeroPadding2D
 from keras.layers import LeakyReLU
-from keras.layers import UpSampling2D, Conv2D
+from keras.layers import UpSampling2D, Conv2D, Conv2DTranspose
 from keras.models import Sequential, Model
 from keras.optimizers import Adam