From 4a534691c7b5875ce7ab50d69e517af7a9db14ce Mon Sep 17 00:00:00 2001 From: nunzip Date: Thu, 7 Mar 2019 21:36:32 +0000 Subject: add dropout --- cgan.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/cgan.py b/cgan.py index 6406244..68bb2cc 100755 --- a/cgan.py +++ b/cgan.py @@ -15,7 +15,7 @@ from tqdm import tqdm import numpy as np class CGAN(): - def __init__(self, dense_layers = 3): + def __init__(self, dense_layers = 3, dropout=0.4): # Input shape self.img_rows = 28 self.img_cols = 28 @@ -24,6 +24,7 @@ class CGAN(): self.num_classes = 10 self.latent_dim = 100 self.dense_layers = dense_layers + self.dropout = dropout optimizer = Adam(0.0002, 0.5) @@ -87,10 +88,10 @@ class CGAN(): model.add(LeakyReLU(alpha=0.2)) model.add(Dense(512)) model.add(LeakyReLU(alpha=0.2)) - model.add(Dropout(0.4)) + model.add(Dropout(self.dropout)) model.add(Dense(512)) model.add(LeakyReLU(alpha=0.2)) - model.add(Dropout(0.4)) + model.add(Dropout(self.dropout)) model.add(Dense(1, activation='sigmoid')) #model.summary() -- cgit v1.2.3