From 3bd669ae1bddd0dfeda3ced44c9042222bb809a2 Mon Sep 17 00:00:00 2001 From: nunzip Date: Thu, 7 Mar 2019 20:26:04 +0000 Subject: Fix dropout --- dcgan.py | 6 +++--- 1 file 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')) -- cgit v1.2.3