aboutsummaryrefslogtreecommitdiff
path: root/models.py
diff options
context:
space:
mode:
Diffstat (limited to 'models.py')
-rw-r--r--models.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/models.py b/models.py
index 5a28b27..1c9a2c3 100644
--- a/models.py
+++ b/models.py
@@ -3,7 +3,7 @@
from keras.models import Model, Sequential
from keras.layers import *
-def get_generator:
+def get_generator():
generator = Sequential([
Dense(128*7*7, input_dim=100, activation=LeakyReLU(0.2)),
BatchNormalization(),
@@ -25,7 +25,7 @@ def get_generator:
])
return generator
-def get_discriminator:
+def get_discriminator():
discriminator = Sequential([
Convolution2D(64, 5, 5, subsample=(2,2), input_shape=(28,28,1), border_mode='same', activation=LeakyReLU(0.2)),
Dropout(0.3),