diff options
| author | nunzip <np.scarh@gmail.com> | 2019-03-07 20:26:04 +0000 |
|---|---|---|
| committer | nunzip <np.scarh@gmail.com> | 2019-03-07 20:26:04 +0000 |
| commit | 3bd669ae1bddd0dfeda3ced44c9042222bb809a2 (patch) | |
| tree | 0d391d5eb16d100ceb43d07d2c767506d80b1720 /dcgan.py | |
| parent | 5c5680085208f941511c9d74447c391124e38f9e (diff) | |
| download | e4-gan-3bd669ae1bddd0dfeda3ced44c9042222bb809a2.tar.gz e4-gan-3bd669ae1bddd0dfeda3ced44c9042222bb809a2.tar.bz2 e4-gan-3bd669ae1bddd0dfeda3ced44c9042222bb809a2.zip | |
Fix dropout
Diffstat (limited to 'dcgan.py')
| -rw-r--r-- | dcgan.py | 6 |
1 files changed, 3 insertions, 3 deletions
@@ -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')) |