aboutsummaryrefslogtreecommitdiff
path: root/dcgan.py
diff options
context:
space:
mode:
Diffstat (limited to 'dcgan.py')
-rw-r--r--dcgan.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/dcgan.py b/dcgan.py
index 4ee5917..75b3de3 100644
--- a/dcgan.py
+++ b/dcgan.py
@@ -94,15 +94,15 @@ class DCGAN():
model.add(ZeroPadding2D(padding=((0,1),(0,1))))
model.add(BatchNormalization())
model.add(LeakyReLU(alpha=0.2))
- model.add(Dropout(dropout))
+ model.add(Dropout(self.dropout))
model.add(Conv2D(128, kernel_size=3, strides=2, padding="same"))
model.add(BatchNormalization())
model.add(LeakyReLU(alpha=0.2))
- model.add(Dropout(dropout))
+ model.add(Dropout(self.dropout))
model.add(Conv2D(256, kernel_size=3, strides=1, padding="same"))
model.add(BatchNormalization())
model.add(LeakyReLU(alpha=0.2))
- model.add(Dropout(dropout))
+ model.add(Dropout(self.dropout))
model.add(Flatten())
model.add(Dense(1, activation='sigmoid'))