aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xcdcgan.py2
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