diff options
Diffstat (limited to 'models.py')
-rw-r--r-- | models.py | 4 |
1 files changed, 2 insertions, 2 deletions
@@ -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), |