From de136df84dc2499c70fce1025d96ca2bebfbe24c Mon Sep 17 00:00:00 2001
From: wangyang59 <wangyang59@baidu.com>
Date: Wed, 30 Nov 2016 18:54:33 -0800
Subject: [PATCH] fixed small bug in demo/gan/mnist and is stable version
---
demo/gan/gan_conf.py | 4 ++--
demo/gan/gan_trainer.py | 2 +-
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/demo/gan/gan_conf.py b/demo/gan/gan_conf.py
index a6943176c22..05eee3a9b9c 100644
--- a/demo/gan/gan_conf.py
+++ b/demo/gan/gan_conf.py
@@ -29,9 +29,9 @@ is_discriminator = mode == "discriminator"
print('mode=%s' % mode)
# the dim of the noise (z) as the input of the generator network
-noise_dim = 10
+noise_dim = 10
# the dim of the hidden layer
-hidden_dim = 15
+hidden_dim = 10
# the dim of the generated sample
sample_dim = 2
diff --git a/demo/gan/gan_trainer.py b/demo/gan/gan_trainer.py
index 3f27f04fc59..72699952b96 100644
--- a/demo/gan/gan_trainer.py
+++ b/demo/gan/gan_trainer.py
@@ -91,7 +91,7 @@ def load_mnist_data(imageFile):
data = data / 255.0 * 2.0 - 1.0
f.close()
- return data
+ return data.astype('float32')
def load_cifar_data(cifar_path):
batch_size = 10000
--
GitLab