Skip to content

Commit de136df

Browse files
author
wangyang59
committed
fixed small bug in demo/gan/mnist and is stable version
1 parent c3ebff5 commit de136df

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

demo/gan/gan_conf.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,9 @@
2929

3030
print('mode=%s' % mode)
3131
# the dim of the noise (z) as the input of the generator network
32-
noise_dim = 10
32+
noise_dim = 10
3333
# the dim of the hidden layer
34-
hidden_dim = 15
34+
hidden_dim = 10
3535
# the dim of the generated sample
3636
sample_dim = 2
3737

demo/gan/gan_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ def load_mnist_data(imageFile):
9191
data = data / 255.0 * 2.0 - 1.0
9292

9393
f.close()
94-
return data
94+
return data.astype('float32')
9595

9696
def load_cifar_data(cifar_path):
9797
batch_size = 10000

0 commit comments

Comments
 (0)