From af9416eed5223621847dce36f0e9dca3042e3833 Mon Sep 17 00:00:00 2001 From: Andy Gao Date: Mon, 23 Jul 2018 10:26:49 -0500 Subject: [PATCH] Update wgan_gp.py --- wgan_gp/wgan_gp.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/wgan_gp/wgan_gp.py b/wgan_gp/wgan_gp.py index 7b789e7a56..42f83e0fc0 100644 --- a/wgan_gp/wgan_gp.py +++ b/wgan_gp/wgan_gp.py @@ -157,15 +157,12 @@ def build_critic(self): model.add(Dropout(0.25)) model.add(Conv2D(32, kernel_size=3, strides=2, padding="same")) model.add(ZeroPadding2D(padding=((0,1),(0,1)))) - model.add(BatchNormalization(momentum=0.8)) model.add(LeakyReLU(alpha=0.2)) model.add(Dropout(0.25)) model.add(Conv2D(64, kernel_size=3, strides=2, padding="same")) - model.add(BatchNormalization(momentum=0.8)) model.add(LeakyReLU(alpha=0.2)) model.add(Dropout(0.25)) model.add(Conv2D(128, kernel_size=3, strides=1, padding="same")) - model.add(BatchNormalization(momentum=0.8)) model.add(LeakyReLU(alpha=0.2)) model.add(Dropout(0.25)) model.add(Flatten())