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