aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--cgan.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/cgan.py b/cgan.py
index f3bdbe5..ad3d194 100644
--- a/cgan.py
+++ b/cgan.py
@@ -1,5 +1,6 @@
from __future__ import print_function, division
-
+import tensorflow.keras as keras
+import tensorflow as tf
from keras.datasets import mnist
from keras.layers import Input, Dense, Reshape, Flatten, Dropout, multiply
from keras.layers import BatchNormalization, Activation, Embedding, ZeroPadding2D